Firebase Cloud Functions用户注册来源识别与用户等级配置问题

这确实是Firebase Auth触发器的一个小局限——默认的用户创建上下文里确实没有注册来源的信息。不过我们有几个靠谱的办法来解决这个问题，下面给你拆解一下：

方案1：客户端注册时传递来源信息到自定义用户声明 #

这个方法最直接，在用户注册成功后，客户端主动把注册来源（比如web或app）添加到用户的自定义声明里，然后云函数触发时读取这个声明来分配userLevel。

网站客户端代码示例： #

// 网站注册流程
firebase.auth().createUserWithEmailAndPassword(email, password)
  .then(async (userCredential) => {
    const user = userCredential.user;
    // 设置自定义声明，标记注册来源为web
    await firebase.auth().currentUser.setCustomUserClaims({
      registrationSource: "web"
    });
  })
  .catch((error) => {
    console.error(error);
  });

APP客户端代码示例（Android Kotlin）： #

// Android APP注册流程
val email = "user@example.com"
val password = "password123"
Firebase.auth.createUserWithEmailAndPassword(email, password)
    .addOnCompleteListener { task ->
        if (task.isSuccessful) {
            val user = task.result?.user
            user?.let {
                Firebase.auth.currentUser?.setCustomUserClaims(mapOf("registrationSource" to "app"))
            }
        } else {
            Log.w(TAG, "createUserWithEmail:failure", task.exception)
        }
    }

云函数处理代码： #

const functions = require("firebase-functions");
const admin = require("firebase-admin");
admin.initializeApp();

exports.onUserCreated = functions.auth.user().onCreate(async (user) => {
  // 获取自定义声明里的注册来源
  const userRecord = await admin.auth().getUser(user.uid);
  const registrationSource = userRecord.customClaims?.registrationSource;

  // 根据来源分配userLevel
  let userLevel;
  if (registrationSource === "web") {
    userLevel = "Power User";
  } else if (registrationSource === "app") {
    userLevel = "Regular User";
  } else {
    // 处理未知来源的情况，设置默认值
    userLevel = "Regular User";
    functions.logger.warn(`Unknown registration source for user ${user.uid}`);
  }

  // 将用户数据写入Firestore
  await admin.firestore().collection("users").doc(user.uid).set({
    email: user.email,
    displayName: user.displayName,
    userLevel: userLevel,
    createdAt: admin.firestore.FieldValue.serverTimestamp()
  });
});

关键注意点：一定要做安全校验！比如结合Firebase App Check来确认请求确实来自你的官方网站/APP，防止恶意用户篡改registrationSource的值。

方案2：注册后调用Callable云函数写入Firestore #

如果不想用自定义声明，也可以让客户端在注册成功后，调用一个Callable云函数，直接传递来源信息并完成Firestore写入。这种方式能让你在云函数里做更严格的身份校验。

客户端调用代码： #

// 客户端调用Callable函数
const setUserLevel = firebase.functions().httpsCallable('setUserLevel');
setUserLevel({ userId: user.uid, source: "web" })
  .then((result) => {
    console.log(result.data);
  });

云函数处理代码： #

exports.setUserLevel = functions.https.onCall(async (data, context) => {
  // 验证用户身份，确保是本人操作
  if (!context.auth || context.auth.uid !== data.userId) {
    throw new functions.https.HttpsError('unauthenticated', '无权限执行此操作');
  }

  const source = data.source;
  const userLevel = source === "web" ? "Power User" : "Regular User";

  // 写入Firestore，用merge避免覆盖已有数据
  await admin.firestore().collection("users").doc(data.userId).set({
    userLevel: userLevel,
    createdAt: admin.firestore.FieldValue.serverTimestamp()
  }, { merge: true });

  return { message: '用户等级设置成功' };
});

方案3：利用提供者信息（仅适用于第三方登录） #

如果你的用户是通过Google、Facebook等第三方登录的，不同端可能配置了不同的提供者（比如网站用web版Google登录，APP用原生Google登录），可以通过user.providerData里的信息判断来源，但这个方法不适用于邮箱密码注册的场景，所以优先级较低。

总结下来，方案1是最常用且高效的，完美适配邮箱密码注册的场景，只要做好安全校验就没问题。

内容的提问来源于stack exchange，提问作者najm