1 | 'use strict';
|
2 |
|
3 | Object.defineProperty(exports, '__esModule', { value: true });
|
4 |
|
5 | var firebase = require('firebase');
|
6 | var request = require('request');
|
7 | var util = require('@firebase/util');
|
8 | var logger = require('@firebase/logger');
|
9 | var component = require('@firebase/component');
|
10 |
|
11 |
|
12 |
|
13 |
|
14 |
|
15 |
|
16 |
|
17 |
|
18 |
|
19 |
|
20 |
|
21 |
|
22 |
|
23 |
|
24 |
|
25 |
|
26 | function __awaiter(thisArg, _arguments, P, generator) {
|
27 | function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
|
28 | return new (P || (P = Promise))(function (resolve, reject) {
|
29 | function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
|
30 | function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
|
31 | function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
|
32 | step((generator = generator.apply(thisArg, _arguments || [])).next());
|
33 | });
|
34 | }
|
35 |
|
36 | function __generator(thisArg, body) {
|
37 | var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g;
|
38 | return g = { next: verb(0), "throw": verb(1), "return": verb(2) }, typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
|
39 | function verb(n) { return function (v) { return step([n, v]); }; }
|
40 | function step(op) {
|
41 | if (f) throw new TypeError("Generator is already executing.");
|
42 | while (_) try {
|
43 | if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
|
44 | if (y = 0, t) op = [op[0] & 2, t.value];
|
45 | switch (op[0]) {
|
46 | case 0: case 1: t = op; break;
|
47 | case 4: _.label++; return { value: op[1], done: false };
|
48 | case 5: _.label++; y = op[1]; op = [0]; continue;
|
49 | case 7: op = _.ops.pop(); _.trys.pop(); continue;
|
50 | default:
|
51 | if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
|
52 | if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
|
53 | if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
|
54 | if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
|
55 | if (t[2]) _.ops.pop();
|
56 | _.trys.pop(); continue;
|
57 | }
|
58 | op = body.call(thisArg, _);
|
59 | } catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
|
60 | if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
|
61 | }
|
62 | }
|
63 |
|
64 |
|
65 |
|
66 |
|
67 |
|
68 |
|
69 |
|
70 |
|
71 |
|
72 |
|
73 |
|
74 |
|
75 |
|
76 |
|
77 |
|
78 |
|
79 |
|
80 |
|
81 | var DATABASE_ADDRESS_ENV = 'FIREBASE_DATABASE_EMULATOR_ADDRESS';
|
82 |
|
83 | var DATABASE_ADDRESS_DEFAULT = 'localhost:9000';
|
84 |
|
85 | var DATABASE_ADDRESS = process.env[DATABASE_ADDRESS_ENV] || DATABASE_ADDRESS_DEFAULT;
|
86 |
|
87 | var FIRESTORE_ADDRESS_ENVS = [
|
88 | 'FIRESTORE_EMULATOR_HOST',
|
89 | 'FIREBASE_FIRESTORE_EMULATOR_ADDRESS'
|
90 | ];
|
91 |
|
92 | var FIRESTORE_ADDRESS_DEFAULT = 'localhost:8080';
|
93 |
|
94 | var FIRESTORE_ADDRESS = FIRESTORE_ADDRESS_ENVS.reduce(function (addr, name) { return process.env[name] || addr; }, FIRESTORE_ADDRESS_DEFAULT);
|
95 |
|
96 | var ADMIN_TOKEN = 'owner';
|
97 |
|
98 | function createUnsecuredJwt(auth) {
|
99 |
|
100 | var header = {
|
101 | alg: 'none',
|
102 | kid: 'fakekid'
|
103 | };
|
104 |
|
105 | auth.iat = auth.iat || 0;
|
106 |
|
107 | auth.sub = auth.sub || auth.uid;
|
108 | if (!auth.sub) {
|
109 | throw new Error("auth must be an object with a 'sub' or 'uid' field");
|
110 | }
|
111 |
|
112 | var signature = '';
|
113 | return [
|
114 | util.base64.encodeString(JSON.stringify(header), false),
|
115 | util.base64.encodeString(JSON.stringify(auth), false),
|
116 | signature
|
117 | ].join('.');
|
118 | }
|
119 | function apps() {
|
120 | return firebase.apps;
|
121 | }
|
122 |
|
123 | function initializeTestApp(options) {
|
124 | return initializeApp(options.auth ? createUnsecuredJwt(options.auth) : undefined, options.databaseName, options.projectId);
|
125 | }
|
126 |
|
127 | function initializeAdminApp(options) {
|
128 | return initializeApp(ADMIN_TOKEN, options.databaseName, options.projectId);
|
129 | }
|
130 | function initializeApp(accessToken, databaseName, projectId) {
|
131 | var _this = this;
|
132 | var appOptions = {};
|
133 | if (databaseName) {
|
134 | appOptions['databaseURL'] = "http://" + DATABASE_ADDRESS + "?ns=" + databaseName;
|
135 | }
|
136 | if (projectId) {
|
137 | appOptions['projectId'] = projectId;
|
138 | }
|
139 | var appName = 'app-' + new Date().getTime() + '-' + Math.random();
|
140 | var app = firebase.initializeApp(appOptions, appName);
|
141 | if (accessToken) {
|
142 | var mockAuthComponent = new component.Component('auth-internal', function () {
|
143 | return ({
|
144 | getToken: function () { return __awaiter(_this, void 0, void 0, function () { return __generator(this, function (_a) {
|
145 | return [2 , ({ accessToken: accessToken })];
|
146 | }); }); },
|
147 | getUid: function () { return null; },
|
148 | addAuthTokenListener: function (listener) {
|
149 |
|
150 | listener(accessToken);
|
151 | },
|
152 | removeAuthTokenListener: function () { }
|
153 | });
|
154 | }, "PRIVATE" );
|
155 | app._addOrOverwriteComponent(mockAuthComponent);
|
156 | }
|
157 | if (databaseName) {
|
158 |
|
159 |
|
160 |
|
161 | app.database().goOffline();
|
162 | app.database().goOnline();
|
163 | }
|
164 | if (projectId) {
|
165 | app.firestore().settings({
|
166 | host: FIRESTORE_ADDRESS,
|
167 | ssl: false
|
168 | });
|
169 | }
|
170 | |
171 |
|
172 |
|
173 |
|
174 | logger.setLogLevel(logger.LogLevel.ERROR);
|
175 | return app;
|
176 | }
|
177 | function loadDatabaseRules(options) {
|
178 | if (!options.databaseName) {
|
179 | throw Error('databaseName not specified');
|
180 | }
|
181 | if (!options.rules) {
|
182 | throw Error('must provide rules to loadDatabaseRules');
|
183 | }
|
184 | return new Promise(function (resolve, reject) {
|
185 | request.put({
|
186 | uri: "http://" + DATABASE_ADDRESS + "/.settings/rules.json?ns=" + options.databaseName,
|
187 | headers: { Authorization: 'Bearer owner' },
|
188 | body: options.rules
|
189 | }, function (err, resp, body) {
|
190 | if (err) {
|
191 | reject(err);
|
192 | }
|
193 | else if (resp.statusCode !== 200) {
|
194 | reject(JSON.parse(body).error);
|
195 | }
|
196 | else {
|
197 | resolve();
|
198 | }
|
199 | });
|
200 | });
|
201 | }
|
202 | function loadFirestoreRules(options) {
|
203 | if (!options.projectId) {
|
204 | throw new Error('projectId not specified');
|
205 | }
|
206 | if (!options.rules) {
|
207 | throw new Error('must provide rules to loadFirestoreRules');
|
208 | }
|
209 | return new Promise(function (resolve, reject) {
|
210 | request.put({
|
211 | uri: "http://" + FIRESTORE_ADDRESS + "/emulator/v1/projects/" + options.projectId + ":securityRules",
|
212 | body: JSON.stringify({
|
213 | rules: {
|
214 | files: [{ content: options.rules }]
|
215 | }
|
216 | })
|
217 | }, function (err, resp, body) {
|
218 | if (err) {
|
219 | reject(err);
|
220 | }
|
221 | else if (resp.statusCode !== 200) {
|
222 | console.log('body', body);
|
223 | reject(JSON.parse(body).error);
|
224 | }
|
225 | else {
|
226 | resolve();
|
227 | }
|
228 | });
|
229 | });
|
230 | }
|
231 | function clearFirestoreData(options) {
|
232 | if (!options.projectId) {
|
233 | throw new Error('projectId not specified');
|
234 | }
|
235 | return new Promise(function (resolve, reject) {
|
236 | request.delete({
|
237 | uri: "http://" + FIRESTORE_ADDRESS + "/emulator/v1/projects/" + options.projectId + "/databases/(default)/documents",
|
238 | body: JSON.stringify({
|
239 | database: "projects/" + options.projectId + "/databases/(default)"
|
240 | })
|
241 | }, function (err, resp, body) {
|
242 | if (err) {
|
243 | reject(err);
|
244 | }
|
245 | else if (resp.statusCode !== 200) {
|
246 | console.log('body', body);
|
247 | reject(JSON.parse(body).error);
|
248 | }
|
249 | else {
|
250 | resolve();
|
251 | }
|
252 | });
|
253 | });
|
254 | }
|
255 | function assertFails(pr) {
|
256 | return pr.then(function (v) {
|
257 | return Promise.reject(new Error('Expected request to fail, but it succeeded.'));
|
258 | }, function (err) { return err; });
|
259 | }
|
260 | function assertSucceeds(pr) {
|
261 | return pr;
|
262 | }
|
263 |
|
264 | Object.defineProperty(exports, 'database', {
|
265 | enumerable: true,
|
266 | get: function () {
|
267 | return firebase.database;
|
268 | }
|
269 | });
|
270 | Object.defineProperty(exports, 'firestore', {
|
271 | enumerable: true,
|
272 | get: function () {
|
273 | return firebase.firestore;
|
274 | }
|
275 | });
|
276 | exports.apps = apps;
|
277 | exports.assertFails = assertFails;
|
278 | exports.assertSucceeds = assertSucceeds;
|
279 | exports.clearFirestoreData = clearFirestoreData;
|
280 | exports.initializeAdminApp = initializeAdminApp;
|
281 | exports.initializeTestApp = initializeTestApp;
|
282 | exports.loadDatabaseRules = loadDatabaseRules;
|
283 | exports.loadFirestoreRules = loadFirestoreRules;
|
284 |
|