SystemController.go 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821
  1. package controllers
  2. import (
  3. "encoding/json"
  4. "fmt"
  5. "strconv"
  6. "strings"
  7. "time"
  8. "git.qianqiusoft.com/qianqiusoft/light-apiengine/models"
  9. "git.qianqiusoft.com/qianqiusoft/light-apiengine/entitys"
  10. sysmodel "git.qianqiusoft.com/qianqiusoft/light-apiengine/models"
  11. "git.qianqiusoft.com/qianqiusoft/light-apiengine/utils"
  12. )
  13. //https://www.kancloud.cn/xormplus/xorm/167077
  14. // _Login
  15. // @Title _Login
  16. // @Description 用户登录
  17. // @Param logininfo false "登录信息"
  18. // @Success 200 {object} sysmodel.Account
  19. // @Failure 403 :id is empty
  20. func System_Login(c *SystemController) {
  21. var logininfo sysmodel.LoginInfo
  22. c.Ctx.BindJSON(&logininfo)
  23. __none_func_system__(logininfo)
  24. //fmt.Println(logininfo)
  25. var user sysmodel.SysUser
  26. ret, err := c.Db.SQL(sysmodel.SqlUserLogin, logininfo.Account).Get(&user)
  27. if ret && err == nil {
  28. //TODO check password
  29. timestamp := uint64(time.Now().UnixNano())
  30. md5Pwd := utils.HashPassword(logininfo.Password, "")
  31. //密码错误
  32. if !strings.EqualFold(user.Password, md5Pwd) {
  33. c.Ctx.JSON(200, sysmodel.SysReturn{400, "passowrd incorrect!", nil})
  34. return
  35. }
  36. token := &entitys.Token{}
  37. timestamp_str := strconv.FormatUint(timestamp, 10)
  38. sec_tooken := utils.GenerateToken(logininfo.Account + timestamp_str)
  39. if v := utils.GetGlobalTokenStore().Get(sec_tooken); v == nil {
  40. token.AccessToken = sec_tooken
  41. token.RefreshToken = sec_tooken
  42. token.LoginID = logininfo.Account
  43. token.UserId = user.Id
  44. token.Result = 200
  45. //token.Password = pwd
  46. token.ServerIp = ""
  47. token.Domain = user.Domain
  48. utils.GetGlobalTokenStore().Set(sec_tooken, token)
  49. } else {
  50. token = v
  51. }
  52. data := sysmodel.LoginReturnInfo{}
  53. data.Userid = user.Id
  54. data.Name = user.LoginId
  55. data.Token = token.AccessToken
  56. data.Type = user.Type
  57. data.SysUserInfo = user
  58. data.SysUserInfo.Password = ""
  59. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", data})
  60. } else {
  61. //fmt.Println(err.Error())
  62. c.Ctx.JSON(200, sysmodel.SysReturn{400, "username or passowrd incorrect!", nil})
  63. }
  64. }
  65. // _Logout
  66. // @Title _Logout
  67. // @Description 用户退出
  68. // @Success 200 {object} sysmodel.Account
  69. // @Failure 403 :id is empty
  70. func System_Logout(c *SystemController) {
  71. token := c.Ctx.GetHeader("token")
  72. fmt.Println("delete token: ", token)
  73. utils.GetGlobalTokenStore().Remove(token)
  74. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", nil})
  75. }
  76. // _GetNavTree
  77. // @Title _GetNavTree
  78. // @Description 获取导航菜单
  79. // @Param user string false "用户id"
  80. // @Success 200 {object} models.Account
  81. // @Failure 403 :id is empty
  82. func System_GetNavTree(c *SystemController) {
  83. // set in the login_ware.go
  84. user_id := c.Ctx.GetString("user_id")
  85. var menus []sysmodel.SysMenuTree
  86. if user_id == "187c50a4-caad-4557-80c8-aca81af9f370"{
  87. c.Db.SQL(sysmodel.Selectall_nav_tree_admin).Find(&menus)
  88. }else{
  89. c.Db.SQL(sysmodel.Selectall_nav_tree, user_id).Find(&menus)
  90. }
  91. var menutree []sysmodel.SysMenuTree
  92. finChildrenMenu(&menus, "", nil, &menutree)
  93. for i := 0; i < len(menutree); i++ {
  94. finChildrenMenu(&menus, menutree[i].Id, &menutree[i], &menutree)
  95. }
  96. fmt.Println(menutree)
  97. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", menutree})
  98. }
  99. // _GetMenuTree
  100. // @Title _GetMenuTree
  101. // @Description 获取系统菜单
  102. // @Param user string false "用户id"
  103. // @Success 200 {object} sysmodel.Account
  104. // @Failure 403 :id is empty
  105. func System_GetMenuTree(c *SystemController) {
  106. user := c.Ctx.Param(":user")
  107. ret := __none_func_system__(user)
  108. var menus []sysmodel.SysMenuTree
  109. c.Db.SQL(sysmodel.Selectall_sys_menu).Find(&menus)
  110. var menutree []sysmodel.SysMenuTree
  111. finChildrenMenu(&menus, "", nil, &menutree)
  112. fmt.Println(menutree)
  113. for i := 0; i < len(menutree); i++ {
  114. finChildrenMenu(&menus, menutree[i].Id, &menutree[i], &menutree)
  115. }
  116. if ret {
  117. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", menutree})
  118. }
  119. }
  120. // _GetOrgTree
  121. // @Title _GetOrgTree
  122. // @Description 获取组织架构树
  123. // @Success 200 {object} sysmodel.Account
  124. // @Failure 403 :id is empty
  125. func System_GetOrgTree(c *SystemController) {
  126. ret := __none_func_system__()
  127. var orgs []sysmodel.SysOrgTree
  128. c.Db.SQL(sysmodel.Selectall_sys_org).Find(&orgs)
  129. var orgtree []sysmodel.SysOrgTree
  130. finChildrenOrg(&orgs, "", nil, &orgtree)
  131. for i := 0; i < len(orgtree); i++ {
  132. finChildrenOrg(&orgs, orgtree[i].Id, &orgtree[i], &orgtree)
  133. }
  134. if ret {
  135. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", orgtree})
  136. } else {
  137. c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
  138. }
  139. }
  140. func finChildrenOrg(menus *[]sysmodel.SysOrgTree, parent string, parent_menu *sysmodel.SysOrgTree, menutree *[]sysmodel.SysOrgTree) {
  141. for _, m := range *menus {
  142. if m.Parent == parent {
  143. if parent_menu == nil {
  144. *menutree = append(*menutree, m)
  145. //finChildrenMenu(menus, m.Id, menutree[len(*menutree)-1], menutree)
  146. } else {
  147. parent_menu.Children = append(parent_menu.Children, m)
  148. finChildrenOrg(menus, m.Id, &parent_menu.Children[len(parent_menu.Children)-1], menutree)
  149. }
  150. }
  151. }
  152. }
  153. func finChildrenMenu(menus *[]sysmodel.SysMenuTree, parent string, parent_menu *sysmodel.SysMenuTree, menutree *[]sysmodel.SysMenuTree) {
  154. for _, m := range *menus {
  155. if m.Parent == parent {
  156. if parent_menu == nil {
  157. m.Level = 0
  158. m.ParentName = ""
  159. *menutree = append(*menutree, m)
  160. //finChildrenMenu(menus, m.Id, menutree[len(*menutree)-1], menutree)
  161. } else {
  162. m.Level = parent_menu.Level + 1
  163. m.ParentName = parent_menu.Name
  164. parent_menu.Children = append(parent_menu.Children, m)
  165. finChildrenMenu(menus, m.Id, &parent_menu.Children[len(parent_menu.Children)-1], menutree)
  166. }
  167. }
  168. }
  169. }
  170. // _FindUserPage
  171. // @Title _FindUserPage
  172. // @Description 获取用户分布数据
  173. // @Param page false "分页参数"
  174. // @Success 200 {object} sysmodel.Account
  175. // @Failure 403 :id is empty
  176. func System_FindUserPage(c *SystemController) {
  177. var getpageinfo sysmodel.GetPageInfo
  178. c.Ctx.BindJSON(&getpageinfo)
  179. __none_func_system__(getpageinfo)
  180. page := sysmodel.PageResult{}
  181. var users []sysmodel.UserInfo
  182. dbSession := c.Db.Table("sys_user")
  183. dbSession.Select("sys_user.*, sys_org.name org_name")
  184. countSession := c.Db.Table("sys_user")
  185. JoinTables := []models.Join{}
  186. //JoinTables = append(JoinTables, models.Join{TabelName: "sys_user_role", Condition: "sys_user_role.user_id = sys_user.id", JoinOperator: "left"})
  187. //JoinTables = append(JoinTables, models.Join{TabelName: "sys_role", Condition: "sys_user_role.role_id = sys_role.id", JoinOperator: "left"})
  188. JoinTables = append(JoinTables, models.Join{TabelName: "sys_org", Condition: "sys_org.id = sys_user.org_id", JoinOperator: "left"})
  189. for _, join := range JoinTables {
  190. dbSession = dbSession.Join(join.JoinOperator, join.TabelName, join.Condition)
  191. countSession = countSession.Join(join.JoinOperator, join.TabelName, join.Condition)
  192. }
  193. whereStr, datas, err := getpageinfo.Filter.Parse("")
  194. dbSession = dbSession.Where(whereStr, datas...)
  195. countSession = countSession.Where(whereStr, datas...)
  196. sidx := strings.Split(getpageinfo.Sidx, ",")
  197. for _, order := range sidx {
  198. order = strings.TrimSpace(order)
  199. if len(order) == 0 {
  200. continue
  201. }
  202. if strings.ToLower(getpageinfo.Sord) == "desc" {
  203. dbSession = dbSession.OrderBy(order + " desc")
  204. } else {
  205. dbSession = dbSession.OrderBy(order + " asc")
  206. }
  207. }
  208. totalPage, _ := countSession.Count()
  209. dbSession.Limit(getpageinfo.PageSize, (getpageinfo.PageNum-1)*getpageinfo.PageSize).Find(&users)
  210. //查找每个用户的角色
  211. for i, user := range users {
  212. roles := []sysmodel.UserInfo{}
  213. err := c.Db.SQL(sysmodel.SqlSelectUserRoles, user.Id).Find(&roles)
  214. if err != nil {
  215. fmt.Println("----", err.Error())
  216. continue
  217. }
  218. users[i].UserRole = roles[0].UserRole
  219. users[i].RoleName = roles[0].RoleName
  220. }
  221. page.Content = users
  222. page.PageSize = getpageinfo.PageSize
  223. page.PageNum = getpageinfo.PageNum
  224. page.TotalSize = int(totalPage)
  225. if err == nil {
  226. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", page})
  227. } else {
  228. fmt.Println(err.Error())
  229. c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
  230. }
  231. }
  232. // _AddUser
  233. // @Title _AddUser
  234. // @Description 添加用户
  235. // @Param login_id string false "登录ID"
  236. // @Param password string false "密码"
  237. // @Param org_id string false "组织ID"
  238. // @Param email string false "邮箱"
  239. // @Success 200 {object} models.Account
  240. // @Failure 403 :id is empty
  241. func System_AddUser(c *SystemController) {
  242. var user sysmodel.SysUser
  243. err := c.Ctx.ShouldBindJSON(&user)
  244. if err != nil{
  245. c.Ctx.JSON(500, sysmodel.SysReturn{200, "c.Ctx.ShouldBindJSON error " + err.Error(), nil})
  246. return;
  247. }
  248. fmt.Println(user.LoginId, "-----add user-----", user.Password, " --- ", user.OrgId)
  249. //查找login_id是否存在
  250. oldUser := new(sysmodel.SysUser)
  251. count, _ := c.Db.SQL("select * from sys_user").Where("login_id = ?", user.LoginId).Count(&oldUser)
  252. if count > 1 {
  253. if oldUser.DelFlag == 1 {
  254. oldUser.DelFlag = 0
  255. _, err := c.Db.Id("id").Cols("del_flag").Update(&oldUser)
  256. if err != nil {
  257. c.Ctx.JSON(500, sysmodel.SysReturn{500, "login_id exit, update err: " + err.Error(), nil})
  258. }
  259. return
  260. } else {
  261. c.Ctx.JSON(200, sysmodel.SysReturn{200, "login_id exit", nil})
  262. return
  263. }
  264. }
  265. user_id := c.Ctx.GetString("user_id")
  266. user.Id = utils.NewUUID()
  267. user.Name = user.LoginId
  268. user.CreateTime = time.Now()
  269. user.LastUpdateDate = time.Now()
  270. user.CreateBy = user_id
  271. user.LastUpdateBy = user_id
  272. user.DelFlag = 0
  273. user.Password = utils.HashPassword(user.Password, "")
  274. _, err = c.Db.Insert(&user)
  275. ret := __none_func_system__(user.LoginId, user.Password, user.OrgId, user.Email)
  276. if ret && err == nil {
  277. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", nil})
  278. } else {
  279. c.Ctx.JSON(500, sysmodel.SysReturn{500, err.Error(), nil})
  280. }
  281. }
  282. // _DelUser
  283. // @Title _DelUser
  284. // @Description 删除用户
  285. // @Param id string false "用户ID"
  286. // @Success 200 {object} models.Account
  287. // @Failure 403 :id is empty
  288. func System_DelUser(c *SystemController) {
  289. type Param struct {
  290. Id string `json:"id"`
  291. }
  292. var params []Param
  293. c.Ctx.BindJSON(&params)
  294. ids := []string{}
  295. for _, param := range params {
  296. ids = append(ids, param.Id)
  297. }
  298. session := c.Db.NewSession()
  299. defer session.Close()
  300. //_, err := c.Db.Table(new(models.SysUser)).ID(param.Id).Update(map[string]interface{}{"del_flag": 0})
  301. _, err := c.Db.Table(new(models.SysUser)).In("id", ids).Update(map[string]interface{}{"del_flag": 1})
  302. if err != nil {
  303. session.Rollback()
  304. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  305. return
  306. }
  307. _, err = c.Db.Table(new(models.SysUserRole)).In("user_id", ids).Update(map[string]interface{}{"del_flag": 1})
  308. if err != nil {
  309. session.Rollback()
  310. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  311. return
  312. }
  313. session.Commit()
  314. ret := __none_func_system__()
  315. if ret {
  316. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", nil})
  317. } else {
  318. c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
  319. }
  320. }
  321. // _UpdateUser
  322. // @Title _UpdateUser
  323. // @Description 修改用户
  324. // @Param login_id string false "登录ID"
  325. // @Param password string false "密码"
  326. // @Param org_id string false "组织ID"
  327. // @Param email string false "邮箱"
  328. // @Success 200 {object} models.Account
  329. // @Failure 403 :id is empty
  330. func System_UpdateUser(c *SystemController) {
  331. user_id := c.Ctx.GetString("user_id")
  332. var userInfo sysmodel.UserInfo
  333. err := c.Ctx.BindJSON(&userInfo)
  334. if err != nil{
  335. c.Ctx.JSON(500, sysmodel.SysReturn{500, err.Error(), nil})
  336. return;
  337. }
  338. session := c.Db.NewSession()
  339. defer session.Close()
  340. //1、删除用户角色
  341. _, err = c.Db.SQL(sysmodel.SqlDeleteUserRoles, userInfo.Id).Execute()
  342. if err != nil {
  343. session.Rollback()
  344. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  345. return
  346. }
  347. //2、修改角色
  348. userRoles := []models.SysUserRole{}
  349. if userInfo.UserRole != "" {
  350. err = json.Unmarshal([]byte(userInfo.UserRole), &userRoles)
  351. if err != nil {
  352. session.Rollback()
  353. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  354. return
  355. }
  356. }
  357. for i, _ := range userRoles {
  358. userRoles[i].Id = utils.NewUUID()
  359. userRoles[i].CreateBy = user_id
  360. userRoles[i].LastUpdateBy = user_id
  361. userRoles[i].CreateTime = time.Now()
  362. userRoles[i].LastUpdateDate = time.Now()
  363. userRoles[i].DelFlag = 0
  364. }
  365. _, err = c.Db.Insert(userRoles)
  366. if err != nil {
  367. session.Rollback()
  368. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  369. return
  370. }
  371. var updateUser models.SysUser
  372. updateUser.Name = userInfo.Name
  373. updateUser.FullName = userInfo.FullName
  374. updateUser.Mobile = userInfo.Mobile
  375. updateUser.Type = userInfo.Type
  376. updateUser.Status = userInfo.Status
  377. updateUser.Domain = userInfo.Domain
  378. updateUser.OptionA = userInfo.OptionA
  379. updateUser.OptionB = userInfo.OptionB
  380. updateUser.LastUpdateBy = user_id
  381. updateUser.LastUpdateDate = time.Now()
  382. //3、更新用户信息
  383. //_, 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": user_id, "last_update_date": time.Now()})
  384. _,err = c.Db.Id(userInfo.Id).Update(&updateUser)
  385. if err != nil {
  386. session.Rollback()
  387. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  388. return
  389. }
  390. session.Commit()
  391. ret := __none_func_system__()
  392. if ret {
  393. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", nil})
  394. } else {
  395. c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
  396. }
  397. }
  398. // _AddRole
  399. // @Title _AddRole
  400. // @Description 添加角色
  401. // @Param name string false "角色名称"
  402. // @Param remark string false "备注"
  403. // @Success 200 {object} models.Account
  404. // @Failure 403 :id is empty
  405. func System_AddRole(c *SystemController) {
  406. var role models.SysRole
  407. c.Ctx.BindJSON(&role)
  408. user_id := c.Ctx.GetString("user_id")
  409. role.Id = utils.NewUUID()
  410. role.DelFlag = 0
  411. role.CreateBy = user_id
  412. role.LastUpdateBy = user_id
  413. role.CreateTime = time.Now()
  414. role.LastUpdateDate = time.Now()
  415. _, err := c.Db.Insert(&role)
  416. ret := __none_func_system__()
  417. if ret && err == nil {
  418. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", nil})
  419. } else {
  420. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  421. }
  422. }
  423. // _DelRole
  424. // @Title _DelRole
  425. // @Description 添加角色
  426. // @Param id string false "角色ID"
  427. // @Success 200 {object} models.Account
  428. // @Failure 403 :id is empty
  429. func System_DelRole(c *SystemController) {
  430. type Param struct {
  431. Id string `json:"id"`
  432. }
  433. var params []Param
  434. c.Ctx.BindJSON(&params)
  435. fmt.Println(params, "-----------")
  436. ids := []string{}
  437. for _, param := range params {
  438. ids = append(ids, param.Id)
  439. }
  440. _, err := c.Db.Table(new(models.SysRole)).In("id", ids).Update(map[string]interface{}{"del_flag": 1})
  441. if err != nil {
  442. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  443. return
  444. }
  445. ret := __none_func_system__()
  446. if ret {
  447. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", nil})
  448. } else {
  449. c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
  450. }
  451. }
  452. // _UpdateRole
  453. // @Title _UpdateRole
  454. // @Description 修改角色
  455. // @Param id string false "角色ID"
  456. // @Param name string false "角色名称"
  457. // @Param remark string false "备注"
  458. // @Success 200 {object} models.Account
  459. // @Failure 403 :id is empty
  460. func System_UpdateRole(c *SystemController) {
  461. var role models.SysRole
  462. c.Ctx.BindJSON(&role)
  463. user_id := c.Ctx.GetString("user_id")
  464. role.LastUpdateBy = user_id
  465. role.LastUpdateDate = time.Now()
  466. c.Db.ID(role.Id).Update(&role)
  467. ret := __none_func_system__()
  468. if ret {
  469. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", nil})
  470. } else {
  471. c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
  472. }
  473. }
  474. // _FindRolePage
  475. // @Title _FindRolePage
  476. // @Description 角色分页
  477. // @Param page false "分页参数"
  478. // @Success 200 {object} models.Account
  479. // @Failure 403 :id is empty
  480. func System_FindRolePage(c *SystemController) {
  481. var getpageinfo models.GetPageInfo
  482. c.Ctx.BindJSON(&getpageinfo)
  483. page := sysmodel.SysRolePageInfo{}
  484. var roles []sysmodel.SysRole
  485. err := c.Db.SQL(sysmodel.Selectall_sys_role).Limit(getpageinfo.PageSize, (getpageinfo.PageNum-1)*getpageinfo.PageSize).Find(&roles)
  486. if err != nil {
  487. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  488. return
  489. }
  490. fmt.Println("-----------", len(roles))
  491. page.Content = roles
  492. page.PageSize = getpageinfo.PageSize
  493. page.PageNum = getpageinfo.PageNum
  494. page.TotalSize = 1
  495. ret := __none_func_system__(getpageinfo)
  496. if ret {
  497. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", page})
  498. } else {
  499. c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
  500. }
  501. }
  502. // _FindRoleMenu
  503. // @Title _FindRoleMenu
  504. // @Description 查找角色权限
  505. // @Param user string false "用户id"
  506. // @Success 200 {object} models.Account
  507. // @Failure 403 :id is empty
  508. func System_FindRoleMenu(c *SystemController) {
  509. role_id, b := c.Ctx.GetQuery("roleId")
  510. if !b {
  511. c.Ctx.JSON(200, sysmodel.SysReturn{500, "roleId is null", nil})
  512. return
  513. }
  514. menus := []models.SysMenu{}
  515. c.Db.SQL(models.SqlFindRoleMenu, role_id).Find(&menus)
  516. ret := __none_func_system__()
  517. if ret {
  518. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", menus})
  519. } else {
  520. c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
  521. }
  522. }
  523. // _SaveRoleMenu
  524. // @Title _SaveRoleMenu
  525. // @Description 保存角色权限
  526. // @Param role_menu string false "角色权限"
  527. // @Success 200 {object} models.Account
  528. // @Failure 403 :id is empty
  529. func System_SaveRoleMenu(c *SystemController) {
  530. var saverolemenu []models.SysRoleMenu
  531. c.Ctx.BindJSON(&saverolemenu)
  532. if len(saverolemenu) < 1 {
  533. c.Ctx.JSON(200, sysmodel.SysReturn{500, "role menu length is error", nil})
  534. return
  535. }
  536. session := c.Db.NewSession()
  537. defer session.Close()
  538. err := session.Begin()
  539. //删除角色下的menu
  540. _, err = c.Db.Exec("delete from sys_role_menu where role_id = ?", saverolemenu[0].RoleId)
  541. //_, err = c.Db.Exec("delete from sys_role_menu where id = 1")
  542. if err != nil {
  543. session.Rollback()
  544. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  545. return
  546. }
  547. for i, _ := range saverolemenu {
  548. saverolemenu[i].Id = utils.NewUUID()
  549. saverolemenu[i].DelFlag = 1
  550. }
  551. _, err = c.Db.Insert(&saverolemenu)
  552. if err != nil {
  553. session.Rollback()
  554. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  555. return
  556. }
  557. session.Commit()
  558. ret := __none_func_system__(&saverolemenu)
  559. if ret {
  560. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", nil})
  561. } else {
  562. c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
  563. }
  564. }
  565. // _RoleAll
  566. // @Title _RoleAll
  567. // @Description 查找所有角色
  568. // @Success 200 {object} models.Account
  569. // @Failure 403 :id is empty
  570. func System_RoleAll(c *SystemController) {
  571. var roles []sysmodel.SysRole
  572. err := c.Db.SQL(sysmodel.Selectall_sys_role).Find(&roles)
  573. ret := __none_func_system__()
  574. if ret && err == nil {
  575. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", roles})
  576. } else {
  577. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  578. }
  579. }
  580. // _FindPermissions
  581. // @Title _FindPermissions
  582. // @Description 查找用户权限
  583. // @Success 200 {object} models.Account
  584. // @Failure 403 :id is empty
  585. func System_FindPermissions(c *SystemController) {
  586. user_id := c.Ctx.GetString("user_id")
  587. //result, err := c.Db.Query(models.SqlFindUserPermissions, op.UserId)
  588. result, err := c.Db.QueryString(models.SqlFindUserPermissions, user_id)
  589. strs := []string{}
  590. for _, v := range result {
  591. strs = append(strs, v["perms"])
  592. }
  593. if err != nil {
  594. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  595. }
  596. ret := __none_func_system__()
  597. //strs = []string{"sys:menu:add", "sys:menu:view", "sys:menu:edit", "sys:menu:delete", "sys:user:edit", "sys:user:delete", "sys:user:add", "sys:role:view", "sys:role:edit", "sys:role:delete", "sys:role:add"}
  598. if ret {
  599. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", strs})
  600. } else {
  601. c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
  602. }
  603. }
  604. // _AddMenu
  605. // @Title _AddMenu
  606. // @Description 添加菜单
  607. // @Param string false "菜单"
  608. // @Success 200 {object} models.Account
  609. // @Failure 403 :id is empty
  610. func System_AddMenu(c *SystemController) {
  611. var sysmenu models.SysMenu
  612. c.Ctx.BindJSON(&sysmenu)
  613. user_id := c.Ctx.GetString("user_id")
  614. sysmenu.Id = utils.NewUUID()
  615. sysmenu.CreateBy = user_id
  616. sysmenu.LastUpdateBy = user_id
  617. sysmenu.LastUpdateDate = time.Now()
  618. sysmenu.CreateTime = time.Now()
  619. sysmenu.DelFlag = 0
  620. _, err := c.Db.Insert(&sysmenu)
  621. if err != nil {
  622. c.Ctx.JSON(500, sysmodel.SysReturn{500, err.Error(), nil})
  623. return
  624. }
  625. ret := __none_func_system__(sysmenu)
  626. if ret {
  627. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", nil})
  628. } else {
  629. c.Ctx.JSON(500, sysmodel.SysReturn{500, "", nil})
  630. }
  631. }
  632. // _DelMenu
  633. // @Title _DelMenu
  634. // @Description 删除菜单
  635. // @Param id string false "菜单ID"
  636. // @Success 200 {object} models.Account
  637. // @Failure 403 :id is empty
  638. func System_DelMenu(c *SystemController) {
  639. type Param struct {
  640. Id string `json:"id"`
  641. }
  642. var params []Param
  643. c.Ctx.BindJSON(&params)
  644. fmt.Println(params, "-----------")
  645. ids := []string{}
  646. for _, param := range params {
  647. ids = append(ids, param.Id)
  648. }
  649. _, err := c.Db.Table(new(models.SysMenu)).In("id", ids).Update(map[string]interface{}{"del_flag": 0})
  650. if err != nil {
  651. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  652. return
  653. }
  654. ret := __none_func_system__()
  655. if ret {
  656. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", nil})
  657. } else {
  658. c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
  659. }
  660. }
  661. // _UpdateMenu
  662. // @Title _UpdateMenu
  663. // @Description 更新菜单
  664. // @Param string false "菜单"
  665. // @Success 200 {object} models.Account
  666. // @Failure 403 :id is empty
  667. func System_UpdateMenu(c *SystemController) {
  668. var sysmenu models.SysMenu
  669. c.Ctx.BindJSON(&sysmenu)
  670. user_id := c.Ctx.GetString("user_id")
  671. sysmenu.LastUpdateBy = user_id
  672. sysmenu.LastUpdateDate = time.Now()
  673. _, err := c.Db.ID(sysmenu.Id).Update(&sysmenu)
  674. if err != nil {
  675. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  676. return
  677. }
  678. ret := __none_func_system__(sysmenu)
  679. if ret {
  680. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", nil})
  681. } else {
  682. c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
  683. }
  684. }
  685. func __none_func_system__(params ...interface{}) bool {
  686. return true
  687. }