projectx/internal/erpserver/http.go

package erpserver

import (
	"net/http"

	"management/internal/erpserver/handler"
	mw "management/internal/pkg/middleware"

	"github.com/go-chi/chi/v5"
	"github.com/go-chi/chi/v5/middleware"
)

func NewRouter(handler handler.IHandler, mw mw.IMiddleware) *chi.Mux {
	r := chi.NewRouter()

	r.Use(middleware.RequestID)
	r.Use(middleware.RealIP)
	// r.Use(middleware.Logger)
	r.Use(middleware.Recoverer)

	staticServer := http.FileServer(http.Dir("./web/statics/"))
	r.Handle("/statics/*", http.StripPrefix("/statics", staticServer))

	uploadServer := http.FileServer(http.Dir("./upload/"))
	r.Handle("/upload/*", http.StripPrefix("/upload", uploadServer))

	r.Group(func(r chi.Router) {
		r.Use(mw.NoSurf)      // CSRF
		r.Use(mw.LoadSession) // Session

		r.Get("/captcha", handler.CommonHandler().CaptchaHandler().Captcha)

		r.Get("/", handler.SystemHandler().UserHandler().Login)
		r.Post("/login", handler.SystemHandler().UserHandler().Login)
		r.Get("/logout", handler.SystemHandler().UserHandler().Logout)

		r.With(mw.Authorize, mw.Audit).Post("/upload/img", handler.CommonHandler().UploadHandler().Img)
		r.With(mw.Authorize, mw.Audit).Post("/upload/file", handler.CommonHandler().UploadHandler().File)
		r.With(mw.Authorize, mw.Audit).Post("/upload/mutilfile", handler.CommonHandler().UploadHandler().MutilFiles)

		r.With(mw.Authorize, mw.Audit).Get("/home.html", handler.SystemHandler().Home)
		r.With(mw.Authorize).Get("/pear.json", handler.SystemHandler().ConfigHandler().Pear)

		r.Route("/system", func(r chi.Router) {
			r.Use(mw.Authorize)

			r.Route("/config", func(r chi.Router) {
				r.Use(mw.Audit)
				r.Get("/list", handler.SystemHandler().ConfigHandler().List)
				r.Post("/list", handler.SystemHandler().ConfigHandler().List)
				r.Get("/add", handler.SystemHandler().ConfigHandler().Add)
				r.Get("/edit", handler.SystemHandler().ConfigHandler().Edit)
				r.Post("/save", handler.SystemHandler().ConfigHandler().Save)
				r.Post("/reset_pear", handler.SystemHandler().ConfigHandler().ResetPear)
				r.Post("/refresh", handler.SystemHandler().ConfigHandler().Refresh)
			})

			r.Route("/department", func(r chi.Router) {
				r.Use(mw.Audit)
				r.Get("/list", handler.SystemHandler().DepartmentHandler().List)
				r.Post("/list", handler.SystemHandler().DepartmentHandler().List)
				r.Get("/add", handler.SystemHandler().DepartmentHandler().Add)
				r.Get("/add_children", handler.SystemHandler().DepartmentHandler().AddChildren)
				r.Get("/edit", handler.SystemHandler().DepartmentHandler().Edit)
				r.Post("/save", handler.SystemHandler().DepartmentHandler().Save)
				r.Post("/tree", handler.SystemHandler().DepartmentHandler().Tree)
				r.Post("/refresh", handler.SystemHandler().DepartmentHandler().Refresh)
				r.Post("/rebuild_parent_path", handler.SystemHandler().DepartmentHandler().RebuildParentPath)
			})

			r.Route("/role", func(r chi.Router) {
				r.Use(mw.Audit)
				r.Get("/list", handler.SystemHandler().RoleHandler().List)
				r.Post("/list", handler.SystemHandler().RoleHandler().List)
				r.Get("/add", handler.SystemHandler().RoleHandler().Add)
				r.Get("/add_children", handler.SystemHandler().RoleHandler().AddChildren)
				r.Get("/edit", handler.SystemHandler().RoleHandler().Edit)
				r.Post("/save", handler.SystemHandler().RoleHandler().Save)
				r.Post("/tree", handler.SystemHandler().RoleHandler().Tree)
				r.Post("/refresh", handler.SystemHandler().RoleHandler().Refresh)
				r.Post("/rebuild_parent_path", handler.SystemHandler().RoleHandler().RebuildParentPath)
				r.Post("/refresh_role_menus", handler.SystemHandler().RoleHandler().RefreshRoleMenus)
				r.Get("/set_menu", handler.SystemHandler().RoleHandler().SetMenu)
				r.Post("/set_menu", handler.SystemHandler().RoleHandler().SetMenu)
			})

			r.Route("/user", func(r chi.Router) {
				r.Get("/list", handler.SystemHandler().UserHandler().List)
				r.Post("/list", handler.SystemHandler().UserHandler().List)
				r.Get("/add", handler.SystemHandler().UserHandler().Add)
				r.Get("/edit", handler.SystemHandler().UserHandler().Edit)
				r.Post("/save", handler.SystemHandler().UserHandler().Save)
				r.Get("/profile", handler.SystemHandler().UserHandler().Profile)
				r.Post("/tree", handler.SystemHandler().UserHandler().Tree)
			})

			r.Route("/login_log", func(r chi.Router) {
				r.Get("/list", handler.SystemHandler().LoginLogHandler().List)
				r.Post("/list", handler.SystemHandler().LoginLogHandler().List)
			})

			r.Route("/audit_log", func(r chi.Router) {
				r.Get("/list", handler.SystemHandler().AuditHandler().List)
				r.Post("/list", handler.SystemHandler().AuditHandler().List)
			})

			r.Get("/menus", handler.SystemHandler().MenuHandler().Menus)
			r.Route("/menu", func(r chi.Router) {
				r.Use(mw.Audit)
				r.Get("/list", handler.SystemHandler().MenuHandler().List)
				r.Post("/list", handler.SystemHandler().MenuHandler().List)
				r.Get("/add", handler.SystemHandler().MenuHandler().Add)
				r.Get("/add_children", handler.SystemHandler().MenuHandler().AddChildren)
				r.Get("/edit", handler.SystemHandler().MenuHandler().Edit)
				r.Post("/save", handler.SystemHandler().MenuHandler().Save)
				r.Post("/tree", handler.SystemHandler().MenuHandler().Tree)
				r.Post("/refresh_cache", handler.SystemHandler().MenuHandler().Refresh)
			})

			r.Route("/category", func(r chi.Router) {
				r.Use(mw.Audit)
				r.Get("/list", handler.SystemHandler().CategoryHandler().List)
				r.Post("/list", handler.SystemHandler().CategoryHandler().List)
				r.Get("/add", handler.SystemHandler().CategoryHandler().Add)
				r.Get("/add_children", handler.SystemHandler().CategoryHandler().AddChildren)
				r.Get("/edit", handler.SystemHandler().CategoryHandler().Edit)
				r.Post("/save", handler.SystemHandler().CategoryHandler().Save)
				r.Post("/tree", handler.SystemHandler().CategoryHandler().Tree)
				r.Post("/refresh", handler.SystemHandler().CategoryHandler().Refresh)
				r.Post("/rebuild_parent_path", handler.SystemHandler().CategoryHandler().RebuildParentPath)
			})
		})
	})

	return r
}