webapp/packages/plugin-sql-generator/src/SqlGenerators/SqlGeneratorsBootstrap.ts - devel - vshmidt/cloudbeaver

webapp/packages/plugin-sql-generator/src/SqlGenerators/SqlGeneratorsBootstrap.ts
100 строк4 KB
sergeyteleshev
[CB] SQL Query generation: front-end part (#4132)
11 фев 2026, 20:16
11 фев 2026, 20:162719f53
/*
 * CloudBeaver - Cloud Database Manager
 * Copyright (C) 2020-2026 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 { importLazyComponent } from '@cloudbeaver/core-blocks';
import { Bootstrap, injectable } from '@cloudbeaver/core-di';
import { CommonDialogService } from '@cloudbeaver/core-dialogs';
import { DATA_CONTEXT_NAV_NODE, EObjectFeature } from '@cloudbeaver/core-navigation-tree';
import { getCachedMapResourceLoaderState } from '@cloudbeaver/core-resource';
import { MenuBaseItem, MenuService } from '@cloudbeaver/core-view';

import { MENU_SQL_GENERATORS } from './MENU_SQL_GENERATORS.js';
import { SqlGeneratorsResource } from './SqlGeneratorsResource.js';
import { NotificationService } from '@cloudbeaver/core-events';

const GeneratedSqlDialog = importLazyComponent(() => import('./GeneratedSqlDialog.js').then(m => m.GeneratedSqlDialog));

@injectable(() => [SqlGeneratorsResource, CommonDialogService, MenuService, NotificationService])
export class SqlGeneratorsBootstrap extends Bootstrap {
  constructor(
    private readonly sqlGeneratorsResource: SqlGeneratorsResource,
    private readonly commonDialogService: CommonDialogService,
    private readonly menuService: MenuService,
    private readonly notificationService: NotificationService,
  ) {
    super();
  }

  override register(): void {
    this.menuService.setHandler({
      id: 'node-sql-generators',
      menus: [MENU_SQL_GENERATORS],
      contexts: [DATA_CONTEXT_NAV_NODE],
      isDisabled: context => {
        const node = context.get(DATA_CONTEXT_NAV_NODE)!;

        return this.sqlGeneratorsResource.get(node.id)?.length === 0;
      },
      getLoader: (context, action) => {
        const node = context.get(DATA_CONTEXT_NAV_NODE)!;

        return getCachedMapResourceLoaderState(this.sqlGeneratorsResource, () => node.id);
      },
    });
    this.menuService.addCreator({
      root: true,
      contexts: [DATA_CONTEXT_NAV_NODE],
      isApplicable: context => {
        const node = context.get(DATA_CONTEXT_NAV_NODE)!;

        if (!(node.objectFeatures.includes(EObjectFeature.entity) || node.objectFeatures.includes(EObjectFeature.script))) {
          return false;
        }

        return true;
      },
      getItems: (context, items) => [...items, MENU_SQL_GENERATORS],
    });

    this.menuService.addCreator({
      menus: [MENU_SQL_GENERATORS],
      contexts: [DATA_CONTEXT_NAV_NODE],
      getItems: (context, items) => {
        const node = context.get(DATA_CONTEXT_NAV_NODE)!;

        const actions = this.sqlGeneratorsResource.get(node.id) || [];

        return [
          ...items,
          ...actions.map(
            action =>
              new MenuBaseItem(
                {
                  id: action.id,
                  label: action.label,
                  tooltip: action.description,
                },
                {
                  onSelect: async () => {
                    try {
                      const query = await this.sqlGeneratorsResource.generateEntityQuery(action.id, node.id);
                      await this.commonDialogService.open(GeneratedSqlDialog, {
                        query,
                        nodeId: node.id,
                      });
                    } catch (e: any) {
                      this.notificationService.logException(e, 'sql_generator_error_title');
                    }
                  },
                },
              ),
          ),
        ];
      },
    });
  }
}