SystemController.go 20 KB

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