Migrate ColorPlatte to ts (#31)

* rename

* Migrate colorPlatte to ts
This commit is contained in:
Chenlei Hu
2024-06-18 14:22:11 -04:00
committed by GitHub
parent 681bb69217
commit f22bcdf2b2
2 changed files with 103 additions and 10 deletions

85
src/types/colorPalette.ts Normal file
View File

@@ -0,0 +1,85 @@
import { z } from 'zod';
const nodeSlotSchema = z.object({
BOOLEAN: z.string().optional(),
CLIP: z.string(),
CLIP_VISION: z.string(),
CLIP_VISION_OUTPUT: z.string(),
CONDITIONING: z.string(),
CONTROL_NET: z.string(),
CONTROL_NET_WEIGHTS: z.string().optional(),
FLOAT: z.string().optional(),
GLIGEN: z.string().optional(),
IMAGE: z.string(),
IMAGEUPLOAD: z.string().optional(),
INT: z.string().optional(),
LATENT: z.string(),
LATENT_KEYFRAME: z.string().optional(),
MASK: z.string(),
MODEL: z.string(),
SAMPLER: z.string().optional(),
SIGMAS: z.string().optional(),
STRING: z.string().optional(),
STYLE_MODEL: z.string(),
T2I_ADAPTER_WEIGHTS: z.string().optional(),
TAESD: z.string(),
TIMESTEP_KEYFRAME: z.string().optional(),
UPSCALE_MODEL: z.string().optional(),
VAE: z.string()
}).passthrough();
const litegraphBaseSchema = z.object({
BACKGROUND_IMAGE: z.string(),
CLEAR_BACKGROUND_COLOR: z.string(),
NODE_TITLE_COLOR: z.string(),
NODE_SELECTED_TITLE_COLOR: z.string(),
NODE_TEXT_SIZE: z.number(),
NODE_TEXT_COLOR: z.string(),
NODE_SUBTEXT_SIZE: z.number(),
NODE_DEFAULT_COLOR: z.string(),
NODE_DEFAULT_BGCOLOR: z.string(),
NODE_DEFAULT_BOXCOLOR: z.string(),
NODE_DEFAULT_SHAPE: z.string(),
NODE_BOX_OUTLINE_COLOR: z.string(),
DEFAULT_SHADOW_COLOR: z.string(),
DEFAULT_GROUP_FONT: z.number(),
WIDGET_BGCOLOR: z.string(),
WIDGET_OUTLINE_COLOR: z.string(),
WIDGET_TEXT_COLOR: z.string(),
WIDGET_SECONDARY_TEXT_COLOR: z.string(),
LINK_COLOR: z.string(),
EVENT_LINK_COLOR: z.string(),
CONNECTING_LINK_COLOR: z.string()
}).passthrough();
const comfyBaseSchema = z.object({
["fg-color"]: z.string(),
["bg-color"]: z.string(),
["comfy-menu-bg"]: z.string(),
["comfy-input-bg"]: z.string(),
["input-text"]: z.string(),
["descrip-text"]: z.string(),
["drag-text"]: z.string(),
["error-text"]: z.string(),
["border-color"]: z.string(),
["tr-even-bg-color"]: z.string(),
["tr-odd-bg-color"]: z.string(),
});
const colorsSchema = z.object({
node_slot: nodeSlotSchema,
litegraph_base: litegraphBaseSchema,
comfy_base: comfyBaseSchema
}).passthrough();
const paletteSchema = z.object({
id: z.string(),
name: z.string(),
colors: colorsSchema
});
const colorPalettesSchema = z.record(paletteSchema);
export type Colors = z.infer<typeof colorsSchema>;
export type Palette = z.infer<typeof paletteSchema>;
export type ColorPalettes = z.infer<typeof colorPalettesSchema>;