cloudbeaver/webapp/packages/core-root/src/ServerConfigResource.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 { action, makeObservable, observable } from 'mobx';

import { injectable } from '@cloudbeaver/core-di';
import { ExecutorInterrupter } from '@cloudbeaver/core-executor';
import { CachedDataResource, type CachedResource } from '@cloudbeaver/core-resource';
import { GraphQLService, NavigatorSettingsInput, ServerConfig as SDKServerConfig, ServerConfigInput } from '@cloudbeaver/core-sdk';
import { isArraysEqual } from '@cloudbeaver/core-utils';

import { isNavigatorViewSettingsEqual } from './ConnectionNavigatorViewSettings';
import { DataSynchronizationQueue } from './DataSynchronization/DataSynchronizationQueue';
import { DataSynchronizationService } from './DataSynchronization/DataSynchronizationService';
import { ServerConfigEventHandler } from './ServerConfigEventHandler';

export const FEATURE_GIT_ID = 'git';

export type ServerConfig = Omit<SDKServerConfig, 'hostName'>;

@injectable()
export class ServerConfigResource extends CachedDataResource<ServerConfig | null> {
  update: ServerConfigInput;
  navigatorSettingsUpdate: NavigatorSettingsInput;

  private readonly syncQueue: DataSynchronizationQueue;

  constructor(
    private readonly graphQLService: GraphQLService,
    private readonly dataSynchronizationService: DataSynchronizationService,
    serverConfigEventHandler: ServerConfigEventHandler,
  ) {
    super(() => null, undefined, []);

    this.syncQueue = new DataSynchronizationQueue(state => {
      if (state) {
        this.markOutdated();
      }
    });
    this.update = {};
    this.navigatorSettingsUpdate = {
      hideFolders: false,
      hideSchemas: false,
      hideVirtualModel: false,
      mergeEntities: false,
      showOnlyEntities: false,
      showSystemObjects: false,
      showUtilityObjects: false,
    };

    makeObservable<this, 'syncUpdateData'>(this, {
      update: observable,
      navigatorSettingsUpdate: observable,
      unlinkUpdate: action,
      syncUpdateData: action,
    });

    serverConfigEventHandler.on(
      () => {
        this.syncQueue.add(this.dataSynchronizationService.requestSynchronization('server-config', 'Server Configuration'));
      },
      () => undefined,
      undefined,
      this,
    );
  }

  requirePublic<T>(resource: CachedResource<any, any, T, any, any>, map?: (param: void) => T): this {
    resource.preloadResource(this, () => {}).before(ExecutorInterrupter.interrupter(() => this.publicDisabled));

    this.outdateResource<T>(resource, map as any);

    return this;
  }

  get redirectOnFederatedAuth(): boolean {
    return this.data?.redirectOnFederatedAuth ?? false;
  }

  get serverVersion(): string {
    return this.data?.version || '';
  }

  get workspaceId(): string {
    return this.data?.workspaceId || '';
  }

  get distributed(): boolean {
    return this.data?.distributed || false;
  }

  get licenseRequired(): boolean {
    return this.data?.licenseRequired ?? false;
  }

  get licenseValid(): boolean {
    return this.data?.licenseValid ?? false;
  }

  get configurationMode(): boolean {
    return !!this.data?.configurationMode;
  }

  get publicDisabled(): boolean {
    if (this.configurationMode || (this.data?.licenseRequired && !this.data.licenseValid)) {
      return true;
    }

    return false;
  }

  get adminCredentialsSaveEnabled(): boolean {
    return this.data?.adminCredentialsSaveEnabled ?? false;
  }

  get publicCredentialsSaveEnabled(): boolean {
    return this.data?.publicCredentialsSaveEnabled ?? false;
  }

  get anonymousAccessEnabled(): boolean {
    return this.data?.anonymousAccessEnabled ?? false;
  }

  get enabledFeatures(): string[] {
    return this.update.enabledFeatures || this.data?.enabledFeatures || [];
  }

  get enabledAuthProviders(): string[] {
    return this.update.enabledAuthProviders || this.data?.enabledAuthProviders || [];
  }

  get disabledDrivers(): string[] {
    return this.update.disabledDrivers || this.data?.disabledDrivers || [];
  }

  get credentialsSaveEnabled(): boolean {
    return this.update.adminCredentialsSaveEnabled ?? this.data?.adminCredentialsSaveEnabled ?? false;
  }

  get userCredentialsSaveEnabled(): boolean {
    return this.update.publicCredentialsSaveEnabled ?? this.data?.publicCredentialsSaveEnabled ?? false;
  }

  get resourceQuotas() {
    return this.data?.resourceQuotas ?? {};
  }

  get passwordPolicy() {
    return this.data?.passwordPolicyConfiguration ?? {};
  }

  get resourceManagerEnabled() {
    return this.update.resourceManagerEnabled ?? this.data?.resourceManagerEnabled ?? false;
  }

  isBetaFeatureDisabled(feature: string): boolean {
    return this.data?.disabledBetaFeatures?.includes(feature) || false;
  }

  isFeatureEnabled(feature: string, serverSide = false): boolean {
    if (serverSide) {
      return this.data?.enabledFeatures.includes(feature) || false;
    }
    return this.enabledFeatures.includes(feature);
  }

  isChanged(): boolean {
    if (!this.data || Object.keys(this.update).length === 0) {
      return false;
    }

    if (this.update.adminName || this.update.adminPassword) {
      return true;
    }

    return (
      this.update.serverName !== this.data.name ||
      this.update.serverURL !== this.data.serverURL ||
      this.update.sessionExpireTime !== this.data.sessionExpireTime ||
      this.update.anonymousAccessEnabled !== this.data.anonymousAccessEnabled ||
      this.update.resourceManagerEnabled !== this.data.resourceManagerEnabled ||
      this.update.adminCredentialsSaveEnabled !== this.data.adminCredentialsSaveEnabled ||
      this.update.publicCredentialsSaveEnabled !== this.data.publicCredentialsSaveEnabled ||
      this.update.customConnectionsEnabled !== this.data.supportsCustomConnections ||
      !isArraysEqual(this.update.enabledAuthProviders || [], this.data.enabledAuthProviders) ||
      !isArraysEqual(this.update.enabledFeatures || [], this.data.enabledFeatures) ||
      !isArraysEqual(this.update.disabledDrivers || [], this.data.disabledDrivers)
    );
  }

  isNavigatorSettingsChanged(): boolean {
    if (!this.data?.defaultNavigatorSettings) {
      return false;
    }

    return !isNavigatorViewSettingsEqual(this.data.defaultNavigatorSettings, this.navigatorSettingsUpdate);
  }

  setDataUpdate(update: ServerConfigInput): void {
    this.update = update;
  }

  setNavigatorSettingsUpdate(update: NavigatorSettingsInput): void {
    this.navigatorSettingsUpdate = update;
  }

  resetUpdate(): void {
    if (this.data) {
      this.syncUpdateData(this.data);
    }
  }

  unlinkUpdate(): void {
    this.update = {};

    if (this.data) {
      Object.assign(this.navigatorSettingsUpdate, this.data.defaultNavigatorSettings);
    } else {
      this.navigatorSettingsUpdate = {
        hideFolders: false,
        hideSchemas: false,
        hideVirtualModel: false,
        mergeEntities: false,
        showOnlyEntities: false,
        showSystemObjects: false,
        showUtilityObjects: false,
      };
    }
  }

  async updateProductConfiguration(configuration: any) {
    await this.performUpdate(undefined, undefined, async () => {
      await this.graphQLService.sdk.updateProductConfiguration({ configuration });
      this.setData(await this.loader());
      this.onDataOutdated.execute();
    });
  }

  async save(skipConfigUpdate = false): Promise<void> {
    await this.performUpdate(
      undefined,
      undefined,
      async () => {
        if (this.isNavigatorSettingsChanged()) {
          await this.graphQLService.sdk.setDefaultNavigatorSettings({ settings: this.navigatorSettingsUpdate });

          if (this.data) {
            this.data.defaultNavigatorSettings = { ...this.navigatorSettingsUpdate };
          } else {
            this.setData(await this.loader());
          }
        }

        if (this.isChanged() && !skipConfigUpdate) {
          await this.graphQLService.sdk.configureServer({
            configuration: this.update,
          });
          this.setData(await this.loader());
        }

        this.onDataOutdated.execute();
      },
      () => !this.isNavigatorSettingsChanged() && (!this.isChanged() || skipConfigUpdate),
    );
  }

  async finishConfiguration(onlyRestart = false): Promise<void> {
    await this.performUpdate(
      undefined,
      undefined,
      async () => {
        await this.graphQLService.sdk.configureServer({
          configuration: !this.isChanged() && onlyRestart ? {} : this.update,
        });

        this.setData(await this.loader());
        this.onDataOutdated.execute();
      },
      () => !this.isChanged() && !onlyRestart,
    );
  }

  protected async loader(): Promise<ServerConfig> {
    const { serverConfig } = await this.graphQLService.sdk.serverConfig();

    this.syncUpdateData(serverConfig);

    return serverConfig as ServerConfig;
  }

  private syncUpdateData(serverConfig: ServerConfig) {
    if (serverConfig.configurationMode) {
      return;
    }

    Object.assign(this.navigatorSettingsUpdate, serverConfig.defaultNavigatorSettings);

    this.update.serverName = serverConfig.name;
    this.update.serverURL = serverConfig.serverURL;
    this.update.sessionExpireTime = serverConfig.sessionExpireTime;

    this.update.adminName = undefined;
    this.update.adminPassword = undefined;

    this.update.anonymousAccessEnabled = serverConfig.anonymousAccessEnabled;

    this.update.adminCredentialsSaveEnabled = serverConfig.adminCredentialsSaveEnabled;
    this.update.publicCredentialsSaveEnabled = serverConfig.publicCredentialsSaveEnabled;

    this.update.resourceManagerEnabled = serverConfig.resourceManagerEnabled;

    this.update.customConnectionsEnabled = serverConfig.supportsCustomConnections;
    this.update.enabledAuthProviders = [...serverConfig.enabledAuthProviders];
    this.update.enabledFeatures = [...serverConfig.enabledFeatures];
    this.update.disabledDrivers = [...serverConfig.disabledDrivers];
  }
}