routes.go 8.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259
  1. /*
  2. * @description: 路由配置
  3. * @Author: CP
  4. * @Date: 2020-08-23 21:32:54
  5. * @FilePath: \construction_management\web\routes\routes.go
  6. */
  7. package routes
  8. import (
  9. "github.com/iris-contrib/middleware/csrf"
  10. "github.com/kataras/iris/v12"
  11. "github.com/kataras/iris/v12/mvc"
  12. "go.mod/bootstrap"
  13. "go.mod/services"
  14. "go.mod/web/api"
  15. "go.mod/web/backstage"
  16. "go.mod/web/external"
  17. "go.mod/web/middleware"
  18. )
  19. func Configure(b *bootstrap.Bootstrapper) {
  20. //service加载
  21. ProjectAccountService := services.NewProjectAccountService()
  22. LoginService := services.NewLoginService()
  23. ProjectService := services.NewProjectService()
  24. TreeService := services.NewTreeService()
  25. BidsectionService := services.NewBidsectionService()
  26. BidAccountService := services.NewBidAccountService()
  27. ContractService := services.NewContractService()
  28. // RpcService := services.NewRpcService()
  29. SafeService := services.NewSafeService()
  30. RuleService := services.NewRuleService()
  31. AnnexService := services.NewAnnexService()
  32. SafeAuditService := services.NewSafeAuditService()
  33. QualityService := services.NewQualityService()
  34. QualityAuditService := services.NewQualityAuditService()
  35. ProjectMessageService := services.NewProjectMessageService()
  36. VersionService := services.NewVersionService()
  37. backstageService := services.NewBackstageService()
  38. managerService := services.NewManagerService()
  39. groupService := services.NewGroupService()
  40. versionService := services.NewVersionService()
  41. //CSRF相关
  42. b.Use(middleware.SetCsrf)
  43. protect := csrf.Protect([]byte("9AB0F421E53A477C084477AEA06096F5"), csrf.FieldName("csrf"), csrf.Secure(false), csrf.Path("/"), csrf.ErrorHandler(func(ctx iris.Context) {
  44. ctx.JSON(iris.Map{"code": -1, "msg": "CSRF token invalid"})
  45. }))
  46. //b.Party("/", protect)
  47. //protect := NewCsrf()
  48. //登录相关
  49. //login := mvc.New(b.Party("/", protect))
  50. // login := mvc.New(b.Party("/"))
  51. // login.Register(ProjectAccountService)
  52. // login.Register(LoginService)
  53. // login.Handle(new(controllers.LoginController))
  54. // todolist相关
  55. // todo := mvc.New(b.Party("/todolist"))
  56. // todo.Router.Use(middleware.SessionsAuth)
  57. // todo.Handle(new(controllers.TodoController))
  58. // // 合同管理
  59. // contract := mvc.New(b.Party("/contract"))
  60. // contract.Router.Use(middleware.SessionsAuth)
  61. // contract.Handle(new(controllers.ContractController))
  62. // 安全巡检
  63. //用户相关
  64. // account := mvc.New(b.Party("/account"))
  65. // account.Register(ProjectAccountService)
  66. // account.Router.Use(middleware.SessionsAuth)
  67. // account.Handle(new(controllers.AccountController))
  68. // 项目相关
  69. // project := mvc.New(b.Party("/project"))
  70. // project.Router.Use(middleware.SessionsAuth)
  71. // project.Register(TreeService)
  72. // project.Register(ProjectService)
  73. // project.Handle(new(controllers.ProjectController))
  74. //标段相关
  75. // bidsection := mvc.New(b.Party("/bidsection"))
  76. // bidsection.Register(ProjectService)
  77. // bidsection.Router.Use(middleware.SessionsAuth)
  78. // bidsection.Handle(new(controllers.BidsectionController))
  79. // u := &websocket.Upgrader{
  80. // CheckOrigin: func(r *http.Request) bool {
  81. // return true
  82. // },
  83. // }
  84. // ws := mvc.New(b.Party("/ws"))
  85. // ws.Handle(lib.Handler(u))
  86. // websocketServer := websocket.New(
  87. // websocket.DefaultGorillaUpgrader, /* DefaultGobwasUpgrader can be used too. */
  88. // serverEvents)
  89. // b.Get("/swagger/{any:path}", )
  90. // apiTree.Router.Use(middleware.JwtAuth().Serve)
  91. // 接口相关
  92. // 登陆接口
  93. apiLogin := mvc.New(b.Party("/api/login", protect))
  94. apiLogin.Register(ProjectAccountService)
  95. apiLogin.Register(LoginService)
  96. apiLogin.Register(ProjectService)
  97. apiLogin.Handle(new(api.LoginApi))
  98. // 项目相关接口
  99. apiProject := mvc.New(b.Party("/api/project", protect))
  100. apiProject.Register(ProjectService)
  101. apiProject.Router.Use(middleware.SessionsAuth)
  102. apiProject.Router.Use(middleware.AccessAuth)
  103. apiProject.Handle(new(api.ProjectApi))
  104. // 项目账号相关接口
  105. apiProjectAccount := mvc.New(b.Party("/api/projectAccount", protect))
  106. apiProjectAccount.Register(ProjectAccountService)
  107. apiProjectAccount.Router.Use(middleware.SessionsAuth)
  108. apiProjectAccount.Router.Use(middleware.AccessAuth)
  109. apiProjectAccount.Handle(new(api.ProjectAccountApi))
  110. // 标段相关接口
  111. apiBidsection := mvc.New(b.Party("/api/bidsection", protect))
  112. apiBidsection.Register(BidsectionService)
  113. apiBidsection.Router.Use(middleware.SessionsAuth)
  114. apiBidsection.Router.Use(middleware.AccessAuth)
  115. apiBidsection.Handle(new(api.BidsectionApi))
  116. // 项目设置接口
  117. apiSetting := mvc.New(b.Party("/api/projectSetting", protect))
  118. apiSetting.Register(ProjectAccountService)
  119. apiSetting.Register(ProjectService)
  120. apiSetting.Register(BidAccountService)
  121. apiSetting.Router.Use(middleware.SessionsAuth)
  122. apiSetting.Router.Use(middleware.AccessAuth)
  123. apiSetting.Handle(new(api.ProjectSettingApi))
  124. // TreeNode相关接口
  125. apiTree := mvc.New(b.Party("/api/tree", protect))
  126. apiTree.Register(TreeService)
  127. apiTree.Router.Use(middleware.SessionsAuth)
  128. apiTree.Router.Use(middleware.AccessAuth)
  129. apiTree.Handle(new(api.TreeApi))
  130. // 合同管理
  131. apiContract := mvc.New(b.Party("/api/contract", protect))
  132. apiContract.Register(TreeService)
  133. apiContract.Register(ContractService)
  134. // 中间件
  135. apiContract.Router.Use(middleware.SessionsAuth)
  136. apiContract.Router.Use(middleware.AccessAuth)
  137. apiContract.Handle(new(api.ContractApi))
  138. // oss相关
  139. apiOss := mvc.New(b.Party("/api/oss", protect))
  140. apiOss.Router.Use(middleware.SessionsAuth)
  141. apiOss.Router.Use(middleware.AccessAuth)
  142. apiOss.Handle(new(api.OssApi))
  143. // rpc相关
  144. // rpc := mvc.New(b.Party("/api/rpc/test"))
  145. // rpc.Register(RpcService)
  146. // rpc.Router.Use(middleware.SessionsAuth)
  147. // rpc.Router.Use(middleware.AccessAuth)
  148. // rpc.Handle(new(api.RpcApi))
  149. // safe
  150. apiSafe := mvc.New(b.Party("/api/safe", protect))
  151. apiSafe.Register(SafeService)
  152. apiSafe.Router.Use(middleware.SessionsAuth)
  153. apiSafe.Router.Use(middleware.AccessAuth)
  154. apiSafe.Handle(new(api.SafeApi))
  155. apiSafeAudit := mvc.New(b.Party("/api/safe_audit", protect))
  156. apiSafeAudit.Register(SafeAuditService)
  157. apiSafeAudit.Router.Use(middleware.SessionsAuth)
  158. apiSafeAudit.Router.Use(middleware.AccessAuth)
  159. apiSafeAudit.Handle(new(api.SafeAuditApi))
  160. // quality
  161. apiQuality := mvc.New(b.Party("/api/quality", protect))
  162. apiQuality.Register(QualityService)
  163. apiQuality.Router.Use(middleware.SessionsAuth)
  164. apiQuality.Router.Use(middleware.AccessAuth)
  165. apiQuality.Handle(new(api.QualityApi))
  166. apiQualityAudit := mvc.New(b.Party("/api/quality_audit", protect))
  167. apiQualityAudit.Register(QualityAuditService)
  168. apiQualityAudit.Router.Use(middleware.SessionsAuth)
  169. apiQualityAudit.Router.Use(middleware.AccessAuth)
  170. apiQualityAudit.Handle(new(api.QualityAuditApi))
  171. // rule
  172. apiRule := mvc.New(b.Party("/api/rule", protect))
  173. apiRule.Register(RuleService)
  174. apiRule.Router.Use(middleware.SessionsAuth)
  175. apiRule.Router.Use(middleware.AccessAuth)
  176. apiRule.Handle(new(api.RuleApi))
  177. // file
  178. apiAnnex := mvc.New(b.Party("/api/file", protect))
  179. apiAnnex.Register(AnnexService)
  180. apiAnnex.Router.Use(middleware.SessionsAuth)
  181. apiAnnex.Router.Use(middleware.AccessAuth)
  182. apiAnnex.Handle(new(api.AnnexApi))
  183. apiDashboard := mvc.New(b.Party("/api/dashboard", protect))
  184. apiDashboard.Register(SafeService)
  185. apiDashboard.Register(QualityService)
  186. apiDashboard.Register(ProjectMessageService)
  187. apiDashboard.Register(VersionService)
  188. apiDashboard.Router.Use(middleware.SessionsAuth)
  189. apiDashboard.Router.Use(middleware.AccessAuth)
  190. apiDashboard.Handle(new(api.DashboardApi))
  191. // 对计量接口
  192. jlEx := mvc.New(b.Party("/api/external/jl"))
  193. jlEx.Handle(new(external.JlEx))
  194. // 后台相关
  195. backstageCM := mvc.New(b.Party("/api/backstage/login"))
  196. backstageCM.Register(backstageService)
  197. backstageCM.Handle(new(backstage.LoginBs))
  198. backstageProjectCM := mvc.New(b.Party("/api/backstage/project"))
  199. backstageProjectCM.Register(ProjectService)
  200. backstageProjectCM.Register(backstageService)
  201. backstageProjectCM.Router.Use(middleware.SessionsBackstageAuth)
  202. backstageProjectCM.Router.Use(middleware.AccessBackstageAuth)
  203. backstageProjectCM.Handle(new(backstage.ProjectBs))
  204. backstageAccountCM := mvc.New(b.Party("/api/backstage/account"))
  205. backstageAccountCM.Register(ProjectAccountService)
  206. backstageAccountCM.Register(ProjectService)
  207. backstageAccountCM.Register(BidAccountService)
  208. backstageAccountCM.Register(backstageService)
  209. backstageAccountCM.Router.Use(middleware.SessionsBackstageAuth)
  210. backstageAccountCM.Router.Use(middleware.AccessBackstageAuth)
  211. backstageAccountCM.Handle(new(backstage.ProjectAccountBs))
  212. managerCM := mvc.New(b.Party("/api/backstage/manager"))
  213. managerCM.Register(managerService)
  214. managerCM.Router.Use(middleware.SessionsBackstageAuth)
  215. managerCM.Handle(new(backstage.ManagerBs))
  216. groupCM := mvc.New(b.Party("/api/backstage/group"))
  217. groupCM.Register(groupService)
  218. groupCM.Router.Use(middleware.SessionsBackstageAuth)
  219. groupCM.Handle(new(backstage.GroupBs))
  220. versionCM := mvc.New(b.Party("/api/backstage/version"))
  221. versionCM.Register(versionService)
  222. versionCM.Router.Use(middleware.SessionsBackstageAuth)
  223. versionCM.Handle(new(backstage.VersionBs))
  224. }