v3
This commit is contained in:
231
internal/erpserver/handler/expense/expense.go
Normal file
231
internal/erpserver/handler/expense/expense.go
Normal file
@@ -0,0 +1,231 @@
|
||||
package expense
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"net/http"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"management/internal/db/model/form"
|
||||
db "management/internal/db/sqlc"
|
||||
"management/internal/erpserver/biz"
|
||||
"management/internal/pkg/convertor"
|
||||
"management/internal/pkg/know"
|
||||
"management/internal/pkg/middleware"
|
||||
"management/internal/pkg/tpl"
|
||||
"management/internal/pkg/tpl/html"
|
||||
|
||||
"github.com/jackc/pgx/v5/pgtype"
|
||||
)
|
||||
|
||||
type ExpenseHandler interface {
|
||||
List(w http.ResponseWriter, r *http.Request)
|
||||
Add(w http.ResponseWriter, r *http.Request)
|
||||
Edit(w http.ResponseWriter, r *http.Request)
|
||||
Save(w http.ResponseWriter, r *http.Request)
|
||||
}
|
||||
|
||||
type expenseHandler struct {
|
||||
render tpl.Renderer
|
||||
biz biz.IBiz
|
||||
mi middleware.IMiddleware
|
||||
}
|
||||
|
||||
var _ ExpenseHandler = (*expenseHandler)(nil)
|
||||
|
||||
func NewExpenseHandler(render tpl.Renderer, biz biz.IBiz, mi middleware.IMiddleware) *expenseHandler {
|
||||
return &expenseHandler{
|
||||
render: render,
|
||||
biz: biz,
|
||||
mi: mi,
|
||||
}
|
||||
}
|
||||
|
||||
func (h *expenseHandler) List(w http.ResponseWriter, r *http.Request) {
|
||||
switch r.Method {
|
||||
case http.MethodGet:
|
||||
ctx := r.Context()
|
||||
pp, _ := h.biz.ProjectV1().ListHtml(ctx)
|
||||
cc, _ := h.biz.SystemV1().CategoryBiz().ListHtmlByLetter(ctx, know.ExpenseCategory)
|
||||
h.render.HTML(w, r, "expense/list.tmpl", map[string]any{
|
||||
"Statuses": html.NewSelectControls(html.Statuses, "0"),
|
||||
"Projects": html.NewSelectControls(pp, "0"),
|
||||
"Categories": html.NewSelectControls(cc, "0"),
|
||||
})
|
||||
case http.MethodPost:
|
||||
ctx := r.Context()
|
||||
project := convertor.ConvertInt[int64](r.PostFormValue("project"), 9999)
|
||||
if project == 0 {
|
||||
project = 9999
|
||||
}
|
||||
budget := convertor.ConvertInt[int64](r.PostFormValue("budget"), 9999)
|
||||
if budget == 0 {
|
||||
budget = 9999
|
||||
}
|
||||
category := convertor.ConvertInt[int32](r.PostFormValue("category"), 9999)
|
||||
if category == 0 {
|
||||
category = 9999
|
||||
}
|
||||
|
||||
title := strings.TrimSpace(r.PostFormValue("title"))
|
||||
var search string
|
||||
if len(title) > 0 {
|
||||
search = "%" + title + "%"
|
||||
if strings.HasSuffix(title, ":") {
|
||||
search = title[:len(title)-1] + "%"
|
||||
}
|
||||
}
|
||||
arg := &db.ListExpenseConditionParam{
|
||||
ProjectID: project,
|
||||
BudgetID: budget,
|
||||
ExpenseType: category,
|
||||
IsTitle: len(search) > 0,
|
||||
Title: search,
|
||||
Status: convertor.ConvertInt[int16](r.PostFormValue("status"), 9999),
|
||||
PageID: convertor.ConvertInt[int32](r.PostFormValue("page"), 1),
|
||||
PageSize: convertor.ConvertInt[int32](r.PostFormValue("rows"), 10),
|
||||
}
|
||||
arg.TimeBegin, arg.TimeEnd = convertor.DefaultStartTimeAndEndTime(r.PostFormValue("timeBegin"), r.PostFormValue("timeEnd"))
|
||||
res, total, err := h.biz.ExpenseV1().List(ctx, arg)
|
||||
if err != nil {
|
||||
http.Error(w, err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
data := tpl.ResponseList{
|
||||
Code: 0,
|
||||
Message: "ok",
|
||||
Count: total,
|
||||
Data: res,
|
||||
}
|
||||
h.render.JSON(w, data)
|
||||
default:
|
||||
http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
|
||||
}
|
||||
}
|
||||
|
||||
func (h *expenseHandler) Add(w http.ResponseWriter, r *http.Request) {
|
||||
h.render.HTML(w, r, "expense/edit.tmpl", map[string]any{
|
||||
"Item": &form.ExpenseForm{},
|
||||
"Statuses": html.NewSelectControls(html.Statuses, "0"),
|
||||
})
|
||||
}
|
||||
|
||||
func (h *expenseHandler) Edit(w http.ResponseWriter, r *http.Request) {
|
||||
vars := r.URL.Query()
|
||||
id := convertor.QueryInt[int64](vars, "id", 0)
|
||||
expense := &form.ExpenseForm{}
|
||||
ctx := r.Context()
|
||||
if id > 0 {
|
||||
if cus, err := h.biz.ExpenseV1().Get(ctx, id); err == nil {
|
||||
expense = expense.ToForm(cus)
|
||||
if u, err := h.biz.SystemV1().UserBiz().Get(ctx, cus.CreatedUserID); err == nil {
|
||||
expense.CreatedName = u.Username
|
||||
}
|
||||
if u, err := h.biz.SystemV1().UserBiz().Get(ctx, cus.UpdatedUserID); err == nil {
|
||||
expense.UpdatedName = u.Username
|
||||
}
|
||||
}
|
||||
}
|
||||
h.render.HTML(w, r, "expense/edit.tmpl", map[string]any{
|
||||
"Item": expense,
|
||||
"Statuses": html.NewSelectControls(html.Statuses, strconv.Itoa(int(expense.Status))),
|
||||
})
|
||||
}
|
||||
|
||||
func (h *expenseHandler) Save(w http.ResponseWriter, r *http.Request) {
|
||||
data, err := validForm(r)
|
||||
if err != nil {
|
||||
h.render.JSONERR(w, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
ctx := r.Context()
|
||||
authUser := h.mi.AuthUser(ctx)
|
||||
if data.ID > 0 {
|
||||
arg := &db.UpdateExpenseParams{
|
||||
ID: data.ID,
|
||||
ProjectID: pgtype.Int8{
|
||||
Int64: data.ProjectID,
|
||||
Valid: true,
|
||||
},
|
||||
BudgetID: pgtype.Int8{
|
||||
Int64: data.BudgetID,
|
||||
Valid: true,
|
||||
},
|
||||
Amount: data.AmountF,
|
||||
ExpensesAt: pgtype.Timestamptz{
|
||||
Time: data.ExpensesAt,
|
||||
Valid: true,
|
||||
},
|
||||
ExpensesType: pgtype.Int4{
|
||||
Int32: data.ExpensesType,
|
||||
Valid: true,
|
||||
},
|
||||
Remark: pgtype.Text{
|
||||
String: data.Remark,
|
||||
Valid: true,
|
||||
},
|
||||
Status: pgtype.Int2{
|
||||
Int16: data.Status,
|
||||
Valid: true,
|
||||
},
|
||||
UpdatedUserID: pgtype.Int4{
|
||||
Int32: authUser.ID,
|
||||
Valid: true,
|
||||
},
|
||||
}
|
||||
_, err := h.biz.ExpenseV1().Update(ctx, arg)
|
||||
if err != nil {
|
||||
h.render.JSONERR(w, err.Error())
|
||||
return
|
||||
}
|
||||
h.render.JSONOK(w, "更新成功")
|
||||
} else {
|
||||
arg := &db.CreateExpenseParams{
|
||||
ProjectID: data.ProjectID,
|
||||
BudgetID: data.BudgetID,
|
||||
Amount: data.AmountF,
|
||||
ExpensesAt: data.ExpensesAt,
|
||||
ExpensesType: data.ExpensesType,
|
||||
Remark: data.Remark,
|
||||
Status: data.Status,
|
||||
CreatedUserID: authUser.ID,
|
||||
}
|
||||
_, err := h.biz.ExpenseV1().Create(ctx, arg)
|
||||
if err != nil {
|
||||
h.render.JSONERR(w, err.Error())
|
||||
return
|
||||
}
|
||||
h.render.JSONOK(w, "添加成功")
|
||||
}
|
||||
}
|
||||
|
||||
func validForm(r *http.Request) (form.ExpenseForm, error) {
|
||||
var err error
|
||||
data := form.ExpenseForm{}
|
||||
|
||||
data.ID = convertor.ConvertInt[int64](r.PostFormValue("ID"), 0)
|
||||
data.ProjectID, err = strconv.ParseInt(r.PostFormValue("ProjectID"), 10, 64)
|
||||
if err != nil || data.ProjectID == 0 {
|
||||
return data, errors.New("项目不能为空")
|
||||
}
|
||||
data.BudgetID, err = strconv.ParseInt(r.PostFormValue("BudgetID"), 10, 64)
|
||||
|
||||
if err := data.AmountF.Scan(r.PostFormValue("Amount")); err != nil {
|
||||
return data, errors.New("报销金额格式错误")
|
||||
}
|
||||
data.ExpensesAt, err = time.ParseInLocation("2006-01-02", r.PostFormValue("ExpensesAt"), time.Local)
|
||||
if err != nil {
|
||||
return data, errors.New("报销时间格式错误")
|
||||
}
|
||||
expensesType, err := strconv.ParseInt(r.PostFormValue("ExpensesType"), 10, 64)
|
||||
if err != nil || expensesType == 0 {
|
||||
return data, errors.New("报销类型不能为空")
|
||||
}
|
||||
data.ExpensesType = int32(expensesType)
|
||||
|
||||
data.Status = convertor.ConvertInt[int16](r.PostFormValue("Status"), 9999)
|
||||
return data, nil
|
||||
}
|
||||
Reference in New Issue
Block a user