初次提交

api/user.go

@@ -0,0 +1,86 @@
+package api
+
+import (
+	"msm/consts/ctxflag"
+	"msm/consts/role"
+	"msm/dao"
+	"msm/model"
+	"msm/utils"
+
+	"github.com/gin-gonic/gin"
+)
+
+type userApi struct{}
+
+var UserApi = new(userApi)
+
+const DEFAULT_ROOT_PASSWORD = "root"
+
+func (u *userApi) LoginHandler(ctx *gin.Context) {
+	info := map[string]string{}
+	ctx.ShouldBindJSON(&info)
+	account := info["account"]
+	password := info["password"]
+	errCheck(ctx, !u.checkLoginInfo(account, password), "登入失败,账号或密码错误")
+	token, err := utils.GenToken(account)
+	errCheck(ctx, err != nil, err)
+	ctx.JSON(200, gin.H{
+		"code":     0,
+		"msg":      "登入成功！",
+		"token":    token,
+		"username": account,
+		"role":     dao.UserDao.GetUserByName(account).Role,
+	})
+}
+
+func (u *userApi) CreateUser(ctx *gin.Context) {
+	user := model.User{}
+	err := ctx.ShouldBindJSON(&user)
+	errCheck(ctx, err != nil, err)
+	errCheck(ctx, user.Role == int(role.ROOT), "不能添加root账号")
+	err = dao.UserDao.CreateUser(user)
+	errCheck(ctx, err != nil, err)
+	rOk(ctx, "注册成功", nil)
+}
+
+func (u *userApi) ChangePassword(ctx *gin.Context) {
+	user := model.User{}
+	err := ctx.ShouldBindJSON(&user)
+	errCheck(ctx, err != nil, err)
+	reqUser := ctx.GetString(ctxflag.USER_NAME)
+	errCheck(ctx, ctx.GetInt(ctxflag.ROLE) != int(role.ROOT) && user.Account != "", "参数错误")
+	var userName string
+	if user.Account != "" {
+		userName = user.Account
+	} else {
+		userName = reqUser
+	}
+	err = dao.UserDao.UpdatePassword(userName, user.Password)
+	errCheck(ctx, err != nil, err)
+	rOk(ctx, "修改密码成功", nil)
+
+}
+
+func (u *userApi) DeleteUser(ctx *gin.Context) {
+	errCheck(ctx, ctx.Query("account") == "root", "无法删除root账户")
+	err := dao.UserDao.DeleteUser(ctx.Query("account"))
+	errCheck(ctx, err != nil, "无法删除root账户")
+	rOk(ctx, "删除成功", nil)
+}
+
+func (u *userApi) GetUserList(ctx *gin.Context) {
+	rOk(ctx, "查询成功", dao.UserDao.GetUserList())
+}
+
+func (u *userApi) checkLoginInfo(account, password string) bool {
+	user := dao.UserDao.GetUserByName(account)
+	if account == "root" && user.Account == "" {
+		dao.UserDao.CreateUser(model.User{
+			Account:  "root",
+			Password: DEFAULT_ROOT_PASSWORD,
+			Role:     int(role.ROOT),
+		})
+		return password == DEFAULT_ROOT_PASSWORD
+	}
+	return user.Password == utils.Md5(password)
+}