Matthias Meister
ed08b976f9
- Introduced a new node type "video-prompt" for AI video generation, including its integration into the canvas command palette and node template picker. - Updated connection validation to allow connections from text nodes to video-prompt and from video-prompt to ai-video nodes. - Enhanced error handling and messaging for video generation failures, including specific cases for provider issues. - Added tests to validate new video-prompt functionality and connection policies. - Updated localization files to include new labels and prompts for video-prompt and ai-video nodes.
109 lines
3.2 KiB
TypeScript
109 lines
3.2 KiB
TypeScript
import { isAdjustmentNodeType } from "@/lib/canvas-node-types";
|
|
|
|
export const CANVAS_NODE_DND_MIME = "application/lemonspace-node-type";
|
|
|
|
const ADJUSTMENT_ALLOWED_SOURCE_TYPES = new Set<string>([
|
|
"image",
|
|
"asset",
|
|
"ai-image",
|
|
"curves",
|
|
"color-adjust",
|
|
"light-adjust",
|
|
"detail-adjust",
|
|
]);
|
|
|
|
const RENDER_ALLOWED_SOURCE_TYPES = new Set<string>([
|
|
"image",
|
|
"asset",
|
|
"ai-image",
|
|
"curves",
|
|
"color-adjust",
|
|
"light-adjust",
|
|
"detail-adjust",
|
|
]);
|
|
|
|
const ADJUSTMENT_DISALLOWED_TARGET_TYPES = new Set<string>(["prompt", "ai-image"]);
|
|
|
|
export type CanvasConnectionValidationReason =
|
|
| "incomplete"
|
|
| "self-loop"
|
|
| "unknown-node"
|
|
| "ai-video-source-invalid"
|
|
| "video-prompt-target-invalid"
|
|
| "adjustment-source-invalid"
|
|
| "adjustment-incoming-limit"
|
|
| "compare-incoming-limit"
|
|
| "adjustment-target-forbidden"
|
|
| "render-source-invalid";
|
|
|
|
export function validateCanvasConnectionPolicy(args: {
|
|
sourceType: string;
|
|
targetType: string;
|
|
targetIncomingCount: number;
|
|
}): CanvasConnectionValidationReason | null {
|
|
const { sourceType, targetType, targetIncomingCount } = args;
|
|
|
|
if (targetType === "ai-video" && sourceType !== "video-prompt") {
|
|
return "ai-video-source-invalid";
|
|
}
|
|
|
|
if (sourceType === "video-prompt" && targetType !== "ai-video") {
|
|
return "video-prompt-target-invalid";
|
|
}
|
|
|
|
if (targetType === "render" && !RENDER_ALLOWED_SOURCE_TYPES.has(sourceType)) {
|
|
return "render-source-invalid";
|
|
}
|
|
|
|
if (isAdjustmentNodeType(targetType) && targetType !== "render") {
|
|
if (!ADJUSTMENT_ALLOWED_SOURCE_TYPES.has(sourceType)) {
|
|
return "adjustment-source-invalid";
|
|
}
|
|
if (targetIncomingCount >= 1) {
|
|
return "adjustment-incoming-limit";
|
|
}
|
|
}
|
|
|
|
if (targetType === "compare" && targetIncomingCount >= 2) {
|
|
return "compare-incoming-limit";
|
|
}
|
|
|
|
if (
|
|
isAdjustmentNodeType(sourceType) &&
|
|
ADJUSTMENT_DISALLOWED_TARGET_TYPES.has(targetType)
|
|
) {
|
|
return "adjustment-target-forbidden";
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
export function getCanvasConnectionValidationMessage(
|
|
reason: CanvasConnectionValidationReason,
|
|
): string {
|
|
switch (reason) {
|
|
case "incomplete":
|
|
return "Unvollstaendige Verbindung.";
|
|
case "self-loop":
|
|
return "Node kann nicht mit sich selbst verbunden werden.";
|
|
case "unknown-node":
|
|
return "Verbindung enthaelt unbekannte Nodes.";
|
|
case "ai-video-source-invalid":
|
|
return "KI-Video-Ausgabe akzeptiert nur Eingaben von KI-Video.";
|
|
case "video-prompt-target-invalid":
|
|
return "KI-Video kann nur mit KI-Video-Ausgabe verbunden werden.";
|
|
case "adjustment-source-invalid":
|
|
return "Adjustment-Nodes akzeptieren nur Bild-, Asset-, KI-Bild- oder Adjustment-Input.";
|
|
case "adjustment-incoming-limit":
|
|
return "Adjustment-Nodes erlauben genau eine eingehende Verbindung.";
|
|
case "compare-incoming-limit":
|
|
return "Compare-Nodes erlauben genau zwei eingehende Verbindungen.";
|
|
case "adjustment-target-forbidden":
|
|
return "Adjustment-Ausgaben koennen nicht an Prompt- oder KI-Bild-Nodes angeschlossen werden.";
|
|
case "render-source-invalid":
|
|
return "Render akzeptiert nur Bild-, Asset-, KI-Bild- oder Adjustment-Input.";
|
|
default:
|
|
return "Verbindung ist fuer diese Node-Typen nicht erlaubt.";
|
|
}
|
|
}
|