ColdP_server/controllers/CompanyList.go

package controllers

import (
	"ColdP_server/conf"
	"ColdP_server/controllers/lib"
	"ColdP_server/models/Account"
	"ColdP_server/models/Company"
	"crypto/md5"
	"encoding/hex"
	"encoding/json"
	"fmt"
	"strconv"

	beego "github.com/beego/beego/v2/server/web"
	uuid "github.com/satori/go.uuid"
)

type CompanyController struct {
	beego.Controller
}

func (c *CompanyController) User_html() {
	b_, _ := lib.Verification(c.Ctx.GetCookie("User_tokey"), c.GetString("User_tokey"))
	if !b_ {
		c.Data["json"] = lib.JSONS{Code: 201, Msg: "User_tokey Err!"}
		c.ServeJSON()
		return
	}
	// 验证登录
	b_, R_u := lib.Verification(c.Ctx.GetCookie("User_tokey"), c.GetString("User_tokey"))
	if !b_ {
		c.Data["json"] = lib.JSONS{Code: 201, Msg: "User_tokey Err!"}
		c.ServeJSON()
		return
	}

	Name := c.GetString("Name")

	c.Data["Class_List"] = Company.Read_CompanyClass_All(R_u.T_pid, "")
	c.Data["Name"] = Name
	c.TplName = "User/useredit.html"
}

// GetCompanyList 获取公司列表
func (c *CompanyController) GetCompanyList() {
	// 验证登录
	b_, user_r := lib.Verification(c.Ctx.GetCookie("User_tokey"), c.GetString("User_tokey"))
	if !b_ {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "no"}
		c.ServeJSON()
		return
	}
	if user_r.T_pid != 0 {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "没有操作权限!"}
		c.ServeJSON()
		return
	}
	fmt.Println(user_r)
	list, err := Account.GetCompanyList(c.GetString("serch"))
	if err != nil {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "查询失败!"}
		c.ServeJSON()
		return
	}
	c.Data["json"] = lib.JSONS{Code: 200, Msg: "OK!", Data: list}
	c.ServeJSON()
	return
}

// Read_Admin_List 获取用户列表
func (c *CompanyController) Read_Admin_List() {
	// 验证登录
	b_, user_r := lib.Verification(c.Ctx.GetCookie("User_tokey"), c.GetString("User_tokey"))
	if !b_ {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "no"}
		c.ServeJSON()
		return
	}
	if user_r.T_pid != 0 {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "没有操作权限!"}
		c.ServeJSON()
		return
	}
	page, _ := c.GetInt("page")
	list, count := Account.Read_ColdUser_List(user_r.T_pid, c.GetString("T_name"), page, conf.Page_size)
	var pageCount int
	if (int(count) % conf.Page_size) != 0 {
		pageCount = int(count) / conf.Page_size
		pageCount++
	} else {
		pageCount = int(count) / conf.Page_size
	}
	c.Data["json"] = lib.PageHelper{int(count), pageCount, int(page), int(page) >= pageCount, page <= 1, list}
	c.ServeJSON()
}

// UpdateUserStateById 修改用户状态
func (c *CompanyController) UpdateUserStateById() {
	// 验证登录
	b_, user_r := lib.Verification(c.Ctx.GetCookie("User_tokey"), c.GetString("User_tokey"))
	if !b_ {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "no"}
		c.ServeJSON()
		return
	}
	if user_r.T_pid != 0 {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "没有操作权限!"}
		c.ServeJSON()
		return
	}
	var coldServe Account.ColdpUser
	var cold []string
	var err error
	coldServe.Id, err = strconv.Atoi(c.GetString("Id"))
	coldServe.T_State, err = strconv.Atoi(c.GetString("T_State"))
	if err != nil {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "获取参数失败!"}
		c.ServeJSON()
		return
	}
	cold = append(cold, "t__state")
	cold = append(cold, "ID")
	state := Account.UpdateByIdState(coldServe, cold...)
	if state {
		c.Data["json"] = lib.JSONS{Code: 200, Msg: "OK!"}
		c.ServeJSON()
		return
	} else {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "修改用户状态失败!"}
		c.ServeJSON()
		return
	}
}
func (c *CompanyController) UpdateUser() {
	// 验证登录
	b_, user_r := lib.Verification(c.Ctx.GetCookie("User_tokey"), c.GetString("User_tokey"))
	if !b_ {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "no"}
		c.ServeJSON()
		return
	}
	if user_r.T_pid != 0 {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "没有操作权限!"}
		c.ServeJSON()
		return
	}
	var UpdateUser Account.UpdateUser
	var err error
	var coldServe Account.ColdpUser
	var cold []string
	err = json.Unmarshal(c.Ctx.Input.RequestBody, &UpdateUser)

	if len(UpdateUser.Id) > 0 {
		coldServe.Id, err = strconv.Atoi(UpdateUser.Id)
		cold = append(cold, "ID")
	} else {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "id不存在!"}
		c.ServeJSON()
		return
	}
	if len(UpdateUser.T_name) > 0 {
		coldServe.T_name = UpdateUser.T_name
		cold = append(cold, "t_name")
	}
	if len(UpdateUser.T_user) > 0 {
		coldServe.T_user = UpdateUser.T_user
		cold = append(cold, "t_user")
	}
	if len(UpdateUser.T_pass) > 0 {
		pass := UpdateUser.T_pass
		bytes := []byte(pass)
		sum := md5.Sum(bytes)
		coldServe.T_pass = hex.EncodeToString(sum[:])
		cold = append(cold, "t_pass")
	}
	if len(UpdateUser.T_pid) > 0 {
		coldServe.T_pid, err = strconv.Atoi(UpdateUser.T_pid)
		cold = append(cold, "t_pid")
	}
	if err != nil {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "获取参数失败!"}
		c.ServeJSON()
		return
	}

	state := Account.UpdateByIdState(coldServe, cold...)
	if state {
		c.Data["json"] = lib.JSONS{Code: 200, Msg: "OK!"}
		c.ServeJSON()
		return
	} else {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "修改用户状态失败!"}
		c.ServeJSON()
		return
	}
}
func (c *CompanyController) GetUserById() {
	// 验证登录
	b_, user_r := lib.Verification(c.Ctx.GetCookie("User_tokey"), c.GetString("User_tokey"))
	if !b_ {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "no"}
		c.ServeJSON()
		return
	}
	if user_r.T_pid != 0 {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "没有操作权限!"}
		c.ServeJSON()
		return
	}
	//var coldServe Account.ColdpUser
	//var cold []string
	//var err error
	var userId string
	if len(c.GetString("Id")) > 0 {
		userId = c.GetString("Id")
	}
	err, user := Account.GetByIdUser(userId)
	if err != nil {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "获取用户信息失败!"}
		c.ServeJSON()
		return
	}
	c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: user}
	c.ServeJSON()
	return
}

// DeleteUserById 删除用户
func (c *CompanyController) DeleteUserById() {
	// 验证登录
	b_, user_r := lib.Verification(c.Ctx.GetCookie("User_tokey"), c.GetString("User_tokey"))
	if !b_ {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "no"}
		c.ServeJSON()
		return
	}
	if user_r.T_pid != 0 {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "没有操作权限!"}
		c.ServeJSON()
		return
	}
	var coldServe Account.ColdpUser
	var err error
	coldServe.Id, err = strconv.Atoi(c.GetString("Id"))
	if err != nil {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "获取参数失败!"}
		c.ServeJSON()
		return
	}
	state := Account.DeleteById(coldServe)
	if state {
		c.Data["json"] = lib.JSONS{Code: 200, Msg: "OK!"}
		c.ServeJSON()
		return
	} else {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "删除用户失败!"}
		c.ServeJSON()
		return
	}
}

// AddUser 添加用户
func (c *CompanyController) AddUser() {
	// 验证登录
	b_, user_r := lib.Verification(c.Ctx.GetCookie("User_tokey"), c.GetString("User_tokey"))
	if !b_ {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "no"}
		c.ServeJSON()
		return
	}
	if user_r.T_pid != 0 {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "没有操作权限!"}
		c.ServeJSON()
		return
	}
	var coldServe Account.ColdpUser
	var addUser Account.AddUser
	var err error
	err = json.Unmarshal(c.Ctx.Input.RequestBody, &addUser)

	if len(addUser.T_name) > 0 {
		coldServe.T_name = addUser.T_name
	}
	if len(addUser.T_user) > 0 {
		coldServe.T_user = addUser.T_user
	}
	if addUser.T_pid == "" {
		coldServe.T_pid = 0
		coldServe.T_pids = "*"
	} else {
		if len(addUser.T_pid) > 0 {
			coldServe.T_pid, err = strconv.Atoi(addUser.T_pid)

		}
	}
	if err != nil {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "获取参数失败!"}
		c.ServeJSON()
		return
	}
	if len(addUser.T_pass) > 0 {
		bytes := []byte(addUser.T_pass)
		sum := md5.Sum(bytes)
		coldServe.T_pass = hex.EncodeToString(sum[:])
	}
	user := Account.Read_Company_ByUser(coldServe.T_user)
	if !user {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "账号重复!"}
		c.ServeJSON()
		return
	}
	coldServe.T_State = 1
	coldServe.T_uuid = uuid.NewV4().String()
	_, err = Account.AddColdpUser(coldServe)
	if err != nil {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "添加用户失败!"}
		c.ServeJSON()
		return
	}
	c.Data["json"] = lib.JSONS{Code: 200, Msg: "添加成功!"}
	c.ServeJSON()
}