feat(core): register with email (#212)

packages/core/src/lib/register.ts

@@ -0,0 +1,99 @@
+import { PasscodeType } from '@logto/schemas';
+import { Context } from 'koa';
+import { Provider } from 'oidc-provider';
+
+import RequestError from '@/errors/RequestError';
+import { WithUserLogContext } from '@/middleware/koa-user-log';
+import { hasUser, hasUserWithEmail, insertUser } from '@/queries/user';
+import assertThat from '@/utils/assert-that';
+import { emailRegEx } from '@/utils/regex';
+
+import { createPasscode, sendPasscode, verifyPasscode } from './passcode';
+import { encryptUserPassword, generateUserId } from './user';
+
+const assignRegistrationResult = async (ctx: Context, provider: Provider, userId: string) => {
+  const redirectTo = await provider.interactionResult(
+    ctx.req,
+    ctx.res,
+    {
+      login: { accountId: userId },
+    },
+    { mergeWithLastSubmission: false }
+  );
+  ctx.body = { redirectTo };
+};
+
+export const registerWithUsernameAndPassword = async (
+  ctx: WithUserLogContext<Context>,
+  provider: Provider,
+  username: string,
+  password: string
+) => {
+  assertThat(
+    username && password,
+    new RequestError({
+      code: 'session.insufficient_info',
+      status: 400,
+    })
+  );
+  assertThat(
+    !(await hasUser(username)),
+    new RequestError({
+      code: 'user.username_exists',
+      status: 422,
+    })
+  );
+
+  const id = await generateUserId();
+
+  const { passwordEncryptionSalt, passwordEncrypted, passwordEncryptionMethod } =
+    encryptUserPassword(id, password);
+
+  await insertUser({
+    id,
+    username,
+    passwordEncrypted,
+    passwordEncryptionMethod,
+    passwordEncryptionSalt,
+  });
+
+  await assignRegistrationResult(ctx, provider, id);
+};
+
+export const sendPasscodeToEmail = async (ctx: Context, jti: string, email: string) => {
+  assertThat(emailRegEx.test(email), new RequestError('user.invalid_email'));
+  assertThat(
+    !(await hasUserWithEmail(email)),
+    new RequestError({
+      code: 'user.email_exists',
+      status: 422,
+    })
+  );
+
+  const passcode = await createPasscode(jti, PasscodeType.Register, { email });
+  await sendPasscode(passcode);
+  ctx.state = 204;
+};
+
+export const registerWithEmailAndPasscode = async (
+  ctx: WithUserLogContext<Context>,
+  provider: Provider,
+  { jti, email, code }: { jti: string; email: string; code: string }
+) => {
+  assertThat(
+    !(await hasUserWithEmail(email)),
+    new RequestError({
+      code: 'user.email_exists',
+      status: 422,
+    })
+  );
+  await verifyPasscode(jti, PasscodeType.Register, code, { email });
+
+  const id = await generateUserId();
+  await insertUser({
+    id,
+    primaryEmail: email,
+  });
+
+  await assignRegistrationResult(ctx, provider, id);
+};

packages/core/src/lib/sign-in.ts

@@ -6,7 +6,7 @@ import RequestError from '@/errors/RequestError';
 import { WithUserLogContext } from '@/middleware/koa-user-log';
 import { findUserByEmail, hasUserWithEmail } from '@/queries/user';
 import assertThat from '@/utils/assert-that';
-import { emailReg } from '@/utils/regex';
+import { emailRegEx } from '@/utils/regex';
 
 import { createPasscode, sendPasscode, verifyPasscode } from './passcode';
 import { findUserByUsernameAndPassword } from './user';
@@ -24,7 +24,7 @@ const assignSignInResult = async (ctx: Context, provider: Provider, userId: stri
 };
 
 export const sendSignInWithEmailPasscode = async (ctx: Context, jti: string, email: string) => {
-  assertThat(emailReg.test(email), new RequestError('user.invalid_email'));
+  assertThat(emailRegEx.test(email), new RequestError('user.invalid_email'));
   assertThat(
     await hasUserWithEmail(email),
     new RequestError({

packages/core/src/routes/session.ts

@@ -4,14 +4,17 @@ import { Provider } from 'oidc-provider';
 import { object, string } from 'zod';
 
 import RequestError from '@/errors/RequestError';
+import {
+  registerWithEmailAndPasscode,
+  registerWithUsernameAndPassword,
+  sendPasscodeToEmail,
+} from '@/lib/register';
 import {
   sendSignInWithEmailPasscode,
   signInWithEmailAndPasscode,
   signInWithUsernameAndPassword,
 } from '@/lib/sign-in';
-import { encryptUserPassword, generateUserId } from '@/lib/user';
 import koaGuard from '@/middleware/koa-guard';
-import { hasUser, insertUser } from '@/queries/user';
 import assertThat from '@/utils/assert-that';
 
 import { AnonymousRouter } from './types';
@@ -108,40 +111,27 @@ export default function sessionRoutes<T extends AnonymousRouter>(router: T, prov
     '/session/register',
     koaGuard({
       body: object({
-        username: string().min(3),
-        password: string().min(6),
+        username: string().min(3).optional(),
+        password: string().min(6).optional(),
+        email: string().optional(),
+        code: string().optional(),
       }),
     }),
     async (ctx, next) => {
-      const { username, password } = ctx.guard.body;
+      const interaction = await provider.interactionDetails(ctx.req, ctx.res);
+      const { jti } = interaction;
+      const { username, password, email, code } = ctx.guard.body;
 
-      if (await hasUser(username)) {
-        throw new RequestError('user.username_exists');
+      if (email && !code) {
+        await sendPasscodeToEmail(ctx, jti, email);
+      } else if (email && code) {
+        await registerWithEmailAndPasscode(ctx, provider, { jti, email, code });
+      } else if (username && password) {
+        await registerWithUsernameAndPassword(ctx, provider, username, password);
+      } else {
+        throw new RequestError('session.insufficient_info');
       }
 
-      const id = await generateUserId();
-
-      const { passwordEncryptionSalt, passwordEncrypted, passwordEncryptionMethod } =
-        encryptUserPassword(id, password);
-
-      await insertUser({
-        id,
-        username,
-        passwordEncrypted,
-        passwordEncryptionMethod,
-        passwordEncryptionSalt,
-      });
-
-      const redirectTo = await provider.interactionResult(
-        ctx.req,
-        ctx.res,
-        {
-          login: { accountId: id },
-        },
-        { mergeWithLastSubmission: false }
-      );
-      ctx.body = { redirectTo };
-
       return next();
     }
   );

packages/core/src/utils/regex.ts

@@ -1 +1 @@
-export const emailReg = /^\S+@\S+\.\S+$/;
+export const emailRegEx = /^\S+@\S+\.\S+$/;

packages/phrases/src/locales/en.ts

@@ -45,6 +45,7 @@ const errors = {
   },
   user: {
     username_exists: 'The username already exists.',
+    email_exists: 'The email already exists.',
     invalid_email: 'Invalid email address.',
     email_not_exists: 'The email address has not been registered yet.',
   },

packages/phrases/src/locales/zh-cn.ts

@@ -46,6 +46,7 @@ const errors = {
   },
   user: {
     username_exists: '用户名已存在。',
+    email_exists: '邮箱地址已存在。',
     invalid_email: '邮箱地址不正确。',
     email_not_exists: '邮箱地址尚未注册。',
   },