0904

2026-09-08 10:10:44 +08:00
commit 2ed27dd55e
288 changed files with 67061 additions and 0 deletions
@@ -0,0 +1,92 @@
+// Copyright 2022 E99p1ant. All rights reserved.
+// Use of this source code is governed by a MIT-style
+// license that can be found in the LICENSE file.
+
+package auth
+
+import (
+	"github.com/flamego/recaptcha"
+	"github.com/pkg/errors"
+	"github.com/sirupsen/logrus"
+
+	"github.com/tamakyi/TamaBox/internal/captcha"
+	"github.com/tamakyi/TamaBox/internal/conf"
+	"github.com/tamakyi/TamaBox/internal/context"
+	"github.com/tamakyi/TamaBox/internal/db"
+	"github.com/tamakyi/TamaBox/internal/form"
+	"github.com/tamakyi/TamaBox/internal/i18n"
+	"github.com/tamakyi/TamaBox/internal/shared"
+)
+
+func Register(ctx context.Context) {
+	ctx.Success("auth/register")
+}
+
+func RegisterAction(ctx context.Context, f form.Register, recaptcha recaptcha.RecaptchaV3) {
+	if ctx.HasError() {
+		ctx.Success("auth/register")
+		return
+	}
+
+	// Check captcha code (unified: recaptcha, turnstile, or math).
+	ok, err := captcha.Verify(f.Recaptcha, shared.ResolveClientIP(ctx.Request().Request), ctx.Session, recaptcha, ctx.Lang())
+	if err != nil {
+		logrus.WithContext(ctx.Request().Context()).WithError(err).Error("Failed to check captcha")
+		ctx.SetErrorFlash(err.Error())
+		ctx.Redirect("/register")
+		return
+	}
+	if !ok {
+		ctx.SetErrorFlash(i18n.T(ctx.Lang(), "captcha.error"))
+		ctx.Redirect("/register")
+		return
+	}
+
+	if err := db.Users.Create(ctx.Request().Context(), db.CreateUserOptions{
+		Name:                 f.Name,
+		Password:             f.Password,
+		Email:                f.Email,
+		Avatar:               conf.Upload.DefaultAvatarURL,
+		Domain:               f.Domain,
+		Background:           conf.Upload.DefaultBackground,
+		Intro:                "问你想问的",
+		Qrcodebackcolor:      "#FDF8F0",
+		Qrcodecolor:          "#C45500",
+		Usernamecolor:        "#000000",
+		Introcolor:           "#000000",
+		Dotscale:             "0.35",
+		BackgroundImage:      conf.Upload.DefaultBackgroundImageURL,
+		Backgroundimagealpha: "0.8",
+		Qrcodepdpcolor:       "#C45500",
+		Acfunlink:            "none",
+		Bililink:             "none",
+		Weibolink:            "none",
+		Bloglink:             "box.shiroko.one",
+		Xiaohongshulink:      "none",
+		Language:             i18n.Resolve(f.Language), // 注册时选择的界面语言；留空回退站点默认
+	}); err != nil {
+		switch {
+		case errors.Is(err, db.ErrDuplicateEmail):
+			ctx.SetError(errors.New(i18n.T(ctx.Lang(), "auth.errDuplicateEmail")))
+		case errors.Is(err, db.ErrDuplicateDomain):
+			ctx.SetError(errors.New(i18n.T(ctx.Lang(), "auth.errDuplicateDomain")))
+		case errors.Is(err, db.ErrBadCredential):
+			ctx.SetError(errors.New(i18n.T(ctx.Lang(), "auth.errBadCredential")))
+
+		default:
+			logrus.WithContext(ctx.Request().Context()).WithError(err).Error("Failed to create new user")
+			ctx.SetInternalError()
+		}
+
+		ctx.Success("auth/register")
+		return
+	}
+
+	// 成功提示跟随用户注册时选择的语言
+	lang := "zh-CN"
+	if l, ok := ctx.Data["Lang"].(string); ok && l != "" {
+		lang = l
+	}
+	ctx.SetSuccessFlash(i18n.T(lang, "auth.register.success"))
+	ctx.Redirect("/login")
+}