package system import ( "net/http" "management/internal/db/model/dto" db "management/internal/db/sqlc" "management/internal/erpserver/biz" "management/internal/erpserver/model/form" "management/internal/pkg/binding" "management/internal/pkg/convertor" "management/internal/pkg/tpl" ) type DepartmentHandler interface { List(w http.ResponseWriter, r *http.Request) Add(w http.ResponseWriter, r *http.Request) AddChildren(w http.ResponseWriter, r *http.Request) Edit(w http.ResponseWriter, r *http.Request) Save(w http.ResponseWriter, r *http.Request) Data(w http.ResponseWriter, r *http.Request) Refresh(w http.ResponseWriter, r *http.Request) RebuildParentPath(w http.ResponseWriter, r *http.Request) } type departmentHandler struct { render tpl.Renderer biz biz.IBiz } var _ DepartmentHandler = (*departmentHandler)(nil) func NewDepartmentHandler(render tpl.Renderer, biz biz.IBiz) *departmentHandler { return &departmentHandler{ render: render, biz: biz, } } func (h *departmentHandler) List(w http.ResponseWriter, r *http.Request) { switch r.Method { case http.MethodGet: h.render.HTML(w, r, "department/list.tmpl", nil) case http.MethodPost: var q dto.SearchDto q.SearchStatus = convertor.ConvertInt(r.PostFormValue("status"), 9999) q.SearchParentID = convertor.ConvertInt(r.PostFormValue("parentId"), 0) q.SearchName = r.PostFormValue("name") q.SearchID = convertor.ConvertInt[int64](r.PostFormValue("id"), 0) q.Page = convertor.ConvertInt(r.PostFormValue("page"), 1) q.Rows = convertor.ConvertInt(r.PostFormValue("rows"), 10) res, count, err := h.biz.SystemV1().DepartmentBiz().List(r.Context(), q) if err != nil { http.Error(w, err.Error(), http.StatusInternalServerError) return } data := tpl.ResponseList{ Code: 0, Message: "ok", Count: count, Data: res, } h.render.JSON(w, data) default: http.Error(w, "Method Not Allowed", http.StatusMethodNotAllowed) } } func (h *departmentHandler) Add(w http.ResponseWriter, r *http.Request) { h.render.HTML(w, r, "department/edit.tmpl", map[string]any{ "Item": &db.SysDepartment{Sort: 6666}, }) } func (h *departmentHandler) AddChildren(w http.ResponseWriter, r *http.Request) { vars := r.URL.Query() parentID := convertor.QueryInt(vars, "parentID", 0) vm := &db.SysDepartment{ParentID: int32(parentID), Sort: 6666} h.render.HTML(w, r, "department/edit.tmpl", map[string]any{ "Item": vm, }) } func (h *departmentHandler) Edit(w http.ResponseWriter, r *http.Request) { vars := r.URL.Query() id := convertor.QueryInt[int32](vars, "id", 0) vm := &db.SysDepartment{Sort: 6666} if id > 0 { vm, _ = h.biz.SystemV1().DepartmentBiz().Get(r.Context(), id) } h.render.HTML(w, r, "department/edit.tmpl", map[string]any{ "Item": vm, }) } func (h *departmentHandler) Save(w http.ResponseWriter, r *http.Request) { var req form.Department if err := binding.Form.Bind(r, &req); err != nil { h.render.JSONERR(w, binding.ValidatorErrors(err)) return } ctx := r.Context() if *req.ID == 0 { err := h.biz.SystemV1().DepartmentBiz().Create(ctx, &req) if err != nil { h.render.JSONERR(w, err.Error()) return } h.render.JSONOK(w, "添加成功") } else { err := h.biz.SystemV1().DepartmentBiz().Update(ctx, &req) if err != nil { h.render.JSONERR(w, err.Error()) return } h.render.JSONOK(w, "更新成功") } } func (h *departmentHandler) Data(w http.ResponseWriter, r *http.Request) { ctx := r.Context() vars := r.URL.Query() t := vars.Get("type") if t == "tree" { res, err := h.biz.SystemV1().DepartmentBiz().Tree(ctx, 0) if err != nil { h.render.JSONERR(w, err.Error()) return } h.render.JSON(w, res) return } else if t == "xmselect_tree" { res, err := h.biz.SystemV1().DepartmentBiz().XmSelectTree(ctx, 0) if err != nil { h.render.JSONERR(w, err.Error()) return } h.render.JSON(w, res) return } } func (h *departmentHandler) Refresh(w http.ResponseWriter, r *http.Request) { _, err := h.biz.SystemV1().DepartmentBiz().Refresh(r.Context()) if err != nil { h.render.JSONERR(w, err.Error()) return } h.render.JSONOK(w, "刷新成功") } func (h *departmentHandler) RebuildParentPath(w http.ResponseWriter, r *http.Request) { ctx := r.Context() err := h.biz.SystemV1().DepartmentBiz().RebuildParentPath(ctx) if err != nil { h.render.JSONERR(w, err.Error()) return } h.render.JSONOK(w, "重建成功") }