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

/*
 * CloudBeaver - Cloud Database Manager
 * Copyright (C) 2020-2022 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 type { ISqlDataSource } from './SqlDataSource/ISqlDataSource';


export function useDataSource(dataSource?: ISqlDataSource) {
  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();
    }
  });
}