fix(sso): auto assign roles for xd.com users only (#955)

leancloud · Dec 25, 2023 · 20a5764 · 20a5764
1 parent 6f911dd
commit 20a5764
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/oauth/xd-cas.js b/oauth/xd-cas.js
@@ -59,11 +59,16 @@ module.exports = (router) => {
                   useMasterKey: true,
                 })
                 console.log('new user created', user.id)
-                const role = await new AV.Query(AV.Role)
-                  .equalTo('name', process.env.AUTO_ASSIGN_XD_STAFF_ROLE ? 'staff' : 'collaborator')
-                  .first()
-                role.getUsers().add(user)
-                await role.save(undefined, { useMasterKey: true })
+                if (email.endsWith('@xd.com')) {
+                  const role = await new AV.Query(AV.Role)
+                    .equalTo(
+                      'name',
+                      process.env.AUTO_ASSIGN_XD_STAFF_ROLE ? 'staff' : 'collaborator'
+                    )
+                    .first()
+                  role.getUsers().add(user)
+                  await role.save(undefined, { useMasterKey: true })
+                }
                 return done(undefined, user)
               }
               return done(error)