kandev/apps/web/lib/ws/handlers/kanban.ts

import type { StoreApi } from 'zustand';
import type { AppState } from '@/lib/state/store';
import type { WsHandlers } from '@/lib/ws/handlers/types';
import type { KanbanState } from '@/lib/state/slices/kanban/types';

type KanbanTask = KanbanState['tasks'][number];
type KanbanStep = KanbanState['steps'][number];

export function registerKanbanHandlers(store: StoreApi<AppState>): WsHandlers {
  return {
    'kanban.update': (message) => {
      const workflowId = message.payload.workflowId;
      // eslint-disable-next-line @typescript-eslint/no-explicit-any
      const steps: KanbanStep[] = message.payload.steps.map((step: any, index: number) => ({
        id: step.id,
        title: step.title,
        color: step.color ?? 'bg-neutral-400',
        position: step.position ?? index,
        events: step.events,
      }));
      // eslint-disable-next-line @typescript-eslint/no-explicit-any
      const tasks: KanbanTask[] = message.payload.tasks.map((task: any) => ({
        id: task.id,
        workflowStepId: task.workflowStepId,
        title: task.title,
        description: task.description,
        position: task.position ?? 0,
        state: task.state,
      }));

      store.setState((state) => {
        const next = {
          ...state,
          kanban: { workflowId, steps, tasks },
        };

        // Also update multi-workflow snapshots if this workflow is tracked
        const snapshot = state.kanbanMulti.snapshots[workflowId];
        if (snapshot) {
          return {
            ...next,
            kanbanMulti: {
              ...next.kanbanMulti,
              snapshots: {
                ...next.kanbanMulti.snapshots,
                [workflowId]: { ...snapshot, steps, tasks },
              },
            },
          };
        }

        return next;
      });
    },
  };
}