"use client"; import { useCallback, useMemo, useState } from "react"; import { Handle, Position, type Node, type NodeProps } from "@xyflow/react"; import { useMutation } from "convex/react"; import { useTranslations } from "next-intl"; import { Sun } from "lucide-react"; import { api } from "@/convex/_generated/api"; import type { Id } from "@/convex/_generated/dataModel"; import { useCanvasAdjustmentPresets } from "@/components/canvas/canvas-presets-context"; import { useCanvasSync } from "@/components/canvas/canvas-sync-context"; import BaseNodeWrapper from "@/components/canvas/nodes/base-node-wrapper"; import AdjustmentPreview from "@/components/canvas/nodes/adjustment-preview"; import { useNodeLocalData } from "@/components/canvas/nodes/use-node-local-data"; import { ParameterSlider, type SliderConfig, type SliderValue, } from "@/src/components/tool-ui/parameter-slider"; import { cloneAdjustmentData, DEFAULT_LIGHT_ADJUST_DATA, normalizeLightAdjustData, type LightAdjustData, } from "@/lib/image-pipeline/adjustment-types"; import { LIGHT_PRESETS } from "@/lib/image-pipeline/presets"; import { Select, SelectContent, SelectItem, SelectTrigger, SelectValue } from "@/components/ui/select"; import { toast } from "@/lib/toast"; type LightAdjustNodeData = LightAdjustData & { _status?: string; _statusMessage?: string; }; export type LightAdjustNodeType = Node; type PresetDoc = { _id: Id<"adjustmentPresets">; name: string; params: unknown; }; export default function LightAdjustNode({ id, data, selected, width }: NodeProps) { const tCommon = useTranslations("common"); const tNodes = useTranslations("nodes"); const tToasts = useTranslations("toasts"); const { queueNodeDataUpdate } = useCanvasSync(); const savePreset = useMutation(api.presets.save); const userPresets = useCanvasAdjustmentPresets("light-adjust") as PresetDoc[]; const [presetSelection, setPresetSelection] = useState("custom"); const normalizeData = useCallback( (value: unknown) => normalizeLightAdjustData({ ...cloneAdjustmentData(DEFAULT_LIGHT_ADJUST_DATA), ...(value as Record), }), [], ); const { localData, applyLocalData, updateLocalData } = useNodeLocalData({ data, normalize: normalizeData, saveDelayMs: 16, onSave: (next) => queueNodeDataUpdate({ nodeId: id as Id<"nodes">, data: next, }), debugLabel: "light-adjust", }); const updateData = (updater: (draft: LightAdjustData) => LightAdjustData) => { setPresetSelection("custom"); updateLocalData(updater); }; const builtinOptions = useMemo(() => Object.entries(LIGHT_PRESETS), []); const sliderConfigs = useMemo( () => [ { id: "brightness", label: tNodes("adjustments.lightAdjust.sliders.brightness"), min: -100, max: 100, value: DEFAULT_LIGHT_ADJUST_DATA.brightness, }, { id: "contrast", label: tNodes("adjustments.lightAdjust.sliders.contrast"), min: -100, max: 100, value: DEFAULT_LIGHT_ADJUST_DATA.contrast, }, { id: "exposure", label: tNodes("adjustments.lightAdjust.sliders.exposure"), min: -5, max: 5, step: 0.01, precision: 2, value: DEFAULT_LIGHT_ADJUST_DATA.exposure, }, { id: "highlights", label: tNodes("adjustments.lightAdjust.sliders.highlights"), min: -100, max: 100, value: DEFAULT_LIGHT_ADJUST_DATA.highlights, }, { id: "shadows", label: tNodes("adjustments.lightAdjust.sliders.shadows"), min: -100, max: 100, value: DEFAULT_LIGHT_ADJUST_DATA.shadows, }, { id: "whites", label: tNodes("adjustments.lightAdjust.sliders.whites"), min: -100, max: 100, value: DEFAULT_LIGHT_ADJUST_DATA.whites, }, { id: "blacks", label: tNodes("adjustments.lightAdjust.sliders.blacks"), min: -100, max: 100, value: DEFAULT_LIGHT_ADJUST_DATA.blacks, }, { id: "vignette-amount", label: tNodes("adjustments.lightAdjust.sliders.vignette"), min: 0, max: 1, step: 0.01, precision: 2, value: DEFAULT_LIGHT_ADJUST_DATA.vignette.amount, }, ], [tNodes], ); const sliderValues = useMemo( () => [ { id: "brightness", value: localData.brightness }, { id: "contrast", value: localData.contrast }, { id: "exposure", value: localData.exposure }, { id: "highlights", value: localData.highlights }, { id: "shadows", value: localData.shadows }, { id: "whites", value: localData.whites }, { id: "blacks", value: localData.blacks }, { id: "vignette-amount", value: localData.vignette.amount }, ], [ localData.blacks, localData.brightness, localData.contrast, localData.exposure, localData.highlights, localData.shadows, localData.vignette.amount, localData.whites, ], ); const applyPresetValue = (value: string) => { if (value === "custom") { setPresetSelection("custom"); return; } if (value.startsWith("builtin:")) { const key = value.replace("builtin:", ""); const preset = LIGHT_PRESETS[key]; if (!preset) return; const next = cloneAdjustmentData(preset); setPresetSelection(value); applyLocalData(next); return; } if (value.startsWith("user:")) { const presetId = value.replace("user:", "") as Id<"adjustmentPresets">; const preset = userPresets.find((entry) => entry._id === presetId); if (!preset) return; const next = normalizeLightAdjustData(preset.params); setPresetSelection(value); applyLocalData(next); } }; const handleSavePreset = async () => { const name = window.prompt(tNodes("adjustments.common.presetNamePrompt")); if (!name) return; await savePreset({ name, nodeType: "light-adjust", params: localData, }); toast.success(tToasts("canvas.adjustmentPresetSaved")); }; return (
{tNodes("adjustments.lightAdjust.title")}
{ const valueById = new Map(values.map((entry) => [entry.id, entry.value])); updateData((current) => ({ ...current, brightness: valueById.get("brightness") ?? current.brightness, contrast: valueById.get("contrast") ?? current.contrast, exposure: valueById.get("exposure") ?? current.exposure, highlights: valueById.get("highlights") ?? current.highlights, shadows: valueById.get("shadows") ?? current.shadows, whites: valueById.get("whites") ?? current.whites, blacks: valueById.get("blacks") ?? current.blacks, vignette: { ...current.vignette, amount: valueById.get("vignette-amount") ?? current.vignette.amount, }, preset: null, })); }} />
); }