cmautobuy/admin/handler/web/others.go

package web

import (
	"context"
	"errors"
	"fmt"
	"log"
	"net/http"
	"net/url"
	"strings"
	"time"

	"github.com/gin-gonic/gin"

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

// ---------- 2. 顺运宝数据 ----------

// SybList 渲染货运单列表页。
// 「匹配状态」是算出来的（shopee_sku_id 是否非空），不是存的字段，
// 本工单（#46）不做规格匹配，这里只如实显示"有没有"。
func (h *Handler) SybList(c *gin.Context) {
	h.renderSybList(c, c.Query("order_no"), c.Query("page"), c.Query("msg"))
}

// renderSybList 是 SybList、SybSync、SybLoginAndSync 跳转回来共用的渲染逻辑。
func (h *Handler) renderSybList(c *gin.Context, keyword, pageRaw, msg string) {
	h.renderSybListWithLoginReason(c, keyword, pageRaw, msg, c.Query("login_reason"))
}

// renderSybListWithLoginReason 同 renderSybList，额外带一条"登录弹窗里要
// 显示的原因"——工单 #47：自动识别验证码失败或降级时，操作员需要知道
// 是配置错了还是服务挂了，不能弹一个空的手工输入框了事。
func (h *Handler) renderSybListWithLoginReason(c *gin.Context, keyword, pageRaw, msg, loginReason string) {
	pageNum := service.ParsePage(pageRaw)
	result, err := service.ListSybOrdersView(h.db, keyword, pageNum)
	if err != nil {
		fail(c, http.StatusInternalServerError,
			"读取顺运宝货运单列表失败，数据没有被改动。刷新页面重试；一直失败请把这句话报给维护者。")
		return
	}

	status := msg
	if status == "" {
		status = sybStatusLine(result)
	}
	if service.GetSybSyncStatus().Running {
		status = "同步进行中，请稍后刷新页面查看结果 · " + status
	}

	values := url.Values{}
	if keyword != "" {
		values.Set("order_no", keyword)
	}

	// `[必须]` 密码不读出来显示、也不回显到 HTML，见工单 #46。
	// 这里只取 username（只读展示）和 base_url 是否配置正确。
	username := ""
	configProblem := ""
	needLogin := false
	cfg, cfgErr := config.Load()
	switch {
	case cfgErr != nil:
		configProblem = cfgErr.Error()
	default:
		username = cfg.Syb.Username
		client, err := syb.New(cfg.Syb.BaseURL)
		if err != nil {
			configProblem = "顺运宝 base_url 配置有误: " + err.Error()
		} else {
			sessErr := service.EnsureSybSession(h.db, client, cfg.Syb.Username, time.Now())
			needLogin = errors.Is(sessErr, service.ErrSybLoginRequired)
		}
	}

	c.HTML(http.StatusOK, "syb/list", page(c, "syb", "顺运宝数据", gin.H{
		"Keyword":         keyword,
		"Rows":            result.Rows,
		"Status":          status,
		"HasAny":          result.HasAny,
		"IsFiltered":      result.IsFiltered,
		"NeedLogin":       needLogin,
		"NeedLoginReason": loginReason,
		"Username":        username,
		"ConfigProblem":   configProblem,
		"Pagination":      service.NewPaginationView(result.Page, result.TotalPages, values.Encode()),
	}))
}

// sybStatusLine 组装底部状态条的默认文案（没有 msg 覆盖时）。
// `[必须]` 显示筛选后的**全量**总数，不是本页行数，见工单 #43 定下的规则。
func sybStatusLine(result *service.SybListResult) string {
	prefix := fmt.Sprintf("共 %d 条货运单明细", result.Total)
	if !result.HasAny {
		return "还没有货运单明细。点上方「同步」从顺运宝拉取。"
	}
	return fmt.Sprintf("%s · 第 %d/%d 页", prefix, result.Page, result.TotalPages)
}

// SybSync 点「同步」按钮的入口。
//
// 流程见工单 #46/#47：
//
//	会话有效 ─────────────────────────→ 直接开始同步（后台跑，立即跳转回列表页）
//	会话无效/过期 ──→ OCR 已配置 ──→ 自动识别验证码登录成功 ──→ 直接开始同步，无人值守
//	              │              └→ 识别失败/OCR 不可达 ──┐
//	              └→ OCR 未配置（留空）────────────────────┴→ 跳回列表页，弹手工登录框
func (h *Handler) SybSync(c *gin.Context) {
	cfg, err := config.Load()
	if err != nil {
		fail(c, http.StatusBadRequest, err.Error())
		return
	}
	client, err := syb.New(cfg.Syb.BaseURL)
	if err != nil {
		fail(c, http.StatusBadRequest, "顺运宝 base_url 配置有误: "+err.Error())
		return
	}

	if err := service.EnsureSybSession(h.db, client, cfg.Syb.Username, time.Now()); err != nil {
		if !errors.Is(err, service.ErrSybLoginRequired) {
			fail(c, http.StatusInternalServerError, "校验顺运宝会话失败："+err.Error())
			return
		}

		// `[必须]` 会话过期后先尝试 OCR 自动登录（工单 #47），
		// 识别不出来或 OCR 不可达再降级弹手工输入框——不是直接报错。
		reason, ok := h.attemptSybAutoLogin(c, client, *cfg)
		if !ok {
			h.sybRedirectLogin(c, reason)
			return
		}
		if !h.startSybSync(client, *cfg) {
			h.sybRedirect(c, "验证码自动识别成功，已自动登录；但已经有一个同步任务在跑，请稍后刷新页面查看结果")
			return
		}
		h.sybRedirect(c, "验证码自动识别成功，已自动登录，同步已开始，请稍后刷新页面查看结果")
		return
	}

	if !h.startSybSync(client, *cfg) {
		h.sybRedirect(c, "已经有一个同步任务在跑，请稍后刷新页面查看结果，不要重复点击")
		return
	}
	h.sybRedirect(c, "同步已开始，请稍后刷新页面查看结果")
}

// attemptSybAutoLogin 尝试用配置的 OCR 服务自动识别验证码并登录。
//
// `[必须]` cfg.Syb.OcrURL 留空 = 禁用，直接返回 ok=false、reason=""——
// 这种情况下弹出的手工登录框不应该带任何"失败原因"文案，因为压根没
// 尝试过自动识别，见工单 #47 验收标准「ocr_url 留空 → 直接走手工，不报错」。
//
// 返回 ok=true 时表示登录已成功、会话已缓存，调用方可以直接发起同步；
// ok=false 时 reason 是给操作员看的降级原因，调用方应该带着它弹手工输入框。
func (h *Handler) attemptSybAutoLogin(c *gin.Context, client *syb.Client, cfg config.Config) (reason string, ok bool) {
	if strings.TrimSpace(cfg.Syb.OcrURL) == "" {
		return "", false
	}

	ocrClient, err := syb.NewOcrClient(cfg.Syb.OcrURL, syb.DefaultOcrTimeout)
	if err != nil {
		// `[必须]` OCR 客户端本身构造失败（配置有误）也是"服务不可用"的
		// 一种，同样降级到手工，不报错，见工单 #47。
		return "验证码识别服务配置有误（" + err.Error() + "），请手工输入", false
	}

	result, degradeReason := client.LoginWithOCR(
		c.Request.Context(), ocrClient, cfg.Syb.Username, cfg.Syb.Password, cfg.Syb.OcrMaxAttempts)
	if degradeReason != "" {
		return degradeReason, false
	}

	// `[必须]` 缓存写失败不能让已经登录的会话失效——只记日志，照样往下走同步，
	// 和 SybLoginAndSync 手工登录路径的处理方式一致。
	if err := service.SaveSybLoginSession(h.db, client, result.User.Username, result.ExpiresAt); err != nil {
		log.Printf("syb_session_save_failed username=%s err=%v", result.User.Username, err)
	}
	return "", true
}

// SybCaptcha 返回一张新的顺运宝验证码图片。
//
// `[必须]` 验证码和随后提交的登录必须用同一个 Cookie Jar（08 §3.2），
// 所以这里新建的 syb.Client 要缓存起来（service.NewPendingSybLogin），
// 供 SybLoginAndSync 复用，不能各请求各建一个客户端。
func (h *Handler) SybCaptcha(c *gin.Context) {
	cfg, err := config.Load()
	if err != nil {
		c.String(http.StatusBadRequest, "%s", err.Error())
		return
	}
	client, err := service.NewPendingSybLogin(cfg.Syb.BaseURL)
	if err != nil {
		c.String(http.StatusBadRequest, "%s", err.Error())
		return
	}
	captcha, err := client.FetchCaptcha(c.Request.Context())
	if err != nil {
		c.String(http.StatusBadGateway, "获取验证码失败：%s", err.Error())
		return
	}
	c.Data(http.StatusOK, captcha.ContentType, captcha.Image)
}

// SybLoginAndSync 提交验证码登录，成功后立即发起同步。
//
// `[必须]` 密码从 config.yaml 读，不接受表单传入、不回显、不进日志——
// 界面上只有验证码是操作员手输的，见工单 #46。
func (h *Handler) SybLoginAndSync(c *gin.Context) {
	cfg, err := config.Load()
	if err != nil {
		fail(c, http.StatusBadRequest, err.Error())
		return
	}

	code := strings.TrimSpace(c.PostForm("code"))
	if code == "" {
		h.sybRedirect(c, "验证码不能为空，请重新输入")
		return
	}

	client := service.PendingSybLoginClient()
	if client == nil {
		h.sybRedirect(c, "验证码已过期，请重新获取后再试")
		return
	}
	defer service.ClearPendingSybLogin()

	result, err := client.Login(c.Request.Context(), cfg.Syb.Username, cfg.Syb.Password, code)
	if err != nil {
		// `[必须]` err 来自 syb.Client，错误信息本身不含密码，可以直接展示。
		h.sybRedirect(c, "登录失败："+err.Error())
		return
	}

	// `[必须]` 缓存写失败不能让已经登录的会话失效——只记日志，照样往下走同步。
	if err := service.SaveSybLoginSession(h.db, client, result.User.Username, result.ExpiresAt); err != nil {
		log.Printf("syb_session_save_failed username=%s err=%v", result.User.Username, err)
	}

	if !h.startSybSync(client, *cfg) {
		h.sybRedirect(c, "登录成功，但已经有一个同步任务在跑，请稍后刷新页面查看结果")
		return
	}
	h.sybRedirect(c, "登录成功，同步已开始，请稍后刷新页面查看结果")
}

// startSybSync 尝试拿互斥标志并在后台协程里跑同步；已经在跑时返回 false。
//
// `[必须]` 同步是长任务，不能阻塞 HTTP 请求线程直到结束，也不引入后台
// 协程池——一次只允许一个同步在跑，靠 service.TryStartSybSync 这个互斥
// 标志挡住重复点击，见工单 #46。
//
// `[必须]` 后台协程用 context.Background()，不能用 c.Request.Context()——
// 那个请求上下文会在这次 HTTP 请求返回后就被取消，同步跑到一半会被打断。
func (h *Handler) startSybSync(client *syb.Client, cfg config.Config) bool {
	if !service.TryStartSybSync() {
		return false
	}
	db := h.db
	go func() {
		report := service.RunSybSync(context.Background(), db, client, cfg.Syb, time.Now())
		service.FinishSybSync(report)
	}()
	return true
}

// sybRedirect 处理完写操作后跳回列表页，带上当前筛选和提示信息。
// 用 303 跳转是为了让浏览器地址栏变成 GET /syb，按 F5 不会重复提交。
func (h *Handler) sybRedirect(c *gin.Context, msg string) {
	params := url.Values{}
	if q := c.PostForm("order_no"); q != "" {
		params.Set("order_no", q)
	}
	if msg != "" {
		params.Set("msg", msg)
	}
	target := "/syb"
	if len(params) > 0 {
		target += "?" + params.Encode()
	}
	c.Redirect(http.StatusSeeOther, target)
}

// sybRedirectLogin 同 sybRedirect，但把 reason 放进 login_reason 参数，
// 渲染时会显示在**登录弹窗里面**（工单 #47 要求：自动识别失败/降级要
// 说明原因，不能弹一个不知道为什么弹出来的空表单）。reason 为空时
// （比如 ocr_url 留空，压根没尝试过自动识别）弹窗不显示任何原因说明。
func (h *Handler) sybRedirectLogin(c *gin.Context, reason string) {
	params := url.Values{}
	if q := c.PostForm("order_no"); q != "" {
		params.Set("order_no", q)
	}
	if reason != "" {
		params.Set("login_reason", reason)
	}
	target := "/syb"
	if len(params) > 0 {
		target += "?" + params.Encode()
	}
	c.Redirect(http.StatusSeeOther, target)
}

// SybMatch 保存规格匹配结果。
//
// 关键：保存的是**可复用的 SKU 映射**（写 sku_mappings），
// 不是这一张订单的临时数据。下次遇到同一个蝦皮 SKU 自动带出，
// 操作员只需确认。见 docs/admin/03-data-model.md §5。
func (h *Handler) SybMatch(c *gin.Context) {
	// TODO(骨架): upsert sku_mappings
	fail(c, http.StatusNotImplemented, "规格匹配尚未实现。")
}

// SybCreateTask 由勾选的货运单创建采购任务。
//
// 六条校验缺一不可，见 docs/admin/01-requirements.md §5：
//  1. 已填 PDD 链接
//  2. 已采集成功（pdd_data 非空）
//  3. 已有 SKU 映射
//  4. 数量 > 0
//  5. **价格上限已填且 > 0**（Client 的价格保护，没有它会拒绝执行）
//  6. 已选择分配的客户端
//
// 校验不过的**不要静默跳过**，要列出来告诉操作员缺什么。
func (h *Handler) SybCreateTask(c *gin.Context) {
	// TODO(骨架): 调 service.CreatePurchaseTasks(sybIDs, clientID)
	fail(c, http.StatusNotImplemented, "创建采购任务尚未实现。")
}

// SybDelete 批量删除货运单。
func (h *Handler) SybDelete(c *gin.Context) {
	// TODO(骨架)
	fail(c, http.StatusNotImplemented, "删除功能尚未实现。")
}

// ---------- 3. 采集采购 ----------
//
// `tasks` 是一张表，用 task_type 区分采集和采购，这个模块把两种任务放在
// 同一个列表里显示，靠「目标」一列概括各自的业务信息，见 #19。
// 拼接和翻译逻辑全在 service/task.go，这里只做取参数 → 调 service → 渲染。

// TaskList 渲染「采集采购」列表页。
func (h *Handler) TaskList(c *gin.Context) {
	filter := repository.TaskFilter{
		Type:    service.ParseTaskType(c.Query("type")),
		Status:  service.ParseTaskStatus(c.Query("status")),
		Keyword: c.Query("q"),
	}

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

	// 底部状态条平时显示统计，刚做完删除操作时先显示操作结果，
	// 跳转带过来的 msg 参数，见 h.taskRedirect。
	statusLine := result.StatusLine()
	if msg := c.Query("msg"); msg != "" {
		statusLine = msg + " · " + statusLine
	}

	c.HTML(http.StatusOK, "task/list", page(c, "tasks", "采集采购", gin.H{
		"Rows":          result.Rows,
		"Keyword":       filter.Keyword,
		"Status":        statusLine,
		"TypeFilter":    string(filter.Type),
		"StatusFilter":  string(filter.Status),
		"TypeOptions":   service.TaskTypeOptions(),
		"StatusOptions": service.TaskStatusOptions(),
		"IsFiltered":    result.IsFiltered,
	}))
}

// TaskDetail 渲染双击行弹出的只读详情弹窗内容（不是整页）。
//
// `[必须]` 只读。改派 / 重试 / 取消是后续工单的范围，这里不提供入口。
// 弹窗机制复用 #18 已有的那套（static/js/app.js 的 setupRowDetail），
// 不新造一套。
func (h *Handler) TaskDetail(c *gin.Context) {
	taskID := c.Query("id")
	if taskID == "" {
		fail(c, http.StatusBadRequest, "任务编号不对，请刷新页面后重试。")
		return
	}

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

	c.HTML(http.StatusOK, "task/detail_modal", gin.H{"D": detail})
}

// TaskDelete 批量删除任务。
//
// `tasks` 表没有软删除列——加一列是数据库结构变更，不在本工单范围内，
// 见 repository.DeleteTasks 的注释。二次确认在前端做（data-confirm-delete）。
func (h *Handler) TaskDelete(c *gin.Context) {
	ids := c.PostFormArray("ids")
	if len(ids) == 0 {
		h.taskRedirect(c, "没有勾选任何任务，没有删除")
		return
	}

	n, err := service.DeleteTasks(h.db, ids)
	if err != nil {
		fail(c, http.StatusInternalServerError,
			"删除失败："+err.Error()+"。没有删除任何记录。")
		return
	}
	h.taskRedirect(c, fmt.Sprintf("已删除 %d 条", n))
}

// taskRedirect 处理完写操作后跳回列表页，带上当前筛选条件，
// 免得操作员每做一次删除就要重新筛一遍。用 303 而不是直接渲染，
// 是为了让浏览器地址栏变成 GET /tasks——按 F5 不会重复提交刚才的删除。
func (h *Handler) taskRedirect(c *gin.Context, msg string) {
	params := url.Values{}
	if t := c.PostForm("type"); t != "" {
		params.Set("type", t)
	}
	if s := c.PostForm("status"); s != "" {
		params.Set("status", s)
	}
	if q := c.PostForm("q"); q != "" {
		params.Set("q", q)
	}
	if msg != "" {
		params.Set("msg", msg)
	}

	target := "/tasks"
	if len(params) > 0 {
		target += "?" + params.Encode()
	}
	c.Redirect(http.StatusSeeOther, target)
}

// ---------- 4. 客户端列表 ----------

// ClientList 渲染客户端列表页。
//
// 「状态」是**算出来的**：last_seen_at 在 N 分钟内为在线，否则离线。
// 数据库里没有 status 字段，存成字段会和真实情况不同步。
//
// 客户端执行长任务期间不调 claim，可能显示为离线，属正常现象
// （没有心跳是有意的，见 docs/admin/04-client-api.md §3）。
func (h *Handler) ClientList(c *gin.Context) {
	keyword := c.Query("name")

	views, err := service.ListClientViews(h.db, keyword, h.onlineThreshold)
	if err != nil {
		fail(c, http.StatusInternalServerError,
			"读取客户端列表失败，数据没有被改动。请稍后重试，或查看 data/logs/ 里的日志。")
		return
	}

	online := 0
	for _, v := range views {
		if v.Status == "在线" {
			online++
		}
	}
	status := fmt.Sprintf("共 %d 台客户端 · 在线 %d · 离线 %d",
		len(views), online, len(views)-online)
	if len(views) == 0 {
		status = "还没有客户端。客户端第一次调用领取接口时会自动登记。"
	}

	c.HTML(http.StatusOK, "client/list", page(c, "clients", "客户端列表", gin.H{
		"Keyword": keyword,
		"Rows":    views,
		"Status":  status,
	}))
}

// ClientDelete 批量删除客户端。
//
// 二次确认在前端做（见 static/js/app.js），这里直接删。
// 删掉之后客户端下次调 claim 会重新登记，属于正常行为。
func (h *Handler) ClientDelete(c *gin.Context) {
	ids := c.PostFormArray("ids")
	if len(ids) == 0 {
		fail(c, http.StatusBadRequest, "没有选中任何客户端，请勾选后再删除。")
		return
	}

	n, err := service.DeleteClients(h.db, ids)
	if err != nil {
		fail(c, http.StatusInternalServerError,
			"删除失败，数据没有被改动。请稍后重试，或查看 data/logs/ 里的日志。")
		return
	}
	log.Printf("clients_deleted count=%d", n)
	c.Redirect(http.StatusSeeOther, "/clients")
}