40 lines
1.4 KiB
TypeScript
40 lines
1.4 KiB
TypeScript
|
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)}`);
|
||
|
|
}
|