dbgate/packages/web/src/commands/registerCommand.ts

import { commands } from '../stores';
import { invalidateCommandDefinitions } from './invalidateCommands';

export interface SubCommand {
  text: string;
  onClick: Function;
}

export interface GlobalCommand {
  id: string;
  category: string; // null for group commands
  isGroupCommand?: boolean;
  name: string;
  text?: string /* category: name */;
  keyText?: string;
  keyTextFromGroup?: string; // automatically filled from group
  group?: string;
  getSubCommands?: () => SubCommand[];
  onClick?: Function;
  testEnabled?: () => boolean;
  // enabledStore?: any;
  icon?: string;
  toolbar?: boolean;
  enabled?: boolean;
  showDisabled?: boolean;
  toolbarName?: string;
  menuName?: string;
  toolbarOrder?: number;
  disableHandleKeyText?: string;
  isRelatedToTab?: boolean,
}

export default function registerCommand(command: GlobalCommand) {
  const { testEnabled } = command;
  commands.update(x => {
    if (x[command.id]) {
      console.error(`Command ${command.id} already registered`);
      return x;
    }
    return {
      ...x,
      [command.id]: {
        text: `${command.category}: ${command.name}`,
        ...command,
        enabled: !testEnabled,
      },
    };
  });
  invalidateCommandDefinitions();
}