ColdP_server/controllers/CommonsController.go

package controllers

import (
	"ColdP_server/controllers/lib"
	"ColdP_server/models/Account"
	"fmt"
	beego "github.com/beego/beego/v2/server/web"
	"net/http"
	"strings"
	"time"
)

type AdminController struct {
	beego.Controller
}

func (c *AdminController) Login() {
	c.TplName = "login.html"
}

//func (c *AdminController) Login_verification() {
//	Admin_user := c.GetString("bzd_username")
//	Admin_pass := c.GetString("bzd_password")
//	companyId := c.GetString("company_id")
//
//	println("Login_verification", Admin_user, Admin_pass)
//	err, admin_r := Account.Read_AdminLogin_verification(Admin_user, Admin_pass)
//
//	if err != nil {
//		c.Data["json"] = lib.JSONS{Code: 202, Msg: "账号密码错误!"}
//		c.ServeJSON()
//		return
//	}
//
//	if admin_r.T_pids == "*" {
//		User_tokey := Account.Add_Tokey_Set(admin_r.T_uuid, companyId)
//		c.Ctx.SetCookie("User_tokey", User_tokey, time.Second*60*60)
//		c.Data["json"] = lib.JSONS{Code: 200, Msg: "OK!", Data: User_tokey}
//		c.ServeJSON()
//		return
//	}
//	//如果自定义了公司ID，则标识是管理员，判断是否pids是否存在要操作的pid，并且要求登录用户的PID一定为0
//	if companyId != "" && admin_r.T_pid == 0 {
//		pids := strings.Split(admin_r.T_pids, "|")
//		for _, v := range pids {
//			newV := strings.Replace(v, "P", "", -1)
//			if newV == companyId {
//				User_tokey := Account.Add_Tokey_Set(admin_r.T_uuid, companyId)
//				c.Ctx.SetCookie("User_tokey", User_tokey, time.Second*60*60)
//				c.Data["json"] = lib.JSONS{Code: 200, Msg: "OK!", Data: User_tokey}
//				c.ServeJSON()
//				return
//			}
//		}
//		//不存在该公司的Pid
//		c.Data["json"] = lib.JSONS{Code: 202, Msg: "没有操作该公司的权限!"}
//		c.ServeJSON()
//		return
//	} else {
//		User_tokey := Account.Add_Tokey_Set(admin_r.T_uuid, fmt.Sprintf("%d", admin_r.T_pid))
//		c.Ctx.SetCookie("User_tokey", User_tokey, time.Second*60*60)
//		c.Data["json"] = lib.JSONS{Code: 200, Msg: "OK!", Data: User_tokey}
//		c.ServeJSON()
//		return
//	}
//
//}

func (c *AdminController) Login_verification() {
	Admin_user := c.GetString("bzd_username")
	Admin_pass := c.GetString("bzd_password")
	companyId := c.GetString("company_id")

	println("Login_verification", Admin_user, Admin_pass)
	err, admin_r := Account.ReadAdminLoginVerification(Admin_user, Admin_pass)

	if err != nil {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "账号密码错误!"}
		c.ServeJSON()
		return
	}

	if admin_r.T_pids == "*" {
		User_tokey := Account.Add_Tokey_Set(admin_r.T_uuid, companyId)
		c.Ctx.SetCookie("User_tokey", User_tokey, time.Second*60*60)
		c.Data["json"] = lib.JSONS{Code: 200, Msg: "OK!", Data: User_tokey}
		c.ServeJSON()
		return
	}
	//如果自定义了公司ID，则标识是管理员，判断是否pids是否存在要操作的pid，并且要求登录用户的PID一定为0
	if companyId != "" && admin_r.T_pid == 0 {
		pids := strings.Split(admin_r.T_pids, "|")
		for _, v := range pids {
			newV := strings.Replace(v, "P", "", -1)
			if newV == companyId {
				User_tokey := Account.Add_Tokey_Set(admin_r.T_uuid, companyId)
				c.Ctx.SetCookie("User_tokey", User_tokey, time.Second*60*60)
				c.Data["json"] = lib.JSONS{Code: 200, Msg: "OK!", Data: User_tokey}
				c.ServeJSON()
				return
			}
		}
		//不存在该公司的Pid
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "没有操作该公司的权限!"}
		c.ServeJSON()
		return
	} else {
		User_tokey := Account.Add_Tokey_Set(admin_r.T_uuid, fmt.Sprintf("%d", admin_r.T_pid))
		c.Ctx.SetCookie("User_tokey", User_tokey, time.Second*60*60)
		c.Data["json"] = lib.JSONS{Code: 200, Msg: "OK!", Data: User_tokey}
		c.ServeJSON()
		return
	}

}

// func (c *AdminController) Info() {
//
//		// 验证登录
//		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
//		}
//
//		c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok", Data: Account.AdminToAdmin_R(user_r)}
//		c.ServeJSON()
//		return
//	}
func (c *AdminController) Info() {

	// 验证登录
	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
	}

	c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok", Data: Account.ColdpUserToColdpUser_R(user_r)}
	c.ServeJSON()
	return
}

func (c *AdminController) Index() {
	// 验证登录

	b_, admin_r := lib.Verification(c.Ctx.GetCookie("User_tokey"), c.GetString("User_tokey"))
	if !b_ {
		c.Ctx.Redirect(302, "Login")
		return
	}

	// 基本信息
	c.Data["Admin_name"] = admin_r.T_name
	c.Data["Admin_root"] = admin_r.Id
	c.TplName = "index.html"

}

func (c *AdminController) Home() {
	b_, admin_r := lib.Verification(c.Ctx.GetCookie("User_tokey"), c.GetString("User_tokey"))
	if !b_ {
		fmt.Println("当前未登录，请先登录!")
		c.Ctx.Redirect(http.StatusFound, "Login")
	}
	c.Data["T_pid"] = admin_r.T_pid
	c.Data["names"] = admin_r.T_name
	fmt.Println("======================>", admin_r.T_pid, admin_r.T_name)
	c.TplName = "home.html"
}