110 lines
3.1 KiB
Go
110 lines
3.1 KiB
Go
package controller
|
||
|
||
import (
|
||
"yitao/config"
|
||
"yitao/ecode"
|
||
"yitao/util"
|
||
"yitao/validate"
|
||
|
||
"github.com/iris-contrib/middleware/jwt"
|
||
"github.com/kataras/iris/v12/mvc"
|
||
)
|
||
|
||
type UserController struct {
|
||
BaseController
|
||
}
|
||
|
||
func (con *UserController) BeforeActivation(b mvc.BeforeActivation) {
|
||
// Login Api
|
||
{
|
||
b.Handle("POST", "/auth/login", "AuthLogin")
|
||
b.Handle("POST", "/auth/signup", "AuthSignup")
|
||
}
|
||
}
|
||
|
||
// @Summary login api
|
||
// @Description the 用户登录接口
|
||
// @Tags auth api
|
||
// @Accept json
|
||
// @Produce json
|
||
// @Param loginParam body validate.LoginParam true "登录参数"
|
||
// @Success 200 {object} map[string]interface{} "{"jwt": "jwt.jwt.jwt"}"
|
||
// @Failure 400 {object} map[string]interface{} "{"msg": "错误信息","code":0}"
|
||
// @Router /api/user/auth/login [post]
|
||
func (con *UserController) AuthLogin() mvc.Result {
|
||
login_param := new(validate.LoginParam)
|
||
e := validate.ReadJSON(con.Ctx, login_param)
|
||
if e.Error() {
|
||
return e.Response()
|
||
}
|
||
|
||
e = con.Service.Captcha.Validate(login_param.CaptchaKey, login_param.CaptchaData)
|
||
if e.Error() {
|
||
return e.Response()
|
||
}
|
||
|
||
user, e := con.Service.User.AuthLogin(login_param.Username, login_param.Password)
|
||
if e.Error() {
|
||
return e.Response()
|
||
}
|
||
|
||
// 写入登录日志
|
||
ip := con.Ctx.Request().RemoteAddr
|
||
// 一个测试ip
|
||
//ip = "109.123.229.220" // 日本 东京
|
||
// 获取 ip 地址
|
||
city, country, _ := util.GetCityAndCountry(ip)
|
||
con.Service.LoginLog.Save(user.ID, ip, "Web", country, city)
|
||
|
||
// 原版 NP 在这里会触发一个对登录地点的检测，但我看数据库它就没成功获取到过国家于城市
|
||
// 所以就先不做异地检测了
|
||
// @TODO 如果有人愿意的话在这做个异地检测
|
||
// 好吧现在好了，但我懒得写了，就这样吧
|
||
|
||
// 开始生成jwt
|
||
conf := config.GetConfig()
|
||
token := jwt.NewTokenWithClaims(jwt.SigningMethodHS256, jwt.MapClaims{
|
||
"uid": user.ID,
|
||
"username": user.Username,
|
||
"time": util.GetTime(),
|
||
"is_admin": user.IsAdmin,
|
||
})
|
||
// 使用设置的秘钥，签名生成jwt字符串
|
||
tokenString, _ := token.SignedString([]byte(conf.JwtSecret))
|
||
|
||
return mvc.Response{
|
||
Object: map[string]interface{}{
|
||
"jwt": tokenString,
|
||
},
|
||
}
|
||
}
|
||
|
||
// @Summary 注册API
|
||
// @Description 用户注册接口
|
||
// @Tags auth api
|
||
// @Accept json
|
||
// @Produce json
|
||
// @Param signupParam body validate.SignupParam true "注册参数"
|
||
// @Success 200 {object} map[string]interface{} "{"msg": "注册成功","code":1}"
|
||
// @Failure 400 {object} map[string]interface{} "{"msg": "错误信息","code":0}"
|
||
// @Router /api/user/auth/signup [post]
|
||
func (con *UserController) AuthSignup() mvc.Result {
|
||
var e *ecode.Ecode
|
||
signup_param := new(validate.SignupParam)
|
||
e = validate.ReadJSON(con.Ctx, signup_param)
|
||
if e.Error() {
|
||
return e.Response()
|
||
}
|
||
|
||
e = con.Service.Captcha.Validate(signup_param.CaptchaKey, signup_param.CaptchaData)
|
||
if e.Error() {
|
||
return e.Response()
|
||
}
|
||
|
||
_, e = con.Service.User.AuthSignup(signup_param.Username, signup_param.Password)
|
||
if e.Error() {
|
||
return e.Response()
|
||
}
|
||
return ecode.OK().Response()
|
||
}
|