0904

internal/context/auth.go

@@ -0,0 +1,57 @@
+// 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 context
+
+import (
+	"github.com/flamego/flamego"
+	"github.com/flamego/session"
+
+	"github.com/tamakyi/TamaBox/internal/db"
+	"github.com/tamakyi/TamaBox/internal/i18n"
+)
+
+// authenticatedUser returns the user object of the authenticated user.
+func authenticatedUser(ctx flamego.Context, sess session.Session) *db.User {
+	uid, ok := sess.Get("uid").(uint)
+	if !ok {
+		return nil
+	}
+
+	user, err := db.Users.GetByID(ctx.Request().Context(), uid)
+	if err != nil || user == nil {
+		return nil
+	}
+
+	// 密码变更后，旧会话保存的版本号不匹配，强制重新登录。
+	ver, _ := sess.Get("pwd_ver").(uint)
+	if user.PasswordVersion != ver {
+		return nil
+	}
+	return user
+}
+
+type ToggleOptions struct {
+	UserSignInRequired  bool
+	UserSignOutRequired bool
+}
+
+func Toggle(options *ToggleOptions) flamego.Handler {
+	return func(ctx Context, endpoint EndpointType) error {
+		if options.UserSignOutRequired && ctx.IsLogged {
+			ctx.Redirect("/")
+			return nil
+		}
+
+		if options.UserSignInRequired && !ctx.IsLogged {
+			if endpoint.IsAPI() {
+				return ctx.JSONError(40100, i18n.T(ctx.Lang(), "auth.errLoginRequired"))
+			}
+			ctx.SetErrorFlash(i18n.T(ctx.Lang(), "auth.errLoginRequired"))
+			ctx.Redirect("/login")
+			return nil
+		}
+		return nil
+	}
+}