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

import type { StoreApi } from "zustand";
import type { AppState } from "@/lib/state/store";
import type { WsHandlers } from "@/lib/ws/handlers/types";
import type { TaskMRAutomationOptions } from "@/lib/types/gitlab";

export function registerGitLabHandlers(store: StoreApi<AppState>): WsHandlers {
  return {
    "gitlab.task_mr.updated": (message) => {
      const mr = message.payload;
      const activeWorkspaceId = store.getState().workspaces.activeId;
      if (!mr.task_id || !mr.workspace_id || mr.workspace_id !== activeWorkspaceId) return;
      store.getState().setTaskMR(mr.workspace_id, mr.task_id, mr);
    },
    "gitlab.task_mr.deleted": (message) => {
      const deleted = message.payload;
      const activeWorkspaceId = store.getState().workspaces.activeId;
      if (
        !deleted.task_id ||
        !deleted.association_id ||
        !deleted.workspace_id ||
        deleted.workspace_id !== activeWorkspaceId
      ) {
        return;
      }
      store.getState().removeTaskMR(deleted.workspace_id, deleted.association_id);
    },
    "gitlab.task_mr_options.updated": (message) => {
      const options = message.payload as TaskMRAutomationOptions;
      if (options.task_id) {
        const current = store.getState().taskMRAutomation.byTaskId[options.task_id];
        if (current && (options.automation_revision ?? 0) < (current.automation_revision ?? 0)) {
          return;
        }
        store.getState().setTaskMRAutomationOptions(options.task_id, options);
        // Marks this write as externally-sourced so a slower in-flight local
        // refresh()/update() in useTaskMRAutomationOptions knows not to
        // overwrite it with a stale response.
        store.getState().markTaskMRAutomationExternalUpdate(options.task_id);
      }
    },
  };
}