@firebase/data-connect 0.0.2-dataconnect-preview.388b61c7e → 0.0.3-canary.beaa4dffb

package/dist/node-esm/src/network/transport/rest.d.ts

@@ -15,11 +15,15 @@
  * limitations under the License.
  */
 import { DataConnectOptions, TransportOptions } from '../../api/DataConnect';
+import { AppCheckTokenProvider } from '../../core/AppCheckTokenProvider';
 import { AuthTokenProvider } from '../../core/FirebaseAuthProvider';
 import { DataConnectTransport } from '.';
 export declare class RESTTransport implements DataConnectTransport {
     private apiKey?;
+    private appId?;
     private authProvider?;
+    private appCheckProvider?;
+    private _isUsingGen;
     private _host;
     private _port;
     private _location;
@@ -28,17 +32,27 @@ export declare class RESTTransport implements DataConnectTransport {
     private _project;
     private _serviceName;
     private _accessToken;
-    private _authInitialized;
-    constructor(options: DataConnectOptions, apiKey?: string | undefined, authProvider?: AuthTokenProvider | undefined, transportOptions?: TransportOptions | undefined);
+    private _appCheckToken;
+    private _lastToken;
+    constructor(options: DataConnectOptions, apiKey?: string | undefined, appId?: string, authProvider?: AuthTokenProvider | undefined, appCheckProvider?: AppCheckTokenProvider | undefined, transportOptions?: TransportOptions | undefined, _isUsingGen?: boolean);
     get endpointUrl(): string;
     useEmulator(host: string, port?: number, isSecure?: boolean): void;
     onTokenChanged(newToken: string | null): void;
-    getWithAuth(): Promise<string>;
-    invokeQuery: <T, U = unknown>(queryName: string, body: U) => {
-        then: any;
-    };
-    invokeMutation: <T, U = unknown>(mutationName: string, body: U) => {
-        then: any;
-        cancel: () => void;
-    };
+    getWithAuth(forceToken?: boolean): Promise<string>;
+    _setLastToken(lastToken: string | null): void;
+    withRetry<T>(promiseFactory: () => Promise<{
+        data: T;
+        errors: Error[];
+    }>, retry?: boolean): Promise<{
+        data: T;
+        errors: Error[];
+    }>;
+    invokeQuery: <T, U>(queryName: string, body?: U) => PromiseLike<{
+        data: T;
+        errors: Error[];
+    }>;
+    invokeMutation: <T, U>(queryName: string, body?: U) => PromiseLike<{
+        data: T;
+        errors: Error[];
+    }>;
 }

package/dist/node-esm/src/util/validateArgs.d.ts

@@ -0,0 +1,33 @@
+/**
+ * @license
+ * Copyright 2024 Google LLC
+ *
+ * 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 { ConnectorConfig, DataConnect } from '../api/DataConnect';
+interface ParsedArgs<Variables> {
+    dc: DataConnect;
+    vars: Variables;
+}
+/**
+ * The generated SDK will allow the user to pass in either the variable or the data connect instance with the variable,
+ * and this function validates the variables and returns back the DataConnect instance and variables based on the arguments passed in.
+ * @param connectorConfig
+ * @param dcOrVars
+ * @param vars
+ * @param validateVars
+ * @returns {DataConnect} and {Variables} instance
+ * @internal
+ */
+export declare function validateArgs<Variables extends object>(connectorConfig: ConnectorConfig, dcOrVars?: DataConnect | Variables, vars?: Variables, validateVars?: boolean): ParsedArgs<Variables>;
+export {};

package/dist/private.d.ts

@@ -4,17 +4,21 @@
  * @packageDocumentation
  */
 
+import { AppCheckInternalComponentName } from '@firebase/app-check-interop-types';
+import { AppCheckTokenListener } from '@firebase/app-check-interop-types';
+import { AppCheckTokenResult } from '@firebase/app-check-interop-types';
 import { FirebaseApp } from '@firebase/app';
 import { FirebaseAuthInternalName } from '@firebase/auth-interop-types';
 import { FirebaseAuthTokenData } from '@firebase/auth-interop-types';
 import { FirebaseError } from '@firebase/util';
-import { FirebaseOptions } from '@firebase/app';
 import { LogLevelString } from '@firebase/logger';
 import { Provider } from '@firebase/component';
 
-export declare type AuthTokenListener = (token: string | null) => void;
+/* Excluded from this release type: AppCheckTokenProvider */
 
-export declare interface AuthTokenProvider {
+declare type AuthTokenListener = (token: string | null) => void;
+
+declare interface AuthTokenProvider {
     getToken(forceRefresh: boolean): Promise<FirebaseAuthTokenData | null>;
     addTokenChangeListener(listener: AuthTokenListener): void;
 }
@@ -50,15 +54,19 @@ export declare class DataConnect {
     readonly app: FirebaseApp;
     private readonly dataConnectOptions;
     private readonly _authProvider;
+    private readonly _appCheckProvider;
     _queryManager: QueryManager;
     _mutationManager: MutationManager;
     isEmulator: boolean;
-    initialized: boolean;
+    _initialized: boolean;
     private _transport;
     private _transportClass;
     private _transportOptions?;
     private _authTokenProvider?;
-    constructor(app: FirebaseApp, dataConnectOptions: DataConnectOptions, _authProvider: Provider<FirebaseAuthInternalName>);
+    _isUsingGeneratedSdk: boolean;
+    private _appCheckTokenProvider?;
+    constructor(app: FirebaseApp, dataConnectOptions: DataConnectOptions, _authProvider: Provider<FirebaseAuthInternalName>, _appCheckProvider: Provider<AppCheckInternalComponentName>);
+    _useGeneratedSdk(): void;
     _delete(): Promise<void>;
     getSettings(): ConnectorConfig;
     setInitialized(): void;
@@ -89,7 +97,7 @@ declare class DataConnectError extends FirebaseError {
     message: string);
 }
 
-declare type DataConnectErrorCode = 'other' | 'already-initialized' | 'not-initialized' | 'not-supported' | 'invalid-argument' | 'partial-error';
+declare type DataConnectErrorCode = 'other' | 'already-initialized' | 'not-initialized' | 'not-supported' | 'invalid-argument' | 'partial-error' | 'unauthorized';
 
 /**
  * DataConnectOptions including project id
@@ -111,18 +119,7 @@ export declare interface DataConnectSubscription<Data, Variables> {
     unsubscribe: () => void;
 }
 
-export declare interface DataConnectTransport {
-    invokeQuery<T, U>(queryName: string, body?: U): PromiseLike<{
-        data: T;
-        errors: Error[];
-    }>;
-    invokeMutation<T, U>(queryName: string, body?: U): PromiseLike<{
-        data: T;
-        errors: Error[];
-    }>;
-    useEmulator(host: string, port?: number, sslEnabled?: boolean): void;
-    onTokenChanged: (token: string | null) => void;
-}
+/* Excluded from this release type: DataConnectTransport */
 
 export declare type DataSource = typeof SOURCE_CACHE | typeof SOURCE_SERVER;
 
@@ -140,19 +137,6 @@ export declare function executeMutation<Data, Variables>(mutationRef: MutationRe
  */
 export declare function executeQuery<Data, Variables>(queryRef: QueryRef<Data, Variables>): QueryPromise<Data, Variables>;
 
-export declare const FIREBASE_DATA_CONNECT_EMULATOR_HOST_VAR = "FIREBASE_DATA_CONNECT_EMULATOR_HOST";
-
-export declare class FirebaseAuthProvider implements AuthTokenProvider {
-    private _appName;
-    private _options;
-    private _authProvider;
-    private _auth;
-    constructor(_appName: string, _options: FirebaseOptions, _authProvider: Provider<FirebaseAuthInternalName>);
-    getToken(forceRefresh: boolean): Promise<FirebaseAuthTokenData | null>;
-    addTokenChangeListener(listener: AuthTokenListener): void;
-    removeTokenChangeListener(listener: (token: string | null) => void): void;
-}
-
 /**
  * Initialize DataConnect instance
  * @param options ConnectorConfig
@@ -195,9 +179,6 @@ export declare function mutationRef<Data>(dcInstance: DataConnect, mutationName:
  */
 export declare function mutationRef<Data, Variables>(dcInstance: DataConnect, mutationName: string, variables: Variables): MutationRef<Data, Variables>;
 
-export declare interface MutationResponse<T> extends CancellableOperation<T> {
-}
-
 /**
  * Mutation Result from `executeMutation`
  */
@@ -233,6 +214,11 @@ export declare interface OpResult<Data> {
     fetchTime: string;
 }
 
+declare interface ParsedArgs<Variables> {
+    dc: DataConnect;
+    vars: Variables;
+}
+
 /* Excluded from this release type: parseOptions */
 
 export declare const QUERY_STR = "query";
@@ -241,7 +227,7 @@ declare class QueryManager {
     private transport;
     _queries: Map<string, TrackedQuery<unknown, unknown>>;
     constructor(transport: DataConnectTransport);
-    track<Data, Variables>(queryName: string, variables: Variables, initialCache?: OpResult<Data>): TrackedQuery<unknown, unknown>;
+    track<Data, Variables>(queryName: string, variables: Variables, initialCache?: OpResult<Data>): TrackedQuery<Data, Variables>;
     addSubscription<Data, Variables>(queryRef: OperationRef<Data, Variables>, onResultCallback: OnResultSubscription<Data, Variables>, onErrorCallback?: OnErrorSubscription, initialCache?: OpResult<Data>): () => void;
     executeQuery<Data, Variables>(queryRef: QueryRef<Data, Variables>): QueryPromise<Data, Variables>;
     enableEmulator(host: string, port: number): void;
@@ -277,9 +263,6 @@ export declare function queryRef<Data>(dcInstance: DataConnect, queryName: strin
  */
 export declare function queryRef<Data, Variables>(dcInstance: DataConnect, queryName: string, variables: Variables): QueryRef<Data, Variables>;
 
-export declare interface QueryResponse<T> extends CancellableOperation<T> {
-}
-
 /**
  * Result of `executeQuery`
  */
@@ -304,11 +287,6 @@ export declare interface RefInfo<Variables> {
     connectorConfig: DataConnectOptions;
 }
 
-export declare interface Sender<T> {
-    abort: () => void;
-    send: () => Promise<T>;
-}
-
 /**
  * Serialized Ref as a result of `QueryResult.toJSON()`
  */
@@ -361,7 +339,7 @@ export declare function terminate(dataConnect: DataConnect): Promise<void>;
  * @param serializedRef ref to convert to `QueryRef`
  * @returns `QueryRef`
  */
-export declare function toQueryRef<Data, Variables>(serializedRef: SerializedRef<Data, Variables>): QueryRef<unknown, Variables>;
+export declare function toQueryRef<Data, Variables>(serializedRef: SerializedRef<Data, Variables>): QueryRef<Data, Variables>;
 
 declare interface TrackedQuery<Data, Variables> {
     ref: Omit<OperationRef<Data, Variables>, 'dataConnect'>;
@@ -370,7 +348,7 @@ declare interface TrackedQuery<Data, Variables> {
     lastError: DataConnectError | null;
 }
 
-export declare type TransportClass = new (options: DataConnectOptions, apiKey?: string, authProvider?: AuthTokenProvider, transportOptions?: TransportOptions) => DataConnectTransport;
+/* Excluded from this release type: TransportClass */
 
 /**
  * Options to connect to emulator
@@ -381,4 +359,8 @@ export declare interface TransportOptions {
     port?: number;
 }
 
+/* Excluded from this release type: validateArgs */
+
+/* Excluded from this release type: validateDCOptions */
+
 export { }

package/dist/public.d.ts

@@ -4,18 +4,13 @@
  * @packageDocumentation
  */
 import { FirebaseApp } from '@firebase/app';
-import { FirebaseOptions } from '@firebase/app';
+import { AppCheckInternalComponentName } from '@firebase/app-check-interop-types';
 import { FirebaseAuthInternalName } from '@firebase/auth-interop-types';
-import { FirebaseAuthTokenData } from '@firebase/auth-interop-types';
 import { Provider } from '@firebase/component';
 import { LogLevelString } from '@firebase/logger';
 import { FirebaseError } from '@firebase/util';
 
-export declare type AuthTokenListener = (token: string | null) => void;
-export declare interface AuthTokenProvider {
-    getToken(forceRefresh: boolean): Promise<FirebaseAuthTokenData | null>;
-    addTokenChangeListener(listener: AuthTokenListener): void;
-}
+
 export declare interface CancellableOperation<T> extends PromiseLike<{
     data: T;
 }> {
@@ -44,8 +39,7 @@ export declare class DataConnect {
     readonly app: FirebaseApp;
     private readonly dataConnectOptions;
     isEmulator: boolean;
-    initialized: boolean;
-    constructor(app: FirebaseApp, dataConnectOptions: DataConnectOptions, _authProvider: Provider<FirebaseAuthInternalName>);
+    constructor(app: FirebaseApp, dataConnectOptions: DataConnectOptions, _authProvider: Provider<FirebaseAuthInternalName>, _appCheckProvider: Provider<AppCheckInternalComponentName>);
     getSettings(): ConnectorConfig;
     setInitialized(): void;
     enableEmulator(transportOptions: TransportOptions): void;
@@ -67,18 +61,7 @@ export declare interface DataConnectSubscription<Data, Variables> {
     errCallback?: (e?: FirebaseError) => void;
     unsubscribe: () => void;
 }
-export declare interface DataConnectTransport {
-    invokeQuery<T, U>(queryName: string, body?: U): PromiseLike<{
-        data: T;
-        errors: Error[];
-    }>;
-    invokeMutation<T, U>(queryName: string, body?: U): PromiseLike<{
-        data: T;
-        errors: Error[];
-    }>;
-    useEmulator(host: string, port?: number, sslEnabled?: boolean): void;
-    onTokenChanged: (token: string | null) => void;
-}
+/* Excluded from this release type: DataConnectTransport */
 export declare type DataSource = typeof SOURCE_CACHE | typeof SOURCE_SERVER;
 /**
  * Execute Mutation
@@ -92,13 +75,6 @@ export declare function executeMutation<Data, Variables>(mutationRef: MutationRe
  * @returns `QueryPromise`
  */
 export declare function executeQuery<Data, Variables>(queryRef: QueryRef<Data, Variables>): QueryPromise<Data, Variables>;
-export declare const FIREBASE_DATA_CONNECT_EMULATOR_HOST_VAR = "FIREBASE_DATA_CONNECT_EMULATOR_HOST";
-export declare class FirebaseAuthProvider implements AuthTokenProvider {
-    constructor(_appName: string, _options: FirebaseOptions, _authProvider: Provider<FirebaseAuthInternalName>);
-    getToken(forceRefresh: boolean): Promise<FirebaseAuthTokenData | null>;
-    addTokenChangeListener(listener: AuthTokenListener): void;
-    removeTokenChangeListener(listener: (token: string | null) => void): void;
-}
 /**
  * Initialize DataConnect instance
  * @param options ConnectorConfig
@@ -133,8 +109,6 @@ export declare function mutationRef<Data>(dcInstance: DataConnect, mutationName:
  * @param variables variables to send with mutation
  */
 export declare function mutationRef<Data, Variables>(dcInstance: DataConnect, mutationName: string, variables: Variables): MutationRef<Data, Variables>;
-export declare interface MutationResponse<T> extends CancellableOperation<T> {
-}
 /**
  * Mutation Result from `executeMutation`
  */
@@ -192,8 +166,6 @@ export declare function queryRef<Data>(dcInstance: DataConnect, queryName: strin
  * @returns `QueryRef`
  */
 export declare function queryRef<Data, Variables>(dcInstance: DataConnect, queryName: string, variables: Variables): QueryRef<Data, Variables>;
-export declare interface QueryResponse<T> extends CancellableOperation<T> {
-}
 /**
  * Result of `executeQuery`
  */
@@ -214,10 +186,6 @@ export declare interface RefInfo<Variables> {
     variables: Variables;
     connectorConfig: DataConnectOptions;
 }
-export declare interface Sender<T> {
-    abort: () => void;
-    send: () => Promise<T>;
-}
 /**
  * Serialized Ref as a result of `QueryResult.toJSON()`
  */
@@ -262,8 +230,8 @@ export declare function terminate(dataConnect: DataConnect): Promise<void>;
  * @param serializedRef ref to convert to `QueryRef`
  * @returns `QueryRef`
  */
-export declare function toQueryRef<Data, Variables>(serializedRef: SerializedRef<Data, Variables>): QueryRef<unknown, Variables>;
-export declare type TransportClass = new (options: DataConnectOptions, apiKey?: string, authProvider?: AuthTokenProvider, transportOptions?: TransportOptions) => DataConnectTransport;
+export declare function toQueryRef<Data, Variables>(serializedRef: SerializedRef<Data, Variables>): QueryRef<Data, Variables>;
+/* Excluded from this release type: TransportClass */
 /**
  * Options to connect to emulator
  */
@@ -272,4 +240,6 @@ export declare interface TransportOptions {
     sslEnabled?: boolean;
     port?: number;
 }
+/* Excluded from this release type: validateArgs */
+/* Excluded from this release type: validateDCOptions */
 export {};

package/dist/src/api/DataConnect.d.ts

@@ -15,6 +15,7 @@
  * limitations under the License.
  */
 import { FirebaseApp } from '@firebase/app';
+import { AppCheckInternalComponentName } from '@firebase/app-check-interop-types';
 import { FirebaseAuthInternalName } from '@firebase/auth-interop-types';
 import { Provider } from '@firebase/component';
 import { QueryManager } from '../core/QueryManager';
@@ -35,7 +36,6 @@ export interface TransportOptions {
     sslEnabled?: boolean;
     port?: number;
 }
-export declare const FIREBASE_DATA_CONNECT_EMULATOR_HOST_VAR = "FIREBASE_DATA_CONNECT_EMULATOR_HOST";
 /**
  *
  * @param fullHost
@@ -56,15 +56,19 @@ export declare class DataConnect {
     readonly app: FirebaseApp;
     private readonly dataConnectOptions;
     private readonly _authProvider;
+    private readonly _appCheckProvider;
     _queryManager: QueryManager;
     _mutationManager: MutationManager;
     isEmulator: boolean;
-    initialized: boolean;
+    _initialized: boolean;
     private _transport;
     private _transportClass;
     private _transportOptions?;
     private _authTokenProvider?;
-    constructor(app: FirebaseApp, dataConnectOptions: DataConnectOptions, _authProvider: Provider<FirebaseAuthInternalName>);
+    _isUsingGeneratedSdk: boolean;
+    private _appCheckTokenProvider?;
+    constructor(app: FirebaseApp, dataConnectOptions: DataConnectOptions, _authProvider: Provider<FirebaseAuthInternalName>, _appCheckProvider: Provider<AppCheckInternalComponentName>);
+    _useGeneratedSdk(): void;
     _delete(): Promise<void>;
     getSettings(): ConnectorConfig;
     setInitialized(): void;
@@ -89,6 +93,13 @@ export declare function getDataConnect(options: ConnectorConfig): DataConnect;
  * @param options ConnectorConfig
  */
 export declare function getDataConnect(app: FirebaseApp, options: ConnectorConfig): DataConnect;
+/**
+ *
+ * @param dcOptions
+ * @returns {void}
+ * @internal
+ */
+export declare function validateDCOptions(dcOptions: ConnectorConfig): boolean;
 /**
  * Delete DataConnect instance
  * @param dataConnect DataConnect instance

package/dist/src/api/index.d.ts

@@ -20,3 +20,4 @@ export * from './Reference';
 export * from './Mutation';
 export * from './query';
 export { setLogLevel } from '../logger';
+export { validateArgs } from '../util/validateArgs';

package/dist/src/api/query.d.ts

@@ -81,7 +81,7 @@ export declare function queryRef<Data, Variables>(dcInstance: DataConnect, query
  * @param serializedRef ref to convert to `QueryRef`
  * @returns `QueryRef`
  */
-export declare function toQueryRef<Data, Variables>(serializedRef: SerializedRef<Data, Variables>): QueryRef<unknown, Variables>;
+export declare function toQueryRef<Data, Variables>(serializedRef: SerializedRef<Data, Variables>): QueryRef<Data, Variables>;
 /**
  * `OnCompleteSubscription`
  */

package/dist/src/core/AppCheckTokenProvider.d.ts

@@ -0,0 +1,30 @@
+/**
+ * @license
+ * Copyright 2024 Google LLC
+ *
+ * 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 { AppCheckInternalComponentName, AppCheckTokenListener, AppCheckTokenResult } from '@firebase/app-check-interop-types';
+import { Provider } from '@firebase/component';
+/**
+ * @internal
+ * Abstraction around AppCheck's token fetching capabilities.
+ */
+export declare class AppCheckTokenProvider {
+    private appName_;
+    private appCheckProvider?;
+    private appCheck?;
+    constructor(appName_: string, appCheckProvider?: Provider<AppCheckInternalComponentName>);
+    getToken(forceRefresh?: boolean): Promise<AppCheckTokenResult>;
+    addTokenChangeListener(listener: AppCheckTokenListener): void;
+}

package/dist/src/core/FirebaseAuthProvider.d.ts

@@ -14,7 +14,7 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-import { FirebaseOptions } from '@firebase/app';
+import { FirebaseOptions } from '@firebase/app-types';
 import { FirebaseAuthInternalName, FirebaseAuthTokenData } from '@firebase/auth-interop-types';
 import { Provider } from '@firebase/component';
 export interface AuthTokenProvider {

package/dist/src/core/QueryManager.d.ts

@@ -28,7 +28,7 @@ export declare class QueryManager {
     private transport;
     _queries: Map<string, TrackedQuery<unknown, unknown>>;
     constructor(transport: DataConnectTransport);
-    track<Data, Variables>(queryName: string, variables: Variables, initialCache?: OpResult<Data>): TrackedQuery<unknown, unknown>;
+    track<Data, Variables>(queryName: string, variables: Variables, initialCache?: OpResult<Data>): TrackedQuery<Data, Variables>;
     addSubscription<Data, Variables>(queryRef: OperationRef<Data, Variables>, onResultCallback: OnResultSubscription<Data, Variables>, onErrorCallback?: OnErrorSubscription, initialCache?: OpResult<Data>): () => void;
     executeQuery<Data, Variables>(queryRef: QueryRef<Data, Variables>): QueryPromise<Data, Variables>;
     enableEmulator(host: string, port: number): void;

package/dist/src/core/error.d.ts

@@ -15,7 +15,7 @@
  * limitations under the License.
  */
 import { FirebaseError } from '@firebase/util';
-export declare type DataConnectErrorCode = 'other' | 'already-initialized' | 'not-initialized' | 'not-supported' | 'invalid-argument' | 'partial-error';
+export declare type DataConnectErrorCode = 'other' | 'already-initialized' | 'not-initialized' | 'not-supported' | 'invalid-argument' | 'partial-error' | 'unauthorized';
 export declare type Code = DataConnectErrorCode;
 export declare const Code: {
     OTHER: DataConnectErrorCode;
@@ -24,6 +24,7 @@ export declare const Code: {
     NOT_SUPPORTED: DataConnectErrorCode;
     INVALID_ARGUMENT: DataConnectErrorCode;
     PARTIAL_ERROR: DataConnectErrorCode;
+    UNAUTHORIZED: DataConnectErrorCode;
 };
 /** An error returned by a DataConnect operation. */
 export declare class DataConnectError extends FirebaseError {

package/dist/src/network/fetch.d.ts

@@ -15,7 +15,7 @@
  * limitations under the License.
  */
 export declare function initializeFetch(fetchImpl: typeof fetch): void;
-export declare function dcFetch<T, U>(url: string, body: U, { signal }: AbortController, accessToken: string | null): Promise<{
+export declare function dcFetch<T, U>(url: string, body: U, { signal }: AbortController, appId: string | null, accessToken: string | null, appCheckToken: string | null, _isUsingGen: boolean): Promise<{
     data: T;
     errors: Error[];
 }>;

package/dist/src/network/transport/index.d.ts

@@ -15,7 +15,11 @@
  * limitations under the License.
  */
 import { DataConnectOptions, TransportOptions } from '../../api/DataConnect';
+import { AppCheckTokenProvider } from '../../core/AppCheckTokenProvider';
 import { AuthTokenProvider } from '../../core/FirebaseAuthProvider';
+/**
+ * @internal
+ */
 export interface DataConnectTransport {
     invokeQuery<T, U>(queryName: string, body?: U): PromiseLike<{
         data: T;
@@ -33,13 +37,7 @@ export interface CancellableOperation<T> extends PromiseLike<{
 }> {
     cancel: () => void;
 }
-export interface QueryResponse<T> extends CancellableOperation<T> {
-}
-export interface MutationResponse<T> extends CancellableOperation<T> {
-}
-export interface Sender<T> {
-    abort: () => void;
-    send: () => Promise<T>;
-}
-export declare type TransportClass = new (options: DataConnectOptions, apiKey?: string, authProvider?: AuthTokenProvider, transportOptions?: TransportOptions) => DataConnectTransport;
-export * from '../../core/FirebaseAuthProvider';
+/**
+ * @internal
+ */
+export declare type TransportClass = new (options: DataConnectOptions, apiKey?: string, appId?: string, authProvider?: AuthTokenProvider, appCheckProvider?: AppCheckTokenProvider, transportOptions?: TransportOptions, _isUsingGen?: boolean) => DataConnectTransport;

package/dist/src/network/transport/rest.d.ts

@@ -15,11 +15,15 @@
  * limitations under the License.
  */
 import { DataConnectOptions, TransportOptions } from '../../api/DataConnect';
+import { AppCheckTokenProvider } from '../../core/AppCheckTokenProvider';
 import { AuthTokenProvider } from '../../core/FirebaseAuthProvider';
 import { DataConnectTransport } from '.';
 export declare class RESTTransport implements DataConnectTransport {
     private apiKey?;
+    private appId?;
     private authProvider?;
+    private appCheckProvider?;
+    private _isUsingGen;
     private _host;
     private _port;
     private _location;
@@ -28,17 +32,27 @@ export declare class RESTTransport implements DataConnectTransport {
     private _project;
     private _serviceName;
     private _accessToken;
-    private _authInitialized;
-    constructor(options: DataConnectOptions, apiKey?: string | undefined, authProvider?: AuthTokenProvider | undefined, transportOptions?: TransportOptions | undefined);
+    private _appCheckToken;
+    private _lastToken;
+    constructor(options: DataConnectOptions, apiKey?: string | undefined, appId?: string, authProvider?: AuthTokenProvider | undefined, appCheckProvider?: AppCheckTokenProvider | undefined, transportOptions?: TransportOptions | undefined, _isUsingGen?: boolean);
     get endpointUrl(): string;
     useEmulator(host: string, port?: number, isSecure?: boolean): void;
     onTokenChanged(newToken: string | null): void;
-    getWithAuth(): Promise<string>;
-    invokeQuery: <T, U = unknown>(queryName: string, body: U) => {
-        then: any;
-    };
-    invokeMutation: <T, U = unknown>(mutationName: string, body: U) => {
-        then: any;
-        cancel: () => void;
-    };
+    getWithAuth(forceToken?: boolean): Promise<string>;
+    _setLastToken(lastToken: string | null): void;
+    withRetry<T>(promiseFactory: () => Promise<{
+        data: T;
+        errors: Error[];
+    }>, retry?: boolean): Promise<{
+        data: T;
+        errors: Error[];
+    }>;
+    invokeQuery: <T, U>(queryName: string, body?: U) => PromiseLike<{
+        data: T;
+        errors: Error[];
+    }>;
+    invokeMutation: <T, U>(queryName: string, body?: U) => PromiseLike<{
+        data: T;
+        errors: Error[];
+    }>;
 }

package/dist/src/util/validateArgs.d.ts

@@ -0,0 +1,33 @@
+/**
+ * @license
+ * Copyright 2024 Google LLC
+ *
+ * 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 { ConnectorConfig, DataConnect } from '../api/DataConnect';
+interface ParsedArgs<Variables> {
+    dc: DataConnect;
+    vars: Variables;
+}
+/**
+ * The generated SDK will allow the user to pass in either the variable or the data connect instance with the variable,
+ * and this function validates the variables and returns back the DataConnect instance and variables based on the arguments passed in.
+ * @param connectorConfig
+ * @param dcOrVars
+ * @param vars
+ * @param validateVars
+ * @returns {DataConnect} and {Variables} instance
+ * @internal
+ */
+export declare function validateArgs<Variables extends object>(connectorConfig: ConnectorConfig, dcOrVars?: DataConnect | Variables, vars?: Variables, validateVars?: boolean): ParsedArgs<Variables>;
+export {};