package web

import (
	"encoding/json"
	"net/http"
	"net/http/httptest"
	"net/url"
	"strings"
	"testing"
)

// taxonomyPost 按页面 fetch 的方式调「活动与类别」那几个端点，返回解析后的 JSON。
func taxonomyPost(t *testing.T, f *movementFixture, path string, form url.Values) map[string]any {
	t.Helper()
	rec := httptest.NewRecorder()
	req := httptest.NewRequest(http.MethodPost, path, strings.NewReader(form.Encode()))
	req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
	ctx := req.WithContext(f.ctx)
	switch path {
	case "/admin/edit/category/rename":
		f.srv.editCategoryRename(rec, ctx)
	case "/admin/edit/category/clear":
		f.srv.editCategoryClear(rec, ctx)
	case "/admin/edit/activity/merge":
		f.srv.editActivityMerge(rec, ctx)
	default:
		t.Fatalf("未接线的端点 %s", path)
	}
	var out map[string]any
	if err := json.Unmarshal(rec.Body.Bytes(), &out); err != nil {
		t.Fatalf("%s 返回的不是 JSON: %s", path, rec.Body.String())
	}
	return out
}

// 类别没有表，「改名」只能是按值批量改地点；填成另一个已有值就是合并。
func TestCategoryRenameMergesIntoExisting(t *testing.T) {
	f := newMovementFixture(t)
	mk := func(pk int, cat string) {
		f.mustExec(t, `INSERT INTO places (dataset_id, src_pk, name, lat, lon, poi_category)
			VALUES ($1,$2,'类别测试点',30,120,$3)`, f.dsID, pk, cat)
	}
	mk(690001, "测试类别甲")
	mk(690002, "测试类别甲")
	mk(690003, "测试类别乙")
	t.Cleanup(func() {
		f.pg.ExecContext(f.ctx, `DELETE FROM places WHERE dataset_id=$1 AND src_pk BETWEEN 690001 AND 690003`, f.dsID)
	})

	got := taxonomyPost(t, f, "/admin/edit/category/rename", url.Values{
		"from": {"测试类别甲"}, "to": {"测试类别乙"},
	})
	if got["places"] != float64(2) {
		t.Fatalf("应报告 2 个地点被改，实际 %v", got["places"])
	}
	if n := f.scalar(t, `SELECT count(*) FROM places WHERE dataset_id=$1 AND poi_category='测试类别乙'
		AND src_pk BETWEEN 690001 AND 690003`, f.dsID); n != 3 {
		t.Fatalf("甲应整体并进乙（共 3 个），实际 %d 个", n)
	}
	if n := f.scalar(t, `SELECT count(*) FROM places WHERE dataset_id=$1 AND poi_category='测试类别甲'`, f.dsID); n != 0 {
		t.Fatalf("甲应一个不剩，还剩 %d 个", n)
	}
}

// 新类别填空不能当成「改名成空」把类别悄悄丢掉——那是「清空」的活，要另走一步。
func TestCategoryRenameRefusesEmptyTarget(t *testing.T) {
	f := newMovementFixture(t)
	f.mustExec(t, `INSERT INTO places (dataset_id, src_pk, name, lat, lon, poi_category)
		VALUES ($1,690011,'类别测试点',30,120,'测试类别丙')`, f.dsID)
	t.Cleanup(func() {
		f.pg.ExecContext(f.ctx, `DELETE FROM places WHERE dataset_id=$1 AND src_pk=690011`, f.dsID)
	})

	got := taxonomyPost(t, f, "/admin/edit/category/rename", url.Values{"from": {"测试类别丙"}, "to": {"   "}})
	if got["error"] == nil {
		t.Fatal("新类别为空时应报错")
	}
	if n := f.scalar(t, `SELECT count(*) FROM places WHERE dataset_id=$1 AND poi_category='测试类别丙'`, f.dsID); n != 1 {
		t.Fatalf("报错就不该动数据，实际只剩 %d 个", n)
	}
}

// 「清空」= 摘掉类别：那些地点退回未指定（NULL），去重列表里也就不再出现。
func TestCategoryClearDetachesPlaces(t *testing.T) {
	f := newMovementFixture(t)
	for _, pk := range []int{690021, 690022} {
		f.mustExec(t, `INSERT INTO places (dataset_id, src_pk, name, lat, lon, poi_category)
			VALUES ($1,$2,'类别测试点',30,120,'测试类别丁')`, f.dsID, pk)
	}
	t.Cleanup(func() {
		f.pg.ExecContext(f.ctx, `DELETE FROM places WHERE dataset_id=$1 AND src_pk BETWEEN 690021 AND 690022`, f.dsID)
	})

	got := taxonomyPost(t, f, "/admin/edit/category/clear", url.Values{"from": {"测试类别丁"}})
	if got["places"] != float64(2) {
		t.Fatalf("应报告 2 个地点被摘掉，实际 %v", got["places"])
	}
	if n := f.scalar(t, `SELECT count(*) FROM places WHERE dataset_id=$1 AND src_pk BETWEEN 690021 AND 690022
		AND poi_category IS NULL`, f.dsID); n != 2 {
		t.Fatalf("应写 NULL 而不是空串，实际 %d 个", n)
	}
	if n := f.scalar(t, `SELECT count(*) FROM places WHERE dataset_id=$1 AND poi_category='测试类别丁'`, f.dsID); n != 0 {
		t.Fatalf("类别应彻底消失，还剩 %d 个", n)
	}
}

// 合并：源的到访整体转到目标，源活动被删，is_home/is_work 跟着目标走
// （那两个是从活动抄到到访上的冗余标记，不跟着改就会留下一个矛盾的到访）。
func TestActivityMergeMovesVisitsThenDeletesSource(t *testing.T) {
	f := newMovementFixture(t)
	placeID, _ := fillupPlace(t, f)

	mkAct := func(pk int, name string, home bool) int64 {
		var id int64
		if err := f.pg.QueryRowContext(f.ctx, `INSERT INTO activities
			(dataset_id, src_pk, name, color, is_home) VALUES ($1,$2,$3,'',$4) RETURNING id`,
			f.dsID, pk, name, home).Scan(&id); err != nil {
			t.Fatalf("建活动失败: %v", err)
		}
		t.Cleanup(func() { f.pg.ExecContext(f.ctx, `DELETE FROM activities WHERE id=$1`, id) })
		return id
	}
	from := mkAct(695001, "合并测试甲", false)
	to := mkAct(695002, "合并测试乙", true)

	f.mustExec(t, `UPDATE visits SET activity_id=$2, is_home=FALSE WHERE dataset_id=$1 AND place_id=$3`,
		f.dsID, from, placeID)
	if n := f.scalar(t, `SELECT count(*) FROM visits WHERE dataset_id=$1 AND activity_id=$2`, f.dsID, from); n == 0 {
		t.Fatal("合并前应有到访挂在源活动上")
	}

	got := taxonomyPost(t, f, "/admin/edit/activity/merge", url.Values{
		"from": {itoa(from)}, "to": {itoa(to)},
	})
	if got["error"] != nil {
		t.Fatalf("合并不该失败: %v", got["error"])
	}
	if n := f.scalar(t, `SELECT count(*) FROM visits WHERE dataset_id=$1 AND activity_id=$2`, f.dsID, to); n == 0 {
		t.Fatal("到访应转到目标活动名下")
	}
	if n := f.scalar(t, `SELECT count(*) FROM visits WHERE dataset_id=$1 AND is_home`, f.dsID); n == 0 {
		t.Fatal("目标活动的 is_home 应抄到转过来的到访上")
	}
	if n := f.scalar(t, `SELECT count(*) FROM activities WHERE dataset_id=$1 AND id=$2`, f.dsID, from); n != 0 {
		t.Fatal("源活动应被删掉")
	}
	if n := f.scalar(t, `SELECT count(*) FROM entity_raw WHERE dataset_id=$1 AND entity='ZACTIVITY' AND src_pk=695001`,
		f.dsID); n != 0 {
		t.Fatal("源活动的原始行留档应一起清掉，否则导出还会带它")
	}
}

// 合并到自己或选空都必须拒绝，别把活动连同它的到访一起清掉。
func TestActivityMergeRefusesDegenerateInput(t *testing.T) {
	f := newMovementFixture(t)
	var id int64
	if err := f.pg.QueryRowContext(f.ctx, `INSERT INTO activities (dataset_id, src_pk, name, color)
		VALUES ($1,695011,'合并测试丙','') RETURNING id`, f.dsID).Scan(&id); err != nil {
		t.Fatalf("建活动失败: %v", err)
	}
	t.Cleanup(func() { f.pg.ExecContext(f.ctx, `DELETE FROM activities WHERE id=$1`, id) })

	for _, form := range []url.Values{
		{"from": {itoa(id)}, "to": {itoa(id)}},
		{"from": {itoa(id)}, "to": {"0"}},
		{"from": {"0"}, "to": {itoa(id)}},
	} {
		if got := taxonomyPost(t, f, "/admin/edit/activity/merge", form); got["error"] == nil {
			t.Fatalf("%v 应被拒绝", form)
		}
	}
	if n := f.scalar(t, `SELECT count(*) FROM activities WHERE dataset_id=$1 AND id=$2`, f.dsID, id); n != 1 {
		t.Fatal("被拒绝的合并不该动数据")
	}
}