cmautobuy/admin/handler/web/shopee.go

package web

import (
	"fmt"
	"io"
	"net/http"
	"net/url"
	"os"
	"path/filepath"
	"strconv"
	"strings"

	"github.com/gin-gonic/gin"

	"cmautobuy/admin/config"
	"cmautobuy/admin/repository"
	"cmautobuy/admin/service"
)

// ShopeeList 渲染蝦皮数据列表页。
//
// **商品级一行**（不是 SKU 级），数据来自 shopee_products 和 shopee_skus
// 聚合联查。列定义见 docs/admin/05-ui-specification.md §4.2、工单 #41。
func (h *Handler) ShopeeList(c *gin.Context) {
	h.renderShopeeList(c, c.Query("goods_id"), c.Query("status"), c.Query("page"), c.Query("msg"), nil)
}

// ShopeeDetail 渲染双击行弹出的那个弹窗的**内容**（不是整页）。
//
// 复用 #18 的弹窗机制：行上写 data-detail-id="{{.GoodsID}}"，
// 前端拿 goods_id 拼到 data-detail-url 后面，以 "id" 作为查询参数名
// （固定写法，见 static/js/app.js「双击行打开详情弹窗」）。
func (h *Handler) ShopeeDetail(c *gin.Context) {
	goodsID := strings.TrimSpace(c.Query("id"))
	if goodsID == "" {
		fail(c, http.StatusBadRequest, "商品编号不对，请刷新页面后重试。")
		return
	}

	detail, err := service.GetShopeeProductDetail(h.db, goodsID)
	if err != nil {
		fail(c, http.StatusInternalServerError, "读取商品详情失败，数据没有被改动。")
		return
	}
	if detail == nil {
		fail(c, http.StatusNotFound, "这个商品不存在，请刷新页面。")
		return
	}

	c.HTML(http.StatusOK, "shopee/detail_modal", gin.H{
		"D": detail, "CSRFToken": csrfToken(c),
		"Keyword": c.Query("goods_id"), "StatusFilter": c.Query("status"),
		"CurrentPage": service.ParsePage(c.Query("page")),
	})
}

// ShopeeImport 处理 Excel 上传导入。
//
// 关键规则（写错会丢数据，见 docs/admin/03-data-model.md §3.3）：
//   - 商品汇总行（商品規格ID 为 "-"）进 shopee_products，
//     SKU 行进 shopee_skus，两者要分开处理；
//   - 按列名找索引，不要写死列号；
//   - 规格原文两种格式各占一半，解析失败留空并把 parse_ok 置 0，不要猜；
//   - **只能 upsert，绝不允许先清空再导入**，
//     否则人工填的 PDD 链接会被洗掉；
//   - upsert 的 DO UPDATE SET 里不得出现 pdd_goods_url / pdd_goods_id。
//
// 导入结果（含全部失败行）直接渲染在返回页面里，**不走 303 跳转**——
// 失败行可能有几十条，塞进跳转用的 URL 查询参数既丑又有长度限制，
// 没法满足"失败行要能全部看到"的要求（见工单 #38）。
// 代价是这个页面刷新（F5）会弹出浏览器自带的"重新提交表单"确认，
// 这是标准行为，比丢失失败行信息更容易接受。
func (h *Handler) ShopeeImport(c *gin.Context) {
	// 限制整个请求体大小：50MB 上限 + 给表单其它字段留一点余量。
	// 放在 FormFile 之前，这样超限时请求体读到一半就会出错，
	// 不会先把整个超大文件读进内存再拒绝。
	c.Request.Body = http.MaxBytesReader(c.Writer, c.Request.Body, service.MaxShopeeUploadBytes+1<<20)

	fileHeader, err := c.FormFile("file")
	if err != nil {
		fail(c, http.StatusBadRequest,
			"没有收到文件（也可能是文件超过了 50MB 上限），请重新选择 Excel 后再提交。未写入数据库。")
		return
	}

	src, err := fileHeader.Open()
	if err != nil {
		fail(c, http.StatusBadRequest, "无法打开上传的文件，请重试。未写入数据库。")
		return
	}
	defer src.Close()

	// 只读文件开头几个字节判断是不是真的 xlsx（zip），不用整份读进内存。
	head := make([]byte, 8)
	n, _ := io.ReadFull(src, head)
	if err := service.ValidateShopeeUpload(fileHeader.Filename, fileHeader.Size, head[:n]); err != nil {
		fail(c, http.StatusBadRequest, err.Error()+"。未写入数据库。")
		return
	}
	if _, err := src.Seek(0, io.SeekStart); err != nil {
		fail(c, http.StatusInternalServerError, "读取上传文件失败，请重试。未写入数据库。")
		return
	}

	uploadDir, err := config.SubDir("uploads")
	if err != nil {
		fail(c, http.StatusInternalServerError, "无法准备上传目录，请重试。未写入数据库。")
		return
	}
	// `[必须]` 自己生成文件名，不用 fileHeader.Filename 拼路径——
	// 用户可控的文件名里塞一个 "../../etc/passwd" 就是路径穿越。
	savePath := filepath.Join(uploadDir, service.NewShopeeUploadFilename())

	dst, err := os.Create(savePath)
	if err != nil {
		fail(c, http.StatusInternalServerError, "保存上传文件失败，请重试。未写入数据库。")
		return
	}
	if _, err := io.Copy(dst, src); err != nil {
		dst.Close()
		fail(c, http.StatusInternalServerError, "保存上传文件失败，请重试。未写入数据库。")
		return
	}
	if err := dst.Close(); err != nil {
		fail(c, http.StatusInternalServerError, "保存上传文件失败，请重试。未写入数据库。")
		return
	}

	result, err := service.ImportShopeeExcel(h.db, savePath)
	if err != nil {
		fail(c, http.StatusBadRequest,
			"导入失败："+err.Error()+"。这次导入整体没有生效（已回滚），可以修好文件后重新上传。")
		return
	}

	msg := fmt.Sprintf("导入完成：%d 个商品 / %d 个 SKU", result.ProductCount, result.SKUCount)
	if len(result.Failures) > 0 {
		msg += fmt.Sprintf("，%d 行解析失败（见下方列表）", len(result.Failures))
	}
	// 导入完成后回到第 1 页、清空筛选——刚导入的数据从第一页就能看到，
	// 沿用筛选反而可能让人以为导入没生效（筛出来的还是老的几条）。
	h.renderShopeeList(c, "", "", "", msg, result.Failures)
}

// renderShopeeList 是 ShopeeList 和 ShopeeImport 共用的渲染逻辑。
//
// `[必须]` 分页控件用 <a href> 纯 GET 导航，翻页要保留 keyword/statusRaw，
// 所以这里把它们原样透传回模板去拼下一页的链接，不在这里丢掉（工单 #43）。
func (h *Handler) renderShopeeList(c *gin.Context, keyword, statusRaw, pageRaw, msg string, failures []service.ImportFailure) {
	filter := repository.ShopeeFilter{
		Keyword: keyword,
		Status:  service.ParseShopeeStatus(statusRaw),
	}
	// 不叫 page：本文件末尾要调用同名的 page(c, ...) 渲染辅助函数，
	// 局部变量会把它遮住导致编译失败。
	pageNum := service.ParsePage(pageRaw)

	result, err := service.ListShopeeProducts(h.db, filter, pageNum)
	if err != nil {
		fail(c, http.StatusInternalServerError,
			"读取蝦皮数据失败，数据没有被改动。刷新页面重试；一直失败请把这句话报给维护者。")
		return
	}

	status := msg
	if status == "" {
		status = shopeeStatusLine(filter.Status, result)
	}

	values := url.Values{}
	if keyword != "" {
		values.Set("goods_id", keyword)
	}
	if filter.Status != "" {
		values.Set("status", filter.Status)
	}

	c.HTML(http.StatusOK, "shopee/list", page(c, "shopee", "蝦皮数据", gin.H{
		"Keyword":        keyword,
		"StatusFilter":   filter.Status,
		"StatusOptions":  service.ShopeeStatusOptions(),
		"Rows":           result.Rows,
		"Status":         status,
		"HasAnyProducts": result.HasAnyProducts,
		"IsFiltered":     result.IsFiltered,
		"Failures":       failures,
		"Pagination":     service.NewPaginationView(result.Page, result.TotalPages, values.Encode()),
		"DetailURL":      "/shopee/detail?" + detailValuesForShopee(values, result.Page),
	}))
}

func detailValuesForShopee(values url.Values, pageNum int) string {
	detailValues := url.Values{}
	for key, entries := range values {
		detailValues[key] = append([]string(nil), entries...)
	}
	detailValues.Set("page", strconv.Itoa(pageNum))
	return detailValues.Encode()
}

// shopeeStatusLine 组装底部状态条的默认文案（没有 msg 覆盖时）。
//
// `[必须]` 显示的是筛选后的**全量**总数（result.Total），不是本页行数，
// 筛选时前缀用状态文字（"待补规格：6 个商品"），不筛选时用"共"，
// 见工单 #43。
func shopeeStatusLine(statusFilter string, result *service.ShopeeListResult) string {
	prefix := fmt.Sprintf("共 %d 个商品", result.Total)
	if text := service.ShopeeStatusLabel(statusFilter); text != "" {
		prefix = fmt.Sprintf("%s：%d 个商品", text, result.Total)
	}
	return fmt.Sprintf("%s · 第 %d/%d 页", prefix, result.Page, result.TotalPages)
}

// ShopeeSave 保存蝦皮商品当前关联的 PDD 商品。
func (h *Handler) ShopeeSave(c *gin.Context) {
	goodsID, err := service.AssociateShopeePdd(h.db, c.PostForm("shopee_goods_id"),
		c.PostForm("pdd_url"), c.PostForm("confirm_replace") == "1")
	if err != nil {
		h.shopeeRedirect(c, "PDD 关联未保存："+err.Error()+"。请核对链接后重试。")
		return
	}
	h.shopeeRedirect(c, fmt.Sprintf("已关联 PDD 商品 %s，可继续创建采集任务", goodsID))
}

// ShopeeDelete 批量删除勾选的行。
//
// `[不做]` 本工单（#38）明确不实现这个接口，保持 501。
func (h *Handler) ShopeeDelete(c *gin.Context) {
	// TODO(骨架): 取 ids，二次确认已在前端做过，这里直接删
	fail(c, http.StatusNotImplemented, "删除功能尚未实现。")
}

// ShopeeCollect 发起采集任务。
//
// 入口在编辑弹窗里，紧挨 PDD 链接输入框——不要放到表格每一行，
// 一个商品有 N 个 SKU 行，放行上就是 N 个按钮干同一件事。
//
// 规则：
//   - PDD 链接为空不允许发起；
//   - collect_status 已经是 collecting 的跳过并说明跳过了几个；
//   - 批量采集要**按商品去重**后再建任务。
//
// `[不做]` 本工单（#38）明确不实现这个接口，保持 501。
func (h *Handler) ShopeeCollect(c *gin.Context) {
	result, err := service.CreateShopeePddCollectTaskForUser(h.db, currentUser(c), c.PostForm("shopee_goods_id"))
	if err != nil {
		h.shopeeRedirect(c, "采集任务未创建："+err.Error())
		return
	}
	h.shopeeRedirect(c, service.FormatCollectTaskMessage(result))
}

func (h *Handler) shopeeRedirect(c *gin.Context, msg string) {
	params := url.Values{}
	if value := strings.TrimSpace(c.PostForm("list_goods_id")); value != "" {
		params.Set("goods_id", value)
	}
	if value := strings.TrimSpace(c.PostForm("status")); value != "" {
		params.Set("status", service.ParseShopeeStatus(value))
	}
	if value := strings.TrimSpace(c.PostForm("page")); value != "" {
		params.Set("page", value)
	}
	if msg != "" {
		params.Set("msg", msg)
	}
	c.Redirect(http.StatusSeeOther, "/shopee?"+params.Encode())
}