QuoteForge/internal/handlers/export.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"`
	ProjectName string `json:"project_name"`
	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)

	// Validate before streaming (can return JSON error)
	if len(data.Items) == 0 {
		c.JSON(http.StatusBadRequest, gin.H{"error": "no items to export"})
		return
	}

	// Set headers before streaming
	projectName := req.ProjectName
	if projectName == "" {
		projectName = req.Name
	}
	filename := fmt.Sprintf("%s (%s) %s BOM.csv", time.Now().Format("2006-01-02"), projectName, req.Name)
	c.Header("Content-Type", "text/csv; charset=utf-8")
	c.Header("Content-Disposition", fmt.Sprintf("attachment; filename=\"%s\"", filename))

	// Stream CSV (cannot return JSON after this point)
	if err := h.exportService.ToCSV(c.Writer, data); err != nil {
		c.Error(err) // Log only
		return
	}
}

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")

	// Get config before streaming (can return JSON error)
	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)

	// Validate before streaming (can return JSON error)
	if len(data.Items) == 0 {
		c.JSON(http.StatusBadRequest, gin.H{"error": "no items to export"})
		return
	}

	// Set headers before streaming
	// For config export, use config name for both project and quotation name
	filename := fmt.Sprintf("%s (%s) %s BOM.csv", config.CreatedAt.Format("2006-01-02"), config.Name, config.Name)
	c.Header("Content-Type", "text/csv; charset=utf-8")
	c.Header("Content-Disposition", fmt.Sprintf("attachment; filename=\"%s\"", filename))

	// Stream CSV (cannot return JSON after this point)
	if err := h.exportService.ToCSV(c.Writer, data); err != nil {
		c.Error(err) // Log only
		return
	}
}