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

export interface Namespace {
  "main-db": {
    AccrualPlan: {
      id: Generated<string>;
      leaveTypeKey: string;
      accrualMethod: "FRONT_LOAD_TENURE";
      grantType: "STATUTORY" | "MANUAL";
      eligibilityDelayMonths: number;
      baseGrantDays: string;
      tenureTiers: {
        yearsOfService: number;
        grantDays: number;
      }[];
      grantCondition: ObjectColumnType<{
        type: "NONE" | "MIN_WORKED_DAYS";
        minWorkedDays?: number | null;
        referenceMonths?: number | null;
      }> | null;
      annualCapDays: string | null;
      expirationMonths: number;
      appliesToEmploymentTypeId: string | null;
      effectiveStart: Timestamp;
      effectiveEnd: Timestamp | null;
      versionOf: string;
      createdAt: Generated<Timestamp>;
      updatedAt: Generated<Timestamp>;
    }

    LeaveConsumption: {
      id: Generated<string>;
      leaveRequestId: string;
      leaveGrantId: string;
      daysConsumed: string;
      restoredAt: Timestamp | null;
      createdAt: Generated<Timestamp>;
      updatedAt: Generated<Timestamp>;
    }

    LeaveGrant: {
      id: Generated<string>;
      workerId: string;
      leaveTypeKey: string;
      grantType: "STATUTORY" | "COMPENSATORY" | "MANUAL";
      grantSource: "HIRE" | "ANNIVERSARY" | "MANUAL";
      grantedDays: string;
      remainingDays: string;
      grantedDate: Timestamp;
      expirationDate: Timestamp;
      expiredAt: Timestamp | null;
      createdAt: Generated<Timestamp>;
      updatedAt: Generated<Timestamp>;
    }

    LeaveRequest: {
      id: Generated<string>;
      workerId: string;
      assignmentId: string;
      leaveTypeKey: string;
      startDate: Timestamp;
      endDate: Timestamp;
      reason: string;
      status: "PENDING" | "APPROVED" | "REJECTED" | "CANCELLED" | "CANCEL_PENDING";
      resolvedAt: Timestamp | null;
      resolvedBy: string | null;
      targetEntityId: string;
      createdAt: Generated<Timestamp>;
      updatedAt: Generated<Timestamp>;
    }

    LeaveType: {
      id: Generated<string>;
      key: string;
      displayName: string;
      category: string;
      requiresBalance: boolean;
      drawsFromCompensatoryGrants: boolean | null;
      timeEntryCodeKey: string;
      isActive: boolean;
      requestUnit: "FULL_DAY" | "HALF_DAY" | "RANGE" | 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>;