supabase/studio/components/grid/services/meta/SqlMetaService.ts

import { IMetaService } from '.'
import {
  getColumnsSql,
  getPrimaryKeysSql,
  getRelationshipsSql,
  getTableSql,
} from './SqlMetaService.sqls'

export class SqlMetaService implements IMetaService {
  constructor(protected onSqlQuery: (query: string) => Promise<{ data?: any; error?: any }>) {}

  async fetchInfo(name: string, schema?: string | undefined) {
    const sql = getTableSql(schema ?? 'public', name)
    const { data, error } = await this.onSqlQuery(sql)
    if (error) {
      return { error }
    } else {
      if (data?.length == 1) {
        return { data: data[0] }
      } else {
        return { error: { message: 'fetch table info failed' } }
      }
    }
  }
  async fetchColumns(name: string, schema?: string | undefined) {
    const sql = getColumnsSql(schema ?? 'public', name)
    const { data, error } = await this.onSqlQuery(sql)
    if (error) {
      return { error }
    } else {
      return { data }
    }
  }
  async fetchPrimaryKeys(name: string, schema?: string | undefined) {
    const sql = getPrimaryKeysSql(schema ?? 'public', name)
    const { data, error } = await this.onSqlQuery(sql)
    if (error) {
      return { error }
    } else {
      return { data }
    }
  }
  async fetchRelationships(name: string, schema?: string | undefined) {
    const sql = getRelationshipsSql(schema ?? 'public', name)
    const { data, error } = await this.onSqlQuery(sql)
    if (error) {
      return { error }
    } else {
      return { data }
    }
  }
}