package export

import (
	"context"
	"fmt"
	"strings"

	"github.com/xuri/excelize/v2"
	"go.uber.org/zap"
)

// ExportConfig 定义了导出所需的配置
type ExportConfig struct {
	SheetName    string          // 工作表名称
	Headers      []string        // 表头
	Data         [][]interface{} // 导出数据
	ColumnWidths []float64       // 列宽
}

// ExportManager 负责管理不同格式的导出
type ExportManager struct {
	logger *zap.Logger
}

// NewExportManager 创建一个新的ExportManager
func NewExportManager(logger *zap.Logger) *ExportManager {
	return &ExportManager{
		logger: logger,
	}
}

// Export 根据配置和格式生成导出文件
func (m *ExportManager) Export(ctx context.Context, config *ExportConfig, format string) ([]byte, error) {
	switch format {
	case "excel":
		return m.generateExcel(ctx, config)
	case "csv":
		return m.generateCSV(ctx, config)
	default:
		return nil, fmt.Errorf("不支持的导出格式: %s", format)
	}
}

// generateExcel 生成Excel导出文件
func (m *ExportManager) generateExcel(ctx context.Context, config *ExportConfig) ([]byte, error) {
	f := excelize.NewFile()
	defer func() {
		if err := f.Close(); err != nil {
			m.logger.Error("关闭Excel文件失败", zap.Error(err))
		}
	}()

	sheetName := config.SheetName
	index, err := f.NewSheet(sheetName)
	if err != nil {
		return nil, err
	}
	f.SetActiveSheet(index)

	// 设置表头
	for i, header := range config.Headers {
		cell := fmt.Sprintf("%c1", 'A'+i)
		f.SetCellValue(sheetName, cell, header)
	}

	// 设置表头样式
	headerStyle, err := f.NewStyle(&excelize.Style{
		Font: &excelize.Font{Bold: true},
		Fill: excelize.Fill{Type: "pattern", Color: []string{"#E6F3FF"}, Pattern: 1},
	})
	if err != nil {
		return nil, err
	}
	headerRange := fmt.Sprintf("A1:%c1", 'A'+len(config.Headers)-1)
	f.SetCellStyle(sheetName, headerRange, headerRange, headerStyle)

	// 批量写入数据
	for i, rowData := range config.Data {
		row := i + 2 // 从第2行开始写入数据
		for j, value := range rowData {
			cell := fmt.Sprintf("%c%d", 'A'+j, row)
			f.SetCellValue(sheetName, cell, value)
		}
	}

	// 设置列宽
	for i, width := range config.ColumnWidths {
		col := fmt.Sprintf("%c", 'A'+i)
		f.SetColWidth(sheetName, col, col, width)
	}

	buf, err := f.WriteToBuffer()
	if err != nil {
		return nil, err
	}

	return buf.Bytes(), nil
}

// generateCSV 生成CSV导出文件
func (m *ExportManager) generateCSV(ctx context.Context, config *ExportConfig) ([]byte, error) {
	var csvData strings.Builder

	// 写入CSV头部
	csvData.WriteString(strings.Join(config.Headers, ",") + "\n")

	// 写入数据行
	for _, rowData := range config.Data {
		rowStrings := make([]string, len(rowData))
		for i, value := range rowData {
			rowStrings[i] = fmt.Sprintf("%v", value) // 使用%v通用格式化
		}
		csvData.WriteString(strings.Join(rowStrings, ",") + "\n")
	}

	return []byte(csvData.String()), nil
}