如何在Supabase OAuth登录时传递自定义用户角色数据？

针对你的需求，有两种可靠的实现方式，分别适合不同场景：

方案一：后端触发器自动处理（推荐，安全可靠） #

通过Supabase的Postgres触发器，在新用户创建时自动为其分配角色，无需前端额外处理，且避免前端篡改角色的风险。

操作步骤： #

1. 登录Supabase控制台，打开SQL编辑器
2. 执行以下SQL代码，创建处理新用户的函数和触发器：

-- 创建函数：为新用户设置默认角色（可根据需求修改默认值）
CREATE OR REPLACE FUNCTION public.handle_new_user()
RETURNS TRIGGER AS $$
BEGIN
  -- 为新用户的元数据添加角色字段，默认设为'user'
  UPDATE auth.users 
  SET raw_user_meta_data = raw_user_meta_data || '{"role": "user"}'::jsonb
  WHERE id = new.id;

  -- 可选：将用户信息同步到自定义的profiles表（如果你有该表）
  INSERT INTO public.profiles (id, full_name, role)
  VALUES (new.id, new.raw_user_meta_data->>'full_name', 'user')
  ON CONFLICT (id) DO NOTHING;
  
  RETURN NEW;
END;
$$ LANGUAGE plpgsql SECURITY DEFINER;

-- 创建触发器：监听auth.users表的新增事件，自动触发上述函数
CREATE TRIGGER on_auth_user_created
AFTER INSERT ON auth.users
FOR EACH ROW EXECUTE FUNCTION public.handle_new_user();

说明： #

• 该方案对所有注册方式（邮箱密码、OAuth）生效，自动为新用户分配角色
• 如果需要区分OAuth和邮箱注册的角色，可在函数中通过new.raw_user_meta_data->>'provider'判断登录来源，再设置对应角色

方案二：前端传递角色参数，回调路由更新元数据 #

如果需要让用户在登录前自主选择角色（如登录页面提供user/organisation选项），可通过以下步骤实现：

1. 修改OAuth登录代码，追加角色参数到跳转地址 #

// 假设用户选择的角色存储在selectedRole变量中
await supabase.auth.signInWithOAuth({
  provider: "google",
  options: {
    queryParams: {
      access_type: "offline",
      prompt: "consent",
    },
    // 将角色作为查询参数追加到回调地址
    redirectTo: `${OAUTH_REDIRECT_URL}?role=${selectedRole}`,
  },
});

2. 修改auth/callback路由，获取角色并更新用户元数据 #

import { createRouteHandlerClient } from "@supabase/auth-helpers-nextjs";
import { cookies } from "next/headers";
import { NextRequest, NextResponse } from "next/server";

export async function GET(req: NextRequest) {
  const requestUrl = new URL(req.url);
  const { searchParams } = requestUrl;
  const code = searchParams.get("code");
  const role = searchParams.get("role");

  if (code) {
    const supabase = createRouteHandlerClient({ cookies });
    const { data: { session } } = await supabase.auth.exchangeCodeForSession(code);
    
    // 验证角色合法性，仅允许预设的两种角色
    const validRoles = ["user", "organisation"];
    if (session?.user && role && validRoles.includes(role)) {
      await supabase.auth.updateUser({
        data: {
          role: role,
        },
      });
    }
  }

  // 跳转至dashboard，移除URL中的角色参数
  return NextResponse.redirect(`${requestUrl.origin}/dashboard`);
}

说明： #

• 必须添加角色合法性校验，防止恶意参数注入
• 可通过session.user.created_at判断是否为新用户，仅对首次登录的用户设置角色

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