package controllers

import (
	"fmt"
	"github.com/gin-gonic/gin"
	"github.com/xuri/excelize/v2"
	"gorm.io/gorm"
	"strings"
	"wms/constvar"
	"wms/extend/code"
	"wms/extend/util"
	"wms/models"
	"wms/pkg/logx"
	"wms/utils/upload"
)

type OtherController struct{}

type saveProductImagesFromExcelRequest struct {
	ExcelFilePath        string `json:"excelFilePath"`        //excel路径 用docker cp传到容器里
	SheetIndex           int    `json:"sheetIndex"`           //第几个sheet
	ProductIdColumnIndex int    `json:"productIdColumnIndex"` //产品ID是第几列(从0开始)
	ImageColumn          string `json:"imageColumn"`          //图片所在列名
}

// SaveProductImagesFromExcel
// @Tags      其他
// @Summary   从excel中获取产品图片并保存
// @Produce   application/json
// @Param     object  body  saveProductImagesFromExcelRequest true  "属性信息"
// @Param     Authorization	header string true "token"
// @Success   200 {object} util.Response "成功"
// @Router    /api-wms/v1/other/saveProductImagesFromExcel [post]
func (slf OtherController) SaveProductImagesFromExcel(c *gin.Context) {
	var params saveProductImagesFromExcelRequest
	if err := c.BindJSON(&params); err != nil {
		util.ResponseFormat(c, code.RequestParamError, "参数解析失败，数据类型错误:"+err.Error())
		return
	}
	if params.ExcelFilePath == "" || params.ImageColumn == "" {
		util.ResponseFormat(c, code.RequestParamError, "参数缺失")
		return
	}

	productImageMap, err := UploadProductImagesFromExcel(params.ExcelFilePath, params.SheetIndex, params.ProductIdColumnIndex, params.ImageColumn)
	if err != nil {
		logx.Errorf("SaveProductImagesFromExcel err:%v", err)
		util.ResponseFormat(c, code.SaveFail, "操作失败")
		return
	}

	for productId, urls := range productImageMap {
		if len(urls) == 0 {
			continue
		}
		attachmentList := make([]*models.Attachment, 0)
		materialAttachmentList := make([]*models.MaterialAttachment, 0)
		for _, url := range urls {
			attachmentList = append(attachmentList, &models.Attachment{
				FileUrl:  url,
				FileType: constvar.FileType_Picture,
			})
		}

		err = models.WithTransaction(func(db *gorm.DB) error {
			//保存附件
			if attachmentList, err = models.NewAttachmentSearch().CreateBatchWithResp(attachmentList); err != nil {
				logx.Errorf("attachment create err: %v", err)
				return err
			}

			//保存物料和附件映射
			for _, v := range attachmentList {
				ma := &models.MaterialAttachment{MaterialID: productId, AttachmentID: v.Id}
				materialAttachmentList = append(materialAttachmentList, ma)
			}
			if err := models.NewMaterialAttachmentSearch().SetOrm(db).CreateBatch(materialAttachmentList); err != nil {
				return err
			}
			return nil
		})
		if err != nil {
			logx.Errorf("SaveProductImagesFromExcel save db  err:%v, productId:%v, urls:%v", err, productId, urls)
		}
	}

	util.ResponseFormat(c, code.Success, "保存成功")
}

// UploadProductImagesFromExcel 从Excel文件中上传产品图片
func UploadProductImagesFromExcel(filePath string, sheetIndex int, productIdColumn int, imageColumn string) (productImageMap map[string][]string, err error) {
	productImageMap = make(map[string][]string)
	// 加载Excel文件
	f, err := excelize.OpenFile(filePath)
	if err != nil {
		logx.Errorf("Failed to open Excel file: %v", err)
		return
	}
	defer f.Close()

	// 获取工作表名称
	sheetName := f.GetSheetName(sheetIndex)
	// 读取工作表中的所有行
	rows, err := f.GetRows(sheetName)
	if err != nil {
		logx.Errorf("Failed to get rows from sheet:%v", err)
		return
	}

	// 获取指定列的合并单元格信息
	mergedCells, err := getColumnMergeCells(f, sheetName, imageColumn)
	if err != nil {
		logx.Errorf("Failed to get merged cells:%v", err)
		return
	}

	mergeCellMap := make(map[string]excelize.MergeCell)
	for _, cell := range mergedCells {
		mergeCellMap[cell.GetStartAxis()] = cell
	}

	var imagePaths []string
	// 遍历Excel行，读取数据
	for rowIndex, row := range rows[1:] { // 假设第一行是标题行，跳过
		if len(row) < 1 {
			continue
		}
		productId := row[productIdColumn]
		currentCell := fmt.Sprintf("%s%d", imageColumn, rowIndex+2)

		// 检查当前单元格是否在合并单元格中
		if mergeCell, ok := mergeCellMap[currentCell]; ok {
			imagePaths = make([]string, 0)
			imageCell := mergeCell.GetStartAxis()

			// 获取图片
			pictures, err := f.GetPictures(sheetName, imageCell)
			if err != nil {
				logx.Errorf("Failed to get picture for cell %s: %v", imageCell, err)
				continue
			}

			if len(pictures) == 0 {
				logx.Errorf("No picture found for cell %s", imageCell)
				continue
			}

			for k, picture := range pictures {
				fileBytes := picture.File
				fileName := fmt.Sprintf("image_%s_%d.png", imageCell, k+1)

				// 保存图片到本地并获取路径
				imagePath, err := upload.UploadFileToSeaWeed(string(constvar.FileType_Picture), fileName, fileBytes)
				if err != nil {
					logx.Errorf("Failed to save image for product %s: %v\n", productId, err)
					continue
				}
				imagePaths = append(imagePaths, imagePath)
			}
		}

		if productId == "" {
			continue
		}

		// imagePaths
		productImageMap[productId] = imagePaths
		logx.Infof("UploadProductImagesFromExcel Product ID: %s, Image Urls: %s\n", productId, imagePaths)
	}
	return productImageMap, nil
}

// getColumnMergeCells 获取指定列的合并单元格
func getColumnMergeCells(f *excelize.File, sheetName string, targetColumn string) ([]excelize.MergeCell, error) {
	// 获取所有合并单元格
	mergeCells, err := f.GetMergeCells(sheetName)
	if err != nil {
		return nil, err
	}

	var columnMergeCells []excelize.MergeCell
	for _, mergeCell := range mergeCells {
		if strings.HasPrefix(mergeCell.GetStartAxis(), targetColumn) {
			columnMergeCells = append(columnMergeCells, mergeCell)
		}
	}
	return columnMergeCells, nil
}