@robsun/create-keystone-app 0.2.13 → 0.4.0

package/template/apps/server/internal/modules/example/api/handler/item_handler.go

@@ -1,165 +1,468 @@
-package handler
-
-import (
-	"errors"
-
-	"github.com/gin-gonic/gin"
-	hcommon "github.com/robsuncn/keystone/api/handler/common"
-	"github.com/robsuncn/keystone/api/response"
-	"github.com/robsuncn/keystone/infra/i18n"
-
-	examplei18n "__APP_NAME__/apps/server/internal/modules/example/i18n"
-	"__APP_NAME__/apps/server/internal/modules/example/domain/models"
-	"__APP_NAME__/apps/server/internal/modules/example/domain/service"
-)
-
-type ItemHandler struct {
-	items *service.ItemService
-}
-
-func NewItemHandler(items *service.ItemService) *ItemHandler {
-	if items == nil {
-		return nil
-	}
-	return &ItemHandler{items: items}
-}
-
-type itemInput struct {
-	Title       string            `json:"title"`
-	Description string            `json:"description"`
-	Status      models.ItemStatus `json:"status"`
-}
-
-type itemUpdateInput struct {
-	Title       *string            `json:"title"`
-	Description *string            `json:"description"`
-	Status      *models.ItemStatus `json:"status"`
-}
-
-const defaultTenantID uint = 1
-
-func (h *ItemHandler) List(c *gin.Context) {
-	if h == nil || h.items == nil {
-		response.ServiceUnavailableI18n(c, examplei18n.MsgServiceUnavailable)
-		return
-	}
-	tenantID := resolveTenantID(c)
-
-	items, err := h.items.List(c.Request.Context(), tenantID)
-	if err != nil {
-		response.InternalErrorI18n(c, examplei18n.MsgItemLoadFailed)
-		return
-	}
-
-	response.Success(c, gin.H{"items": items})
-}
-
-func (h *ItemHandler) Create(c *gin.Context) {
-	if h == nil || h.items == nil {
-		response.ServiceUnavailableI18n(c, examplei18n.MsgServiceUnavailable)
-		return
-	}
-	tenantID := resolveTenantID(c)
-
-	var input itemInput
-	if err := c.ShouldBindJSON(&input); err != nil {
-		response.BadRequestI18n(c, examplei18n.MsgInvalidPayload)
-		return
-	}
-
-	item, err := h.items.Create(c.Request.Context(), tenantID, service.ItemInput{
-		Title:       input.Title,
-		Description: input.Description,
-		Status:      input.Status,
-	})
-	if err != nil {
-		var i18nErr *i18n.I18nError
-		if errors.As(err, &i18nErr) {
-			response.BadRequestI18n(c, i18nErr.Key)
-			return
-		}
-		response.InternalErrorI18n(c, examplei18n.MsgItemCreateFailed)
-		return
-	}
-
-	response.CreatedI18n(c, examplei18n.MsgItemCreated, item)
-}
-
-func (h *ItemHandler) Update(c *gin.Context) {
-	if h == nil || h.items == nil {
-		response.ServiceUnavailableI18n(c, examplei18n.MsgServiceUnavailable)
-		return
-	}
-	tenantID := resolveTenantID(c)
-
-	id, err := hcommon.ParseUintParam(c, "id")
-	if err != nil || id == 0 {
-		response.BadRequestI18n(c, examplei18n.MsgInvalidID)
-		return
-	}
-
-	var input itemUpdateInput
-	if err := c.ShouldBindJSON(&input); err != nil {
-		response.BadRequestI18n(c, examplei18n.MsgInvalidPayload)
-		return
-	}
-
-	item, err := h.items.Update(c.Request.Context(), tenantID, id, service.ItemUpdateInput{
-		Title:       input.Title,
-		Description: input.Description,
-		Status:      input.Status,
-	})
-	if err != nil {
-		var i18nErr *i18n.I18nError
-		if errors.As(err, &i18nErr) {
-			if i18nErr.Key == examplei18n.MsgItemNotFound {
-				response.NotFoundI18n(c, i18nErr.Key)
-			} else {
-				response.BadRequestI18n(c, i18nErr.Key)
-			}
-			return
-		}
-		response.InternalErrorI18n(c, examplei18n.MsgItemUpdateFailed)
-		return
-	}
-
-	response.SuccessI18n(c, examplei18n.MsgItemUpdated, item)
-}
-
-func (h *ItemHandler) Delete(c *gin.Context) {
-	if h == nil || h.items == nil {
-		response.ServiceUnavailableI18n(c, examplei18n.MsgServiceUnavailable)
-		return
-	}
-	tenantID := resolveTenantID(c)
-
-	id, err := hcommon.ParseUintParam(c, "id")
-	if err != nil || id == 0 {
-		response.BadRequestI18n(c, examplei18n.MsgInvalidID)
-		return
-	}
-
-	if err := h.items.Delete(c.Request.Context(), tenantID, id); err != nil {
-		var i18nErr *i18n.I18nError
-		if errors.As(err, &i18nErr) {
-			if i18nErr.Key == examplei18n.MsgItemNotFound {
-				response.NotFoundI18n(c, i18nErr.Key)
-				return
-			}
-		}
-		response.InternalErrorI18n(c, examplei18n.MsgItemDeleteFailed)
-		return
-	}
-
-	response.SuccessI18n(c, examplei18n.MsgItemDeleted, gin.H{"id": id})
-}
-
-func resolveTenantID(c *gin.Context) uint {
-	if c == nil {
-		return defaultTenantID
-	}
-	if tenantID, ok := hcommon.GetTenantID(c); ok && tenantID > 0 {
-		return tenantID
-	}
-	return defaultTenantID
-}
+package handler
+
+import (
+	"bytes"
+	"encoding/json"
+	"errors"
+	"fmt"
+	"time"
+
+	"github.com/gin-gonic/gin"
+	"github.com/xuri/excelize/v2"
+	"gorm.io/datatypes"
+
+	hcommon "github.com/robsuncn/keystone/api/handler/common"
+	"github.com/robsuncn/keystone/api/response"
+	approvalmodels "github.com/robsuncn/keystone/domain/approval/models"
+	"github.com/robsuncn/keystone/infra/i18n"
+	"github.com/robsuncn/keystone/infra/pdf"
+
+	examplei18n "__APP_NAME__/apps/server/internal/modules/example/i18n"
+	"__APP_NAME__/apps/server/internal/modules/example/domain/models"
+	"__APP_NAME__/apps/server/internal/modules/example/domain/service"
+)
+
+type ItemHandler struct {
+	items *service.ItemService
+}
+
+func NewItemHandler(items *service.ItemService) *ItemHandler {
+	if items == nil {
+		return nil
+	}
+	return &ItemHandler{items: items}
+}
+
+type itemInput struct {
+	Title       string            `json:"title"`
+	Description string            `json:"description"`
+	Category    models.ItemCategory `json:"category"`
+	Amount      float64           `json:"amount"`
+	Status      models.ItemStatus `json:"status"`
+	Attachment  json.RawMessage   `json:"attachment"`
+}
+
+type itemUpdateInput struct {
+	Title       *string             `json:"title"`
+	Description *string             `json:"description"`
+	Category    *models.ItemCategory `json:"category"`
+	Amount      *float64            `json:"amount"`
+	Status      *models.ItemStatus  `json:"status"`
+	Attachment  *json.RawMessage    `json:"attachment"`
+}
+
+type cancelInput struct {
+	Reason string `json:"reason"`
+}
+
+const defaultTenantID uint = 1
+
+func (h *ItemHandler) List(c *gin.Context) {
+	if h == nil || h.items == nil {
+		response.ServiceUnavailableI18n(c, examplei18n.MsgServiceUnavailable)
+		return
+	}
+	tenantID := resolveTenantID(c)
+	page, pageSize := hcommon.ParsePagination(c)
+
+	filter, err := buildItemListFilter(c)
+	if err != nil {
+		response.BadRequestI18n(c, examplei18n.MsgInvalidPayload)
+		return
+	}
+	filter.Page = page
+	filter.PageSize = pageSize
+
+	items, total, err := h.items.List(c.Request.Context(), tenantID, filter)
+	if err != nil {
+		response.InternalErrorI18n(c, examplei18n.MsgItemLoadFailed)
+		return
+	}
+	response.Paginated(c, items, total, page, pageSize)
+}
+
+func (h *ItemHandler) ExportPDF(c *gin.Context) {
+	if h == nil || h.items == nil {
+		response.ServiceUnavailableI18n(c, examplei18n.MsgServiceUnavailable)
+		return
+	}
+	tenantID := resolveTenantID(c)
+	filter, err := buildItemListFilter(c)
+	if err != nil {
+		response.BadRequestI18n(c, examplei18n.MsgInvalidPayload)
+		return
+	}
+	filter.Page = 1
+	filter.PageSize = 0
+
+	items, _, err := h.items.List(c.Request.Context(), tenantID, filter)
+	if err != nil {
+		response.InternalErrorI18n(c, examplei18n.MsgItemExportFailed)
+		return
+	}
+
+	data, err := buildItemsPDF(items)
+	if err != nil {
+		response.InternalErrorI18n(c, examplei18n.MsgItemExportFailed)
+		return
+	}
+	filename := fmt.Sprintf("example-items-%s.pdf", time.Now().Format("20060102-150405"))
+	response.DownloadPDF(c, filename, data)
+}
+
+func (h *ItemHandler) DownloadImportTemplate(c *gin.Context) {
+	if h == nil || h.items == nil {
+		response.ServiceUnavailableI18n(c, examplei18n.MsgServiceUnavailable)
+		return
+	}
+	file := excelize.NewFile()
+	defer func() { _ = file.Close() }()
+
+	sheet := file.GetSheetName(0)
+	if sheet == "" {
+		sheet = "Sheet1"
+		file.NewSheet(sheet)
+	}
+	headers := []string{"title", "description", "category", "amount", "status"}
+	if err := file.SetSheetRow(sheet, "A1", &headers); err != nil {
+		response.InternalErrorI18n(c, examplei18n.MsgItemExportFailed)
+		return
+	}
+	_ = file.SetColWidth(sheet, "A", "E", 18)
+
+	if _, err := file.NewSheet("README"); err == nil {
+		_ = file.SetSheetRow("README", "A1", &[]string{"Columns: title, description, category, amount, status"})
+		_ = file.SetSheetRow("README", "A3", &[]string{"category: general | operations | finance | growth"})
+		_ = file.SetSheetRow("README", "A4", &[]string{"status: draft | pending | approved | rejected | cancelled"})
+	}
+
+	buf, err := file.WriteToBuffer()
+	if err != nil {
+		response.InternalErrorI18n(c, examplei18n.MsgItemExportFailed)
+		return
+	}
+	response.DownloadFile(
+		c,
+		"application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
+		"example_items_import_template.xlsx",
+		buf.Bytes(),
+	)
+}
+
+func (h *ItemHandler) Get(c *gin.Context) {
+	if h == nil || h.items == nil {
+		response.ServiceUnavailableI18n(c, examplei18n.MsgServiceUnavailable)
+		return
+	}
+	tenantID := resolveTenantID(c)
+	id, err := hcommon.ParseUintParam(c, "id")
+	if err != nil || id == 0 {
+		response.BadRequestI18n(c, examplei18n.MsgInvalidID)
+		return
+	}
+	item, err := h.items.Get(c.Request.Context(), tenantID, id)
+	if err != nil {
+		var i18nErr *i18n.I18nError
+		if errors.As(err, &i18nErr) {
+			response.NotFoundI18n(c, i18nErr.Key)
+			return
+		}
+		response.InternalErrorI18n(c, examplei18n.MsgItemLoadFailed)
+		return
+	}
+	response.Success(c, item)
+}
+
+func (h *ItemHandler) Create(c *gin.Context) {
+	if h == nil || h.items == nil {
+		response.ServiceUnavailableI18n(c, examplei18n.MsgServiceUnavailable)
+		return
+	}
+	tenantID := resolveTenantID(c)
+
+	var input itemInput
+	if err := c.ShouldBindJSON(&input); err != nil {
+		response.BadRequestI18n(c, examplei18n.MsgInvalidPayload)
+		return
+	}
+
+	item, err := h.items.Create(c.Request.Context(), tenantID, service.ItemInput{
+		Title:       input.Title,
+		Description: input.Description,
+		Category:    input.Category,
+		Amount:      input.Amount,
+		Status:      input.Status,
+		Attachment:  normalizeAttachment(input.Attachment),
+	})
+	if err != nil {
+		var i18nErr *i18n.I18nError
+		if errors.As(err, &i18nErr) {
+			response.BadRequestI18n(c, i18nErr.Key)
+			return
+		}
+		response.InternalErrorI18n(c, examplei18n.MsgItemCreateFailed)
+		return
+	}
+
+	response.CreatedI18n(c, examplei18n.MsgItemCreated, item)
+}
+
+func (h *ItemHandler) Update(c *gin.Context) {
+	if h == nil || h.items == nil {
+		response.ServiceUnavailableI18n(c, examplei18n.MsgServiceUnavailable)
+		return
+	}
+	tenantID := resolveTenantID(c)
+
+	id, err := hcommon.ParseUintParam(c, "id")
+	if err != nil || id == 0 {
+		response.BadRequestI18n(c, examplei18n.MsgInvalidID)
+		return
+	}
+
+	var input itemUpdateInput
+	if err := c.ShouldBindJSON(&input); err != nil {
+		response.BadRequestI18n(c, examplei18n.MsgInvalidPayload)
+		return
+	}
+
+	item, err := h.items.Update(c.Request.Context(), tenantID, id, service.ItemUpdateInput{
+		Title:       input.Title,
+		Description: input.Description,
+		Category:    input.Category,
+		Amount:      input.Amount,
+		Status:      input.Status,
+		Attachment:  normalizeAttachmentPtr(input.Attachment),
+	})
+	if err != nil {
+		var i18nErr *i18n.I18nError
+		if errors.As(err, &i18nErr) {
+			if i18nErr.Key == examplei18n.MsgItemNotFound {
+				response.NotFoundI18n(c, i18nErr.Key)
+			} else {
+				response.BadRequestI18n(c, i18nErr.Key)
+			}
+			return
+		}
+		response.InternalErrorI18n(c, examplei18n.MsgItemUpdateFailed)
+		return
+	}
+
+	response.SuccessI18n(c, examplei18n.MsgItemUpdated, item)
+}
+
+func (h *ItemHandler) Delete(c *gin.Context) {
+	if h == nil || h.items == nil {
+		response.ServiceUnavailableI18n(c, examplei18n.MsgServiceUnavailable)
+		return
+	}
+	tenantID := resolveTenantID(c)
+
+	id, err := hcommon.ParseUintParam(c, "id")
+	if err != nil || id == 0 {
+		response.BadRequestI18n(c, examplei18n.MsgInvalidID)
+		return
+	}
+
+	if err := h.items.Delete(c.Request.Context(), tenantID, id); err != nil {
+		var i18nErr *i18n.I18nError
+		if errors.As(err, &i18nErr) {
+			if i18nErr.Key == examplei18n.MsgItemNotFound {
+				response.NotFoundI18n(c, i18nErr.Key)
+				return
+			}
+			response.BadRequestI18n(c, i18nErr.Key)
+			return
+		}
+		response.InternalErrorI18n(c, examplei18n.MsgItemDeleteFailed)
+		return
+	}
+
+	response.SuccessI18n(c, examplei18n.MsgItemDeleted, gin.H{"id": id})
+}
+
+func (h *ItemHandler) Submit(c *gin.Context) {
+	if h == nil || h.items == nil {
+		response.ServiceUnavailableI18n(c, examplei18n.MsgServiceUnavailable)
+		return
+	}
+	tenantID := resolveTenantID(c)
+
+	id, err := hcommon.ParseUintParam(c, "id")
+	if err != nil || id == 0 {
+		response.BadRequestI18n(c, examplei18n.MsgInvalidID)
+		return
+	}
+
+	item, err := h.items.Submit(c.Request.Context(), tenantID, id)
+	if err != nil {
+		var i18nErr *i18n.I18nError
+		if errors.As(err, &i18nErr) {
+			response.BadRequestI18n(c, i18nErr.Key)
+			return
+		}
+		response.InternalErrorI18n(c, examplei18n.MsgItemSubmitFailed)
+		return
+	}
+
+	response.SuccessI18n(c, examplei18n.MsgItemSubmitted, item)
+}
+
+func (h *ItemHandler) Cancel(c *gin.Context) {
+	if h == nil || h.items == nil {
+		response.ServiceUnavailableI18n(c, examplei18n.MsgServiceUnavailable)
+		return
+	}
+	tenantID := resolveTenantID(c)
+
+	id, err := hcommon.ParseUintParam(c, "id")
+	if err != nil || id == 0 {
+		response.BadRequestI18n(c, examplei18n.MsgInvalidID)
+		return
+	}
+
+	var input cancelInput
+	_ = c.ShouldBindJSON(&input)
+
+	item, err := h.items.Cancel(c.Request.Context(), tenantID, id, input.Reason)
+	if err != nil {
+		var i18nErr *i18n.I18nError
+		if errors.As(err, &i18nErr) {
+			response.BadRequestI18n(c, i18nErr.Key)
+			return
+		}
+		response.InternalErrorI18n(c, examplei18n.MsgItemCancelFailed)
+		return
+	}
+
+	response.SuccessI18n(c, examplei18n.MsgItemCancelled, item)
+}
+
+func resolveTenantID(c *gin.Context) uint {
+	if c == nil {
+		return defaultTenantID
+	}
+	if tenantID, ok := hcommon.GetTenantID(c); ok && tenantID > 0 {
+		return tenantID
+	}
+	return defaultTenantID
+}
+
+func buildItemListFilter(c *gin.Context) (service.ItemListFilter, error) {
+	filter := service.ItemListFilter{
+		Keyword: c.Query("keyword"),
+	}
+	if raw := c.Query("status"); raw != "" {
+		status := models.ItemStatus(raw)
+		filter.Status = &status
+	}
+	if raw := c.Query("approval_status"); raw != "" {
+		status := approvalmodels.InstanceStatus(raw)
+		filter.ApprovalStatus = &status
+	}
+	if raw := c.Query("category"); raw != "" {
+		category := models.ItemCategory(raw)
+		filter.Category = &category
+	}
+	if raw := c.Query("start_date"); raw != "" {
+		parsed, err := parseDate(raw, false)
+		if err != nil {
+			return filter, err
+		}
+		filter.StartDate = parsed
+	}
+	if raw := c.Query("end_date"); raw != "" {
+		parsed, err := parseDate(raw, true)
+		if err != nil {
+			return filter, err
+		}
+		filter.EndDate = parsed
+	}
+	return filter, nil
+}
+
+func buildItemsPDF(items []models.ExampleItem) ([]byte, error) {
+	doc, err := pdf.NewDocument(pdf.Config{})
+	if err != nil {
+		return nil, err
+	}
+	doc.AddPage()
+	p := doc.PDF()
+	p.SetFont(doc.FontFamily(), "", 16)
+	p.CellFormat(0, 10, "Example Items", "", 1, "L", false, 0, "")
+	p.SetFont(doc.FontFamily(), "", 11)
+	p.CellFormat(0, 7, fmt.Sprintf("Generated at: %s", time.Now().Format("2006-01-02 15:04")), "", 1, "L", false, 0, "")
+	p.CellFormat(0, 7, fmt.Sprintf("Total: %d", len(items)), "", 1, "L", false, 0, "")
+	p.Ln(2)
+
+	columns := []pdf.TableColumn{
+		{Header: "Number", Width: 28, Align: "C"},
+		{Header: "Title", Width: 62, Align: "L"},
+		{Header: "Category", Width: 22, Align: "C"},
+		{Header: "Amount", Width: 20, Align: "R"},
+		{Header: "Status", Width: 20, Align: "C"},
+		{Header: "Approval", Width: 28, Align: "C"},
+	}
+	table := pdf.NewTable(doc, columns)
+	table.DrawHeader()
+	for _, item := range items {
+		status := string(item.Status)
+		if status == "" {
+			status = "-"
+		}
+		approvalStatus := string(item.ApprovalStatus)
+		if approvalStatus == "" {
+			approvalStatus = "-"
+		}
+		category := string(item.Category)
+		if category == "" {
+			category = "-"
+		}
+		if err := table.DrawRow(
+			[]string{
+				item.Number,
+				item.Title,
+				category,
+				fmt.Sprintf("%.2f", item.Amount),
+				status,
+				approvalStatus,
+			},
+			12,
+			table.DrawHeader,
+		); err != nil {
+			return nil, err
+		}
+	}
+	return doc.OutputBytes()
+}
+
+func normalizeAttachment(raw json.RawMessage) datatypes.JSON {
+	clean := bytes.TrimSpace(raw)
+	if len(clean) == 0 || string(clean) == "null" {
+		return nil
+	}
+	return datatypes.JSON(clean)
+}
+
+func normalizeAttachmentPtr(raw *json.RawMessage) *datatypes.JSON {
+	if raw == nil {
+		return nil
+	}
+	normalized := normalizeAttachment(*raw)
+	return &normalized
+}
+
+func parseDate(value string, endOfDay bool) (*time.Time, error) {
+	layouts := []string{time.RFC3339, "2006-01-02"}
+	for _, layout := range layouts {
+		parsed, err := time.Parse(layout, value)
+		if err != nil {
+			continue
+		}
+		if endOfDay && layout == "2006-01-02" {
+			parsed = parsed.Add(23*time.Hour + 59*time.Minute + 59*time.Second)
+		}
+		return &parsed, nil
+	}
+	return nil, errors.New("invalid date")
+}