XT_New/controllers/base_api_controller.go

package controllers

import (
	"XT_New/enums"
	"XT_New/models"
	"XT_New/service"
	"encoding/json"
	"os"
	"path"
	"runtime"
	"strconv"
	"strings"
	"time"
)

type BaseAPIController struct {
	BaseController
}

// func (this *BaseAPIController) Prepare() {
// 	this.BaseController.Prepare()
// 	beego.Trace("============================================================")
// 	beego.Trace("session ID: %v", this.Ctx.Input.Cookie("beegosessionID"))
// 	beego.Trace("session : %v", this.GetSession("info"))
// 	this.SetSession("info", time.Now().Format("2006/01/02 15:04:05"))
// 	beego.Trace("============================================================")
// }

// 输出数据格式化
/*
	success json:
	{
		"state": 1,
		"code": 0,
		"data": json,
	}

	fail json:
	{
		"state": 0,
		"code": int,
		"msg": string,
	}
*/
func (this *BaseAPIController) ServeSuccessJSON(data map[string]interface{}) {
	this.Data["json"] = enums.MakeSuccessResponseJSON(data)
	this.ServeJSON()
}

func (this *BaseAPIController) ServeFailJSONWithSGJErrorCode(code int) {
	this.Data["json"] = enums.MakeFailResponseJSONWithSGJErrorCode(code)
	this.ServeJSON()
}

func (this *BaseAPIController) ServeFailJSONWithSGJError(err *enums.SGJError) {
	this.Data["json"] = enums.MakeFailResponseJSONWithSGJError(err)
	this.ServeJSON()
}

func (this *BaseAPIController) ServeFailJsonSend(code int, msg string) {
	this.Data["json"] = enums.MakeFailResponseJSON(msg, code)
	this.ServeJSON()
}

func (this *BaseAPIController) ServeDynamicFailJsonSend(msg string) {
	this.Data["json"] = enums.MakeDynamicFailResponseJSON(msg)
	this.ServeJSON()
}

type BaseAuthAPIController struct {
	BaseAPIController
}

func (this *BaseAuthAPIController) Prepare() {
	this.BaseAPIController.Prepare()
	if this.GetAdminUserInfo() == nil {
		var userAdmin models.AdminUser
		userAdmin.Id = 1448
		userAdmin.Mobile = "13318599895"

		userAdmin.Id = 597
		userAdmin.Mobile = "19874122664"
		userAdmin.IsSuperAdmin = false
		userAdmin.Status = 1
		userAdmin.CreateTime = 1530786071
		userAdmin.ModifyTime = 1530786071
		var subscibe models.ServeSubscibe
		subscibe.ID = 1
		subscibe.OrgId = 10215
		subscibe.PeriodStart = 1538035409
		subscibe.PeriodEnd = 1569571409
		subscibe.State = 1
		subscibe.Status = 1
		subscibe.CreatedTime = 1538035409
		subscibe.UpdatedTime = 1538035409
		subscibes := make(map[int64]*models.ServeSubscibe, 0)
		subscibes[4] = &subscibe
		var adminUserInfo service.AdminUserInfo
		adminUserInfo.CurrentOrgId = 10215
		adminUserInfo.CurrentAppId = 4
		adminUserInfo.AdminUser = &userAdmin
		adminUserInfo.Subscibes = subscibes
		this.SetSession("admin_user_info", &adminUserInfo)

		//this.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeNotLogin)
		//this.StopRun()

	}

	adminUserInfo := this.GetAdminUserInfo()

	if this.Ctx.Request.Header.Get("Permission") == "2" {
		err_msgs := LoadErrMsgConfig("./err_msg.json").Msgs

		org, _ := service.GetOrgById(adminUserInfo.CurrentOrgId)
		if adminUserInfo.AdminUser.Id != org.Creator { //超级管理员不受此限制

			isPermission := false
			adminUserInfo := this.GetAdminUserInfo()

			redisClient := service.RedisClient()
			defer redisClient.Close()
			key := strconv.FormatInt(adminUserInfo.CurrentOrgId, 64) + "_" + strconv.FormatInt(adminUserInfo.AdminUser.Id, 64) + "_role_ids"
			result, _ := redisClient.Get(key).Result()
			var role models.App_Role
			var roles []string
			if len(result) == 0 {
				//该机构下该用户有多少个
				role, _ = service.GetUserAllRole(adminUserInfo.CurrentOrgId, adminUserInfo.AdminUser.Id)
				redisClient.Set(key, role, time.Second*60*60*18)
				if len(role.RoleIds) > 0 { //该用户没有设置角色
					roles = strings.Split(role.RoleIds, ",")
				}
			} else {
				json.Unmarshal([]byte(result), &role)
				if len(role.RoleIds) > 0 { //该用户没有设置角色
					roles = strings.Split(role.RoleIds, ",")
				}
			}

			//key := strconv.FormatInt(role_id, 64) + "_" + strconv.FormatInt(adminUserInfo.AdminUser.Id, 64) + "_role_ids"
			//result, _ := redisClient.Get(key).Result()

			//获取该用户下所有角色的权限总集
			var userRolePurviews string
			var userRolePurviewsArr []string
			for _, item := range roles {
				role_id, _ := strconv.ParseInt(item, 10, 64)
				purviews, _ := service.GetRoleFuncPurviewIds(role_id)
				if len(userRolePurviews) == 0 {
					userRolePurviews = purviews
				} else {
					userRolePurviews = userRolePurviews + "," + purviews
				}
			}
			//该用户所拥有角色的权限的总集
			userRolePurviewsArr = RemoveRepeatedPurviewElement2(strings.Split(userRolePurviews, ","))

			//系统所记录的权限列表
			allPermission, _ := service.GetAllFunctionPurview()

			for _, item := range allPermission {
				// fmt.Println(len(strings.Split(item.Urlfor, ",")))

				// fmt.Println(strings.Split(item.Urlfor, ","))
				//fmt.Println(strings.Split(this.Ctx.Request.RequestURI, "?")[0] + "?" + "mode=" + this.GetString("mode"))

				//判断当前路由是否在权限路由列表里面
				if strings.Split(item.Urlfor, ",")[1] == strings.Split(this.Ctx.Request.RequestURI, "?")[0]+"?"+"mode="+this.GetString("mode") {

					//获取该角色的所有权限
					for _, items := range userRolePurviewsArr {
						id, _ := strconv.ParseInt(items, 10, 64)
						if id == item.ID {
							isPermission = true
						}
					}
					if !isPermission {
						var msg string
						for _, item := range err_msgs {
							if strings.Index(item.Url, strings.Split(this.Ctx.Request.RequestURI, "?")[0]+"?"+"mode="+this.GetString("mode")) != -1 {
								msg = item.ErrMsg
							}
						}
						//msg, _ := service.FindErrorMsgByStr(strings.Split(this.Ctx.Request.RequestURI, "?")[0] + "?" + "mode=" + this.GetString("mode"))
						json := make(map[string]interface{})
						json["msg"] = msg
						json["code"] = 0
						json["state"] = 0
						this.Data["json"] = json
						this.ServeJSON()
						this.StopRun()
					}
				}

			}
		}
	}

	if this.Ctx.Request.Header.Get("Permission") == "3" {
		err_msgs := LoadErrMsgConfig("./err_msg.json").Msgs

		org, _ := service.GetOrgById(adminUserInfo.CurrentOrgId)
		if adminUserInfo.AdminUser.Id != org.Creator { //超级管理员不受此限制

			isPermission := false
			adminUserInfo := this.GetAdminUserInfo()
			//该机构下该用户有多少个
			redisClient := service.RedisClient()
			defer redisClient.Close()
			key := strconv.FormatInt(adminUserInfo.CurrentOrgId, 64) + "_" + strconv.FormatInt(adminUserInfo.AdminUser.Id, 64) + "_role_ids"
			result, _ := redisClient.Get(key).Result()
			var role models.App_Role
			var roles []string
			if len(result) == 0 {
				//该机构下该用户有多少个
				role, _ = service.GetUserAllRole(adminUserInfo.CurrentOrgId, adminUserInfo.AdminUser.Id)
				redisClient.Set(key, role, time.Second*60*60*18)
				if len(role.RoleIds) > 0 { //该用户没有设置角色
					roles = strings.Split(role.RoleIds, ",")
				}
			} else {
				json.Unmarshal([]byte(result), &role)
				if len(role.RoleIds) > 0 { //该用户没有设置角色
					roles = strings.Split(role.RoleIds, ",")
				}
			}

			//获取该用户下所有角色的权限总集
			var userRolePurviews string
			var userRolePurviewsArr []string
			for _, item := range roles {
				role_id, _ := strconv.ParseInt(item, 10, 64)
				purviews, _ := service.GetRoleFuncPurviewIds(role_id)
				if len(userRolePurviews) == 0 {
					userRolePurviews = purviews
				} else {
					userRolePurviews = userRolePurviews + "," + purviews
				}
			}
			//该用户所拥有角色的权限的总集
			userRolePurviewsArr = RemoveRepeatedPurviewElement2(strings.Split(userRolePurviews, ","))
			//系统所记录的权限列表
			allPermission, _ := service.GetAllFunctionPurview()

			for _, item := range allPermission {
				// fmt.Println(strings.Split(item.Urlfor, ",")[2])
				// fmt.Println(strings.Split(this.Ctx.Request.RequestURI, "?")[0] + "?" + "mode=" + this.GetString("mode"))

				//判断当前路由是否在权限路由列表里面
				if strings.Split(item.Urlfor, ",")[2] == strings.Split(this.Ctx.Request.RequestURI, "?")[0]+"?"+"mode="+this.GetString("mode") {

					//获取该角色的所有权限
					for _, items := range userRolePurviewsArr {
						id, _ := strconv.ParseInt(items, 10, 64)
						if id == item.ID {
							isPermission = true
						}
					}
					if !isPermission {
						var msg string
						for _, item := range err_msgs {
							if strings.Index(item.Url, strings.Split(this.Ctx.Request.RequestURI, "?")[0]+"?"+"mode="+this.GetString("mode")) != -1 {
								msg = item.ErrMsg
							}
						}
						//msg, _ := service.FindErrorMsgByStr(strings.Split(this.Ctx.Request.RequestURI, "?")[0] + "?" + "mode=" + this.GetString("mode"))
						json := make(map[string]interface{})
						json["msg"] = msg
						json["code"] = 0
						json["state"] = 0
						this.Data["json"] = json
						this.ServeJSON()
						this.StopRun()
					}
				}
			}
		}
	}

	//if this.Ctx.Request.Method != "GET" {
	//	adminUserInfo := this.GetAdminUserInfo()
	//
	//	err := service.GetOrgSubscibeState(adminUserInfo.Subscibes[adminUserInfo.CurrentOrgId])
	//	if err != nil || adminUserInfo.Subscibes[adminUserInfo.CurrentOrgId].State == 3 {
	//		this.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeNotSubscibe)
	//		this.StopRun()
	//	}
	//}

	//if this.Ctx.Request.Header.Get("Permission") == "1" {
	//	isPermission := false
	//
	//	//adminUserInfo := this.GetAdminUserInfo()
	//
	//	//service.GetUserAllRole(adminUserInfo.CurrentOrgId)
	//
	//
	//
	//
	//	roles := []int64{1,2}		//模拟该用户有多少角色
	//	var targetRole models.RolePurview
	//	var userRolePurview []string
	//
	//	//用户角色1对应的权限
	//	role1 := models.RolePurview{PurviewIds:"11,12,13"}
	//
	//	//用户角色2对应的权限
	//	role2 := models.RolePurview{PurviewIds:"11,15,16"}
	//
	//
	//	//模拟角色2
	//	//判断该用户有多少个角色，合并最大角色权限
	//	if len(roles) == 1{ //单个
	//		targetRole = role1
	//		userRolePurview = strings.Split(targetRole.PurviewIds, ",")
	//
	//	}else{ //多个
	//		targetRole = role1
	//		targetRole.PurviewIds = targetRole.PurviewIds + "," + role2.PurviewIds
	//		userRolePurview = strings.Split(targetRole.PurviewIds, ",")
	//	}
	//
	//	userRolePurview = RemoveRepeatedElement2(userRolePurview)
	//
	//	fmt.Println(userRolePurview)
	//	//所有权限列表
	//	allPermission := []models.Purview{{Id:10,Urlfor:"/m/api/dialysis/dialysisPrescription-/api/dialysis/prescription"},
	//		{Id:11,Urlfor:"/m/api/dialysis/dialysisPrescription1-/api/dialysis/prescription1"},
	//		{Id:12,Urlfor:"/m/api/dialysis/dialysisPrescription2-/api/dialysis/prescription2"},
	//		{Id:13,Urlfor:"/m/api/dialysis/dialysisPrescription3-/api/dialysis/prescription3"},
	//		{Id:14,Urlfor:"/m/api/dialysis/dialysisPrescription4-/api/dialysis/prescription4"},
	//		{Id:15,Urlfor:"/m/api/dialysis/dialysisPrescription5-/api/dialysis/prescription5"},
	//		{Id:16,Urlfor:"/m/api/dialysis/dialysisPrescription6-/api/dialysis/prescription6"}}
	//	for _, item := range allPermission {
	//		//判断当前路由是否在权限路由列表里面
	//		if strings.Split(item.Urlfor, "-")[1] == strings.Split(this.Ctx.Request.RequestURI , "?")[0]{
	//			fmt.Println(strings.Split(this.Ctx.Request.RequestURI , "?")[0])
	//
	//			//获取该角色的所有权限
	//			for _, items := range userRolePurview{
	//				id, _ := strconv.ParseInt(items, 10, 64)
	//				if id == item.Id{
	//					isPermission = true
	//				}
	//			}
	//			if !isPermission{
	//				this.ServeFailJSONWithSGJErrorCode(enums.ErrorCodePreExist)
	//				this.StopRun()
	//			}
	//		}
	//
	//	}
	//}
}

type BaseServeAPIController struct {
	BaseAPIController
}

func (this *BaseServeAPIController) Prepare() {
	this.BaseAPIController.Prepare()
	if this.GetAdminUserInfo() == nil {
		var userAdmin models.AdminUser
		userAdmin.Id = 1448
		userAdmin.Mobile = "13318599895"

		userAdmin.Id = 597
		userAdmin.Mobile = "19874122664"
		userAdmin.IsSuperAdmin = false
		userAdmin.Status = 1
		userAdmin.CreateTime = 1530786071
		userAdmin.ModifyTime = 1530786071
		var subscibe models.ServeSubscibe
		subscibe.ID = 1
		subscibe.OrgId = 10215
		subscibe.PeriodStart = 1538035409
		subscibe.PeriodEnd = 1569571409
		subscibe.State = 1
		subscibe.Status = 1
		subscibe.CreatedTime = 1538035409
		subscibe.UpdatedTime = 1538035409
		subscibes := make(map[int64]*models.ServeSubscibe, 0)
		subscibes[4] = &subscibe
		var adminUserInfo service.AdminUserInfo
		adminUserInfo.CurrentOrgId = 10215
		adminUserInfo.CurrentAppId = 4
		adminUserInfo.AdminUser = &userAdmin
		adminUserInfo.Subscibes = subscibes
		this.SetSession("admin_user_info", &adminUserInfo)
		//this.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeNotLogin)
		//this.StopRun()
	}

	//if adminUserInfo.AppRole != nil {
	//	if adminUserInfo.AppRole.Id > 0 {
	//		app_role, _ := service.FindAppRoleById(adminUserInfo.AppRole.Id)
	//		if app_role != nil {
	//			if app_role.Status != 1 {
	//				this.DelSession("mobile_admin_user_info")
	//				this.Ctx.SetCookie("token_cookie", "")
	//				this.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeForbidden)
	//				this.StopRun()
	//			}
	//		}
	//	}
	//
	//}
	//fmt.Println("222222222")

}

func RemoveRepeatedPurviewElement2(arr []string) (newArr []string) {
	newArr = make([]string, 0)
	for i := 0; i < len(arr); i++ {
		repeat := false
		for j := i + 1; j < len(arr); j++ {
			if arr[i] == arr[j] {
				repeat = true
				break
			}
		}
		if !repeat {
			newArr = append(newArr, arr[i])
		}
	}
	return
}

type ErrMsgConfig struct {
	Msgs []*models.ErrMsg "json:msg"
}

func LoadErrMsgConfig(dataFile string) *ErrMsgConfig {
	var config ErrMsgConfig
	_, filename, _, _ := runtime.Caller(1)
	datapath := path.Join(path.Dir(filename), dataFile)
	config_file, err := os.Open(datapath)
	if err != nil {
		emit("Failed to open config file '%s': %s\n", datapath, err)
		return &config
	}
	fi, _ := config_file.Stat()
	buffer := make([]byte, fi.Size())
	_, err = config_file.Read(buffer)
	buffer, err = StripComments(buffer) //去掉注释
	if err != nil {
		emit("Failed to strip comments from json: %s\n", err)
		return &config
	}
	buffer = []byte(os.ExpandEnv(string(buffer))) //特殊
	err = json.Unmarshal(buffer, &config)         //解析json格式数据
	if err != nil {
		emit("Failed unmarshalling json: %s\n", err)
		return &config
	}
	return &config
}