forked from pinks/eris
1
0
Fork 0

move files

This commit is contained in:
pinks 2023-09-23 20:49:05 +02:00
parent 27eb471f9e
commit 9dbeb37fd7
20 changed files with 28 additions and 34 deletions

View File

@ -1,4 +1,4 @@
import * as SdApi from "../common/sdApi.ts"; import * as SdApi from "../sd/sdApi.ts";
import { db } from "./db.ts"; import { db } from "./db.ts";
export interface ConfigData { export interface ConfigData {

View File

View File

@ -1,10 +1,10 @@
import { bot } from "../bot/mod.ts"; import { bot } from "../bot/mod.ts";
import { PngInfo } from "../common/parsePngInfo.ts"; import { PngInfo } from "../sd/parsePngInfo.ts";
import * as SdApi from "../common/sdApi.ts"; import * as SdApi from "../sd/sdApi.ts";
import { formatUserChat } from "../common/formatUserChat.ts"; import { formatUserChat } from "../utils/formatUserChat.ts";
import { getConfig, SdInstanceData } from "../db/config.ts"; import { getConfig, SdInstanceData } from "./config.ts";
import { db } from "../db/db.ts"; import { db } from "./db.ts";
import { generationStore, SdGenerationInfo } from "../db/jobStore.ts"; import { generationStore, SdGenerationInfo } from "./generationStore.ts";
import { import {
Async, Async,
AsyncX, AsyncX,
@ -18,9 +18,8 @@ import {
KVMQ, KVMQ,
Log, Log,
} from "../deps.ts"; } from "../deps.ts";
import { formatOrdinal } from "../common/formatOrdinal.ts"; import { formatOrdinal } from "../utils/formatOrdinal.ts";
import { deadline } from "../common/deadline.ts"; import { SdError } from "../sd/SdError.ts";
import { SdError } from "../common/SdError.ts";
const logger = () => Log.getLogger(); const logger = () => Log.getLogger();
@ -66,7 +65,7 @@ export async function restartGenerationWorkers() {
// check if worker is up // check if worker is up
const activeWorkerStatus = await activeWorkerSdClient.GET("/sdapi/v1/memory", { const activeWorkerStatus = await activeWorkerSdClient.GET("/sdapi/v1/memory", {
signal: deadline(10_000), signal: AbortSignal.timeout(10_000),
}) })
.then((response) => { .then((response) => {
if (!response.data) { if (!response.data) {
@ -233,7 +232,7 @@ async function processGenerationJob(
await Async.delay(3000); await Async.delay(3000);
const progressResponse = await workerSdClient.GET("/sdapi/v1/progress", { const progressResponse = await workerSdClient.GET("/sdapi/v1/progress", {
params: {}, params: {},
signal: deadline(15_000), signal: AbortSignal.timeout(15_000),
}); });
if (!progressResponse.data) { if (!progressResponse.data) {
throw new SdError( throw new SdError(

View File

@ -1,5 +1,5 @@
import { GrammyTypes, IKV } from "../deps.ts"; import { GrammyTypes, IKV } from "../deps.ts";
import { PngInfo } from "../common/parsePngInfo.ts"; import { PngInfo } from "../sd/parsePngInfo.ts";
import { db } from "./db.ts"; import { db } from "./db.ts";
export interface GenerationSchema { export interface GenerationSchema {

View File

@ -1,4 +1,4 @@
import { generationQueue } from "../tasks/generationQueue.ts"; import { generationQueue } from "../app/generationQueue.ts";
import { Context } from "./mod.ts"; import { Context } from "./mod.ts";
export async function cancelCommand(ctx: Context) { export async function cancelCommand(ctx: Context) {

View File

@ -1,9 +1,9 @@
import { Collections, Grammy, GrammyStatelessQ } from "../deps.ts"; import { Collections, Grammy, GrammyStatelessQ } from "../deps.ts";
import { formatUserChat } from "../common/formatUserChat.ts"; import { formatUserChat } from "../utils/formatUserChat.ts";
import { parsePngInfo, PngInfo } from "../common/parsePngInfo.ts"; import { parsePngInfo, PngInfo } from "../sd/parsePngInfo.ts";
import { Context, logger } from "./mod.ts"; import { Context, logger } from "./mod.ts";
import { generationQueue } from "../tasks/generationQueue.ts"; import { generationQueue } from "../app/generationQueue.ts";
import { getConfig } from "../db/config.ts"; import { getConfig } from "../app/config.ts";
export const img2imgQuestion = new GrammyStatelessQ.StatelessQuestion<Context>( export const img2imgQuestion = new GrammyStatelessQ.StatelessQuestion<Context>(
"img2img", "img2img",

View File

@ -1,11 +1,11 @@
import { Grammy, GrammyAutoQuote, GrammyFiles, GrammyParseMode, Log } from "../deps.ts"; import { Grammy, GrammyAutoQuote, GrammyFiles, GrammyParseMode, Log } from "../deps.ts";
import { formatUserChat } from "../common/formatUserChat.ts"; import { formatUserChat } from "../utils/formatUserChat.ts";
import { queueCommand } from "./queueCommand.ts"; import { queueCommand } from "./queueCommand.ts";
import { txt2imgCommand, txt2imgQuestion } from "./txt2imgCommand.ts"; import { txt2imgCommand, txt2imgQuestion } from "./txt2imgCommand.ts";
import { pnginfoCommand, pnginfoQuestion } from "./pnginfoCommand.ts"; import { pnginfoCommand, pnginfoQuestion } from "./pnginfoCommand.ts";
import { img2imgCommand, img2imgQuestion } from "./img2imgCommand.ts"; import { img2imgCommand, img2imgQuestion } from "./img2imgCommand.ts";
import { cancelCommand } from "./cancelCommand.ts"; import { cancelCommand } from "./cancelCommand.ts";
import { getConfig, setConfig } from "../db/config.ts"; import { getConfig, setConfig } from "../app/config.ts";
export const logger = () => Log.getLogger(); export const logger = () => Log.getLogger();

View File

@ -1,5 +1,5 @@
import { Grammy, GrammyParseMode, GrammyStatelessQ } from "../deps.ts"; import { Grammy, GrammyParseMode, GrammyStatelessQ } from "../deps.ts";
import { getPngInfo, parsePngInfo } from "../common/parsePngInfo.ts"; import { getPngInfo, parsePngInfo } from "../sd/parsePngInfo.ts";
import { Context } from "./mod.ts"; import { Context } from "./mod.ts";
export const pnginfoQuestion = new GrammyStatelessQ.StatelessQuestion<Context>( export const pnginfoQuestion = new GrammyStatelessQ.StatelessQuestion<Context>(

View File

@ -1,8 +1,8 @@
import { Grammy, GrammyParseMode } from "../deps.ts"; import { Grammy, GrammyParseMode } from "../deps.ts";
import { Context, logger } from "./mod.ts"; import { Context, logger } from "./mod.ts";
import { getFlagEmoji } from "../common/getFlagEmoji.ts"; import { getFlagEmoji } from "../utils/getFlagEmoji.ts";
import { activeGenerationWorkers, generationQueue } from "../tasks/generationQueue.ts"; import { activeGenerationWorkers, generationQueue } from "../app/generationQueue.ts";
import { getConfig } from "../db/config.ts"; import { getConfig } from "../app/config.ts";
export async function queueCommand(ctx: Grammy.CommandContext<Context>) { export async function queueCommand(ctx: Grammy.CommandContext<Context>) {
let formattedMessage = await getMessageText(); let formattedMessage = await getMessageText();

View File

@ -1,9 +1,9 @@
import { Grammy, GrammyStatelessQ } from "../deps.ts"; import { Grammy, GrammyStatelessQ } from "../deps.ts";
import { formatUserChat } from "../common/formatUserChat.ts"; import { formatUserChat } from "../utils/formatUserChat.ts";
import { getPngInfo, parsePngInfo, PngInfo } from "../common/parsePngInfo.ts"; import { getPngInfo, parsePngInfo, PngInfo } from "../sd/parsePngInfo.ts";
import { Context, logger } from "./mod.ts"; import { Context, logger } from "./mod.ts";
import { generationQueue } from "../tasks/generationQueue.ts"; import { generationQueue } from "../app/generationQueue.ts";
import { getConfig } from "../db/config.ts"; import { getConfig } from "../app/config.ts";
export const txt2imgQuestion = new GrammyStatelessQ.StatelessQuestion<Context>( export const txt2imgQuestion = new GrammyStatelessQ.StatelessQuestion<Context>(
"txt2img", "txt2img",

View File

@ -1,5 +0,0 @@
export function deadline(timeout: number): AbortSignal {
const controller = new AbortController();
setTimeout(() => controller.abort(), timeout);
return controller.signal;
}

View File

@ -14,7 +14,7 @@ Log.setup({
// Main program logic // Main program logic
import { bot } from "./bot/mod.ts"; import { bot } from "./bot/mod.ts";
import { runAllTasks } from "./tasks/mod.ts"; import { runAllTasks } from "./app/mod.ts";
await Promise.all([ await Promise.all([
bot.start(), bot.start(),
runAllTasks(), runAllTasks(),