NetMesh/components/vault/useRegisterVaultHostTreeActions.ts

import { useEffect } from 'react';

import { activeTabStore } from '../../application/state/activeTabStore';
import {
  vaultHostTreeActionsStore,
  type VaultHostTreeActions,
} from '../../application/state/vaultHostTreeActionsStore';
import type { Host } from '../../types';
import type { VaultOrderPosition } from '../../domain/vaultOrder';

type RegisterVaultHostTreeActionsParams = {
  handleCopyCredentials: (host: Host) => void;
  handleCopyHostname?: (host: Host) => void;
  handleDuplicateHost: (host: Host) => void;
  startInlineRenameHost: (host: Host) => void;
  onDeleteHost: (hostId: string) => void;
  handleUnmanageGroup?: (groupPath: string) => void;
  moveHostToGroup: (hostId: string, groupPath: string | null) => void;
  moveGroup: (sourcePath: string, targetParent: string | null) => void;
  reorderHost: (sourceHostId: string, targetHostId: string, position: VaultOrderPosition) => void;
  reorderGroup: (sourcePath: string, targetPath: string, position: VaultOrderPosition) => boolean;
  managedGroupPaths?: Set<string>;
  startInlineNewGroup: (parentPath?: string) => void;
  startInlineRenameGroup: (groupPath: string) => void;
  startInlineDeleteGroup: (groupPath: string) => void;
  commitInlineGroupRename: (name: string) => boolean | void | Promise<boolean | void>;
  cancelInlineGroupEdit: () => void;
  commitInlineHostRename: (name: string) => void;
  cancelInlineHostEdit: () => void;
};

function focusVaultTab() {
  activeTabStore.setActiveTabId('vault');
}

function withVaultFocus<T extends (...args: never[]) => void>(fn: T): T {
  return ((...args: Parameters<T>) => {
    focusVaultTab();
    fn(...args);
  }) as T;
}

export function useRegisterVaultHostTreeActions({
  handleCopyCredentials,
  handleCopyHostname,
  handleDuplicateHost,
  startInlineRenameHost,
  onDeleteHost,
  handleUnmanageGroup,
  moveHostToGroup,
  moveGroup,
  reorderHost,
  reorderGroup,
  managedGroupPaths,
  startInlineNewGroup,
  startInlineRenameGroup,
  startInlineDeleteGroup,
  commitInlineGroupRename,
  cancelInlineGroupEdit,
  commitInlineHostRename,
  cancelInlineHostEdit,
}: RegisterVaultHostTreeActionsParams) {
  useEffect(() => {
    const actions: VaultHostTreeActions = {
      onCopyCredentials: handleCopyCredentials,
      onCopyHostname: handleCopyHostname,
      onDuplicateHost: withVaultFocus(handleDuplicateHost),
      onRenameHost: startInlineRenameHost,
      onDeleteHost: (host) => onDeleteHost(host.id),
      onNewGroup: startInlineNewGroup,
      onRenameGroup: startInlineRenameGroup,
      onDeleteGroup: startInlineDeleteGroup,
      commitInlineGroupRename,
      cancelInlineGroupEdit,
      commitInlineHostRename,
      cancelInlineHostEdit,
      moveHostToGroup,
      moveGroup,
      reorderHost,
      reorderGroup,
      managedGroupPaths,
      onUnmanageGroup: handleUnmanageGroup
        ? withVaultFocus(handleUnmanageGroup)
        : undefined,
    };

    vaultHostTreeActionsStore.setActions(actions);
    return () => vaultHostTreeActionsStore.setActions(null);
  }, [
    cancelInlineGroupEdit,
    cancelInlineHostEdit,
    commitInlineGroupRename,
    commitInlineHostRename,
    handleCopyCredentials,
    handleCopyHostname,
    handleDuplicateHost,
    handleUnmanageGroup,
    managedGroupPaths,
    moveGroup,
    moveHostToGroup,
    reorderGroup,
    reorderHost,
    onDeleteHost,
    startInlineRenameHost,
    startInlineDeleteGroup,
    startInlineNewGroup,
    startInlineRenameGroup,
  ]);
}