import { GrammyTypes, IKV } from "../deps.ts";
import { SdTxt2ImgInfo } from "../common/sdApi.ts";
import { PngInfo } from "../common/parsePngInfo.ts";
import { db } from "./db.ts";

export interface JobSchema {
  task:
    | {
      type: "txt2img";
      params: Partial<PngInfo>;
    }
    | {
      type: "img2img";
      params: Partial<PngInfo>;
      fileId: string;
    };
  from: GrammyTypes.User;
  chat: GrammyTypes.Chat;
  requestMessageId: number;
  status:
    | {
      type: "waiting";
      message?: GrammyTypes.Message.TextMessage;
      lastErrorDate?: Date;
    }
    | {
      type: "processing";
      progress: number;
      worker: string;
      updatedDate: Date;
      message?: GrammyTypes.Message.TextMessage;
    }
    | {
      type: "done";
      info?: SdTxt2ImgInfo;
      startDate?: Date;
      endDate?: Date;
    };
}

type JobIndices = {
  "status.type": JobSchema["status"]["type"];
};

export const jobStore = new IKV.Store<JobSchema, JobIndices>(db, "job", {
  indices: {
    "status.type": { getValue: (job) => job.status.type },
  },
});