-
Notifications
You must be signed in to change notification settings - Fork 875
/
context.ts
111 lines (101 loc) · 2.91 KB
/
context.ts
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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
/**
* @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 {
FirebaseMessaging,
FirebaseMessagingName
} from '@firebase/messaging-types';
import {
FirebaseAuthInternal,
FirebaseAuthInternalName
} from '@firebase/auth-interop-types';
import { Provider } from '@firebase/component';
/**
* The metadata that should be supplied with function calls.
*/
export interface Context {
authToken?: string;
instanceIdToken?: string;
}
/**
* Helper class to get metadata that should be included with a function call.
*/
export class ContextProvider {
private auth: FirebaseAuthInternal | null = null;
private messaging: FirebaseMessaging | null = null;
constructor(
authProvider: Provider<FirebaseAuthInternalName>,
messagingProvider: Provider<FirebaseMessagingName>
) {
this.auth = authProvider.getImmediate({ optional: true });
this.messaging = messagingProvider.getImmediate({
optional: true
});
if (!this.auth) {
authProvider.get().then(
auth => (this.auth = auth),
() => {
/* get() never rejects */
}
);
}
if (!this.messaging) {
messagingProvider.get().then(
messaging => (this.messaging = messaging),
() => {
/* get() never rejects */
}
);
}
}
async getAuthToken(): Promise<string | undefined> {
if (!this.auth) {
return undefined;
}
try {
const token = await this.auth.getToken();
if (!token) {
return undefined;
}
return token.accessToken;
} catch (e) {
// If there's any error when trying to get the auth token, leave it off.
return undefined;
}
}
async getInstanceIdToken(): Promise<string | undefined> {
if (
!this.messaging ||
!('Notification' in self) ||
Notification.permission !== 'granted'
) {
return undefined;
}
try {
return this.messaging.getToken();
} catch (e) {
// We don't warn on this, because it usually means messaging isn't set up.
// console.warn('Failed to retrieve instance id token.', e);
// If there's any error when trying to get the token, leave it off.
return undefined;
}
}
async getContext(): Promise<Context> {
const authToken = await this.getAuthToken();
const instanceIdToken = await this.getInstanceIdToken();
return { authToken, instanceIdToken };
}
}