cloudbeaver/webapp/packages/data-export-plugin/src/ExportNotification/ExportNotificationController.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 { observable, computed } from 'mobx';

import { ErrorDetailsDialog, NodesManagerService } from '@dbeaver/core/app';
import { IInitializableController, injectable } from '@dbeaver/core/di';
import { CommonDialogService } from '@dbeaver/core/dialogs';
import { INotification } from '@dbeaver/core/eventsLog';
import { LocalizationService } from '@dbeaver/core/localization';
import { EDeferredState } from '@dbeaver/core/utils';

import { DataExportProcessService } from '../DataExportProcessService';

@injectable()
export class ExportNotificationController implements IInitializableController {
  @observable isDetailsDialogOpen = false;

  get isSuccess() {
    return this.process?.getState() === EDeferredState.RESOLVED;
  }

  get isPending() {
    return this.process?.isInProgress;
  }

  get process() {
    return this.task?.process;
  }

  get task() {
    return this.dataExportProcessService.exportProcesses.get(this.notification.source);
  }

  get hasDetails() {
    return !!this.process?.getRejectionReason();
  }

  @computed get sourceName() {
    if (!this.task) {
      return '';
    }
    if (this.task.context.containerNodePath) {
      const node = this.nodesManagerService.getNode(this.task.context.containerNodePath);

      return `${this.localization.translate('data_transfer_exporting_table')} ${node?.name}`;
    }

    return this.localization.translate('data_transfer_exporting_sql');
  }

  get status() {
    switch (this.process?.getState()) {
      case EDeferredState.PENDING:
        return 'data_transfer_notification_preparation';
      case EDeferredState.CANCELLING:
        return 'ui_processing_canceling';
      case EDeferredState.RESOLVED:
        return 'data_transfer_notification_ready';
      default:
        return 'data_transfer_notification_error';
    }
  }

  get downloadUrl() {
    return this.dataExportProcessService.downloadUrl(this.notification.source);
  }

  private notification!: INotification<any>;

  constructor(
    private commonDialogService: CommonDialogService,
    private dataExportProcessService: DataExportProcessService,
    private nodesManagerService: NodesManagerService,
    private localization: LocalizationService,
  ) {}

  init(notification: INotification<any>) {
    this.notification = notification;
  }

  delete = () => {
    this.dataExportProcessService.delete(this.notification.source);
    this.notification.close();
  }

  download = () => {
    this.dataExportProcessService.download(this.notification.source);
    this.notification.close();
  }

  cancel = () => {
    this.dataExportProcessService.cancel(this.notification.source);
  }

  showDetails= async () => {
    this.isDetailsDialogOpen = true;
    try {
      this.notification.showDetails();
      await this.commonDialogService.open(ErrorDetailsDialog, this.process?.getRejectionReason());
    } finally {
      this.isDetailsDialogOpen = false;
    }
  }
}