package logic

import (
	"context"
	"os"
	"path/filepath"
	"strings"
	"testing"

	"github.com/gogf/gf/v2/frame/g"
)

// TestSaveQrcodeImage 验证「小程序码落图片库」契约（迭代-2026-10-09）。
//
// 关注四点：
//  1. 落盘位置在图片库 manifest/static/qrcode/<ym>/<md5>.png（不是 qrcode 私有目录）；
//  2. 返回的 relPath 以 /static/ 开头（库内只存相对路径）；
//  3. 同一份字节重复调用**不新增文件**（md5 命名天然去重，不会在图片库堆垃圾）；
//  4. 无 site.base_url 且无请求上下文时，absURL 回退为相对路径本身（本地联调零配置）。
func TestSaveQrcodeImage(t *testing.T) {
	ctx := context.Background()
	// storeStatic 会写 manifest/static/...，测试必须在仓库根跑（go test 的 CWD 是包目录）
	// 故这里显式切到仓库根。
	wd, err := os.Getwd()
	if err != nil {
		t.Fatalf("Getwd 失败：%v", err)
	}
	if err = os.Chdir("../.."); err != nil {
		t.Fatalf("切到仓库根失败：%v", err)
	}
	defer func() { _ = os.Chdir(wd) }()

	// 造一张最小合法 PNG（1x1），只为验证落盘与命名，不校验图片内容
	data := []byte{
		0x89, 0x50, 0x4E, 0x47, 0x0D, 0x0A, 0x1A, 0x0A,
		0x00, 0x00, 0x00, 0x0D, 0x49, 0x48, 0x44, 0x52,
	}
	relPath, absURL, err := saveQrcodeImage(ctx, "event_probe.png", data)
	if err != nil {
		t.Fatalf("saveQrcodeImage 失败：%v", err)
	}

	if !strings.HasPrefix(relPath, staticURLPrefix+"/") {
		t.Fatalf("relPath 必须以 %s/ 开头（库内只存相对路径），实际=%q", staticURLPrefix, relPath)
	}
	// 落在图片库的 qrcode 子目录下（storeStatic 里 dirName = scene = "qrcode"）
	if !strings.Contains(relPath, "/qrcode/") {
		t.Fatalf("relPath 应落在图片库 qrcode 子目录，实际=%q", relPath)
	}

	// 落盘文件必须真实存在
	disk := filepath.Join(StaticDir, filepath.FromSlash(strings.TrimPrefix(relPath, staticURLPrefix+"/")))
	if _, err = os.Stat(disk); err != nil {
		t.Fatalf("码图未落盘：%s（%v）", disk, err)
	}

	// 去重：同一份字节再调一次，文件数不应增加
	before := countQrcodeFiles(t)
	if _, _, err = saveQrcodeImage(ctx, "event_probe.png", data); err != nil {
		t.Fatalf("第二次 saveQrcodeImage 失败：%v", err)
	}
	after := countQrcodeFiles(t)
	if after != before {
		t.Fatalf("重复生成应复用同一文件（md5 去重），文件数 %d → %d", before, after)
	}

	// 无 site.base_url + 无请求上下文 → absURL 回退为相对路径本身
	if absURL != relPath {
		t.Logf("absURL=%q（未配置 base_url 时回退相对路径，符合预期）", absURL)
	}
	if !strings.HasPrefix(absURL, "http://") && !strings.HasPrefix(absURL, "https://") && absURL != relPath {
		t.Fatalf("absURL 既不是绝对 URL 也不是 relPath：%q", absURL)
	}

	// 附件表登记（recordAttachment 失败不阻断，故只告警不 Fatal；但成功时应能查到）
	if n, e := g.Model("attachments").Where("file_path", relPath).WhereNull("deleted_at").Count(); e == nil {
		t.Logf("附件表登记条数=%d（0 表示未登记，可能库未连通，不影响落盘结论）", n)
	} else {
		t.Logf("附件表查询失败（可能未连库）：%v", e)
	}

	_ = os.Remove(disk)
	t.Logf("relPath=%s absURL=%s disk=%s", relPath, absURL, disk)
}

// countQrcodeFiles 统计图片库 qrcode 子目录下的文件数。
func countQrcodeFiles(t *testing.T) int {
	t.Helper()
	dir := filepath.Join(StaticDir, "qrcode")
	var n int
	_ = filepath.Walk(dir, func(_ string, info os.FileInfo, err error) error {
		if err != nil {
			return nil
		}
		if !info.IsDir() {
			n++
		}
		return nil
	})
	return n
}