forked from pinks/eris
1
0
Fork 0
eris/app/config.ts

50 lines
1.6 KiB
TypeScript
Raw Permalink Normal View History

2023-09-22 02:59:22 +00:00
import { db } from "./db.ts";
2023-10-05 09:00:51 +00:00
import { JsonSchema, jsonType } from "t_rest/server";
2023-09-22 02:59:22 +00:00
2023-10-05 09:00:51 +00:00
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" },
},
},
},
2023-10-13 11:47:57 +00:00
required: ["adminUsernames", "maxUserJobs", "maxJobs", "defaultParams"],
2023-10-05 09:00:51 +00:00
} as const satisfies JsonSchema;
export type Config = jsonType<typeof configSchema>;
2023-09-22 02:59:22 +00:00
2023-10-05 09:00:51 +00:00
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 ?? {},
};
2023-09-22 02:59:22 +00:00
}
2023-10-05 09:00:51 +00:00
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,
};
2023-09-22 02:59:22 +00:00
await db.set(["config"], config);
}