v2

internal/erpserver/handler/common/captcha.go

@@ -0,0 +1,56 @@
+package common
+
+import (
+	"net/http"
+
+	"management/internal/config"
+	commonv1 "management/internal/erpserver/biz/v1/common"
+	"management/internal/pkg/tpl"
+)
+
+type CaptchaHandler interface {
+	Captcha(w http.ResponseWriter, r *http.Request)
+}
+
+// captchaHandler 是 CaptchaHandler 接口的实现.
+type captchaHandler struct {
+	conf   *config.Captcha
+	render tpl.Renderer
+	biz    commonv1.CaptchaBiz
+}
+
+// 确保 captchaHandler 实现了 CaptchaHandler 接口.
+var _ CaptchaHandler = (*captchaHandler)(nil)
+
+func NewCaptchaHandler(conf *config.Captcha, render tpl.Renderer, biz commonv1.CaptchaBiz) *captchaHandler {
+	return &captchaHandler{
+		conf:   conf,
+		render: render,
+		biz:    biz,
+	}
+}
+
+type CaptchaResponse struct {
+	CaptchaID     string `json:"captcha_id"`
+	PicPath       string `json:"pic_path"`
+	CaptchaLength int    `json:"captcha_length"`
+	OpenCaptcha   int    `json:"open_captcha"`
+}
+
+func (h *captchaHandler) Captcha(w http.ResponseWriter, r *http.Request) {
+	keyLong := h.conf.KeyLong
+	oc := h.conf.OpenCaptcha
+	id, b64s, _, err := h.biz.Generate(h.conf.ImgHeight, h.conf.ImgWidth, keyLong, 0.7, 80)
+	if err != nil {
+		h.render.JSON(w, tpl.Response{Success: false, Message: "获取验证码失败"})
+		return
+	}
+
+	rsp := CaptchaResponse{
+		CaptchaID:     id,
+		PicPath:       b64s,
+		CaptchaLength: keyLong,
+		OpenCaptcha:   oc,
+	}
+	h.render.JSON(w, tpl.Response{Success: true, Message: "ok", Data: rsp})
+}

internal/erpserver/handler/common/common.go

@@ -0,0 +1,31 @@
+package common
+
+import (
+	"management/internal/config"
+	commonv1 "management/internal/erpserver/biz/v1/common"
+	"management/internal/pkg/tpl"
+)
+
+type CommonHandler interface {
+	CaptchaHandler() CaptchaHandler
+}
+
+type commonHandler struct {
+	conf   *config.Config
+	render tpl.Renderer
+	biz    commonv1.CommonBiz
+}
+
+var _ CommonHandler = (*commonHandler)(nil)
+
+func NewCommonHandler(conf *config.Config, render tpl.Renderer, biz commonv1.CommonBiz) *commonHandler {
+	return &commonHandler{
+		conf:   conf,
+		render: render,
+		biz:    biz,
+	}
+}
+
+func (h *commonHandler) CaptchaHandler() CaptchaHandler {
+	return NewCaptchaHandler(&h.conf.Captcha, h.render, h.biz.CaptchaBiz())
+}

internal/erpserver/handler/handler.go

@@ -0,0 +1,55 @@
+package handler
+
+import (
+	"net/http"
+
+	"management/internal/config"
+	"management/internal/erpserver/biz"
+	"management/internal/erpserver/handler/common"
+	"management/internal/erpserver/handler/system"
+	"management/internal/pkg/session"
+	"management/internal/pkg/tpl"
+)
+
+// IHandler 定义了Handler需要实现的方法.
+type IHandler interface {
+	// 获取 Common Handler 接口.
+	CommonHandler() common.CommonHandler
+
+	// 获取首页
+	Home(w http.ResponseWriter, req *http.Request)
+
+	// 获取 System Handler 接口.
+	SystemHandler() system.SystemHandler
+}
+
+// handler 是 IHandler 的一个具体实现.
+type handler struct {
+	conf    *config.Config
+	render  tpl.Renderer
+	session session.ISession
+	biz     biz.IBiz
+}
+
+// 确保 handler 实现了 IHandler 接口.
+var _ IHandler = (*handler)(nil)
+
+// NewHandler 创建一个 IHandler 类型的实例.
+func NewHandler(conf *config.Config, render tpl.Renderer, session session.ISession, biz biz.IBiz) *handler {
+	return &handler{
+		conf:    conf,
+		render:  render,
+		session: session,
+		biz:     biz,
+	}
+}
+
+// CommonHandler 返回一个实现了 CommonHandler 接口的实例.
+func (h *handler) CommonHandler() common.CommonHandler {
+	return common.NewCommonHandler(h.conf, h.render, h.biz.CommonV1())
+}
+
+// SystemHandler 返回一个实现了 SystemHandler 接口的实例.
+func (h *handler) SystemHandler() system.SystemHandler {
+	return system.NewSystemHandler(h.render, h.session, h.biz)
+}

internal/erpserver/handler/home.go

@@ -0,0 +1,7 @@
+package handler
+
+import "net/http"
+
+func (h *handler) Home(w http.ResponseWriter, r *http.Request) {
+	h.render.HTML(w, r, "home/home.tmpl", nil)
+}

internal/erpserver/handler/system/config.go

@@ -0,0 +1,50 @@
+package system
+
+import (
+	"net/http"
+
+	"management/internal/erpserver/biz"
+	"management/internal/pkg/session"
+	"management/internal/pkg/tpl"
+)
+
+type ConfigHandler interface {
+	// Add(w http.ResponseWriter, r *http.Request)
+	// Edit(w http.ResponseWriter, r *http.Request)
+	// Save(w http.ResponseWriter, r *http.Request)
+	// List(w http.ResponseWriter, r *http.Request)
+
+	ConfigExpansion
+}
+
+type ConfigExpansion interface {
+	Pear(w http.ResponseWriter, r *http.Request)
+}
+
+// configHandler 是 ConfigHandler 接口的实现.
+type configHandler struct {
+	render  tpl.Renderer
+	session session.ISession
+	biz     biz.IBiz
+}
+
+// 确保 userHandler 实现了 ConfigHandler 接口.
+var _ ConfigHandler = (*configHandler)(nil)
+
+func NewConfigHandler(render tpl.Renderer, session session.ISession, biz biz.IBiz) *configHandler {
+	return &configHandler{
+		render:  render,
+		session: session,
+		biz:     biz,
+	}
+}
+
+func (h *configHandler) Pear(w http.ResponseWriter, r *http.Request) {
+	pear, err := h.biz.SystemV1().ConfigBiz().Pear(r.Context())
+	if err != nil {
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+
+	h.render.JSON(w, pear)
+}

internal/erpserver/handler/system/department.go

@@ -0,0 +1,206 @@
+package system
+
+import (
+	"fmt"
+	"net/http"
+	"time"
+
+	"management/internal/db/model/dto"
+	db "management/internal/db/sqlc"
+	"management/internal/erpserver/biz"
+	"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)
+	Tree(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) {
+	if r.Method == http.MethodGet {
+		h.render.HTML(w, r, "department/list.tmpl", nil)
+		return
+	} else if r.Method == 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)
+		return
+	}
+
+	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) {
+	id := convertor.ConvertInt[int32](r.PostFormValue("ID"), 0)
+	ParentID := convertor.ConvertInt[int32](r.PostFormValue("ParentID"), 0)
+	name := r.PostFormValue("Name")
+	sort := convertor.ConvertInt[int32](r.PostFormValue("Sort"), 6666)
+	status := convertor.ConvertInt[int32](r.PostFormValue("Status"), 9999)
+
+	ctx := r.Context()
+	var parent *db.SysDepartment
+	if ParentID > 0 {
+		var err error
+		parent, err = h.biz.SystemV1().DepartmentBiz().Get(ctx, ParentID)
+		if err != nil {
+			h.render.JSONERR(w, "父级节点错误")
+			return
+		}
+	}
+
+	if id == 0 {
+		arg := db.CreateSysDepartmentParams{
+			Name:       name,
+			ParentID:   ParentID,
+			ParentPath: fmt.Sprintf("%s,%d,", parent.ParentPath, parent.ID),
+			Status:     status,
+			Sort:       sort,
+			CreatedAt:  time.Now(),
+			UpdatedAt:  time.Now(),
+		}
+		_, err := h.biz.SystemV1().DepartmentBiz().Create(ctx, &arg)
+		if err != nil {
+			if db.IsUniqueViolation(err) {
+				h.render.JSONERR(w, "部门名称已存在")
+				return
+			}
+			h.render.JSONERR(w, err.Error())
+			return
+		}
+
+		h.render.JSONOK(w, "添加成功")
+	} else {
+		res, err := h.biz.SystemV1().DepartmentBiz().Get(ctx, id)
+		if err != nil {
+			h.render.JSONERR(w, err.Error())
+			return
+		}
+
+		arg := &db.UpdateSysDepartmentParams{
+			ID:         res.ID,
+			Name:       name,
+			ParentID:   ParentID,
+			ParentPath: fmt.Sprintf("%s,%d,", parent.ParentPath, parent.ID),
+			Status:     status,
+			Sort:       sort,
+			UpdatedAt:  time.Now(),
+		}
+		_, err = h.biz.SystemV1().DepartmentBiz().Update(ctx, arg)
+		if err != nil {
+			h.render.JSONERR(w, err.Error())
+			return
+		}
+
+		h.render.JSONOK(w, "更新成功")
+	}
+}
+
+func (h *departmentHandler) Tree(w http.ResponseWriter, r *http.Request) {
+	ctx := r.Context()
+	vars := r.URL.Query()
+	if vars.Get("type") == "xmselect" {
+		res, err := h.biz.SystemV1().DepartmentBiz().XmSelect(ctx, 0)
+		if err != nil {
+			h.render.JSONERR(w, err.Error())
+			return
+		}
+
+		h.render.JSON(w, res)
+		return
+	} else {
+		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
+	}
+}
+
+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, "重建成功")
+}

internal/erpserver/handler/system/menu.go

@@ -0,0 +1,53 @@
+package system
+
+import (
+	"encoding/json"
+	"net/http"
+
+	"management/internal/db/model/dto"
+	"management/internal/erpserver/biz"
+	"management/internal/global/know"
+	"management/internal/pkg/session"
+	"management/internal/pkg/tpl"
+)
+
+type MenuHandler interface {
+	MenuExpansion
+}
+
+type MenuExpansion interface {
+	Menus(w http.ResponseWriter, r *http.Request)
+}
+
+type menuHandler struct {
+	render  tpl.Renderer
+	session session.ISession
+	biz     biz.IBiz
+}
+
+var _ MenuHandler = (*menuHandler)(nil)
+
+func NewMenuHandler(render tpl.Renderer, session session.ISession, biz biz.IBiz) *menuHandler {
+	return &menuHandler{
+		render:  render,
+		session: session,
+		biz:     biz,
+	}
+}
+
+func (h *menuHandler) Menus(w http.ResponseWriter, r *http.Request) {
+	ctx := r.Context()
+	b := h.session.GetBytes(ctx, know.StoreName)
+	var u dto.AuthorizeUser
+	if err := json.Unmarshal(b, &u); err != nil {
+		h.render.JSONERR(w, err.Error())
+		return
+	}
+	menus, err := h.biz.SystemV1().MenuBiz().RecursiveSysMenus(ctx, u.RoleID)
+	if err != nil {
+		h.render.JSONERR(w, err.Error())
+		return
+	}
+
+	h.render.JSON(w, menus)
+}

internal/erpserver/handler/system/system.go

@@ -0,0 +1,46 @@
+package system
+
+import (
+	"management/internal/erpserver/biz"
+	"management/internal/pkg/session"
+	"management/internal/pkg/tpl"
+)
+
+type SystemHandler interface {
+	UserHandler() UserHandler
+	MenuHandler() MenuHandler
+	DepartmentHandler() DepartmentHandler
+	ConfigHandler() ConfigHandler
+}
+
+type systemHandler struct {
+	render  tpl.Renderer
+	session session.ISession
+	biz     biz.IBiz
+}
+
+var _ SystemHandler = (*systemHandler)(nil)
+
+func NewSystemHandler(render tpl.Renderer, session session.ISession, biz biz.IBiz) *systemHandler {
+	return &systemHandler{
+		render:  render,
+		session: session,
+		biz:     biz,
+	}
+}
+
+func (h *systemHandler) UserHandler() UserHandler {
+	return NewUserHandler(h.render, h.session, h.biz)
+}
+
+func (h *systemHandler) MenuHandler() MenuHandler {
+	return NewMenuHandler(h.render, h.session, h.biz)
+}
+
+func (h *systemHandler) DepartmentHandler() DepartmentHandler {
+	return NewDepartmentHandler(h.render, h.biz)
+}
+
+func (h *systemHandler) ConfigHandler() ConfigHandler {
+	return NewConfigHandler(h.render, h.session, h.biz)
+}

internal/erpserver/handler/system/user.go

@@ -0,0 +1,127 @@
+package system
+
+import (
+	"encoding/json"
+	"net/http"
+	"strings"
+
+	"management/internal/db/model/dto"
+	"management/internal/erpserver/biz"
+	"management/internal/erpserver/model/req"
+	"management/internal/global/know"
+	"management/internal/pkg/session"
+	"management/internal/pkg/tpl"
+
+	"github.com/zhang2092/browser"
+)
+
+type UserHandler interface {
+	Add(w http.ResponseWriter, r *http.Request)
+	Edit(w http.ResponseWriter, r *http.Request)
+	Save(w http.ResponseWriter, r *http.Request)
+	List(w http.ResponseWriter, r *http.Request)
+
+	UserExpansion
+}
+
+type UserExpansion interface {
+	Login(w http.ResponseWriter, r *http.Request)
+	Logout(w http.ResponseWriter, r *http.Request)
+}
+
+// userHandler 是 UserHandler 接口的实现.
+type userHandler struct {
+	render  tpl.Renderer
+	session session.ISession
+	biz     biz.IBiz
+}
+
+// 确保 userHandler 实现了 UserHandler 接口.
+var _ UserHandler = (*userHandler)(nil)
+
+func NewUserHandler(render tpl.Renderer, session session.ISession, biz biz.IBiz) *userHandler {
+	return &userHandler{
+		render:  render,
+		session: session,
+		biz:     biz,
+	}
+}
+
+func (h *userHandler) Add(w http.ResponseWriter, r *http.Request) {}
+
+func (h *userHandler) Edit(w http.ResponseWriter, r *http.Request) {}
+
+func (h *userHandler) Save(w http.ResponseWriter, r *http.Request) {}
+
+func (h *userHandler) List(w http.ResponseWriter, r *http.Request) {}
+
+func (h *userHandler) Login(w http.ResponseWriter, r *http.Request) {
+	ctx := r.Context()
+	if r.Method == http.MethodGet {
+		var user dto.AuthorizeUser
+		u := h.session.GetBytes(ctx, know.StoreName)
+		if err := json.Unmarshal(u, &user); err == nil {
+			// 判断租户是否一致, 一致则刷新令牌，跳转到首页
+			if err := h.session.RenewToken(ctx); err == nil {
+				h.session.Put(ctx, know.StoreName, u)
+				http.Redirect(w, r, "/home.html", http.StatusFound)
+				return
+			}
+		}
+
+		h.session.Destroy(ctx)
+		h.render.HTML(w, r, "oauth/login.tmpl", nil)
+		return
+	} else if r.Method == http.MethodPost {
+		req := &req.Login{
+			Email:     strings.TrimSpace(r.PostFormValue("email")),
+			Password:  strings.TrimSpace(r.PostFormValue("password")),
+			CaptchaID: strings.TrimSpace(r.PostFormValue("captcha_id")),
+			Captcha:   strings.TrimSpace(r.PostFormValue("captcha")),
+			Ip:        r.RemoteAddr,
+			Referrer:  r.Header.Get("Referer"),
+			Url:       r.URL.RequestURI(),
+		}
+
+		if len(req.Email) == 0 {
+			h.render.JSON(w, tpl.Response{Success: false, Message: "请填写邮箱"})
+			return
+		}
+		if len(req.Password) == 0 {
+			h.render.JSON(w, tpl.Response{Success: false, Message: "请填写密码"})
+			return
+		}
+		if len(req.Captcha) == 0 {
+			h.render.JSON(w, tpl.Response{Success: false, Message: "请填写验证码"})
+			return
+		}
+		if !h.biz.CommonV1().CaptchaBiz().Verify(req.CaptchaID, req.Captcha, true) {
+			h.render.JSON(w, tpl.Response{Success: false, Message: "验证码错误"})
+			return
+		}
+
+		br, err := browser.NewBrowser(r.Header.Get("User-Agent"))
+		if err != nil {
+			h.render.JSONERR(w, "平台信息获取错误")
+			return
+		}
+
+		req.Os = br.Platform().Name()
+		req.Browser = br.Name()
+		err = h.biz.SystemV1().UserBiz().Login(ctx, req)
+		if err != nil {
+			h.render.JSONERR(w, err.Error())
+			return
+		}
+
+		h.render.JSONOK(w, "login successful")
+		return
+	}
+
+	http.Error(w, "Method Not Allowed", http.StatusMethodNotAllowed)
+}
+
+func (h *userHandler) Logout(w http.ResponseWriter, r *http.Request) {
+	h.session.Destroy(r.Context())
+	http.Redirect(w, r, "/", http.StatusFound)
+}