using System;
using System.IO;
using System.Text;
using System.Text.RegularExpressions;
using UnityEditor;
using UnityEditor.Build;
using UnityEditor.Build.Reporting;
using UnityEngine;

namespace ProjSG.Editor
{
    /// <summary>
    /// 抖音小游戏（TTSDK 6.8.0）构建后自动修复。
    ///
    /// 背景：
    /// 1. 本项目固定 CDN 模式（dataLoadType: 0），首包数据部署在 DATA_CDN 上。
    ///    但 TTSDK 构建工具在生成 tt-minigame/tt-minigame/game.js 时，
    ///    会把 loadDataPackageFromSubpackage 写成 true（无论模板/dataLoadType 如何），
    ///    导致引擎去空的 data-package 子包找首包数据，报 "processDataJob: unknown data format"。
    /// 2. TTSDK 生成的子包 wasmcode/game.js 与 data-package/game.js 是 0 字节空文件，
    ///    真机会报 "not loaded pkg:game.js"，必须补成 module.exports = {};。
    ///
    /// 用法：
    /// - 构建 WebGL（抖音小游戏）完成后自动执行（IPostprocessBuildWithReport + 延迟重试）。
    /// - 也可手动执行菜单 Tools/抖音小游戏/修复 game.js。
    /// </summary>
    public class DouyinBuildPostFixer : IPostprocessBuildWithReport
    {
        public int callbackOrder => 9999; // 尽量晚执行

        private static readonly UTF8Encoding Utf8NoBom = new UTF8Encoding(false);
        private const int MaxRetry = 20; // 最多重试 20 次（每次间隔 1 帧），应对转换异步的情况

        // ---------------------------------------------------------------
        // 自动 Hook
        // ---------------------------------------------------------------
        public void OnPostprocessBuild(BuildReport report)
        {
            bool isMiniGame = report.summary.platform == BuildTarget.WebGL;
            if (!isMiniGame)
            {
                var outputPath = report.summary.outputPath ?? string.Empty;
                isMiniGame = outputPath.IndexOf("tt-minigame", StringComparison.OrdinalIgnoreCase) >= 0;
            }

            if (!isMiniGame)
                return;

            // TTSDK 转小游戏（生成 game.js）发生在 Unity WebGL 构建之后，
            // 且是同步的，所以把修复延后到下一帧，确保产物已落盘。
            EditorApplication.delayCall += () => PatchWithRetry(0);
        }

        [MenuItem("Tools/抖音小游戏/修复 game.js（构建后自动/手动）")]
        public static void PatchMenu()
        {
            PatchWithRetry(0);
        }

        // ---------------------------------------------------------------
        // 修复入口
        // ---------------------------------------------------------------
        private static void PatchWithRetry(int attempt)
        {
            var miniRoot = GetMiniGameRoot();

            // 主 game.js 还没生成（构建尚未完成/转换异步），稍后重试
            if (!File.Exists(Path.Combine(miniRoot, "game.js")))
            {
                if (attempt < MaxRetry)
                {
                    var next = attempt + 1;
                    EditorApplication.delayCall += () => PatchWithRetry(next);
                }
                else
                {
                    Debug.LogWarning("[DouyinBuildPostFixer] 未找到构建产物 " + miniRoot +
                                     "，请确认已用 TTSDK 打包，或手动执行菜单 Tools/抖音小游戏/修复 game.js。");
                }
                return;
            }

            try
            {
                PatchAll(miniRoot);
            }
            catch (Exception e)
            {
                Debug.LogError("[DouyinBuildPostFixer] 修复失败: " + e);
            }
        }

        // ---------------------------------------------------------------
        // 实际修复逻辑
        // ---------------------------------------------------------------
        public static void PatchAll(string miniRoot)
        {
            PatchMainGameJs(Path.Combine(miniRoot, "game.js"));
            PatchSubpackageGameJs(Path.Combine(miniRoot, "wasmcode", "game.js"), "wasmcode");
            PatchSubpackageGameJs(Path.Combine(miniRoot, "data-package", "game.js"), "data-package");
            // wasm 分包（开发者工具「启用代码分包」）会生成 .dy-game-cache 缓存，里面的 wasmcode 子包
            // 可能缺 game.js，导致真机预览报 "not loaded pkg:game.js"，同样补上。
            PatchSubpackageGameJs(Path.Combine(miniRoot, ".dy-game-cache", "wasmcode", "game.js"), ".dy-game-cache/wasmcode");
            PatchGameJson(Path.Combine(miniRoot, "game.json"));
            PatchLoadingImages(miniRoot);
            EnsureXingwanSdkFile(miniRoot);
            EnsureXingwanImport(Path.Combine(miniRoot, "game.js"));
            Debug.Log("[DouyinBuildPostFixer] 构建产物 game.js/game.json 修复完成。");
        }

        /// <summary>
        /// 主 game.js：loadDataPackageFromSubpackage 必须为 false（CDN 模式）。
        /// </summary>
        private static void PatchMainGameJs(string path)
        {
            if (!File.Exists(path))
            {
                Debug.LogWarning("[DouyinBuildPostFixer] 主 game.js 不存在: " + path);
                return;
            }

            string text = File.ReadAllText(path); // 自动识别 BOM
            string newText = Regex.Replace(
                text,
                @"loadDataPackageFromSubpackage\s*:\s*true",
                "loadDataPackageFromSubpackage: false");

            if (newText == text)
            {
                Debug.Log("[DouyinBuildPostFixer] 主 game.js 的 loadDataPackageFromSubpackage 已是 false，无需修改。");
                return;
            }

            File.WriteAllText(path, newText, Utf8NoBom);
            Debug.Log("[DouyinBuildPostFixer] 已修复主 game.js：loadDataPackageFromSubpackage true → false。");
        }

        /// <summary>
        /// 子包 game.js：0 字节 → module.exports = {};（已有内容则跳过，幂等）。
        /// </summary>
        private static void PatchSubpackageGameJs(string path, string subName)
        {
            Directory.CreateDirectory(Path.GetDirectoryName(path) ?? string.Empty);

            if (File.Exists(path) && new FileInfo(path).Length > 0)
            {
                Debug.Log("[DouyinBuildPostFixer] 子包 " + subName + "/game.js 已有内容，跳过。");
                return;
            }

            File.WriteAllText(path, "module.exports = {};\n", Utf8NoBom);
            Debug.Log("[DouyinBuildPostFixer] 已补全子包 " + subName + "/game.js：module.exports = {};");
        }

        /// <summary>
        /// game.json：补 networkTimeout，提高 tt.request 超时（抖音默认 60s，慢 CDN 下会超时）。
        /// </summary>
        private static void PatchGameJson(string path)
        {
            if (!File.Exists(path))
            {
                Debug.LogWarning("[DouyinBuildPostFixer] game.json 不存在: " + path);
                return;
            }

            string text = File.ReadAllText(path);
            if (text.Contains("\"networkTimeout\""))
            {
                Debug.Log("[DouyinBuildPostFixer] game.json 已有 networkTimeout，跳过。");
                return;
            }

            const string block =
                "  \"networkTimeout\": {\n" +
                "    \"request\": 300000,\n" +
                "    \"connectSocket\": 300000,\n" +
                "    \"uploadFile\": 300000,\n" +
                "    \"downloadFile\": 300000\n" +
                "  },\n";

            // 在 "subpackages" 前插入，若找不到则插到 "plugins" 前，再不行插到第一个 { 后。
            int idx = text.IndexOf("\"subpackages\"", StringComparison.Ordinal);
            if (idx < 0) idx = text.IndexOf("\"plugins\"", StringComparison.Ordinal);
            if (idx < 0)
            {
                idx = text.IndexOf('{');
                if (idx >= 0) idx += 1;
            }
            if (idx < 0)
            {
                Debug.LogWarning("[DouyinBuildPostFixer] game.json 结构异常，跳过 networkTimeout 补丁。");
                return;
            }

            text = text.Insert(idx, block);
            File.WriteAllText(path, text, Utf8NoBom);
            Debug.Log("[DouyinBuildPostFixer] 已为 game.json 补 networkTimeout（request/downloadFile = 300000ms）。");
        }

        /// <summary>
        /// 加载页图片：用工程内 Assets/Editor/DouyinLoading 下的自定义图覆盖产物 images 目录，
        /// 不修改 DefaultTemplate。background.png 为背景图、logo.png 为 LOGO 图标
        /// （产物里叫 unity_logo.png，保持文件名不变，game.js 无需改动）。源图不存在则跳过。
        /// </summary>
        private static void PatchLoadingImages(string miniRoot)
        {
            var sourceDir = Path.Combine(Application.dataPath, "Editor", "DouyinLoading");
            var targetDir = Path.Combine(miniRoot, "images");

            CopyIfExists(Path.Combine(sourceDir, "background.png"), Path.Combine(targetDir, "background.png"));
            CopyIfExists(Path.Combine(sourceDir, "logo.png"), Path.Combine(targetDir, "unity_logo.png"));
        }

        /// <summary>
        /// 星玩渠道 SDK：确保小游戏根目录存在 xingwanTTGameSdk.js。
        /// 构建工具可能未把 DefaultTemplate 下的自定义 js 拷进产物，这里兜底复制。
        /// </summary>
        private static void EnsureXingwanSdkFile(string miniRoot)
        {
            const string fileName = "xingwanTTGameSdk.js";
            var target = Path.Combine(miniRoot, fileName);
            if (File.Exists(target) && new FileInfo(target).Length > 0)
                return;

            var source = Path.Combine(Application.dataPath, "Plugins", "ByteGame",
                "com.bytedance.starksdk", "DefaultTemplate", fileName);
            if (!File.Exists(source))
            {
                Debug.LogWarning("[DouyinBuildPostFixer] 未找到星玩 SDK 源文件: " + source);
                return;
            }

            Directory.CreateDirectory(miniRoot);
            File.Copy(source, target, true);
            Debug.Log("[DouyinBuildPostFixer] 已复制星玩 SDK 到产物目录: " + fileName);
        }

        /// <summary>
        /// 星玩渠道 SDK：确保产物 game.js 已 import 并把 SDK 挂到全局作用域。
        /// 模板被还原/覆盖时自动补回，避免 jslib 找不到 SDK。
        /// </summary>
        private static void EnsureXingwanImport(string gameJsPath)
        {
            if (!File.Exists(gameJsPath))
                return;

            string text = File.ReadAllText(gameJsPath);
            if (text.Contains("xingwanTTGameSdk"))
                return;

            const string block =
                "// 星玩渠道 SDK 注入（构建后自动补回）\n" +
                "try {\n" +
                "  const xingwanModule = require('./xingwanTTGameSdk.js');\n" +
                "  const xingwanTTGameSdk = (xingwanModule && xingwanModule.default) || xingwanModule;\n" +
                "  GameGlobal.xingwanTTGameSdk = xingwanTTGameSdk;\n" +
                "  if (typeof globalThis !== 'undefined') globalThis.xingwanTTGameSdk = xingwanTTGameSdk;\n" +
                "  if (typeof window !== 'undefined') window.xingwanTTGameSdk = xingwanTTGameSdk;\n" +
                "} catch (e) {\n" +
                "  console.error('[Xingwan] 星玩 SDK 注入失败: ' + (e && e.message ? e.message : e));\n" +
                "}\n";

            File.WriteAllText(gameJsPath, block + text, Utf8NoBom);
            Debug.Log("[DouyinBuildPostFixer] 已为 game.js 补回星玩 SDK 注入代码。");
        }

        private static void CopyIfExists(string source, string target)
        {
            if (!File.Exists(source))
                return;

            Directory.CreateDirectory(Path.GetDirectoryName(target) ?? string.Empty);
            File.Copy(source, target, true);
            Debug.Log("[DouyinBuildPostFixer] 已覆盖加载页图片: " + Path.GetFileName(target) + " <- " + source);
        }

        // ---------------------------------------------------------------
        // 路径解析
        // ---------------------------------------------------------------
        private static string GetMiniGameRoot()
        {
            string outputRoot = ReadStarkOutputDir();
            if (string.IsNullOrEmpty(outputRoot))
            {
                outputRoot = Path.GetFullPath(Path.Combine(Application.dataPath, "..", "tt-minigame"));
            }

            return Path.Combine(outputRoot, "tt-minigame");
        }

        /// <summary>
        /// 从 Assets/Editor/StarkBuilderSetting.asset 读取 OutputDir，
        /// 避免写死路径；读取失败时回退到 工程根/tt-minigame。
        /// </summary>
        private static string ReadStarkOutputDir()
        {
            string settingPath = Path.Combine(Application.dataPath, "Editor", "StarkBuilderSetting.asset");
            if (!File.Exists(settingPath))
                return null;

            string txt = File.ReadAllText(settingPath);
            var m = Regex.Match(txt, @"^OutputDir:\s*(.+)$", RegexOptions.Multiline);
            if (!m.Success)
                return null;

            string dir = m.Groups[1].Value.Trim();
            return string.IsNullOrEmpty(dir) ? null : dir;
        }
    }
}