SystemController.go 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794
  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. if v := utils.GlobalTokenStore.Get(logininfo.Account + user.Domain); v == nil {
  38. timestamp_str := strconv.FormatUint(timestamp, 10)
  39. sec_tooken := utils.GenerateToken(logininfo.Account + timestamp_str)
  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.GlobalTokenStore.Set(sec_tooken, token)
  49. //设置session
  50. //fmt.Println("----------set token ----------", token.AccessToken)
  51. //session := sessions.Default(c.Ctx)
  52. //session.Set("token", token.AccessToken)
  53. //session.Save()
  54. } else {
  55. token = v
  56. }
  57. data := sysmodel.LoginReturnInfo{user.Id, user.LoginId, token.AccessToken}
  58. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", data})
  59. } else {
  60. //fmt.Println(err.Error())
  61. c.Ctx.JSON(200, sysmodel.SysReturn{400, "username or passowrd incorrect!", nil})
  62. }
  63. }
  64. // _Logout
  65. // @Title _Logout
  66. // @Description 用户退出
  67. // @Success 200 {object} sysmodel.Account
  68. // @Failure 403 :id is empty
  69. func System_Logout(c *SystemController) {
  70. token := c.Ctx.GetHeader("token")
  71. fmt.Println("delete token: ", token)
  72. utils.GlobalTokenStore.Remove(token)
  73. ret := __none_func_system__()
  74. if ret {
  75. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", nil})
  76. } else {
  77. c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
  78. }
  79. }
  80. // _GetNavTree
  81. // @Title _GetNavTree
  82. // @Description 获取导航菜单
  83. // @Param user string false "用户id"
  84. // @Success 200 {object} models.Account
  85. // @Failure 403 :id is empty
  86. func System_GetNavTree(c *SystemController) {
  87. user_id := c.Ctx.GetString("user_id")
  88. ret := __none_func_system__(user_id)
  89. var menus []sysmodel.SysMenuTree
  90. c.Db.SQL(sysmodel.Selectall_nav_tree, user_id).Find(&menus)
  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. if ret {
  97. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", menutree})
  98. }
  99. }
  100. // _GetMenuTree
  101. // @Title _GetMenuTree
  102. // @Description 获取系统菜单
  103. // @Param user string false "用户id"
  104. // @Success 200 {object} sysmodel.Account
  105. // @Failure 403 :id is empty
  106. func System_GetMenuTree(c *SystemController) {
  107. user := c.Ctx.Param(":user")
  108. ret := __none_func_system__(user)
  109. var menus []sysmodel.SysMenuTree
  110. c.Db.SQL(sysmodel.Selectall_sys_menu).Find(&menus)
  111. var menutree []sysmodel.SysMenuTree
  112. finChildrenMenu(&menus, "", nil, &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. c.Ctx.BindJSON(&user)
  244. fmt.Println(user.LoginId, "-----add user-----", user.Password, " --- ", user.OrgId)
  245. //查找login_id是否存在
  246. oldUser := new(sysmodel.SysUser)
  247. count, _ := c.Db.SQL("select * from sys_user").Where("login_id = ?", user.LoginId).Count(&oldUser)
  248. if count > 1 {
  249. if oldUser.DelFlag == 1 {
  250. oldUser.DelFlag = 0
  251. _, err := c.Db.Id("id").Cols("del_flag").Update(&oldUser)
  252. if err != nil {
  253. c.Ctx.JSON(200, sysmodel.SysReturn{500, "login_id exit, update err: " + err.Error(), nil})
  254. }
  255. return
  256. } else {
  257. c.Ctx.JSON(200, sysmodel.SysReturn{500, "login_id exit", nil})
  258. return
  259. }
  260. }
  261. user_id := c.Ctx.GetString("user_id")
  262. user.Id = utils.NewUUID()
  263. user.Name = user.LoginId
  264. user.CreateTime = time.Now()
  265. user.LastUpdateDate = time.Now()
  266. user.CreateBy = user_id
  267. user.LastUpdateBy = user_id
  268. user.DelFlag = 0
  269. user.Password = utils.HashPassword(user.Password, "")
  270. _, err := c.Db.Insert(&user)
  271. ret := __none_func_system__(user.LoginId, user.Password, user.OrgId, user.Email)
  272. if ret && err == nil {
  273. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", nil})
  274. } else {
  275. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  276. }
  277. }
  278. // _DelUser
  279. // @Title _DelUser
  280. // @Description 删除用户
  281. // @Param id string false "用户ID"
  282. // @Success 200 {object} models.Account
  283. // @Failure 403 :id is empty
  284. func System_DelUser(c *SystemController) {
  285. type Param struct {
  286. Id string `json:"id"`
  287. }
  288. var params []Param
  289. c.Ctx.BindJSON(&params)
  290. ids := []string{}
  291. for _, param := range params {
  292. ids = append(ids, param.Id)
  293. }
  294. session := c.Db.NewSession()
  295. defer session.Close()
  296. //_, err := c.Db.Table(new(models.SysUser)).ID(param.Id).Update(map[string]interface{}{"del_flag": 0})
  297. _, err := c.Db.Table(new(models.SysUser)).In("id", ids).Update(map[string]interface{}{"del_flag": 1})
  298. if err != nil {
  299. session.Rollback()
  300. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  301. return
  302. }
  303. _, err = c.Db.Table(new(models.SysUserRole)).In("user_id", ids).Update(map[string]interface{}{"del_flag": 1})
  304. if err != nil {
  305. session.Rollback()
  306. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  307. return
  308. }
  309. session.Commit()
  310. ret := __none_func_system__()
  311. if ret {
  312. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", nil})
  313. } else {
  314. c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
  315. }
  316. }
  317. // _UpdateUser
  318. // @Title _UpdateUser
  319. // @Description 修改用户
  320. // @Param login_id string false "登录ID"
  321. // @Param password string false "密码"
  322. // @Param org_id string false "组织ID"
  323. // @Param email string false "邮箱"
  324. // @Success 200 {object} models.Account
  325. // @Failure 403 :id is empty
  326. func System_UpdateUser(c *SystemController) {
  327. user_id := c.Ctx.GetString("user_id")
  328. var userInfo sysmodel.UserInfo
  329. c.Ctx.BindJSON(&userInfo)
  330. session := c.Db.NewSession()
  331. defer session.Close()
  332. //1、删除用户角色
  333. _, err := c.Db.SQL(sysmodel.SqlDeleteUserRoles, userInfo.Id).Execute()
  334. if err != nil {
  335. session.Rollback()
  336. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  337. return
  338. }
  339. //2、修改角色
  340. userRoles := []models.SysUserRole{}
  341. err = json.Unmarshal([]byte(userInfo.UserRole), &userRoles)
  342. if err != nil {
  343. session.Rollback()
  344. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  345. return
  346. }
  347. for i, _ := range userRoles {
  348. userRoles[i].Id = utils.NewUUID()
  349. userRoles[i].CreateBy = user_id
  350. userRoles[i].LastUpdateBy = user_id
  351. userRoles[i].CreateTime = time.Now()
  352. userRoles[i].LastUpdateDate = time.Now()
  353. userRoles[i].DelFlag = 0
  354. }
  355. _, err = c.Db.Insert(userRoles)
  356. if err != nil {
  357. session.Rollback()
  358. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  359. return
  360. }
  361. //3、更新用户信息
  362. _, 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()})
  363. if err != nil {
  364. session.Rollback()
  365. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  366. return
  367. }
  368. session.Commit()
  369. ret := __none_func_system__()
  370. if ret {
  371. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", nil})
  372. } else {
  373. c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
  374. }
  375. }
  376. // _AddRole
  377. // @Title _AddRole
  378. // @Description 添加角色
  379. // @Param name string false "角色名称"
  380. // @Param remark string false "备注"
  381. // @Success 200 {object} models.Account
  382. // @Failure 403 :id is empty
  383. func System_AddRole(c *SystemController) {
  384. var role models.SysRole
  385. c.Ctx.BindJSON(&role)
  386. user_id := c.Ctx.GetString("user_id")
  387. role.Id = utils.NewUUID()
  388. role.DelFlag = 0
  389. role.CreateBy = user_id
  390. role.LastUpdateBy = user_id
  391. role.CreateTime = time.Now()
  392. role.LastUpdateDate = time.Now()
  393. _, err := c.Db.Insert(&role)
  394. ret := __none_func_system__()
  395. if ret && err == nil {
  396. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", nil})
  397. } else {
  398. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  399. }
  400. }
  401. // _DelRole
  402. // @Title _DelRole
  403. // @Description 添加角色
  404. // @Param id string false "角色ID"
  405. // @Success 200 {object} models.Account
  406. // @Failure 403 :id is empty
  407. func System_DelRole(c *SystemController) {
  408. type Param struct {
  409. Id string `json:"id"`
  410. }
  411. var params []Param
  412. c.Ctx.BindJSON(&params)
  413. fmt.Println(params, "-----------")
  414. ids := []string{}
  415. for _, param := range params {
  416. ids = append(ids, param.Id)
  417. }
  418. _, err := c.Db.Table(new(models.SysRole)).In("id", ids).Update(map[string]interface{}{"del_flag": 1})
  419. if err != nil {
  420. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  421. return
  422. }
  423. ret := __none_func_system__()
  424. if ret {
  425. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", nil})
  426. } else {
  427. c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
  428. }
  429. }
  430. // _UpdateRole
  431. // @Title _UpdateRole
  432. // @Description 修改角色
  433. // @Param id string false "角色ID"
  434. // @Param name string false "角色名称"
  435. // @Param remark string false "备注"
  436. // @Success 200 {object} models.Account
  437. // @Failure 403 :id is empty
  438. func System_UpdateRole(c *SystemController) {
  439. var role models.SysRole
  440. c.Ctx.BindJSON(&role)
  441. user_id := c.Ctx.GetString("user_id")
  442. role.LastUpdateBy = user_id
  443. role.LastUpdateDate = time.Now()
  444. c.Db.ID(role.Id).Update(&role)
  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. // _FindRolePage
  453. // @Title _FindRolePage
  454. // @Description 角色分页
  455. // @Param page false "分页参数"
  456. // @Success 200 {object} models.Account
  457. // @Failure 403 :id is empty
  458. func System_FindRolePage(c *SystemController) {
  459. var getpageinfo models.GetPageInfo
  460. c.Ctx.BindJSON(&getpageinfo)
  461. page := sysmodel.SysRolePageInfo{}
  462. var roles []sysmodel.SysRole
  463. err := c.Db.SQL(sysmodel.Selectall_sys_role).Limit(getpageinfo.PageSize, (getpageinfo.PageNum-1)*getpageinfo.PageSize).Find(&roles)
  464. if err != nil {
  465. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  466. return
  467. }
  468. fmt.Println("-----------", len(roles))
  469. page.Content = roles
  470. page.PageSize = getpageinfo.PageSize
  471. page.PageNum = getpageinfo.PageNum
  472. page.TotalSize = 1
  473. ret := __none_func_system__(getpageinfo)
  474. if ret {
  475. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", page})
  476. } else {
  477. c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
  478. }
  479. }
  480. // _FindRoleMenu
  481. // @Title _FindRoleMenu
  482. // @Description 查找角色权限
  483. // @Param user string false "用户id"
  484. // @Success 200 {object} models.Account
  485. // @Failure 403 :id is empty
  486. func System_FindRoleMenu(c *SystemController) {
  487. role_id, b := c.Ctx.GetQuery("roleId")
  488. if !b {
  489. c.Ctx.JSON(200, sysmodel.SysReturn{500, "roleId is null", nil})
  490. return
  491. }
  492. menus := []models.SysMenu{}
  493. c.Db.SQL(models.SqlFindRoleMenu, role_id).Find(&menus)
  494. ret := __none_func_system__()
  495. if ret {
  496. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", menus})
  497. } else {
  498. c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
  499. }
  500. }
  501. // _SaveRoleMenu
  502. // @Title _SaveRoleMenu
  503. // @Description 保存角色权限
  504. // @Param role_menu string false "角色权限"
  505. // @Success 200 {object} models.Account
  506. // @Failure 403 :id is empty
  507. func System_SaveRoleMenu(c *SystemController) {
  508. var saverolemenu []models.SysRoleMenu
  509. c.Ctx.BindJSON(&saverolemenu)
  510. if len(saverolemenu) < 1 {
  511. c.Ctx.JSON(200, sysmodel.SysReturn{500, "role menu length is error", nil})
  512. return
  513. }
  514. session := c.Db.NewSession()
  515. defer session.Close()
  516. err := session.Begin()
  517. //删除角色下的menu
  518. _, err = c.Db.Exec("delete from sys_role_menu where role_id = ?", saverolemenu[0].RoleId)
  519. //_, err = c.Db.Exec("delete from sys_role_menu where id = 1")
  520. if err != nil {
  521. session.Rollback()
  522. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  523. return
  524. }
  525. for i, _ := range saverolemenu {
  526. saverolemenu[i].Id = utils.NewUUID()
  527. saverolemenu[i].DelFlag = 1
  528. }
  529. _, err = c.Db.Insert(&saverolemenu)
  530. if err != nil {
  531. session.Rollback()
  532. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  533. return
  534. }
  535. session.Commit()
  536. ret := __none_func_system__(&saverolemenu)
  537. if ret {
  538. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", nil})
  539. } else {
  540. c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
  541. }
  542. }
  543. // _RoleAll
  544. // @Title _RoleAll
  545. // @Description 查找所有角色
  546. // @Success 200 {object} models.Account
  547. // @Failure 403 :id is empty
  548. func System_RoleAll(c *SystemController) {
  549. var roles []sysmodel.SysRole
  550. err := c.Db.SQL(sysmodel.Selectall_sys_role).Find(&roles)
  551. ret := __none_func_system__()
  552. if ret && err == nil {
  553. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", roles})
  554. } else {
  555. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  556. }
  557. }
  558. // _FindPermissions
  559. // @Title _FindPermissions
  560. // @Description 查找用户权限
  561. // @Success 200 {object} models.Account
  562. // @Failure 403 :id is empty
  563. func System_FindPermissions(c *SystemController) {
  564. user_id := c.Ctx.GetString("user_id")
  565. //result, err := c.Db.Query(models.SqlFindUserPermissions, op.UserId)
  566. result, err := c.Db.QueryString(models.SqlFindUserPermissions, user_id)
  567. strs := []string{}
  568. for _, v := range result {
  569. strs = append(strs, v["perms"])
  570. }
  571. if err != nil {
  572. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  573. }
  574. ret := __none_func_system__()
  575. //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"}
  576. if ret {
  577. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", strs})
  578. } else {
  579. c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
  580. }
  581. }
  582. // _AddMenu
  583. // @Title _AddMenu
  584. // @Description 添加菜单
  585. // @Param string false "菜单"
  586. // @Success 200 {object} models.Account
  587. // @Failure 403 :id is empty
  588. func System_AddMenu(c *SystemController) {
  589. var sysmenu models.SysMenu
  590. c.Ctx.BindJSON(&sysmenu)
  591. user_id := c.Ctx.GetString("user_id")
  592. sysmenu.Id = utils.NewUUID()
  593. sysmenu.CreateBy = user_id
  594. sysmenu.LastUpdateBy = user_id
  595. sysmenu.LastUpdateDate = time.Now()
  596. sysmenu.CreateTime = time.Now()
  597. sysmenu.DelFlag = 0
  598. _, err := c.Db.Insert(&sysmenu)
  599. if err != nil {
  600. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  601. return
  602. }
  603. ret := __none_func_system__(sysmenu)
  604. if ret {
  605. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", nil})
  606. } else {
  607. c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
  608. }
  609. }
  610. // _DelMenu
  611. // @Title _DelMenu
  612. // @Description 删除菜单
  613. // @Param id string false "菜单ID"
  614. // @Success 200 {object} models.Account
  615. // @Failure 403 :id is empty
  616. func System_DelMenu(c *SystemController) {
  617. type Param struct {
  618. Id string `json:"id"`
  619. }
  620. var params []Param
  621. c.Ctx.BindJSON(&params)
  622. fmt.Println(params, "-----------")
  623. ids := []string{}
  624. for _, param := range params {
  625. ids = append(ids, param.Id)
  626. }
  627. _, err := c.Db.Table(new(models.SysMenu)).In("id", ids).Update(map[string]interface{}{"del_flag": 0})
  628. if err != nil {
  629. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  630. return
  631. }
  632. ret := __none_func_system__()
  633. if ret {
  634. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", nil})
  635. } else {
  636. c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
  637. }
  638. }
  639. // _UpdateMenu
  640. // @Title _UpdateMenu
  641. // @Description 更新菜单
  642. // @Param string false "菜单"
  643. // @Success 200 {object} models.Account
  644. // @Failure 403 :id is empty
  645. func System_UpdateMenu(c *SystemController) {
  646. var sysmenu models.SysMenu
  647. c.Ctx.BindJSON(&sysmenu)
  648. user_id := c.Ctx.GetString("user_id")
  649. sysmenu.LastUpdateBy = user_id
  650. sysmenu.LastUpdateDate = time.Now()
  651. _, err := c.Db.ID(sysmenu.Id).Update(&sysmenu)
  652. if err != nil {
  653. c.Ctx.JSON(200, sysmodel.SysReturn{500, err.Error(), nil})
  654. return
  655. }
  656. ret := __none_func_system__(sysmenu)
  657. if ret {
  658. c.Ctx.JSON(200, sysmodel.SysReturn{200, "", nil})
  659. } else {
  660. c.Ctx.JSON(200, sysmodel.SysReturn{500, "", nil})
  661. }
  662. }
  663. func __none_func_system__(params ...interface{}) bool {
  664. return true
  665. }