base_api_controller.go 12KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383
  1. package controllers
  2. import (
  3. "Xcx_New/enums"
  4. "Xcx_New/service"
  5. "fmt"
  6. "strconv"
  7. "strings"
  8. )
  9. type BaseAPIController struct {
  10. BaseController
  11. }
  12. // func (this *BaseAPIController) Prepare() {
  13. // this.BaseController.Prepare()
  14. // beego.Trace("============================================================")
  15. // beego.Trace("session ID: %v", this.Ctx.Input.Cookie("beegosessionID"))
  16. // beego.Trace("session : %v", this.GetSession("info"))
  17. // this.SetSession("info", time.Now().Format("2006/01/02 15:04:05"))
  18. // beego.Trace("============================================================")
  19. // }
  20. // 输出数据格式化
  21. /*
  22. success json:
  23. {
  24. "state": 1,
  25. "code": 0,
  26. "data": json,
  27. }
  28. fail json:
  29. {
  30. "state": 0,
  31. "code": int,
  32. "msg": string,
  33. }
  34. */
  35. func (this *BaseAPIController) ServeSuccessJSON(data map[string]interface{}) {
  36. this.Data["json"] = enums.MakeSuccessResponseJSON(data)
  37. this.ServeJSON()
  38. }
  39. func (this *BaseAPIController) ServeFailJSONWithSGJErrorCode(code int) {
  40. this.Data["json"] = enums.MakeFailResponseJSONWithSGJErrorCode(code)
  41. this.ServeJSON()
  42. }
  43. func (this *BaseAPIController) ServeFailJSONWithSGJError(err *enums.SGJError) {
  44. this.Data["json"] = enums.MakeFailResponseJSONWithSGJError(err)
  45. this.ServeJSON()
  46. }
  47. func (this *BaseAPIController) ServeFailJsonSend(code int, msg string) {
  48. this.Data["json"] = enums.MakeFailResponseJSON(msg, code)
  49. this.ServeJSON()
  50. }
  51. func (this *BaseAPIController) ServeDynamicFailJsonSend(msg string) {
  52. this.Data["json"] = enums.MakeDynamicFailResponseJSON(msg)
  53. this.ServeJSON()
  54. }
  55. type BaseAuthAPIController struct {
  56. BaseAPIController
  57. }
  58. func (this *BaseAuthAPIController) Prepare() {
  59. this.BaseAPIController.Prepare()
  60. if this.GetAdminUserInfo() == nil {
  61. //var userAdmin models.AdminUser
  62. //userAdmin.Id = 1448
  63. //userAdmin.Mobile = "13318599895"
  64. //
  65. //userAdmin.Id = 597
  66. //userAdmin.Mobile = "19874122664"
  67. //userAdmin.IsSuperAdmin = false
  68. //userAdmin.Status = 1
  69. //userAdmin.CreateTime = 1530786071
  70. //userAdmin.ModifyTime = 1530786071
  71. //var subscibe models.ServeSubscibe
  72. //subscibe.ID = 1
  73. //subscibe.OrgId = 4
  74. //subscibe.PeriodStart = 1538035409
  75. //subscibe.PeriodEnd = 1569571409
  76. //subscibe.State = 1
  77. //subscibe.Status = 1
  78. //subscibe.CreatedTime = 1538035409
  79. //subscibe.UpdatedTime = 1538035409
  80. //subscibes := make(map[int64]*models.ServeSubscibe, 0)
  81. //subscibes[4] = &subscibe
  82. //var adminUserInfo service.AdminUserInfo
  83. //adminUserInfo.CurrentOrgId = 4
  84. //adminUserInfo.CurrentAppId = 4
  85. //adminUserInfo.AdminUser = &userAdmin
  86. //adminUserInfo.Subscibes = subscibes
  87. //this.SetSession("admin_user_info", &adminUserInfo)
  88. this.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeNotLogin)
  89. this.StopRun()
  90. }
  91. adminUserInfo := this.GetAdminUserInfo()
  92. if this.Ctx.Request.Header.Get("Permission") == "2" {
  93. org, _ := service.GetOrgById(adminUserInfo.CurrentOrgId)
  94. if adminUserInfo.AdminUser.Id != org.Creator { //超级管理员不受此限制
  95. isPermission := false
  96. adminUserInfo := this.GetAdminUserInfo()
  97. //该机构下该用户有多少个
  98. role, _ := service.GetUserAllRole(adminUserInfo.CurrentOrgId, adminUserInfo.AdminUser.Id)
  99. var roles []string
  100. if len(role.RoleIds) <= 0 { //该用户没有设置角色
  101. } else {
  102. roles = strings.Split(role.RoleIds, ",")
  103. }
  104. fmt.Println(roles)
  105. //获取该用户下所有角色的权限总集
  106. var userRolePurviews string
  107. var userRolePurviewsArr []string
  108. for _, item := range roles {
  109. role_id, _ := strconv.ParseInt(item, 10, 64)
  110. purviews, _ := service.GetRoleFuncPurviewIds(role_id)
  111. if len(userRolePurviews) == 0 {
  112. userRolePurviews = purviews
  113. } else {
  114. userRolePurviews = userRolePurviews + "," + purviews
  115. }
  116. }
  117. //该用户所拥有角色的权限的总集
  118. userRolePurviewsArr = RemoveRepeatedPurviewElement2(strings.Split(userRolePurviews, ","))
  119. fmt.Println(userRolePurviewsArr)
  120. //系统所记录的权限列表
  121. allPermission, _ := service.GetAllFunctionPurview()
  122. for _, item := range allPermission {
  123. // fmt.Println(len(strings.Split(item.Urlfor, ",")))
  124. // fmt.Println(strings.Split(item.Urlfor, ","))
  125. //fmt.Println(strings.Split(this.Ctx.Request.RequestURI, "?")[0] + "?" + "mode=" + this.GetString("mode"))
  126. //判断当前路由是否在权限路由列表里面
  127. if strings.Split(item.Urlfor, ",")[1] == strings.Split(this.Ctx.Request.RequestURI, "?")[0]+"?"+"mode="+this.GetString("mode") {
  128. //获取该角色的所有权限
  129. for _, items := range userRolePurviewsArr {
  130. id, _ := strconv.ParseInt(items, 10, 64)
  131. if id == item.ID {
  132. isPermission = true
  133. }
  134. }
  135. if !isPermission {
  136. msg, _ := service.FindErrorMsgByStr(strings.Split(this.Ctx.Request.RequestURI, "?")[0] + "?" + "mode=" + this.GetString("mode"))
  137. json := make(map[string]interface{})
  138. json["msg"] = msg
  139. json["code"] = 0
  140. json["state"] = 0
  141. this.Data["json"] = json
  142. this.ServeJSON()
  143. this.StopRun()
  144. }
  145. }
  146. }
  147. }
  148. }
  149. if this.Ctx.Request.Header.Get("Permission") == "3" {
  150. org, _ := service.GetOrgById(adminUserInfo.CurrentOrgId)
  151. if adminUserInfo.AdminUser.Id != org.Creator { //超级管理员不受此限制
  152. isPermission := false
  153. adminUserInfo := this.GetAdminUserInfo()
  154. //该机构下该用户有多少个
  155. role, _ := service.GetUserAllRole(adminUserInfo.CurrentOrgId, adminUserInfo.AdminUser.Id)
  156. var roles []string
  157. if len(role.RoleIds) <= 0 { //该用户没有设置角色
  158. } else {
  159. roles = strings.Split(role.RoleIds, ",")
  160. }
  161. fmt.Println(roles)
  162. //获取该用户下所有角色的权限总集
  163. var userRolePurviews string
  164. var userRolePurviewsArr []string
  165. for _, item := range roles {
  166. role_id, _ := strconv.ParseInt(item, 10, 64)
  167. purviews, _ := service.GetRoleFuncPurviewIds(role_id)
  168. if len(userRolePurviews) == 0 {
  169. userRolePurviews = purviews
  170. } else {
  171. userRolePurviews = userRolePurviews + "," + purviews
  172. }
  173. }
  174. //该用户所拥有角色的权限的总集
  175. userRolePurviewsArr = RemoveRepeatedPurviewElement2(strings.Split(userRolePurviews, ","))
  176. fmt.Println(userRolePurviewsArr)
  177. //系统所记录的权限列表
  178. allPermission, _ := service.GetAllFunctionPurview()
  179. for _, item := range allPermission {
  180. // fmt.Println(strings.Split(item.Urlfor, ",")[2])
  181. // fmt.Println(strings.Split(this.Ctx.Request.RequestURI, "?")[0] + "?" + "mode=" + this.GetString("mode"))
  182. //判断当前路由是否在权限路由列表里面
  183. if strings.Split(item.Urlfor, ",")[2] == strings.Split(this.Ctx.Request.RequestURI, "?")[0]+"?"+"mode="+this.GetString("mode") {
  184. //获取该角色的所有权限
  185. for _, items := range userRolePurviewsArr {
  186. id, _ := strconv.ParseInt(items, 10, 64)
  187. if id == item.ID {
  188. isPermission = true
  189. }
  190. }
  191. if !isPermission {
  192. msg, _ := service.FindErrorMsgByStr(strings.Split(this.Ctx.Request.RequestURI, "?")[0] + "?" + "mode=" + this.GetString("mode"))
  193. json := make(map[string]interface{})
  194. json["msg"] = msg
  195. json["code"] = 0
  196. json["state"] = 0
  197. this.Data["json"] = json
  198. this.ServeJSON()
  199. this.StopRun()
  200. }
  201. }
  202. }
  203. }
  204. }
  205. //if this.Ctx.Request.Method != "GET" {
  206. // adminUserInfo := this.GetAdminUserInfo()
  207. //
  208. // err := service.GetOrgSubscibeState(adminUserInfo.Subscibes[adminUserInfo.CurrentOrgId])
  209. // if err != nil || adminUserInfo.Subscibes[adminUserInfo.CurrentOrgId].State == 3 {
  210. // this.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeNotSubscibe)
  211. // this.StopRun()
  212. // }
  213. //}
  214. //if this.Ctx.Request.Header.Get("Permission") == "1" {
  215. // isPermission := false
  216. //
  217. // //adminUserInfo := this.GetAdminUserInfo()
  218. //
  219. // //service.GetUserAllRole(adminUserInfo.CurrentOrgId)
  220. //
  221. //
  222. //
  223. //
  224. // roles := []int64{1,2} //模拟该用户有多少角色
  225. // var targetRole models.RolePurview
  226. // var userRolePurview []string
  227. //
  228. // //用户角色1对应的权限
  229. // role1 := models.RolePurview{PurviewIds:"11,12,13"}
  230. //
  231. // //用户角色2对应的权限
  232. // role2 := models.RolePurview{PurviewIds:"11,15,16"}
  233. //
  234. //
  235. // //模拟角色2
  236. // //判断该用户有多少个角色,合并最大角色权限
  237. // if len(roles) == 1{ //单个
  238. // targetRole = role1
  239. // userRolePurview = strings.Split(targetRole.PurviewIds, ",")
  240. //
  241. // }else{ //多个
  242. // targetRole = role1
  243. // targetRole.PurviewIds = targetRole.PurviewIds + "," + role2.PurviewIds
  244. // userRolePurview = strings.Split(targetRole.PurviewIds, ",")
  245. // }
  246. //
  247. // userRolePurview = RemoveRepeatedElement2(userRolePurview)
  248. //
  249. // fmt.Println(userRolePurview)
  250. // //所有权限列表
  251. // allPermission := []models.Purview{{Id:10,Urlfor:"/m/api/dialysis/dialysisPrescription-/api/dialysis/prescription"},
  252. // {Id:11,Urlfor:"/m/api/dialysis/dialysisPrescription1-/api/dialysis/prescription1"},
  253. // {Id:12,Urlfor:"/m/api/dialysis/dialysisPrescription2-/api/dialysis/prescription2"},
  254. // {Id:13,Urlfor:"/m/api/dialysis/dialysisPrescription3-/api/dialysis/prescription3"},
  255. // {Id:14,Urlfor:"/m/api/dialysis/dialysisPrescription4-/api/dialysis/prescription4"},
  256. // {Id:15,Urlfor:"/m/api/dialysis/dialysisPrescription5-/api/dialysis/prescription5"},
  257. // {Id:16,Urlfor:"/m/api/dialysis/dialysisPrescription6-/api/dialysis/prescription6"}}
  258. // for _, item := range allPermission {
  259. // //判断当前路由是否在权限路由列表里面
  260. // if strings.Split(item.Urlfor, "-")[1] == strings.Split(this.Ctx.Request.RequestURI , "?")[0]{
  261. // fmt.Println(strings.Split(this.Ctx.Request.RequestURI , "?")[0])
  262. //
  263. // //获取该角色的所有权限
  264. // for _, items := range userRolePurview{
  265. // id, _ := strconv.ParseInt(items, 10, 64)
  266. // if id == item.Id{
  267. // isPermission = true
  268. // }
  269. // }
  270. // if !isPermission{
  271. // this.ServeFailJSONWithSGJErrorCode(enums.ErrorCodePreExist)
  272. // this.StopRun()
  273. // }
  274. // }
  275. //
  276. // }
  277. //}
  278. }
  279. type BaseServeAPIController struct {
  280. BaseAPIController
  281. }
  282. func (this *BaseServeAPIController) Prepare() {
  283. this.BaseAPIController.Prepare()
  284. if this.GetAdminUserInfo() == nil {
  285. //var userAdmin models.AdminUser
  286. //userAdmin.Id = 1448
  287. //userAdmin.Mobile = "13318599895"
  288. //
  289. //userAdmin.Id = 597
  290. //userAdmin.Mobile = "19874122664"
  291. //userAdmin.IsSuperAdmin = false
  292. //userAdmin.Status = 1
  293. //userAdmin.CreateTime = 1530786071
  294. //userAdmin.ModifyTime = 1530786071
  295. //var subscibe models.ServeSubscibe
  296. //subscibe.ID = 1
  297. //subscibe.OrgId = 4
  298. //subscibe.PeriodStart = 1538035409
  299. //subscibe.PeriodEnd = 1569571409
  300. //subscibe.State = 1
  301. //subscibe.Status = 1
  302. //subscibe.CreatedTime = 1538035409
  303. //subscibe.UpdatedTime = 1538035409
  304. //subscibes := make(map[int64]*models.ServeSubscibe, 0)
  305. //subscibes[4] = &subscibe
  306. //var adminUserInfo service.AdminUserInfo
  307. //adminUserInfo.CurrentOrgId = 4
  308. //adminUserInfo.CurrentAppId = 4
  309. //adminUserInfo.AdminUser = &userAdmin
  310. //adminUserInfo.Subscibes = subscibes
  311. //this.SetSession("admin_user_info", &adminUserInfo)
  312. this.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeNotLogin)
  313. this.StopRun()
  314. }
  315. //if adminUserInfo.AppRole != nil {
  316. // if adminUserInfo.AppRole.Id > 0 {
  317. // app_role, _ := service.FindAppRoleById(adminUserInfo.AppRole.Id)
  318. // if app_role != nil {
  319. // if app_role.Status != 1 {
  320. // this.DelSession("mobile_admin_user_info")
  321. // this.Ctx.SetCookie("token_cookie", "")
  322. // this.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeForbidden)
  323. // this.StopRun()
  324. // }
  325. // }
  326. // }
  327. //
  328. //}
  329. //fmt.Println("222222222")
  330. }
  331. func RemoveRepeatedPurviewElement2(arr []string) (newArr []string) {
  332. newArr = make([]string, 0)
  333. for i := 0; i < len(arr); i++ {
  334. repeat := false
  335. for j := i + 1; j < len(arr); j++ {
  336. if arr[i] == arr[j] {
  337. repeat = true
  338. break
  339. }
  340. }
  341. if !repeat {
  342. newArr = append(newArr, arr[i])
  343. }
  344. }
  345. return
  346. }