cloudbeaver/webapp/packages/plugin-sql-editor/src/SqlEditorGroupTabsBootstrap.ts

/*
 * CloudBeaver - Cloud Database Manager
 * Copyright (C) 2020-2025 DBeaver Corp and others
 *
 * Licensed under the Apache License, Version 2.0.
 * you may not use this file except in compliance with the License.
 */
import type { IDataContextProvider } from '@cloudbeaver/core-data-context';
import { Bootstrap, injectable } from '@cloudbeaver/core-di';
import { DATA_CONTEXT_TABS_CONTEXT, MENU_TAB } from '@cloudbeaver/core-ui';
import { ActionService, menuExtractItems, MenuService } from '@cloudbeaver/core-view';

import { ACTION_TAB_CLOSE_SQL_RESULT_GROUP } from './ACTION_TAB_CLOSE_SQL_RESULT_GROUP.js';
import { DATA_CONTEXT_SQL_EDITOR_STATE } from './DATA_CONTEXT_SQL_EDITOR_STATE.js';
import { DATA_CONTEXT_SQL_EDITOR_RESULT_ID } from './SqlResultTabs/DATA_CONTEXT_SQL_EDITOR_RESULT_ID.js';
import { SqlResultTabsService } from './SqlResultTabs/SqlResultTabsService.js';

@injectable(() => [ActionService, MenuService, SqlResultTabsService])
export class SqlEditorGroupTabsBootstrap extends Bootstrap {
  constructor(
    private readonly actionService: ActionService,
    private readonly menuService: MenuService,
    private readonly sqlResultTabsService: SqlResultTabsService,
  ) {
    super();
  }

  override register(): void {
    this.menuService.addCreator({
      menus: [MENU_TAB],
      contexts: [DATA_CONTEXT_SQL_EDITOR_RESULT_ID, DATA_CONTEXT_SQL_EDITOR_STATE],
      isApplicable: context => {
        const state = context.get(DATA_CONTEXT_TABS_CONTEXT);
        return !!state?.enabledBaseActions;
      },
      getItems: (context, items) => [...items, ACTION_TAB_CLOSE_SQL_RESULT_GROUP],
      orderItems: (context, items) => {
        items.push(...menuExtractItems(items, [ACTION_TAB_CLOSE_SQL_RESULT_GROUP]));
        return items;
      },
    });

    this.actionService.addHandler({
      id: 'result-tabs-group-base-handler',
      actions: [ACTION_TAB_CLOSE_SQL_RESULT_GROUP],
      menus: [MENU_TAB],
      contexts: [DATA_CONTEXT_SQL_EDITOR_RESULT_ID, DATA_CONTEXT_SQL_EDITOR_STATE, DATA_CONTEXT_TABS_CONTEXT],
      isActionApplicable: context => {
        const tab = context.get(DATA_CONTEXT_SQL_EDITOR_RESULT_ID)!;
        const sqlEditorState = context.get(DATA_CONTEXT_SQL_EDITOR_STATE)!;

        const groupId = sqlEditorState?.resultTabs.find(tabState => tabState.tabId === tab.id)?.groupId;
        const hasTabsInGroup = sqlEditorState.resultTabs.filter(tabState => tabState.groupId === groupId).length > 1;

        return hasTabsInGroup;
      },
      handler: async (context, action) => {
        switch (action) {
          case ACTION_TAB_CLOSE_SQL_RESULT_GROUP:
            this.closeResultTabGroup(context);
            break;
          default:
            break;
        }
      },
    });
  }

  async closeResultTabGroup(context: IDataContextProvider) {
    const tab = context.get(DATA_CONTEXT_SQL_EDITOR_RESULT_ID)!;
    const sqlEditorState = context.get(DATA_CONTEXT_SQL_EDITOR_STATE)!;
    const tabsContext = context.get(DATA_CONTEXT_TABS_CONTEXT)!;
    const resultTabs = this.sqlResultTabsService.getResultTabs(sqlEditorState);

    const resultTab = resultTabs.find(tabState => tabState.tabId === tab.id);
    const groupResultTabs = resultTabs.filter(tab => tab.groupId === resultTab?.groupId);

    groupResultTabs.forEach(groupTab => {
      tabsContext.close(groupTab.tabId);
    });
  }
}