feat: updateLoginMethod

controllers/common.go

@@ -1,5 +1,11 @@
 package controllers
 
+import (
+	"beego-members-api/models"
+
+	"github.com/beego/beego/v2/client/orm"
+)
+
 // DefaultSuccessResponse is ...
 type DefaultSuccessResponse struct {
 	// TODO: enumをサポートしたらenumに変更する
@@ -12,18 +18,21 @@ type DefaultErrorResponse struct {
 	Message string `json:"message" required:"true" example:"snaked_params" description:"result status"`
 }
 
-// func IsUserLoggedIn(c *BaseController) {
-// 	GetCurrentUser(c)
-// 	log.Print("-----------------")
-// 	log.Print("here")
-// 	log.Print("-----------------")
-// }
+// IsUserLoggedIn is ...
+func IsUserLoggedIn(userID int) bool {
+	if userID == 0 {
+		return false
+	}
+	return true
+}
 
 // GetCurrentUser is ...
-func GetCurrentUser(token string) {
-	// token := c.Ctx.GetCookie("access_token")
-	// log.Print("-----------------")
-	// log.Print(token)
-	// log.Print("-----------------")
-
+func GetCurrentUser(token string) int {
+	var accessToken models.AccessToken
+	o := orm.NewOrm()
+	err := o.QueryTable("access_token").Filter("token", token).One(&accessToken)
+	if err == orm.ErrNoRows {
+		return 0
+	}
+	return int(accessToken.User.Id)
 }

controllers/login.go

@@ -72,6 +72,6 @@ func (c *LoginController) Login() {
 	c.Data["json"] = mes
 	c.Ctx.SetCookie("token", token)
 	// swaggerでの開発確認用でcookieにもセットしています。
-	c.Ctx.SetCookie("access_token", token)
+	c.Ctx.SetCookie("access_token", accessToken)
 	c.ServeJSON()
 }

controllers/user.go

@@ -39,7 +39,14 @@ func (c *UserController) URLMapping() {
 // Prepare ...
 func (c *UserController) Prepare() {
 	token := c.Ctx.GetCookie("access_token")
-	GetCurrentUser(token)
+	if token == "" {
+		token = c.Ctx.Request.Header.Get("access_token")
+	}
+	if !IsUserLoggedIn(GetCurrentUser(token)) {
+		c.Ctx.Output.SetStatus(401)
+		c.Data["json"] = DefaultErrorResponse{Message: "user_un_authorized"}
+		c.ServeJSON()
+	}
 }
 
 // Post ...