Feat: User/Me

routers/controllers/main.go

@@ -1,8 +1,10 @@
 package controllers
 
 import (
+	"cloudreve/models"
 	"cloudreve/pkg/serializer"
 	"encoding/json"
+	"github.com/gin-gonic/gin"
 	"gopkg.in/go-playground/validator.v8"
 )
 
@@ -47,3 +49,13 @@ func ErrorResponse(err error) serializer.Response {
 
 	return serializer.ParamErr("参数错误", err)
 }
+
+// CurrentUser 获取当前用户
+func CurrentUser(c *gin.Context) *model.User {
+	if user, _ := c.Get("user"); user != nil {
+		if u, ok := user.(*model.User); ok {
+			return u
+		}
+	}
+	return nil
+}