43 lines
2.8 KiB
TypeScript
43 lines
2.8 KiB
TypeScript
import {filterLoraStackForUse} from '~/utils/loras';
|
|
import { validateRequest } from '~/shared/studio2/contracts.mjs';
|
|
import { resolveRequestSize } from '../../utils/studio2/size';
|
|
import { parsePostedLoraStack, assertImageV2LoraStack } from '../../utils/loras';
|
|
import { saveRecord, readRecord } from '../../utils/studio2/store';
|
|
import { addStudioJob, kickStudioQueue, type StudioJobPayload } from '../../utils/studioQueue';
|
|
import { isXaigenStudio } from '~/utils/videoModels';
|
|
export default defineEventHandler(async (event) => {
|
|
const { owner } = assertLibraryOwner(event), raw = await readBody(event);
|
|
const xaigen = isXaigenStudio();
|
|
let request: any;
|
|
try {
|
|
request = validateRequest(raw, xaigen);
|
|
}
|
|
catch (e: any) {
|
|
throw createError({ statusCode: e.statusCode || 400, statusMessage: e.message });
|
|
}
|
|
const video = ['video', 'extend'].includes(request.mode);
|
|
request.settings.loraStack = parsePostedLoraStack(request.settings.loraStack, video ? 'video' : 'image');
|
|
request.settings.loraStack = filterLoraStackForUse(request.settings.loraStack, request.engine, xaigen);
|
|
if (!video && request.engine !== 'qwen21')
|
|
assertImageV2LoraStack(request.settings.loraStack, request.engine);
|
|
assertFolderAccess(event, request.folderId);
|
|
for (const id of [request.identityStillId, request.imageAId, request.imageBId, request.maskId, request.endStillId, ...request.guides.map((g: any) => g.stillId)].filter(Boolean))
|
|
assertFolderAccess(event, getStill(owner, id).folderId);
|
|
if (request.startClipId)
|
|
assertFolderAccess(event, getClip(owner, request.startClipId).folderId);
|
|
// Header-only size probe — never decode the archive PNG on Generate click
|
|
await resolveRequestSize(owner,request);
|
|
const id = crypto.randomUUID(), record = { id, owner, request, state: 'waiting', queuedAt: Date.now(), familyId: crypto.randomUUID(), outputs: [], purgeResult: 'Not yet saved' };
|
|
saveRecord(record);
|
|
const ack = { id, seed:request.settings.seed, width:request.settings.width, height:request.settings.height };
|
|
// Enqueue + kick after the client has the id — do not block the response on the worker wake path
|
|
void (async () => {
|
|
try {
|
|
const row = await addStudioJob({ ownerKey: owner, kind: ['video', 'extend'].includes(request.mode) ? 'video' : 'edit', familyId: record.familyId, payload: { ...request.settings, studio2Id: id, studio2Request: structuredClone(request), prompt: request.compiledPrompt, folderId: request.folderId, extensions: [], referenceStillIds: [], useIdentityRefs: false, queueAutoRun: true } as StudioJobPayload });
|
|
saveRecord({ ...readRecord(owner, id), queueId: row.id });
|
|
} catch { /* poll / resume will pick up waiting studio2 records */ }
|
|
kickStudioQueue();
|
|
})();
|
|
return ack;
|
|
});
|