import { isDefined } from "prostgles-types";
import type { DboBuilder } from "../DboBuilder";

export const getAllViewRelatedTables = async (dboBuilder: DboBuilder, viewName: string) => {
  const db = dboBuilder.db;
  const tables = await db.any<{
    table_name: string;
    table_schema: string;
    table_oid: number;
  }>(
    `
    SELECT DISTINCT 
        vcu.table_name, 
        vcu.table_schema,
        c.oid AS table_oid
    FROM information_schema.view_column_usage vcu
    JOIN pg_class c ON c.relname = vcu.table_name
    JOIN pg_namespace n ON n.oid = c.relnamespace AND n.nspname = vcu.table_schema
    WHERE vcu.view_name = \${viewName}
  `,
    { viewName },
  );
  return tables
    .map((t) => {
      const tableInfo = dboBuilder.tablesOrViews?.find((tov) => tov.oid === t.table_oid);
      return tableInfo;
    })
    .filter(isDefined);
};