cloudbeaver/webapp/packages/plugin-sql-generator/src/SqlGenerators/SqlGeneratorsResource.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 { injectable } from '@cloudbeaver/core-di';
import { NavNodeInfoResource } from '@cloudbeaver/core-navigation-tree';
import { CachedMapResource, isResourceAlias, type ResourceKey, resourceKeyList, ResourceKeyUtils } from '@cloudbeaver/core-resource';
import { GraphQLService, SqlQueryGenerator } from '@cloudbeaver/core-sdk';

export const MAX_GENERATORS_LENGTH = 15;

@injectable()
export class SqlGeneratorsResource extends CachedMapResource<string, SqlQueryGenerator[]> {
  constructor(private readonly graphQLService: GraphQLService, private readonly navNodeInfoResource: NavNodeInfoResource) {
    super();

    this.navNodeInfoResource.outdateResource(this);
    this.navNodeInfoResource.deleteInResource(this);
  }

  async generateEntityQuery(generatorId: string, nodePathList: string | string[]): Promise<string> {
    const result = await this.graphQLService.sdk.sqlGenerateEntityQuery({
      generatorId,
      nodePathList,
      options: {},
    });

    return result.sqlGenerateEntityQuery;
  }

  protected async loader(key: ResourceKey<string>): Promise<Map<string, SqlQueryGenerator[]>> {
    if (isResourceAlias(key)) {
      throw new Error('Aliases not supported by this resource.');
    }
    const values = new Map();

    await ResourceKeyUtils.forEachAsync(key, async key => {
      const { generators } = await this.graphQLService.sdk.sqlEntityQueryGenerators({
        nodePathList: key,
      });
      values.set(key, generators);
    });

    this.set(resourceKeyList(Array.from(values.keys())), Array.from(values.values()));
    return this.data;
  }

  protected validateKey(key: string): boolean {
    return typeof key === 'string';
  }
}