package web

import (
	"database/sql"
	"fmt"
	"os"
	"path/filepath"
	"strconv"
	"strings"
	"testing"
	"time"

	"rond-with-you/internal/rond"
)

// ---------- 图片挂载恢复 ----------
//
// 守的是这么一件事：图片的「认地点」靠坐标键、与数据集无关，但早期 `place_images.dataset_id`
// 上带 ON DELETE CASCADE ⇒ 删掉旧数据集时行被级联删掉，**文件却留在磁盘上**。
// 2026-09-30 实测：数据集 9 被删后 place_images 从 N 行变 0 行，4 个 .webp 还在盘上。
// 这三条用例分别守：文件能挂回来、历史对象键布局也认得出、以及「以后不会再丢」。

// imageURLs 读某个地点在页面上会看到哪些图片（与 placeImagesOf 同一口径）。
func (f *movementFixture) imageURLs(t *testing.T, srcPK int) []string {
	t.Helper()
	imgs, err := f.srv.placeImagesOf(f.ctx, f.dsID, srcPK)
	if err != nil {
		t.Fatalf("读地点图片失败: %v", err)
	}
	var out []string
	for _, im := range imgs {
		out = append(out, im.URL)
	}
	return out
}

// isolateImageState 让这一轮用例独占「这个地点的图片」这份状态。
//
// 必须隔离，两处都要：
//  1. **盘上的文件**：media 目录是共用的，而图片按坐标键跨数据集读，目录里只要有一张
//     「坐标相同」的图（上一轮跑剩的、别的用例留下的）就会被算成这个地点的图——
//     没隔离时它报「应当 2 张、实际 10 张」。
//  2. **库里的行**：`recoverPlaceImages` 按 object_key 判重、`placeImagesOf` 按 place_key 读，
//     都不看数据集；库里留着上一轮的行时，新文件会被当成「已经在库里」直接跳过——
//     实测过 `Already:1` 让兜底分支根本没被执行到，用例却只看「挂回 1 张」就差点放过。
//
// 隔离不了的部分（别的坐标的残留）不影响这些断言；本轮插的行归本数据集，夹具收尾会清掉。
func isolateImageState(t *testing.T, f *movementFixture, lat, lon float64) string {
	t.Helper()
	f.cfg.DataDir = t.TempDir()
	key := rond.PlaceCoordKey(lat, lon)
	if _, err := f.pg.ExecContext(f.ctx, `DELETE FROM place_images WHERE place_key=$1`, key); err != nil {
		t.Fatalf("清理历史图片行失败: %v", err)
	}
	t.Cleanup(func() {
		f.pg.ExecContext(f.ctx, `DELETE FROM place_images WHERE place_key=$1`, key)
	})
	return key
}

// fakeWebP 造一段最小的 WebP 字节：只需要 RIFF....WEBP 头骗过 isWebP（不真的解码）。
func fakeWebP(seed byte, n int) []byte {
	if n < 32 {
		n = 32
	}
	b := make([]byte, n)
	copy(b, "RIFF")
	copy(b[8:], "WEBP")
	for i := 12; i < n; i++ {
		b[i] = seed
	}
	return b
}

// pickPlaceWithCoords 挑一个有坐标的地点（图片只能挂到有坐标的地点上）。
func pickPlaceWithCoords(t *testing.T, f *movementFixture) (srcPK int, name string, lat, lon float64) {
	t.Helper()
	if err := f.pg.QueryRowContext(f.ctx, `SELECT src_pk, COALESCE(name,''), lat, lon FROM places
		WHERE dataset_id=$1 AND lat IS NOT NULL AND lon IS NOT NULL AND (lat<>0 OR lon<>0)
		ORDER BY src_pk LIMIT 1`, f.dsID).Scan(&srcPK, &name, &lat, &lon); err != nil {
		t.Fatalf("取地点失败: %v", err)
	}
	return
}

// imageRowCount 数某个 key 在库里的行数。
func imageRows(t *testing.T, f *movementFixture, objectKey string) int {
	t.Helper()
	var n int
	if err := f.pg.QueryRowContext(f.ctx, `SELECT count(*) FROM place_images WHERE object_key=$1`,
		objectKey).Scan(&n); err != nil {
		t.Fatalf("查图片行失败: %v", err)
	}
	return n
}

// saveTwoImages 按正常路径给一个地点存两张图，返回它们的对象键。
func saveTwoImages(t *testing.T, f *movementFixture, srcPK int) []string {
	t.Helper()
	var keys []string
	for i, seed := range []byte{0x11, 0x22} {
		_, url, err := f.srv.savePlaceImage(f.ctx, f.dsID, srcPK, fakeWebP(seed, 64+i), 4, 4)
		if err != nil {
			t.Fatalf("存第 %d 张图失败: %v", i+1, err)
		}
		key := strings.TrimPrefix(url, "/media/")
		if key == "" || key == url {
			t.Fatalf("本地存储应当返回 /media/<key>，实际 %q", url)
		}
		if _, err := os.Stat(filepath.Join(f.cfg.MediaDir(), filepath.FromSlash(key))); err != nil {
			t.Fatalf("图片文件没落盘（%s）: %v", key, err)
		}
		keys = append(keys, key)
	}
	return keys
}

// TestRecoverPlaceImagesReattachesFilesOnDisk 行没了、文件还在 ⇒ 按坐标挂回来。
func TestRecoverPlaceImagesReattachesFilesOnDisk(t *testing.T) {
	f := newMovementFixture(t)
	ctx, dsID, srv := f.ctx, f.dsID, f.srv
	srcPK, name, lat, lon := pickPlaceWithCoords(t, f)
	wantKey := isolateImageState(t, f, lat, lon)
	keys := saveTwoImages(t, f, srcPK)

	// 模拟早期行为：删掉数据集时行被级联带走（文件留在盘上）
	if _, err := f.pg.ExecContext(ctx, `DELETE FROM place_images WHERE dataset_id=$1`, dsID); err != nil {
		t.Fatalf("清理图片行失败: %v", err)
	}
	for _, k := range keys {
		if _, err := os.Stat(filepath.Join(f.cfg.MediaDir(), filepath.FromSlash(k))); err != nil {
			t.Fatalf("用例前提不成立：文件应当还留在盘上（%s）: %v", k, err)
		}
	}

	// 预演不许写库
	rep, err := srv.recoverPlaceImages(ctx, false)
	if err != nil {
		t.Fatalf("预演失败: %v", err)
	}
	if rep.Attached != 2 {
		t.Errorf("预演应当报 2 张可挂回，实际 %d（认不出 %d）", rep.Attached, len(rep.Unresolved))
	}
	if imageRows(t, f, keys[0]) != 0 {
		t.Error("预演就把行写进去了")
	}

	rep, err = srv.recoverPlaceImages(ctx, true)
	if err != nil {
		t.Fatalf("恢复失败: %v", err)
	}
	if rep.Attached != 2 {
		t.Errorf("应当挂回 2 张，实际 %d", rep.Attached)
	}
	for _, k := range keys {
		if imageRows(t, f, k) != 1 {
			t.Errorf("%s 没有挂回（或挂重了）", k)
		}
	}
	// 挂在当前数据集上、地点对得上、坐标键正确：这三条决定它能不能在页面上显示
	var gotPK int
	var gotKey, gotStorage string
	if err := f.pg.QueryRowContext(ctx, `SELECT src_pk, place_key, storage FROM place_images
		WHERE object_key=$1`, keys[0]).Scan(&gotPK, &gotKey, &gotStorage); err != nil {
		t.Fatal(err)
	}
	if gotPK != srcPK || gotKey != wantKey || gotStorage != "local" {
		t.Errorf("挂回的行不对：src_pk=%d（想 %d）place_key=%q（想 %q）storage=%q",
			gotPK, srcPK, gotKey, wantKey, gotStorage)
	}
	// 页面口径（按坐标键跨数据集读）必须能看到这两张图
	urls := f.imageURLs(t, srcPK)
	for _, k := range keys {
		if !containsStr(urls, "/media/"+k) {
			t.Errorf("地点「%s」的图片里没有挂回的 %s（实际 %v）", name, k, urls)
		}
	}
	if len(urls) != 2 {
		t.Errorf("地点「%s」应当正好 2 张图，实际 %d：%v", name, len(urls), urls)
	}

	// 幂等：再跑一次不该重复插
	rep, err = srv.recoverPlaceImages(ctx, true)
	if err != nil {
		t.Fatalf("重复恢复失败: %v", err)
	}
	if rep.Attached != 0 || rep.Already < 2 {
		t.Errorf("重复执行应当全部跳过，实际新挂 %d、已在库 %d", rep.Attached, rep.Already)
	}
	for _, k := range keys {
		if imageRows(t, f, k) != 1 {
			t.Errorf("重复执行把 %s 插重了", k)
		}
	}
}

// TestRecoverPlaceImagesUnderstandsLegacyKeyLayout 早期对象键是
// `<数据集>/<地点号>/<哈希>.webp`（中间那层目录就是地点号），也要认得出。
func TestRecoverPlaceImagesUnderstandsLegacyKeyLayout(t *testing.T) {
	for _, tc := range []struct {
		name string
		key  string
		ds   int
		pk   int
		ok   bool
	}{
		{"现在的写法", "9/202609/309-042beb5e786707bb.webp", 9, 309, true},
		{"早期写法", "9/286/ebe4914ed6d18f6b626d.webp", 9, 286, true},
		{"地点号刚好 6 位（早期写法）", "9/100001/beef.webp", 9, 100001, true},
		{"看不出数据集", "x/202609/1-aa.webp", 0, 0, false},
		{"层数不对", "9/309.webp", 0, 0, false},
		{"文件名没有地点号前缀、中间也不是数字", "9/xx/aa.webp", 0, 0, false},
	} {
		ds, pk, ok := parseImageKey(tc.key)
		if ok != tc.ok || (ok && (ds != tc.ds || pk != tc.pk)) {
			t.Errorf("parseImageKey(%q) = (%d,%d,%v)，想 (%d,%d,%v)", tc.key, ds, pk, ok, tc.ds, tc.pk, tc.ok)
		}
	}

	// 真数据：把文件改名成早期布局，仍然挂得回去
	f := newMovementFixture(t)
	ctx, dsID, srv := f.ctx, f.dsID, f.srv
	srcPK, _, lat, lon := pickPlaceWithCoords(t, f)
	isolateImageState(t, f, lat, lon)
	keys := saveTwoImages(t, f, srcPK)
	if _, err := f.pg.ExecContext(ctx, `DELETE FROM place_images WHERE dataset_id=$1`, dsID); err != nil {
		t.Fatal(err)
	}
	legacy := filepath.ToSlash(filepath.Join(strconv.Itoa(int(dsID)), strconv.Itoa(srcPK), fmt.Sprintf("abc123-%d.webp", time.Now().UnixNano())))
	if err := os.MkdirAll(filepath.Dir(filepath.Join(f.cfg.MediaDir(), filepath.FromSlash(legacy))), 0o755); err != nil {
		t.Fatal(err)
	}
	old := filepath.Join(f.cfg.MediaDir(), filepath.FromSlash(keys[0]))
	if err := os.Rename(old, filepath.Join(f.cfg.MediaDir(), filepath.FromSlash(legacy))); err != nil {
		t.Fatal(err)
	}

	rep, err := srv.recoverPlaceImages(ctx, true)
	if err != nil {
		t.Fatalf("恢复失败: %v", err)
	}
	if rep.Attached != 2 {
		t.Errorf("两个文件（含早期布局那个）都该挂回，实际 %d（认不出 %d）", rep.Attached, len(rep.Unresolved))
	}
	if imageRows(t, f, legacy) != 1 {
		t.Error("早期布局那个文件没挂回")
	}
	urls := f.imageURLs(t, srcPK)
	if !containsStr(urls, "/media/"+legacy) || !containsStr(urls, "/media/"+keys[1]) {
		t.Errorf("早期布局与现在布局的文件都该看得见，实际 %v", urls)
	}
	if len(urls) != 2 {
		t.Errorf("应当正好 2 张图，实际 %d：%v", len(urls), urls)
	}
}

// TestRecoverPlaceImagesFallsBackToBaseline 数据集已经被删（places 行没了）时，
// 靠它留下的底座库认回地点——这正是「换基线后图片看不见」的那个场景。
func TestRecoverPlaceImagesFallsBackToBaseline(t *testing.T) {
	f := newMovementFixture(t)
	ctx, dsID, srv := f.ctx, f.dsID, f.srv
	srcPK, _, lat, lon := pickPlaceWithCoords(t, f)
	isolateImageState(t, f, lat, lon)
	keys := saveTwoImages(t, f, srcPK)
	if _, err := f.pg.ExecContext(ctx, `DELETE FROM place_images WHERE dataset_id=$1`, dsID); err != nil {
		t.Fatal(err)
	}

	// 造一个「已经被删掉的数据集」的对象键：编号库里没有，只有一份底座库认得它
	const ghost = 987654
	ghostBase := srv.baselinePath(ghost)
	if err := os.MkdirAll(filepath.Dir(ghostBase), 0o755); err != nil {
		t.Fatal(err)
	}
	writeGhostBaseline(t, ghostBase, srcPK, lat, lon)
	t.Cleanup(func() { os.Remove(ghostBase) })

	key := filepath.ToSlash(filepath.Join(strconv.Itoa(ghost), strconv.Itoa(srcPK), fmt.Sprintf("deadbeef-%d.webp", time.Now().UnixNano())))
	dst := filepath.Join(f.cfg.MediaDir(), filepath.FromSlash(key))
	if err := os.MkdirAll(filepath.Dir(dst), 0o755); err != nil {
		t.Fatal(err)
	}
	if err := os.Rename(filepath.Join(f.cfg.MediaDir(), filepath.FromSlash(keys[0])), dst); err != nil {
		t.Fatal(err)
	}

	rep, err := srv.recoverPlaceImages(ctx, true)
	if err != nil {
		t.Fatalf("恢复失败: %v", err)
	}
	if imageRows(t, f, key) != 1 {
		t.Fatalf("没靠底座库认回地点（认不出 %d 张，理由：%v）", len(rep.Unresolved), rep.Unresolved)
	}
	var gotKey string
	if err := f.pg.QueryRowContext(ctx, `SELECT place_key FROM place_images WHERE object_key=$1`, key).
		Scan(&gotKey); err != nil {
		t.Fatal(err)
	}
	if gotKey != placeKeyOfCoords(lat, lon) {
		t.Errorf("认出的地点不对：%q", gotKey)
	}
	if !strings.Contains(strings.Join(rep.Notes, " "), "底座库") {
		t.Errorf("应当说明这份坐标是从底座库来的，实际 %+v", rep)
	}
}

// TestDeleteDatasetKeepsImageRows 防复发：删数据集不再带走图片行。
//
// 外键去掉之前，删数据集会把 place_images 连带删掉、文件却留在盘上——两套口径打架。
// 现在图片只认坐标键，dataset_id 退化成「出处」这一列。
func TestDeleteDatasetKeepsImageRows(t *testing.T) {
	f := newMovementFixture(t)
	ctx, dsID := f.ctx, f.dsID
	srcPK, _, lat, lon := pickPlaceWithCoords(t, f)
	isolateImageState(t, f, lat, lon)
	keys := saveTwoImages(t, f, srcPK)

	if _, err := f.pg.ExecContext(ctx, `DELETE FROM datasets WHERE id=$1`, dsID); err != nil {
		t.Fatalf("删数据集失败: %v", err)
	}
	for _, k := range keys {
		if imageRows(t, f, k) != 1 {
			t.Errorf("删数据集把图片行 %s 带走了（外键没摘干净？）", k)
		}
		if _, err := os.Stat(filepath.Join(f.cfg.MediaDir(), filepath.FromSlash(k))); err != nil {
			t.Errorf("文件也不该被删: %v", err)
		}
	}
}

// writeGhostBaseline 造一份只有 ZLOCATION 的最小底座库（模拟「数据集已删、底座还在」）。
func writeGhostBaseline(t *testing.T, path string, srcPK int, lat, lon float64) {
	t.Helper()
	db, err := sql.Open("sqlite", "file:"+filepath.ToSlash(path))
	if err != nil {
		t.Fatal(err)
	}
	defer db.Close()
	if _, err := db.Exec(`CREATE TABLE ZLOCATION (
		Z_PK INTEGER PRIMARY KEY, ZLATITUDE REAL, ZLONGITUDE REAL, ZNAME_ TEXT)`); err != nil {
		t.Fatal(err)
	}
	if _, err := db.Exec(`INSERT INTO ZLOCATION (Z_PK, ZLATITUDE, ZLONGITUDE, ZNAME_)
		VALUES (?,?,?,'旧地点')`, srcPK, lat, lon); err != nil {
		t.Fatal(err)
	}
}

// placeKeyOfCoords 与生产同口径的坐标键（图片「认地点」靠的就是它）。
func placeKeyOfCoords(lat, lon float64) string {
	return rond.PlaceCoordKey(lat, lon)
}