import { db } from "./db.ts";
import { JsonSchema, jsonType } from "t_rest/server";

export const configSchema = {
  type: "object",
  properties: {
    pausedReason: { type: ["string", "null"] },
    maxUserJobs: { type: "number" },
    maxJobs: { type: "number" },
    maxDailyJobs: { type: "number" },
    defaultParams: {
      type: "object",
      properties: {
        batch_size: { type: "number" },
        n_iter: { type: "number" },
        width: { type: "number" },
        height: { type: "number" },
        steps: { type: "number" },
        cfg_scale: { type: "number" },
        sampler_name: { type: "string" },
        negative_prompt: { type: "string" },
      },
    },
  },
  required: ["adminUsernames", "maxUserJobs", "maxJobs", "maxDailyJobs", "defaultParams"],
} as const satisfies JsonSchema;

export type Config = jsonType<typeof configSchema>;

export async function getConfig(): Promise<Config> {
  const configEntry = await db.get<Config>(["config"]);
  const config = configEntry?.value;
  return {
    pausedReason: config?.pausedReason ?? null,
    maxUserJobs: config?.maxUserJobs ?? Infinity,
    maxJobs: config?.maxJobs ?? Infinity,
    maxDailyJobs: config?.maxDailyJobs ?? Infinity,
    defaultParams: config?.defaultParams ?? {},
  };
}

export async function setConfig(newConfig: Partial<Config>): Promise<void> {
  const oldConfig = await getConfig();
  const config: Config = {
    pausedReason: newConfig.pausedReason ?? oldConfig.pausedReason,
    maxUserJobs: newConfig.maxUserJobs ?? oldConfig.maxUserJobs,
    maxJobs: newConfig.maxJobs ?? oldConfig.maxJobs,
    maxDailyJobs: newConfig.maxDailyJobs ?? oldConfig.maxDailyJobs,
    defaultParams: newConfig.defaultParams ?? oldConfig.defaultParams,
  };
  await db.set(["config"], config);
}