forked from pinks/eris
50 lines
1.6 KiB
TypeScript
50 lines
1.6 KiB
TypeScript
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" },
|
|
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", "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,
|
|
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,
|
|
defaultParams: newConfig.defaultParams ?? oldConfig.defaultParams,
|
|
};
|
|
await db.set(["config"], config);
|
|
}
|