// pvfrawroundtrip 验证：把 Script.pvf 里某个 type-1 文件按**原始字节**原样写回
// （RepackRaw），产物是否仍能被正常读取。这是"改客户端 PVF"是否安全的前置检查。
//
//	go run ./cmd/scratch/pvfrawroundtrip -pvf <in.pvf> -file <path> -out <out.pvf>
package main

import (
	"flag"
	"fmt"
	"os"

	platformpvf "longheng.io/server/internal/platform/pvf"
)

func main() {
	pvfPath := flag.String("pvf", "", "源 Script.pvf")
	file := flag.String("file", "", "目标文件（PVF 相对路径）")
	outPath := flag.String("out", "", "输出 PVF")
	mode := flag.String("mode", "identity", "identity=原样写回；zeroitems=把奖励物品引用清零")
	flag.Parse()

	archive, err := platformpvf.Open(*pvfPath)
	if err != nil {
		fmt.Fprintln(os.Stderr, "open:", err)
		os.Exit(1)
	}

	raw, err := archive.ReadRaw(*file)
	if err != nil {
		fmt.Fprintln(os.Stderr, "read raw:", err)
		os.Exit(1)
	}
	fmt.Printf("原始 raw %d 字节\n", len(raw))

	patched := make([]byte, len(raw))
	copy(patched, raw)

	if *mode == "zeroitems" {
		// 把奖励物品名的字符串引用整组清零：把 4 字节引用换成 0，
		// 使客户端把该奖励位解析为"空"。
		//
		// 已实测：这样做**不会**破坏 type-1 令牌流（产物内该文件仍可正常解码，
		// 行数与原文一致），只有被清零的位置变成 0。
		n := 0
		for i := 0; i+3 < len(patched); i++ {
			if patched[i+1] == 0xE8 && patched[i+2] == 0x34 && patched[i+3] == 0x1D {
				patched[i] = 0
				patched[i+1] = 0
				patched[i+2] = 0
				patched[i+3] = 0
				n++
				i += 3
			}
		}
		fmt.Printf("已清零 %d 个索引引用\n", n)
	}

	repacked, err := archive.RepackRaw([]platformpvf.RawReplacement{{Path: *file, Data: patched}})
	if err != nil {
		fmt.Fprintln(os.Stderr, "repack:", err)
		os.Exit(1)
	}
	if err := os.WriteFile(*outPath, repacked, 0o644); err != nil {
		fmt.Fprintln(os.Stderr, "write:", err)
		os.Exit(1)
	}
	fmt.Printf("已写出 %s (%d 字节)\n", *outPath, len(repacked))

	// 复核：重新打开产物并读同一个文件
	verify, err := platformpvf.Open(*outPath)
	if err != nil {
		fmt.Fprintln(os.Stderr, "!! 产物无法打开:", err)
		os.Exit(1)
	}
	got, err := verify.ReadRaw(*file)
	if err != nil {
		fmt.Fprintln(os.Stderr, "!! 产物内读该文件失败:", err)
		os.Exit(1)
	}
	fmt.Printf("复核：产物内该文件 %d 字节，与写入一致=%v\n", len(got), len(got) == len(patched))

	txt, err := verify.ReadText(*file)
	if err != nil {
		fmt.Fprintf(os.Stderr, "!! 产物内解码该文件失败（说明令牌流被破坏）: %v\n", err)
		os.Exit(1)
	}
	lines := 0
	for _, c := range txt {
		if c == '\n' {
			lines++
		}
	}
	fmt.Printf("复核：解码得到 %d 字符 / %d 行\n", len([]rune(txt)), lines)
	if lines >= 15 {
		fmt.Println("=> 解码行数正常，归档未被破坏")
	} else {
		fmt.Println("=> 解码行数异常（原文 21 行），归档可能已被破坏")
	}
}