122 lines
3.5 KiB
Go
122 lines
3.5 KiB
Go
package handlers
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"time"
|
|
|
|
"git.mchus.pro/mchus/quoteforge/internal/middleware"
|
|
"git.mchus.pro/mchus/quoteforge/internal/services"
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
type ExportHandler struct {
|
|
exportService *services.ExportService
|
|
configService services.ConfigurationGetter
|
|
componentService *services.ComponentService
|
|
}
|
|
|
|
func NewExportHandler(
|
|
exportService *services.ExportService,
|
|
configService services.ConfigurationGetter,
|
|
componentService *services.ComponentService,
|
|
) *ExportHandler {
|
|
return &ExportHandler{
|
|
exportService: exportService,
|
|
configService: configService,
|
|
componentService: componentService,
|
|
}
|
|
}
|
|
|
|
type ExportRequest struct {
|
|
Name string `json:"name" binding:"required"`
|
|
Items []struct {
|
|
LotName string `json:"lot_name" binding:"required"`
|
|
Quantity int `json:"quantity" binding:"required,min=1"`
|
|
UnitPrice float64 `json:"unit_price"`
|
|
} `json:"items" binding:"required,min=1"`
|
|
Notes string `json:"notes"`
|
|
}
|
|
|
|
func (h *ExportHandler) ExportCSV(c *gin.Context) {
|
|
var req ExportRequest
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
data := h.buildExportData(&req)
|
|
|
|
csvData, err := h.exportService.ToCSV(data)
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
filename := fmt.Sprintf("%s %s SPEC.csv", time.Now().Format("2006-01-02"), req.Name)
|
|
c.Header("Content-Disposition", fmt.Sprintf("attachment; filename=\"%s\"", filename))
|
|
c.Data(http.StatusOK, "text/csv; charset=utf-8", csvData)
|
|
}
|
|
|
|
func (h *ExportHandler) buildExportData(req *ExportRequest) *services.ExportData {
|
|
items := make([]services.ExportItem, len(req.Items))
|
|
var total float64
|
|
|
|
for i, item := range req.Items {
|
|
itemTotal := item.UnitPrice * float64(item.Quantity)
|
|
|
|
// Получаем информацию о компоненте для заполнения категории и описания
|
|
componentView, err := h.componentService.GetByLotName(item.LotName)
|
|
if err != nil {
|
|
// Если не удалось получить информацию о компоненте, используем только основные данные
|
|
items[i] = services.ExportItem{
|
|
LotName: item.LotName,
|
|
Quantity: item.Quantity,
|
|
UnitPrice: item.UnitPrice,
|
|
TotalPrice: itemTotal,
|
|
}
|
|
} else {
|
|
items[i] = services.ExportItem{
|
|
LotName: item.LotName,
|
|
Description: componentView.Description,
|
|
Category: componentView.Category,
|
|
Quantity: item.Quantity,
|
|
UnitPrice: item.UnitPrice,
|
|
TotalPrice: itemTotal,
|
|
}
|
|
}
|
|
total += itemTotal
|
|
}
|
|
|
|
return &services.ExportData{
|
|
Name: req.Name,
|
|
Items: items,
|
|
Total: total,
|
|
Notes: req.Notes,
|
|
CreatedAt: time.Now(),
|
|
}
|
|
}
|
|
|
|
func (h *ExportHandler) ExportConfigCSV(c *gin.Context) {
|
|
username := middleware.GetUsername(c)
|
|
uuid := c.Param("uuid")
|
|
|
|
config, err := h.configService.GetByUUID(uuid, username)
|
|
if err != nil {
|
|
c.JSON(http.StatusNotFound, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
data := h.exportService.ConfigToExportData(config, h.componentService)
|
|
|
|
csvData, err := h.exportService.ToCSV(data)
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
filename := fmt.Sprintf("%s %s SPEC.csv", config.CreatedAt.Format("2006-01-02"), config.Name)
|
|
c.Header("Content-Disposition", fmt.Sprintf("attachment; filename=\"%s\"", filename))
|
|
c.Data(http.StatusOK, "text/csv; charset=utf-8", csvData)
|
|
}
|