cloudbeaver/webapp/packages/plugin-sql-editor/src/useDataSource.ts

/*
 * CloudBeaver - Cloud Database Manager
 * Copyright (C) 2020-2023 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 { useEffect } from 'react';

import { useObservableRef } from '@cloudbeaver/core-blocks';
import { useService } from '@cloudbeaver/core-di';

import type { ISqlDataSource } from './SqlDataSource/ISqlDataSource';
import { SqlEditorService } from './SqlEditorService';

export function useDataSource(dataSource?: ISqlDataSource) {
  const sqlEditorService = useService(SqlEditorService);
  const refObj = useObservableRef(() => ({
    async load(refresh?: boolean) {
      if (!this.dataSource || this.dataSource.isLoading()) {
        return;
      }

      if (refresh) {
        this.dataSource.markOutdated();
      }

      await this.dataSource.load();
    },
  }), { }, {
    dataSource,
  });

  // TODO: getComputed skips update somehow ...
  const outdated = dataSource && (
    (dataSource.isOutdated() || !dataSource.isLoaded())
    && !dataSource.isLoading()
  );


  useEffect(() => {
    if (!outdated) {
      return;
    }

    if (!dataSource.exception || (Array.isArray(dataSource.exception) && !dataSource.exception.some(Boolean))) {
      refObj.load();
    }
  });

  const executionContextId = dataSource?.executionContext?.id;

  useEffect(() => () => {
    if (
      executionContextId !== undefined
      && dataSource?.executionContext?.id !== executionContextId
    ) {
      sqlEditorService.destroyContext({ id: executionContextId, connectionId: '', projectId: '' });
    }
  }, [executionContextId, dataSource]);
}