Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(ui): better types for validateConnection

Browse Source
This commit is contained in:
psychedelicious 2024-05-19 01:10:49 +10:00
parent 972398d203
commit 78f9f3ee95
1 changed files with 24 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
invokeai/frontend/web/src/features/nodes/store/util/validateConnection.ts
View File

@ -6,13 +6,19 @@ import { validateConnectionTypes } from 'features/nodes/store/util/validateConne
import type { AnyNode } from 'features/nodes/types/invocation';
import type { Connection as NullableConnection, Edge } from 'reactflow';
import type { O } from 'ts-toolbelt';
import { assert } from 'tsafe';
type Connection = O.NonNullable<NullableConnection>;
export type ValidateConnectionResult = {
isValid: boolean;
export type ValidateConnectionResult =
| {
isValid: true;
messageTKey?: string;
};
}
| {
isValid: false;
messageTKey: string;
};
export type ValidateConnectionFunc = (
connection: Connection,
@ -22,10 +28,20 @@ export type ValidateConnectionFunc = (
ignoreEdge: Edge | null
) => ValidateConnectionResult;
export const buildResult = (isValid: boolean, messageTKey?: string): ValidateConnectionResult => ({
export const buildResult = (isValid: boolean, messageTKey?: string): ValidateConnectionResult => {
if (isValid) {
return {
isValid,
messageTKey,
});
};
} else {
assert(messageTKey !== undefined);
return {
isValid,
messageTKey,
};
}
};
const getEqualityPredicate =
(c: Connection) =>