light-apiengine/controllers/SystemController.go

package controllers

import (
	"fmt"
	"light-apiengine/models"
	"strconv"
	"strings"
	"time"

	"git.qianqiusoft.com/qianqiusoft/light-apiengine/entitys"
	sysmodel "git.qianqiusoft.com/qianqiusoft/light-apiengine/models"
	"git.qianqiusoft.com/qianqiusoft/light-apiengine/utils"
)

//https://www.kancloud.cn/xormplus/xorm/167077

// _Login
// @Title _Login
// @Description 用户登录
// @Param	logininfo      false  "登录信息"
// @Success 200 {object} sysmodel.Account
// @Failure 403 :id is empty
func System_Login(c *SystemController) {
	var logininfo sysmodel.LoginInfo
	c.Ctx.BindJSON(&logininfo)
	__none_func_system__(logininfo)
	fmt.Println(logininfo)
	var user sysmodel.SysUser
	ret, err := c.Db.SQL(sysmodel.SqlUserLogin, logininfo.Account).Get(&user)
	if ret && err == nil {
		//TODO check password
		timestamp := uint64(time.Now().UnixNano())
		md5Pwd := utils.HashPassword(logininfo.Password, "")

		//密码错误
		if !strings.EqualFold(user.Password, md5Pwd) {
			c.Ctx.JSON(200, sysmodel.SysReturn{400, "passowrd incorrect!", nil})
			return
		}

		token := &entitys.Token{}
		if v := utils.GlobalTokenStore.Get(logininfo.Account + user.Domain); v == nil {
			timestamp_str := strconv.FormatUint(timestamp, 10)
			sec_tooken := utils.GenerateToken(logininfo.Account + timestamp_str)
			token.AccessToken = sec_tooken
			token.RefreshToken = sec_tooken
			token.LoginID = logininfo.Account
			token.UserId = user.Id
			token.Result = 200
			//token.Password = pwd
			token.ServerIp = ""
			token.Domain = user.Domain
			utils.GlobalTokenStore.Set(sec_tooken, token)

			//设置session
			//fmt.Println("----------set token ----------", token.AccessToken)
			//session := sessions.Default(c.Ctx)
			//session.Set("token", token.AccessToken)
			//session.Save()

		} else {
			token = v
		}

		data := sysmodel.LoginReturnInfo{user.Id, user.LoginId, token.AccessToken}
		c.Ctx.JSON(200, sysmodel.SysReturn{200, "", data})
	} else {
		//fmt.Println(err.Error())
		c.Ctx.JSON(200, sysmodel.SysReturn{400, "username or passowrd incorrect!", nil})
	}

}

// _Logout
// @Title _Logout
// @Description 用户退出
// @Success 200 {object} sysmodel.Account
// @Failure 403 :id is empty
func System_Logout(c *SystemController) {

	token := c.Ctx.GetHeader("token")
	fmt.Println("delete token: ", token)
	utils.GlobalTokenStore.Remove(token)

	ret := __none_func_system__()
	if ret {
		c.Ctx.JSON(200, sysmodel.SysReturn{200, "", nil})
	} else {
		c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
	}
}

// _GetMenuTree
// @Title _GetMenuTree
// @Description 获取系统菜单
// @Param	user    string  false  "用户id"
// @Success 200 {object} sysmodel.Account
// @Failure 403 :id is empty
func System_GetMenuTree(c *SystemController) {
	user := c.Ctx.Param(":user")
	ret := __none_func_system__(user)
	var menus []sysmodel.SysMenuTree
	c.Db.SQL(sysmodel.Selectall_sys_menu).Find(&menus)

	var menutree []sysmodel.SysMenuTree
	finChildrenMenu(&menus, "", nil, &menutree)
	for i := 0; i < len(menutree); i++ {
		finChildrenMenu(&menus, menutree[i].Id, &menutree[i], &menutree)
	}
	if ret {
		c.Ctx.JSON(200, sysmodel.SysReturn{200, "", menutree})
	}
}

// _GetOrgTree
// @Title _GetOrgTree
// @Description 获取组织架构树
// @Success 200 {object} sysmodel.Account
// @Failure 403 :id is empty
func System_GetOrgTree(c *SystemController) {

	ret := __none_func_system__()

	var orgs []sysmodel.SysOrgTree
	c.Db.SQL(sysmodel.Selectall_sys_org).Find(&orgs)

	var orgtree []sysmodel.SysOrgTree
	finChildrenOrg(&orgs, "", nil, &orgtree)
	for i := 0; i < len(orgtree); i++ {
		finChildrenOrg(&orgs, orgtree[i].Id, &orgtree[i], &orgtree)
	}

	if ret {
		c.Ctx.JSON(200, sysmodel.SysReturn{200, "", orgtree})
	} else {
		c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
	}
}

func finChildrenOrg(menus *[]sysmodel.SysOrgTree, parent string, parent_menu *sysmodel.SysOrgTree, menutree *[]sysmodel.SysOrgTree) {
	for _, m := range *menus {
		if m.Parent == parent {
			if parent_menu == nil {
				*menutree = append(*menutree, m)
				//finChildrenMenu(menus, m.Id, menutree[len(*menutree)-1], menutree)
			} else {
				parent_menu.Children = append(parent_menu.Children, m)
				finChildrenOrg(menus, m.Id, &parent_menu.Children[len(parent_menu.Children)-1], menutree)
			}
		}
	}
}

func finChildrenMenu(menus *[]sysmodel.SysMenuTree, parent string, parent_menu *sysmodel.SysMenuTree, menutree *[]sysmodel.SysMenuTree) {
	for _, m := range *menus {
		if m.Parent == parent {
			if parent_menu == nil {
				*menutree = append(*menutree, m)
				//finChildrenMenu(menus, m.Id, menutree[len(*menutree)-1], menutree)
			} else {
				parent_menu.Children = append(parent_menu.Children, m)
				finChildrenMenu(menus, m.Id, &parent_menu.Children[len(parent_menu.Children)-1], menutree)
			}
		}
	}
}

// _FindUserPage
// @Title _FindUserPage
// @Description 获取用户分布数据
// @Param	page      false  "分页参数"
// @Success 200 {object} sysmodel.Account
// @Failure 403 :id is empty
func System_FindUserPage(c *SystemController) {

	var getpageinfo sysmodel.GetPageInfo
	c.Ctx.BindJSON(&getpageinfo)
	__none_func_system__(getpageinfo)
	fmt.Println(getpageinfo)
	page := sysmodel.SysUserPageInfo{}
	var users []sysmodel.UserInfo
	err := c.Db.SQL(sysmodel.SqlUserInfoPage).Limit(getpageinfo.PageSize, (getpageinfo.PageNum-1)*getpageinfo.PageSize).Find(&users)

	//查找每个用户的角色
	for i, user := range users {
		roles := []sysmodel.SysUserRole{}
		c.Db.SQL(sysmodel.SqlSelectUserRoles, user.Id).Find(&roles)
		users[i].UserRole = roles
	}

	page.Content = users
	page.PageSize = getpageinfo.PageSize
	page.PageNum = getpageinfo.PageNum
	page.TotalSize = 1
	if err == nil {
		c.Ctx.JSON(200, sysmodel.SysReturn{200, "", page})
	} else {
		fmt.Println(err.Error())
		c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
	}

}

// _AddUser
// @Title _AddUser
// @Description 添加用户
// @Param	login_id    string  false  "登录ID"
// @Param	password    string  false  "密码"
// @Param	org_id    string  false  "组织ID"
// @Param	email    string  false  "邮箱"
// @Success 200 {object} models.Account
// @Failure 403 :id is empty
func System_AddUser(c *SystemController) {

	var user sysmodel.SysUser
	c.Ctx.BindJSON(&user)

	fmt.Println(user.LoginId, "-----add user-----", user.Password, " --- ", user.OrgId)

	//查找login_id是否存在
	oldUser := new(sysmodel.SysUser)
	count, _ := c.Db.SQL("select * from sys_user").Where("login_id = ?", user.LoginId).Count(&oldUser)

	if count > 1 {
		if oldUser.DelFlag == 0 {
			oldUser.DelFlag = 1
			_, err := c.Db.Id("id").Cols("del_flag").Update(&oldUser)
			if err != nil {
				c.Ctx.JSON(200, sysmodel.SysReturn{500, "login_id exit, update err: " + err.Error(), nil})
			}
		} else {
			c.Ctx.JSON(200, sysmodel.SysReturn{500, "login_id exit", nil})
			return
		}
	}

	tk, _ := c.Ctx.Get("token")
	op := tk.(*entitys.Token)

	user.Id = utils.NewUUID()
	user.Name = user.LoginId
	user.CreateTime = time.Now()
	user.LastUpdateDate = time.Now()
	user.CreateBy = op.UserId
	user.LastUpdateBy = op.UserId
	user.DelFlag = 1
	user.Password = utils.HashPassword(user.Password, "")

	_, err := c.Db.Insert(&user)

	ret := __none_func_system__(user.LoginId, user.Password, user.OrgId, user.Email)
	if ret && err == nil {
		c.Ctx.JSON(200, sysmodel.SysReturn{200, "", nil})
	} else {
		c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
	}
}

// _DelUser
// @Title _DelUser
// @Description 删除用户
// @Param	id    string  false  "用户ID"
// @Success 200 {object} models.Account
// @Failure 403 :id is empty
func System_DelUser(c *SystemController) {
	type Param struct {
		Id string `json:"id"`
	}

	var params []Param
	c.Ctx.BindJSON(&params)

	ids := []string{}
	for _, param := range params {
		ids = append(ids, param.Id)
	}

	session := c.Db.NewSession()
	defer session.Close()

	//_, err := c.Db.Table(new(models.SysUser)).ID(param.Id).Update(map[string]interface{}{"del_flag": 0})
	_, err := c.Db.Table(new(models.SysUser)).In("id", ids).Update(map[string]interface{}{"del_flag": 0})
	if err != nil {
		session.Rollback()
		c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
		return
	}

	_, err = c.Db.Table(new(models.SysUserRole)).In("user_id", ids).Update(map[string]interface{}{"del_flag": 0})
	if err != nil {
		session.Rollback()
		c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
		return
	}

	session.Commit()

	ret := __none_func_system__()
	if ret {
		c.Ctx.JSON(200, sysmodel.SysReturn{200, "", nil})
	} else {
		c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
	}
}

// _UpdateUser
// @Title _UpdateUser
// @Description 修改用户
// @Param	login_id    string  false  "登录ID"
// @Param	password    string  false  "密码"
// @Param	org_id    string  false  "组织ID"
// @Param	email    string  false  "邮箱"
// @Success 200 {object} models.Account
// @Failure 403 :id is empty
func System_UpdateUser(c *SystemController) {
	tk, _ := c.Ctx.Get("token")
	op := tk.(*entitys.Token)

	var userInfo sysmodel.UserInfo
	c.Ctx.BindJSON(&userInfo)

	session := c.Db.NewSession()
	defer session.Close()
	//1、删除用户角色
	_, err := c.Db.SQL(sysmodel.SqlDeleteUserRoles, userInfo.Id).Execute()
	if err != nil {
		session.Rollback()
		c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
		return
	}

	//2、修改角色
	for i, _ := range userInfo.UserRole {
		userInfo.UserRole[i].Id = utils.NewUUID()
		userInfo.UserRole[i].CreateBy = op.UserId
		userInfo.UserRole[i].LastUpdateBy = op.UserId
		userInfo.UserRole[i].CreateTime = time.Now()
		userInfo.UserRole[i].LastUpdateDate = time.Now()
		userInfo.UserRole[i].DelFlag = 1
	}
	_, err = c.Db.Insert(userInfo.UserRole)
	if err != nil {
		session.Rollback()
		c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
		return
	}

	//3、更新用户信息
	_, err = c.Db.Table(new(models.SysUser)).ID(userInfo.Id).Update(map[string]interface{}{"name": userInfo.Name, "email": userInfo.Email, "org_id": userInfo.OrgId, "last_update_by": op.UserId, "last_update_date": time.Now()})
	if err != nil {
		session.Rollback()
		c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
		return
	}

	session.Commit()

	ret := __none_func_system__()
	if ret {
		c.Ctx.JSON(200, sysmodel.SysReturn{200, "", nil})
	} else {
		c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
	}
}

// _FindRoleMenu
// @Title _FindRoleMenu
// @Description 查找角色权限
// @Param	user    string  false  "用户id"
// @Success 200 {object} models.Account
// @Failure 403 :id is empty
func System_FindRoleMenu(c *SystemController) {
	user := c.Ctx.Param(":user")
	ret := __none_func_system__(user)
	if ret {
		c.Ctx.JSON(200, sysmodel.SysReturn{200, "", nil})
	} else {
		c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
	}
}

// _RoleAll
// @Title _RoleAll
// @Description 查找所有角色
// @Success 200 {object} models.Account
// @Failure 403 :id is empty
func System_RoleAll(c *SystemController) {

	var roles []sysmodel.SysRole
	err := c.Db.SQL(sysmodel.Selectall_sys_role).Find(&roles)

	ret := __none_func_system__()
	if ret && err == nil {
		c.Ctx.JSON(200, sysmodel.SysReturn{200, "", roles})
	} else {
		c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
	}
}

// _AddMenu
// @Title _AddMenu
// @Description 增加菜单
// @Param	perms    string  false  "权限标识"
// @Param	domain    string  false  "域"
// @Success 200 {object} models.Account
// @Failure 403 :id is empty
func System_AddMenu(c *SystemController) {
	perms := c.Ctx.Param(":perms")
	domain := c.Ctx.Param(":domain")
	ret := __none_func_system__(perms, domain)
	if ret {
		c.Ctx.JSON(200, sysmodel.SysReturn{200, "", nil})
	} else {
		c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
	}
}

// _DelMenu
// @Title _DelMenu
// @Description 删除菜单
// @Param	id    string  false  "ID"
// @Success 200 {object} models.Account
// @Failure 403 :id is empty
func System_DelMenu(c *SystemController) {
	id := c.Ctx.Param(":id")
	ret := __none_func_system__(id)
	if ret {
		c.Ctx.JSON(200, sysmodel.SysReturn{200, "", nil})
	} else {
		c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
	}
}

// _FindPermissions
// @Title _FindPermissions
// @Description 查找用户权限
// @Success 200 {object} models.Account
// @Failure 403 :id is empty
func System_FindPermissions(c *SystemController) {

	ret := __none_func_system__()

	strs := []string{"sys:user:edit", "sys:user:delete", "sys:user:add"}

	if ret {
		c.Ctx.JSON(200, sysmodel.SysReturn{200, "", strs})
	} else {
		c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
	}
}

func __none_func_system__(params ...interface{}) bool {
	return true
}