forked from pinks/eris
80 lines
2.6 KiB
TypeScript
80 lines
2.6 KiB
TypeScript
import { db } from "./db.ts";
|
|
import { JsonSchema, jsonType } from "t_rest/server";
|
|
|
|
export const configSchema = {
|
|
type: "object",
|
|
properties: {
|
|
adminUsernames: { type: "array", items: { type: "string" } },
|
|
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" },
|
|
},
|
|
},
|
|
sdInstances: {
|
|
type: "object",
|
|
additionalProperties: {
|
|
type: "object",
|
|
properties: {
|
|
name: { type: "string" },
|
|
api: {
|
|
type: "object",
|
|
properties: {
|
|
url: { type: "string" },
|
|
auth: { type: "string" },
|
|
},
|
|
required: ["url"],
|
|
},
|
|
maxResolution: { type: "number" },
|
|
},
|
|
required: ["api", "maxResolution"],
|
|
},
|
|
},
|
|
},
|
|
required: ["adminUsernames", "maxUserJobs", "maxJobs", "defaultParams", "sdInstances"],
|
|
} 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 {
|
|
adminUsernames: config?.adminUsernames ?? Deno.env.get("TG_ADMIN_USERNAMES")?.split(",") ?? [],
|
|
pausedReason: config?.pausedReason ?? null,
|
|
maxUserJobs: config?.maxUserJobs ?? Infinity,
|
|
maxJobs: config?.maxJobs ?? Infinity,
|
|
defaultParams: config?.defaultParams ?? {},
|
|
sdInstances: config?.sdInstances ??
|
|
{
|
|
"local": {
|
|
api: { url: Deno.env.get("SD_API_URL") ?? "http://127.0.0.1:7860/" },
|
|
maxResolution: 1024 * 1024,
|
|
},
|
|
},
|
|
};
|
|
}
|
|
|
|
export async function setConfig(newConfig: Partial<Config>): Promise<void> {
|
|
const oldConfig = await getConfig();
|
|
const config: Config = {
|
|
adminUsernames: newConfig.adminUsernames ?? oldConfig.adminUsernames,
|
|
pausedReason: newConfig.pausedReason ?? oldConfig.pausedReason,
|
|
maxUserJobs: newConfig.maxUserJobs ?? oldConfig.maxUserJobs,
|
|
maxJobs: newConfig.maxJobs ?? oldConfig.maxJobs,
|
|
defaultParams: newConfig.defaultParams ?? oldConfig.defaultParams,
|
|
sdInstances: newConfig.sdInstances ?? oldConfig.sdInstances,
|
|
};
|
|
await db.set(["config"], config);
|
|
}
|