import type {
  InferEnum,
  InferSelectModel,
  InferInsertModel,
  AnySchemas,
  AnyRelations,
} from "../orm/index.ts"

import type { PgUnionDatabase } from "./database.ts"
import type { PgEnum, PgTable, TableConfig } from "./pgcore.ts"

export type AnyTableSchema = PgTable<TableConfig>
export type AnyTableSchemas = Record<string, AnyTableSchema>

type RemoveNeverValue<T> = {
  [K in keyof T as T[K] extends never ? never : K]: T[K]
}

export type SchemaToData<T> =
  T extends PgTable<infer _>
    ? InferSelectModel<T>
    : T extends PgEnum<infer __>
      ? InferEnum<T>
      : never
export type SchemasToData<T extends AnySchemas> = RemoveNeverValue<{
  [K in keyof T]: SchemaToData<T[K]>
}>

export type TableSchemaToSelect<T extends AnyTableSchema> =
  T extends PgTable<infer _> ? InferSelectModel<T> : never
export type TableSchemasToSelect<T extends AnyTableSchemas> = RemoveNeverValue<{
  [K in keyof T]: TableSchemaToSelect<T[K]>
}>

export type TableSchemaToInsert<T extends AnyTableSchema> =
  T extends PgTable<infer _> ? InferInsertModel<T> : never
export type TableSchemasToInsert<T extends AnyTableSchemas> = RemoveNeverValue<{
  [K in keyof T]: TableSchemaToInsert<T[K]>
}>

export type TableSchemaToUpdate<T extends AnyTableSchema> =
  T extends PgTable<infer _> ? Partial<InferInsertModel<T>> : never
export type TableSchemasToUpdate<T extends AnyTableSchemas> = RemoveNeverValue<{
  [K in keyof T]: TableSchemaToUpdate<T[K]>
}>

export type TableSchemaToUpsert<T extends AnyTableSchema> =
  T extends PgTable<infer _> ? InferInsertModel<T> : never
export type TableSchemasToUpsert<T extends AnyTableSchemas> = RemoveNeverValue<{
  [K in keyof T]: TableSchemaToUpsert<T[K]>
}>

type TableSchemasToQuery<
  Schemas extends AnyTableSchemas,
  Relations extends AnyRelations,
> = PgUnionDatabase<Schemas, Relations>["query"]
type TableSchemasToFindConfig<Schemas extends AnyTableSchemas, Relations extends AnyRelations> = {
  [K in keyof TableSchemasToQuery<Schemas, Relations>]: NonNullable<
    Parameters<TableSchemasToQuery<Schemas, Relations>[K]["findMany"]>[0]
  >
}

export type TableSchemasToQueryColumns<
  Schemas extends AnyTableSchemas,
  Relations extends AnyRelations,
> = {
  [K in keyof TableSchemasToFindConfig<Schemas, Relations>]: TableSchemasToFindConfig<
    Schemas,
    Relations
  >[K] extends {
    columns?: infer Columns
  }
    ? NonNullable<Columns>
    : never
}
export type TableSchemasToQueryWhere<
  Schemas extends AnyTableSchemas,
  Relations extends AnyRelations,
> = {
  [K in keyof TableSchemasToFindConfig<Schemas, Relations>]: TableSchemasToFindConfig<
    Schemas,
    Relations
  >[K] extends {
    where?: infer Where
  }
    ? NonNullable<Where>
    : never
}
export type TableSchemasToQueryWith<
  Schemas extends AnyTableSchemas,
  Relations extends AnyRelations,
> = {
  [K in keyof TableSchemasToFindConfig<Schemas, Relations>]: TableSchemasToFindConfig<
    Schemas,
    Relations
  >[K] extends {
    with?: infer With
  }
    ? NonNullable<With>
    : never
}
export type TableSchemasToQueryOrderBy<
  Schemas extends AnyTableSchemas,
  Relations extends AnyRelations,
> = {
  [K in keyof TableSchemasToFindConfig<Schemas, Relations>]: TableSchemasToFindConfig<
    Schemas,
    Relations
  >[K] extends {
    orderBy?: infer OrderBy
  }
    ? NonNullable<OrderBy>
    : never
}