cloudbeaver/webapp/packages/plugin-devtools/src/DevToolsService.ts

/*
 * CloudBeaver - Cloud Database Manager
 * Copyright (C) 2020-2024 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 { makeObservable, observable } from 'mobx';

import { injectable } from '@cloudbeaver/core-di';
import { ServerConfigResource } from '@cloudbeaver/core-root';
import { StorageService } from '@cloudbeaver/core-storage';

interface IDevToolsSettings {
  enabled: boolean;
  distributed: boolean;
  configuration: boolean;
}

const DEVTOOLS = 'devtools';

@injectable()
export class DevToolsService {
  get isEnabled(): boolean {
    return this.settings.enabled;
  }

  get isDistributed(): boolean {
    return this.settings.distributed;
  }

  get isConfiguration(): boolean {
    return this.settings.configuration;
  }

  private readonly settings: IDevToolsSettings;

  constructor(
    private readonly serverConfigResource: ServerConfigResource,
    private readonly storageService: StorageService,
  ) {
    this.settings = getDefaultDevToolsSettings();

    makeObservable<this, 'settings'>(this, {
      settings: observable,
    });
    this.storageService.registerSettings(DEVTOOLS, this.settings, getDefaultDevToolsSettings);
    this.serverConfigResource.onDataUpdate.addHandler(this.syncSettingsOverride.bind(this));
  }

  switch() {
    this.settings.enabled = !this.settings.enabled;
    this.syncSettingsOverride();
  }

  setDistributedMode(distributed: boolean) {
    this.settings.distributed = distributed;
    this.syncSettingsOverride();
  }

  setConfigurationMode(configuration: boolean) {
    this.settings.configuration = configuration;
    this.syncSettingsOverride();
  }

  private syncSettingsOverride() {
    if (this.isEnabled) {
      if (this.serverConfigResource.data) {
        this.serverConfigResource.data.distributed = this.isDistributed;
        this.serverConfigResource.data.configurationMode = this.isConfiguration;
      }
    }
  }
}

function getDefaultDevToolsSettings(): IDevToolsSettings {
  return {
    enabled: process.env.NODE_ENV === 'development',
    distributed: false,
    configuration: false,
  };
}