SystemController.go 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799
  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. c.Ctx.BindJSON(&userInfo)
  334. session := c.Db.NewSession()
  335. defer session.Close()
  336. //1、删除用户角色
  337. _, err := c.Db.SQL(sysmodel.SqlDeleteUserRoles, userInfo.Id).Execute()
  338. if err != nil {
  339. session.Rollback()
  340. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  341. return
  342. }
  343. //2、修改角色
  344. userRoles := []models.SysUserRole{}
  345. err = json.Unmarshal([]byte(userInfo.UserRole), &userRoles)
  346. if err != nil {
  347. session.Rollback()
  348. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  349. return
  350. }
  351. for i, _ := range userRoles {
  352. userRoles[i].Id = utils.NewUUID()
  353. userRoles[i].CreateBy = user_id
  354. userRoles[i].LastUpdateBy = user_id
  355. userRoles[i].CreateTime = time.Now()
  356. userRoles[i].LastUpdateDate = time.Now()
  357. userRoles[i].DelFlag = 0
  358. }
  359. _, err = c.Db.Insert(userRoles)
  360. if err != nil {
  361. session.Rollback()
  362. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  363. return
  364. }
  365. //3、更新用户信息
  366. _, 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()})
  367. if err != nil {
  368. session.Rollback()
  369. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  370. return
  371. }
  372. session.Commit()
  373. ret := __none_func_system__()
  374. if ret {
  375. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", nil})
  376. } else {
  377. c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
  378. }
  379. }
  380. // _AddRole
  381. // @Title _AddRole
  382. // @Description 添加角色
  383. // @Param name string false "角色名称"
  384. // @Param remark string false "备注"
  385. // @Success 200 {object} models.Account
  386. // @Failure 403 :id is empty
  387. func System_AddRole(c *SystemController) {
  388. var role models.SysRole
  389. c.Ctx.BindJSON(&role)
  390. user_id := c.Ctx.GetString("user_id")
  391. role.Id = utils.NewUUID()
  392. role.DelFlag = 0
  393. role.CreateBy = user_id
  394. role.LastUpdateBy = user_id
  395. role.CreateTime = time.Now()
  396. role.LastUpdateDate = time.Now()
  397. _, err := c.Db.Insert(&role)
  398. ret := __none_func_system__()
  399. if ret && err == nil {
  400. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", nil})
  401. } else {
  402. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  403. }
  404. }
  405. // _DelRole
  406. // @Title _DelRole
  407. // @Description 添加角色
  408. // @Param id string false "角色ID"
  409. // @Success 200 {object} models.Account
  410. // @Failure 403 :id is empty
  411. func System_DelRole(c *SystemController) {
  412. type Param struct {
  413. Id string `json:"id"`
  414. }
  415. var params []Param
  416. c.Ctx.BindJSON(&params)
  417. fmt.Println(params, "-----------")
  418. ids := []string{}
  419. for _, param := range params {
  420. ids = append(ids, param.Id)
  421. }
  422. _, err := c.Db.Table(new(models.SysRole)).In("id", ids).Update(map[string]interface{}{"del_flag": 1})
  423. if err != nil {
  424. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  425. return
  426. }
  427. ret := __none_func_system__()
  428. if ret {
  429. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", nil})
  430. } else {
  431. c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
  432. }
  433. }
  434. // _UpdateRole
  435. // @Title _UpdateRole
  436. // @Description 修改角色
  437. // @Param id string false "角色ID"
  438. // @Param name string false "角色名称"
  439. // @Param remark string false "备注"
  440. // @Success 200 {object} models.Account
  441. // @Failure 403 :id is empty
  442. func System_UpdateRole(c *SystemController) {
  443. var role models.SysRole
  444. c.Ctx.BindJSON(&role)
  445. user_id := c.Ctx.GetString("user_id")
  446. role.LastUpdateBy = user_id
  447. role.LastUpdateDate = time.Now()
  448. c.Db.ID(role.Id).Update(&role)
  449. ret := __none_func_system__()
  450. if ret {
  451. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", nil})
  452. } else {
  453. c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
  454. }
  455. }
  456. // _FindRolePage
  457. // @Title _FindRolePage
  458. // @Description 角色分页
  459. // @Param page false "分页参数"
  460. // @Success 200 {object} models.Account
  461. // @Failure 403 :id is empty
  462. func System_FindRolePage(c *SystemController) {
  463. var getpageinfo models.GetPageInfo
  464. c.Ctx.BindJSON(&getpageinfo)
  465. page := sysmodel.SysRolePageInfo{}
  466. var roles []sysmodel.SysRole
  467. err := c.Db.SQL(sysmodel.Selectall_sys_role).Limit(getpageinfo.PageSize, (getpageinfo.PageNum-1)*getpageinfo.PageSize).Find(&roles)
  468. if err != nil {
  469. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  470. return
  471. }
  472. fmt.Println("-----------", len(roles))
  473. page.Content = roles
  474. page.PageSize = getpageinfo.PageSize
  475. page.PageNum = getpageinfo.PageNum
  476. page.TotalSize = 1
  477. ret := __none_func_system__(getpageinfo)
  478. if ret {
  479. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", page})
  480. } else {
  481. c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
  482. }
  483. }
  484. // _FindRoleMenu
  485. // @Title _FindRoleMenu
  486. // @Description 查找角色权限
  487. // @Param user string false "用户id"
  488. // @Success 200 {object} models.Account
  489. // @Failure 403 :id is empty
  490. func System_FindRoleMenu(c *SystemController) {
  491. role_id, b := c.Ctx.GetQuery("roleId")
  492. if !b {
  493. c.Ctx.JSON(200, sysmodel.SysReturn{500, "roleId is null", nil})
  494. return
  495. }
  496. menus := []models.SysMenu{}
  497. c.Db.SQL(models.SqlFindRoleMenu, role_id).Find(&menus)
  498. ret := __none_func_system__()
  499. if ret {
  500. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", menus})
  501. } else {
  502. c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
  503. }
  504. }
  505. // _SaveRoleMenu
  506. // @Title _SaveRoleMenu
  507. // @Description 保存角色权限
  508. // @Param role_menu string false "角色权限"
  509. // @Success 200 {object} models.Account
  510. // @Failure 403 :id is empty
  511. func System_SaveRoleMenu(c *SystemController) {
  512. var saverolemenu []models.SysRoleMenu
  513. c.Ctx.BindJSON(&saverolemenu)
  514. if len(saverolemenu) < 1 {
  515. c.Ctx.JSON(200, sysmodel.SysReturn{500, "role menu length is error", nil})
  516. return
  517. }
  518. session := c.Db.NewSession()
  519. defer session.Close()
  520. err := session.Begin()
  521. //删除角色下的menu
  522. _, err = c.Db.Exec("delete from sys_role_menu where role_id = ?", saverolemenu[0].RoleId)
  523. //_, err = c.Db.Exec("delete from sys_role_menu where id = 1")
  524. if err != nil {
  525. session.Rollback()
  526. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  527. return
  528. }
  529. for i, _ := range saverolemenu {
  530. saverolemenu[i].Id = utils.NewUUID()
  531. saverolemenu[i].DelFlag = 1
  532. }
  533. _, err = c.Db.Insert(&saverolemenu)
  534. if err != nil {
  535. session.Rollback()
  536. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  537. return
  538. }
  539. session.Commit()
  540. ret := __none_func_system__(&saverolemenu)
  541. if ret {
  542. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", nil})
  543. } else {
  544. c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
  545. }
  546. }
  547. // _RoleAll
  548. // @Title _RoleAll
  549. // @Description 查找所有角色
  550. // @Success 200 {object} models.Account
  551. // @Failure 403 :id is empty
  552. func System_RoleAll(c *SystemController) {
  553. var roles []sysmodel.SysRole
  554. err := c.Db.SQL(sysmodel.Selectall_sys_role).Find(&roles)
  555. ret := __none_func_system__()
  556. if ret && err == nil {
  557. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", roles})
  558. } else {
  559. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  560. }
  561. }
  562. // _FindPermissions
  563. // @Title _FindPermissions
  564. // @Description 查找用户权限
  565. // @Success 200 {object} models.Account
  566. // @Failure 403 :id is empty
  567. func System_FindPermissions(c *SystemController) {
  568. user_id := c.Ctx.GetString("user_id")
  569. //result, err := c.Db.Query(models.SqlFindUserPermissions, op.UserId)
  570. result, err := c.Db.QueryString(models.SqlFindUserPermissions, user_id)
  571. strs := []string{}
  572. for _, v := range result {
  573. strs = append(strs, v["perms"])
  574. }
  575. if err != nil {
  576. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  577. }
  578. ret := __none_func_system__()
  579. //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"}
  580. if ret {
  581. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", strs})
  582. } else {
  583. c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
  584. }
  585. }
  586. // _AddMenu
  587. // @Title _AddMenu
  588. // @Description 添加菜单
  589. // @Param string false "菜单"
  590. // @Success 200 {object} models.Account
  591. // @Failure 403 :id is empty
  592. func System_AddMenu(c *SystemController) {
  593. var sysmenu models.SysMenu
  594. c.Ctx.BindJSON(&sysmenu)
  595. user_id := c.Ctx.GetString("user_id")
  596. sysmenu.Id = utils.NewUUID()
  597. sysmenu.CreateBy = user_id
  598. sysmenu.LastUpdateBy = user_id
  599. sysmenu.LastUpdateDate = time.Now()
  600. sysmenu.CreateTime = time.Now()
  601. sysmenu.DelFlag = 0
  602. _, err := c.Db.Insert(&sysmenu)
  603. if err != nil {
  604. c.Ctx.JSON(500, sysmodel.SysReturn{500, err.Error(), nil})
  605. return
  606. }
  607. ret := __none_func_system__(sysmenu)
  608. if ret {
  609. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", nil})
  610. } else {
  611. c.Ctx.JSON(500, sysmodel.SysReturn{500, "", nil})
  612. }
  613. }
  614. // _DelMenu
  615. // @Title _DelMenu
  616. // @Description 删除菜单
  617. // @Param id string false "菜单ID"
  618. // @Success 200 {object} models.Account
  619. // @Failure 403 :id is empty
  620. func System_DelMenu(c *SystemController) {
  621. type Param struct {
  622. Id string `json:"id"`
  623. }
  624. var params []Param
  625. c.Ctx.BindJSON(&params)
  626. fmt.Println(params, "-----------")
  627. ids := []string{}
  628. for _, param := range params {
  629. ids = append(ids, param.Id)
  630. }
  631. _, err := c.Db.Table(new(models.SysMenu)).In("id", ids).Update(map[string]interface{}{"del_flag": 0})
  632. if err != nil {
  633. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  634. return
  635. }
  636. ret := __none_func_system__()
  637. if ret {
  638. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", nil})
  639. } else {
  640. c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
  641. }
  642. }
  643. // _UpdateMenu
  644. // @Title _UpdateMenu
  645. // @Description 更新菜单
  646. // @Param string false "菜单"
  647. // @Success 200 {object} models.Account
  648. // @Failure 403 :id is empty
  649. func System_UpdateMenu(c *SystemController) {
  650. var sysmenu models.SysMenu
  651. c.Ctx.BindJSON(&sysmenu)
  652. user_id := c.Ctx.GetString("user_id")
  653. sysmenu.LastUpdateBy = user_id
  654. sysmenu.LastUpdateDate = time.Now()
  655. _, err := c.Db.ID(sysmenu.Id).Update(&sysmenu)
  656. if err != nil {
  657. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  658. return
  659. }
  660. ret := __none_func_system__(sysmenu)
  661. if ret {
  662. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", nil})
  663. } else {
  664. c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
  665. }
  666. }
  667. func __none_func_system__(params ...interface{}) bool {
  668. return true
  669. }