All files / functions get-token.ts

92.31% Statements 12/13
100% Branches 0/0
100% Functions 0/0
91.67% Lines 11/12

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51                                    1x 1x 1x     1x   1x 42x   84x       41x           42x       33x   18x            
/**
 * @license
 * Copyright 2019 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 { FirebaseApp } from '@firebase/app-types';
import { extractAppConfig } from '../helpers/extract-app-config';
import { getInstallationEntry } from '../helpers/get-installation-entry';
import { refreshAuthToken } from '../helpers/refresh-auth-token';
import { AppConfig } from '../interfaces/app-config';
import { RequestStatus } from '../interfaces/installation-entry';
import { ERROR_FACTORY, ErrorCode } from '../util/errors';
 
export async function getToken(app: FirebaseApp): Promise<string> {
  const appConfig = extractAppConfig(app);
 
  await completeInstallationRegistration(appConfig);
 
  // At this point we either have a Registered Installation in the DB, or we've
  // already thrown an error.
  return refreshAuthToken(appConfig);
}
 
async function completeInstallationRegistration(
  appConfig: AppConfig
): Promise<void> {
  const { installationEntry, registrationPromise } = await getInstallationEntry(
    appConfig
  );
 
  if (registrationPromise) {
    // A createInstallation request is in progress. Wait until it finishes.
    await registrationPromise;
  } else if (installationEntry.registrationStatus !== RequestStatus.COMPLETED) {
    // Installation ID can't be registered.
    throw ERROR_FACTORY.create(ErrorCode.CREATE_INSTALLATION_FAILED);
  }
}