From d0bea74645f17537726d64429d5e053dcfbab397 Mon Sep 17 00:00:00 2001
From: undefined <i@undefined.moe>
Date: Sun, 28 Nov 2021 17:14:39 +0800
Subject: [PATCH] core: allow set registration_redirect for domain

---
 packages/hydrooj/package.json        | 2 +-
 packages/hydrooj/src/handler/user.ts | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/packages/hydrooj/package.json b/packages/hydrooj/package.json
index d9e8556bb..7303b6858 100644
--- a/packages/hydrooj/package.json
+++ b/packages/hydrooj/package.json
@@ -1,6 +1,6 @@
 {
     "name": "hydrooj",
-    "version": "3.1.40",
+    "version": "3.1.41",
     "bin": "bin/hydrooj.js",
     "main": "src/loader",
     "module": "src/loader",
diff --git a/packages/hydrooj/src/handler/user.ts b/packages/hydrooj/src/handler/user.ts
index 887602066..860a30865 100644
--- a/packages/hydrooj/src/handler/user.ts
+++ b/packages/hydrooj/src/handler/user.ts
@@ -162,7 +162,7 @@ export class UserRegisterHandler extends Handler {
             const t = await token.add(
                 token.TYPE_REGISTRATION,
                 system.get('session.unsaved_expire_seconds'),
-                { mail },
+                { mail, redirect: this.domain.registerRedirect },
             );
             if (system.get('smtp.verify') && system.get('smtp.user')) {
                 const m = await this.renderHTML('user_register_mail.html', {
@@ -223,7 +223,7 @@ class UserRegisterWithCodeHandler extends Handler {
         this.session.viewLang = '';
         this.session.uid = uid;
         this.session.scpoe = PERM.PERM_ALL.toString();
-        this.response.redirect = this.url('home_settings', { category: 'preference' });
+        this.response.redirect = tdoc.redirect || this.url('home_settings', { category: 'preference' });
     }
 }