import {
  createGetDB,
  type Generated,
  type Timestamp,
  type ObjectColumnType,
  type ArrayColumnType,
  type NamespaceDB,
  type NamespaceInsertable,
  type NamespaceSelectable,
  type NamespaceTable,
  type NamespaceTableName,
  type NamespaceTransaction,
  type NamespaceUpdateable,
} from "@tailor-platform/sdk/kysely";

export interface Namespace {
  "main-db": {
    Shift: {
      id: Generated<string>;
      shiftScheduleId: string;
      date: Timestamp;
      shiftType: "NORMAL" | "SPLIT" | "THROUGH" | "ON_CALL" | "SUPPORT";
      cancelledAt: Timestamp | null;
      plannedStartAt: Timestamp;
      plannedEndAt: Timestamp;
      segments: ArrayColumnType<ObjectColumnType<{
        sequence: number;
        plannedStartAt: Timestamp;
        plannedEndAt: Timestamp;
        breakMinutes: number;
      }>>;
      shiftPatternId: string | null;
      siteId: string | null;
      createdAt: Generated<Timestamp>;
      updatedAt: Generated<Timestamp>;
    }

    ShiftPattern: {
      id: Generated<string>;
      code: string;
      scopeId: string | null;
      name: string;
      kind: "DAY" | "EARLY" | "LATE" | "NIGHT" | "ON_CALL" | "OFF";
      segments: {
        sequence: number;
        startTime: number;
        endTime: number;
        breakMinutes: number;
        spansMidnight: boolean;
      }[];
      createdAt: Generated<Timestamp>;
      updatedAt: Generated<Timestamp>;
    }

    ShiftPlacement: {
      id: Generated<string>;
      shiftId: string;
      assignmentId: string;
      status: "ACTIVE" | "SUPERSEDED" | "CANCELLED";
      supersededById: string | null;
      assignedAt: Timestamp;
      releasedAt: Timestamp | null;
      provenance: "GENERATED" | "MANUAL";
      createdAt: Generated<Timestamp>;
      updatedAt: Generated<Timestamp>;
    }

    ShiftSchedule: {
      id: Generated<string>;
      startDate: Timestamp;
      endDate: Timestamp;
      status: "DRAFT" | "CONFIRMED";
      confirmedAt: Timestamp | null;
      createdAt: Generated<Timestamp>;
      updatedAt: Generated<Timestamp>;
    }
  }
}

export const getDB = createGetDB<Namespace>();

export type DB<N extends keyof Namespace = keyof Namespace> = NamespaceDB<Namespace, N>;

export type Transaction<K extends keyof Namespace | DB = keyof Namespace> =
  NamespaceTransaction<Namespace, K>;

type TableName = NamespaceTableName<Namespace>;
export type Table<T extends TableName> = NamespaceTable<Namespace, T>;

export type Insertable<T extends TableName> = NamespaceInsertable<Namespace, T>;
export type Selectable<T extends TableName> = NamespaceSelectable<Namespace, T>;
export type Updateable<T extends TableName> = NamespaceUpdateable<Namespace, T>;