data-connect.ts

/*!
 * @license
 * Copyright 2024 Google Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import { App } from '../app';
import { DataConnectApiClient } from './data-connect-api-client-internal';

import {
  ConnectorConfig,
  ExecuteGraphqlResponse,
  GraphqlOptions,
} from './data-connect-api';

export class DataConnectService {

  private readonly appInternal: App;
  private dataConnectInstances: Map<string, DataConnect> = new Map();

  constructor(app: App) {
    this.appInternal = app;
  }

  getDataConnect(connectorConfig: ConnectorConfig): DataConnect {
    // N.B. For a while, Admin used "serviceId" as its field even though the autogenerated
    // SDKs created "service". Let's reconcile.
    const service = 'service' in connectorConfig ? connectorConfig.service : connectorConfig.serviceId;
    const id = `${connectorConfig.location}-${service}`;
    const dc = this.dataConnectInstances.get(id);
    if (typeof dc !== 'undefined') {
      return dc;
    }

    const newInstance = new DataConnect(connectorConfig, this.appInternal);
    this.dataConnectInstances.set(id, newInstance);
    return newInstance;
  }

  /**
 * Returns the app associated with this `DataConnectService` instance.
 *
 * @returns The app associated with this `DataConnectService` instance.
 */
  get app(): App {
    return this.appInternal;
  }
}

/**
 * The Firebase `DataConnect` service interface.
 */
export class DataConnect {

  private readonly client: DataConnectApiClient;

  /**
 * @param connectorConfig - The connector configuration.
 * @param app - The app for this `DataConnect` service.
 * @constructor
 * @internal
 */
  constructor(readonly connectorConfig: ConnectorConfig, readonly app: App) {
    this.client = new DataConnectApiClient(connectorConfig, app);
  }

  /**
 * Execute an arbitrary GraphQL query or mutation
 *
 * @param query - The GraphQL query or mutation.
 * @param options - Optional {@link GraphqlOptions} when executing a GraphQL query or mutation.
 *
 * @returns A promise that fulfills with a `ExecuteGraphqlResponse`.
 * @beta
 */
  public executeGraphql<GraphqlResponse, Variables>(
    query: string,
    options?: GraphqlOptions<Variables>,
  ): Promise<ExecuteGraphqlResponse<GraphqlResponse>> {
    return this.client.executeGraphql(query, options);
  }

  /**
 * Execute an arbitrary read-only GraphQL query
 *
 * @param query - The GraphQL read-only query.
 * @param options - Optional {@link GraphqlOptions} when executing a read-only GraphQL query.
 *
 * @returns A promise that fulfills with a `ExecuteGraphqlResponse`.
 * @beta
 */
  public executeGraphqlRead<GraphqlResponse, Variables>(
    query: string,
    options?: GraphqlOptions<Variables>,
  ): Promise<ExecuteGraphqlResponse<GraphqlResponse>> {
    return this.client.executeGraphqlRead(query, options);
  }
}