eris/bot/txt2imgCommand.ts

40 lines
1.4 KiB
TypeScript
Raw Normal View History

2023-09-10 18:56:17 +00:00
import { Grammy } from "../deps.ts";
import { formatUserChat } from "../utils.ts";
import { jobStore } from "../db/jobStore.ts";
import { parsePngInfo } from "../sd.ts";
import { Context, logger } from "./mod.ts";
export async function txt2imgCommand(ctx: Grammy.CommandContext<Context>) {
if (!ctx.from?.id) {
return ctx.reply("I don't know who you are");
}
const config = ctx.session.global;
if (config.pausedReason != null) {
return ctx.reply(`I'm paused: ${config.pausedReason || "No reason given"}`);
}
const jobs = await jobStore.getBy("status.type", "waiting");
if (jobs.length >= config.maxJobs) {
return ctx.reply(
`The queue is full. Try again later. (Max queue size: ${config.maxJobs})`,
);
}
const userJobs = jobs.filter((job) => job.value.request.from.id === ctx.from?.id);
if (userJobs.length >= config.maxUserJobs) {
return ctx.reply(
`You already have ${config.maxUserJobs} jobs in queue. Try again later.`,
);
}
const params = parsePngInfo(ctx.match);
if (!params.prompt) {
return ctx.reply("Please describe what you want to see after the command");
}
const reply = await ctx.reply("Accepted. You are now in queue.");
await jobStore.create({
params,
request: ctx.message,
reply,
status: { type: "waiting" },
});
logger().debug(`Job enqueued for ${formatUserChat(ctx)}`);
}