create-initial-migration.ts

import { EOL } from "node:os";

import type { RelationalTable } from "../create-relational-structure";
import { getForeignKeys, getPrimaryKeys } from "../create-relational-structure";
import { getFullTableName } from "../utils";
import { getFieldType } from "./get-field-type";

export function createInitialMigration({
  tables,
  schemaName,
}: {
  tables: RelationalTable[];
  schemaName?: string | undefined;
}): string {
  return tables
    .map((table) => {
      const foreignKeys = getForeignKeys(table);
      return [
        `CREATE TABLE ${getFullTableName({ tableName: table.name, schemaName })} (`,
        [
          ...table.fields.map(
            (field) =>
              `  ${field.key} ${getFieldType(field)}${field.isNullable ? "" : " NOT NULL"}`,
          ),
          `  PRIMARY KEY (${getPrimaryKeys(table).join(", ")})`,
          ...(foreignKeys.length
            ? foreignKeys.map(
                ({ key, reference }) =>
                  `  FOREIGN KEY (${key}) REFERENCES ${getFullTableName({ tableName: reference.table, schemaName })} (${reference.key})`,
              )
            : []),
        ].join(`,${EOL}`),
        `);`,
        ...foreignKeys.map(
          ({ key }) =>
            `CREATE INDEX idx_${table.name}_${key} ON ${getFullTableName({
              tableName: table.name,
              schemaName,
            })} (${key});`,
        ),
      ].join(EOL);
    })
    .join(`${EOL}${EOL}`);
}