forked from pinks/eris
move files
This commit is contained in:
parent
27eb471f9e
commit
9dbeb37fd7
|
@ -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 {
|
|
@ -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(
|
|
@ -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 {
|
|
@ -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) {
|
||||||
|
|
|
@ -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",
|
||||||
|
|
|
@ -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();
|
||||||
|
|
||||||
|
|
|
@ -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>(
|
||||||
|
|
|
@ -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();
|
||||||
|
|
|
@ -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",
|
||||||
|
|
|
@ -1,5 +0,0 @@
|
||||||
export function deadline(timeout: number): AbortSignal {
|
|
||||||
const controller = new AbortController();
|
|
||||||
setTimeout(() => controller.abort(), timeout);
|
|
||||||
return controller.signal;
|
|
||||||
}
|
|
2
main.ts
2
main.ts
|
@ -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(),
|
||||||
|
|
Loading…
Reference in New Issue