cloudbeaver/webapp/packages/basic-connection-plugin/src/BasicConnectionPluginBootstrap.ts

/*
 * cloudbeaver - Cloud Database Manager
 * Copyright (C) 2020 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 { MainMenuService, ConnectionDialogsService } from '@dbeaver/core/app';
import { injectable } from '@dbeaver/core/di';
import { CommonDialogService } from '@dbeaver/core/dialogs';
import { NotificationService } from '@dbeaver/core/eventsLog';
import { PermissionsService, EPermission } from '@dbeaver/core/root';

import { BasicConnectionService } from './BasicConnectionService';
import { ConnectionDialog } from './ConnectionDialog/ConnectionDialog';

@injectable()
export class BasicConnectionPluginBootstrap {

  constructor(
    private connectionDialogsService: ConnectionDialogsService,
    private mainMenuService: MainMenuService,
    private basicConnectionService: BasicConnectionService,
    private commonDialogService: CommonDialogService,
    private notificationService: NotificationService,
    private permissionsService: PermissionsService
  ) {
  }

  bootstrap() {
    this.loadDbSources();
    this.mainMenuService.registerMenuItem(
      this.connectionDialogsService.newConnectionMenuToken,
      {
        id: 'mainMenuConnect',
        order: 2,
        title: 'basicConnection_main_menu_item',
        onClick: () => this.openConnectionsDialog(),
        isHidden: () => !this.permissionsService.has(EPermission.public),
        isDisabled: () => !this.basicConnectionService.dbSources.data.length,
      }
    );
  }

  private async openConnectionsDialog() {
    await this.commonDialogService.open(ConnectionDialog, null);
  }

  private async loadDbSources() {
    try {
      await this.basicConnectionService.dbSources.load();
    } catch (error) {
      this.notificationService.logException(error, 'DBSources loading failed');
    }
  }
}