Add Firestore, initializeFirestore, getFirestore to Firestore lite

Author: schmidt-sebastian

packages/firestore/lite/index.node.ts

@@ -0,0 +1,58 @@
+/**
+ * @license
+ * Copyright 2020 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 firebase from '@firebase/app';
+import { FirebaseNamespace } from '@firebase/app-types';
+import {
+  Firestore,
+  initializeFirestore,
+  getFirestore
+} from './src/api/database';
+import { version } from '../package.json';
+import { Component, ComponentType } from '@firebase/component';
+import { _FirebaseNamespace } from '@firebase/app-types/private';
+import { makeConstructorPrivate } from '../src/util/api';
+
+export const PublicFirestore = makeConstructorPrivate(
+  Firestore,
+  'Use getFirestore() instead.'
+);
+
+const firestoreNamespace = {
+  Firestore: PublicFirestore,
+  initializeFirestore,
+  getFirestore
+};
+
+export function registerFirestore(instance: FirebaseNamespace): void {
+  (instance as _FirebaseNamespace).INTERNAL.registerComponent(
+    new Component(
+      'firestore/lite',
+      container => {
+        const app = container.getProvider('app').getImmediate()!;
+        return ((app, auth) => new Firestore(app, auth))(
+          app,
+          container.getProvider('auth-internal')
+        );
+      },
+      ComponentType.PUBLIC
+    ).setServiceProps({ ...firestoreNamespace })
+  );
+  instance.registerVersion('@firebase/firestore/lite', version, 'node');
+}
+
+registerFirestore(firebase);

packages/firestore/lite/src/api/database.ts

@@ -0,0 +1,104 @@
+/**
+ * @license
+ * Copyright 2017 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 * as firestore from '../../';
+
+import { _getProvider } from '@firebase/app-exp';
+import { Provider } from '@firebase/component';
+import { FirebaseApp } from '@firebase/app-types';
+import { FirebaseService } from '@firebase/app-types/private';
+
+import { Code, FirestoreError } from '../../../src/util/error';
+import { DatabaseId } from '../../../src/core/database_info';
+import { FirebaseAuthInternalName } from '@firebase/auth-interop-types';
+import {
+  CredentialsProvider,
+  FirebaseCredentialsProvider
+} from '../../../src/api/credentials';
+
+/**
+ * The root reference to the Firestore Lite database.
+ */
+export class Firestore implements firestore.FirebaseFirestore, FirebaseService {
+  readonly _databaseId: DatabaseId;
+  private readonly _firebaseApp: FirebaseApp;
+  private readonly _credentials: CredentialsProvider;
+  private _settings?: firestore.Settings;
+
+  constructor(
+    app: FirebaseApp,
+    authProvider: Provider<FirebaseAuthInternalName>
+  ) {
+    this._firebaseApp = app;
+    this._databaseId = Firestore.databaseIdFromApp(app);
+    this._credentials = new FirebaseCredentialsProvider(authProvider);
+  }
+
+  get app(): FirebaseApp {
+    return this._firebaseApp;
+  }
+
+  _configureClient(settings: firestore.Settings): void {
+    if (this._settings) {
+      throw new FirestoreError(
+        Code.FAILED_PRECONDITION,
+        'Firestore has already been started and its settings can no longer ' +
+          'be changed. initializeFirestore() can only be called before calling ' +
+          'getFirestore().'
+      );
+    }
+    this._settings = settings;
+  }
+
+  _ensureClientConfigured(): void {
+    if (!this._settings) {
+      this._settings = {};
+    }
+  }
+
+  private static databaseIdFromApp(app: FirebaseApp): DatabaseId {
+    if (!Object.prototype.hasOwnProperty.apply(app.options, ['projectId'])) {
+      throw new FirestoreError(
+        Code.INVALID_ARGUMENT,
+        '"projectId" not provided in firebase.initializeApp.'
+      );
+    }
+
+    return new DatabaseId(app.options.projectId!);
+  }
+}
+
+export function initializeFirestore(
+  app: FirebaseApp,
+  settings: firestore.Settings
+): Firestore {
+  const firestore = _getProvider(
+    app,
+    'firestore/lite'
+  ).getImmediate() as Firestore;
+  firestore._configureClient(settings);
+  return firestore;
+}
+
+export function getFirestore(app: FirebaseApp): Firestore {
+  const firestore = _getProvider(
+    app,
+    'firestore/lite'
+  ).getImmediate() as Firestore;
+  firestore._ensureClientConfigured();
+  return firestore;
+}

packages/firestore/lite/test/integration.test.ts

@@ -0,0 +1,41 @@
+/**
+ * @license
+ * Copyright 2017 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 { expect } from 'chai';
+
+import firebase from '../../test/integration/util/firebase_export';
+import { getFirestore, initializeFirestore } from '../src/api/database';
+
+describe('Firestore', () => {
+  it('can provide setting', () => {
+    const app = firebase.initializeApp(
+      { apiKey: 'fake-api-key', projectId: 'test-project' },
+      'test-app-initializeFirestore'
+    );
+    initializeFirestore(app, { host: 'localhost', ssl: false });
+  });
+
+  it('returns same instance', () => {
+    const app = firebase.initializeApp(
+      { apiKey: 'fake-api-key', projectId: 'test-project' },
+      'test-app-getFirestore'
+    );
+    const fs1 = getFirestore(app);
+    const fs2 = getFirestore(app);
+    expect(fs1 === fs2).to.be.true;
+  });
+});

packages/firestore/package.json

@@ -11,7 +11,8 @@
     "build": "rollup -c rollup.config.es2017.js && rollup -c rollup.config.es5.js",
     "build:deps": "lerna run --scope @firebase/'{app,firestore}' --include-dependencies build",
     "build:console": "node tools/console.build.js",
-    "build:exp": "rollup -c rollup.config.exp.js",
+    "build:exp": "rollup -c rollup.config.exp.js", 
+    "build:lite": "lerna run --scope @firebase/app-exp --include-dependencies build && rollup -c rollup.config.lite.js",
     "predev": "yarn prebuild",
     "dev": "rollup -c -w",
     "lint": "eslint -c .eslintrc.js '**/*.ts' --ignore-path '../../.gitignore'",
@@ -32,8 +33,11 @@
     "test:node:persistence:prod": "USE_MOCK_PERSISTENCE=YES TS_NODE_CACHE=NO TS_NODE_COMPILER_OPTIONS='{\"module\":\"commonjs\"}' nyc --reporter lcovonly -- mocha 'test/{,!(browser)/**/}*.test.ts' --require ts-node/register --require index.node.ts --require test/util/node_persistence.ts --config ../../config/mocharc.node.js",
     "test:travis": "ts-node --compiler-options='{\"module\":\"commonjs\"}' ../../scripts/emulator-testing/firestore-test-runner.ts",
     "test:minified": "(cd ../../integration/firestore ; yarn test)",
+    "pretest:lite": "yarn build:lite",
+    "test:lite":  "TS_NODE_CACHE=NO TS_NODE_COMPILER_OPTIONS='{\"module\":\"commonjs\"}' nyc --reporter lcovonly -- mocha 'lite/test/**/*.test.ts' --file lite/index.node.ts --config ../../config/mocharc.node.js",
     "prepare": "yarn build"
   },
+  "lite": "dist/lite/index.node.cjs.js",
   "main": "dist/index.node.cjs.js",
   "main-esm2017": "dist/index.node.esm2017.js",
   "browser": "dist/index.cjs.js",

packages/firestore/rollup.config.lite.js

@@ -0,0 +1,54 @@
+/**
+ * @license
+ * Copyright 2018 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 json from 'rollup-plugin-json';
+import typescriptPlugin from 'rollup-plugin-typescript2';
+import typescript from 'typescript';
+
+import { resolveNodeExterns } from './rollup.shared';
+
+import pkg from './package.json';
+
+const defaultPlugins = [
+  typescriptPlugin({
+    typescript,
+    tsconfigOverride: {
+      compilerOptions: {
+        target: 'es2017'
+      }
+    },
+    clean: true
+  }),
+  json({ preferConst: true })
+];
+
+const nodeBuilds = [
+  {
+    input: 'lite/index.node.ts',
+    output: {
+      file: pkg.lite,
+      format: 'es'
+    },
+    plugins: defaultPlugins,
+    external: resolveNodeExterns,
+    treeshake: {
+      tryCatchDeoptimization: false
+    }
+  }
+];
+
+export default [...nodeBuilds];