import type { CoreAccountSchema } from "./schemaTypes/AccountSchema.js";
import type { CoreCoFeedSchema } from "./schemaTypes/CoFeedSchema.js";
import type { CoreCoListSchema } from "./schemaTypes/CoListSchema.js";
import type { CoreCoMapSchema } from "./schemaTypes/CoMapSchema.js";
import type { CoreCoValueSchema } from "./schemaTypes/CoValueSchema.js";
import type { SnapshotRefSchema } from "./schemaTypes/SnapshotRefSchema.js";

type ConstructorWithSchema<S extends CoreCoValueSchema = CoreCoValueSchema> = {
  name?: string;
  coValueSchema?: S;
};

function assertCoreCoValueSchema<C extends ConstructorWithSchema>(
  constructor: C,
  expectedSchemaType: string,
  operation: "create" | "load" | "resolve",
): NonNullable<C["coValueSchema"]> {
  const schema = constructor.coValueSchema;
  if (!schema) {
    const className = constructor.name || "AnonymousCoValue";
    throw new Error(
      `[schema-invariant] ${className}.${operation} requires a coValueSchema. ` +
        `Attach a schema via co.map/co.list/co.feed/co.account before using this class.`,
    );
  }

  if (schema.builtin !== expectedSchemaType) {
    const className = constructor.name || "AnonymousCoValue";
    throw new Error(
      `[schema-invariant] ${className}.${operation} requires a ${expectedSchemaType} schema. ` +
        `Got ${schema.builtin} instead.`,
    );
  }

  return schema as NonNullable<C["coValueSchema"]>;
}

type CoValueSchema =
  | CoreCoMapSchema
  | CoreCoListSchema
  | CoreCoFeedSchema
  | CoreAccountSchema
  | SnapshotRefSchema;

export function assertCoValueSchema<
  T extends CoValueSchema["builtin"],
  C extends ConstructorWithSchema,
>(
  constructor: C,
  type: T,
  operation: "create" | "load" | "resolve",
): Extract<CoValueSchema, { builtin: T }> {
  const schema = assertCoreCoValueSchema(constructor, type, operation);

  return schema as Extract<CoValueSchema, { builtin: T }>;
}