auth-functions

Author: samtstern

auth/functions/custom_claims.js

@@ -1,8 +1,10 @@
 // [START auth_custom_claims_cloud_function]
 const functions = require('firebase-functions');
-const admin = require('firebase-admin');
+const { initializeApp } = require('firebase-admin/app');
+const { getAuth } = require('firebase-admin/auth');
+const { getDatabase } = require('firebase-admin/database');
 
-admin.initializeApp();
+initializeApp();
 
 // On sign up.
 exports.processSignUp = functions.auth.user().onCreate(async (user) => {
@@ -19,10 +21,10 @@ exports.processSignUp = functions.auth.user().onCreate(async (user) => {
 
     try {
       // Set custom user claims on this newly created user.
-      await admin.auth().setCustomUserClaims(user.uid, customClaims);
+      await getAuth().setCustomUserClaims(user.uid, customClaims);
 
       // Update real-time database to notify client to force refresh.
-      const metadataRef = admin.database().ref('metadata/' + user.uid);
+      const metadataRef = getDatabase().ref('metadata/' + user.uid);
 
       // Set the refresh time to the current UTC timestamp.
       // This will be captured on the client to force a token refresh.