Function Login

api/users.go:268–311

(c *gin.Context)

}

func Login(c *gin.Context) {
	response := gin.H{"code": http.StatusOK}
	defer c.JSON(http.StatusOK, response)
	var (
		tokenString string
		err         error
	)
	switch c.Request.Method {
	case "GET", "get":
		tokenString = c.DefaultQuery("token", "")
		if _, err = ValidateToken(tokenString); err != nil {
			fmt.Println("validateToken error:", err.Error())
			response["code"] = http.StatusBadRequest
			response["message"] = err.Error()
			return
		}

	case "POST", "post":
		username, _ := c.GetPostForm("username")
		password, _ := c.GetPostForm("password")
		username, password = strings.TrimSpace(username), strings.TrimSpace(password)
		user := mydb.getUserProfile(username)
		if user == nil {
			response["code"] = http.StatusUnauthorized
			response["message"] = "user not found"
			return
		}
		if user.Password != password || user.Username != username {
			response["code"] = http.StatusUnauthorized
			response["message"] = "username or password is wrong"
			return
		}
		tokenString, err = GenerateToken(username, user)
		if err != nil {
			response["code"] = http.StatusInternalServerError
			response["message"] = "generate token failed"
			return
		}
		response["token"] = tokenString
	}
	c.SetCookie("token", tokenString, 86400, "/", "", false, true)
	response["code"] = http.StatusTemporaryRedirect
	response["link"] = "/"
}

func Logout(c *gin.Context) {
	tokenString, err := c.Cookie("token")