// ============================================================
//  UmiOcrHost:Umi-OCR 本体(网盘 zip)下载/安装/启停/检测 管理器
//  · 本体 zip 直链(用户网盘,~179MB,zip 根直接是 Umi-OCR.exe 无顶层目录)
//  · 安装 = 下载 zip 到临时文件 → 解压到 exe 旁 Umi-OCR\ → 找到主程序验证 → 删 zip
//  · 启动 = Process.Start(主程序,Umi-OCR 启动即带 HTTP 服务 1224 端口) → 轮询端口就绪
//  · 停止 = 只停由本程序启动的实例(CloseMainWindow→超时 Kill),不碰用户自己开的
// ============================================================
using System;
using System.Diagnostics;
using System.IO;
using System.IO.Compression;
using System.Net;
using System.Threading.Tasks;
using System.Windows.Forms;

namespace pdfbt
{
    internal static class UmiOcrHost
    {
        /// <summary>本体 zip 直链(用户网盘直连,GET 才有内容,HEAD 会 404,不支持断点续传)。</summary>
        public const string DownloadUrl = "http://s333.odn.cc/s/845f64a239b665ec";

        private const string InstallDirName = "Umi-OCR";
        private const int HttpPort = 1224;

        /// <summary>由本程序启动的实例(停止时只关它,不碰用户自己开的)。</summary>
        private static Process _startedProc;

        /// <summary>本体安装目录(exe 旁 Umi-OCR\)。</summary>
        public static string InstallDir
        {
            get { return Path.Combine(AppDomain.CurrentDomain.BaseDirectory, InstallDirName); }
        }

        /// <summary>找安装目录里的主程序(Umi-OCR.exe,大小写不敏感;找不到返回 null)。</summary>
        public static string FindExe()
        {
            try
            {
                if (!Directory.Exists(InstallDir)) return null;
                foreach (var f in Directory.GetFiles(InstallDir, "*.exe", SearchOption.TopDirectoryOnly))
                {
                    string name = Path.GetFileName(f);
                    if (name.Equals("Umi-OCR.exe", StringComparison.OrdinalIgnoreCase) ||
                        name.Equals("Umi-OCR.exe", StringComparison.OrdinalIgnoreCase))
                        return f;
                }
                // 目录名不完全一致时兜底:取第一个 exe
                string[] all = Directory.GetFiles(InstallDir, "*.exe", SearchOption.TopDirectoryOnly);
                return all.Length > 0 ? all[0] : null;
            }
            catch { return null; }
        }

        /// <summary>本体是否已安装(目录里能找到主程序)。</summary>
        public static bool IsInstalled
        {
            get { return FindExe() != null; }
        }

        /// <summary>HTTP 服务(1224 端口)是否在监听。</summary>
        public static bool IsHttpUp()
        {
            try
            {
                using (var c = new System.Net.Sockets.TcpClient())
                {
                    var ar = c.BeginConnect("127.0.0.1", HttpPort, null, null);
                    if (ar.AsyncWaitHandle.WaitOne(800))
                    {
                        c.EndConnect(ar);
                        return true;
                    }
                    c.Close();
                }
            }
            catch { }
            return false;
        }

        /// <summary>三态:0=未安装,1=已安装未运行,2=运行中(HTTP 在线)。</summary>
        public static int Detect()
        {
            if (!IsInstalled) return 0;
            return IsHttpUp() ? 2 : 1;
        }

        /// <summary>状态文字(设置窗口显示用)。</summary>
        public static string DetectText()
        {
            int st = Detect();
            if (st == 0) return "未检测到 OCR 本体(点\"下载 OCR 本体\"自动下载安装)";
            if (st == 1) return "已检测到 OCR 本体,服务未运行(识别时会自动启动)";
            return "OCR 本体运行中(HTTP " + HttpPort + " 端口在线)";
        }

        /// <summary>下载 zip → 解压到 Umi-OCR\ → 验证主程序 → 删 zip。
        /// progress: 0~100 下载百分比;解压与收尾阶段回调 100。失败抛异常(内部已清理残留)。
        /// ★ 必须用 DownloadFileAsync:同步版 DownloadFile 不触发 DownloadProgressChanged
        /// (进度事件只在异步路径发,同步下载会一直停在 0%)。</summary>
        public static Task DownloadAndInstallAsync(Action<int> progress)
        {
            return Task.Factory.StartNew(delegate
            {
                string tmpZip = Path.Combine(Path.GetTempPath(), "Umi-OCR_" + Guid.NewGuid().ToString("N") + ".zip");
                try
                {
                    if (progress != null) progress(0);
                    // 异步下载包成同步等待(TaskCompletionSource),进度事件才会触发
                    var dlTcs = new TaskCompletionSource<bool>();
                    using (var wc = new WebClient())
                    {
                        wc.Proxy = null;
                        wc.DownloadProgressChanged += (s, e) =>
                        {
                            if (progress != null) progress(e.ProgressPercentage);
                        };
                        wc.DownloadFileCompleted += (s, e) =>
                        {
                            if (e.Error != null) dlTcs.TrySetException(e.Error);
                            else if (e.Cancelled) dlTcs.TrySetException(new Exception("下载已取消"));
                            else dlTcs.TrySetResult(true);
                        };
                        wc.DownloadFileAsync(new Uri(DownloadUrl), tmpZip);
                    }
                    dlTcs.Task.GetAwaiter().GetResult();   // 等下载结束(异常原样抛出)
                    if (progress != null) progress(100);

                    // 校验是 zip(PK 头)而不是错误页/JSON
                    using (var fs = File.OpenRead(tmpZip))
                    {
                        int b0 = fs.ReadByte(), b1 = fs.ReadByte();
                        if (b0 != 'P' || b1 != 'K')
                            throw new Exception("下载的内容不是有效的压缩包(可能是网盘临时故障),请稍后重试");
                    }

                    // 解压(覆盖);zip 根直接是 Umi-OCR.exe,目标目录即安装目录
                    string dst = InstallDir;
                    if (!Directory.Exists(dst)) Directory.CreateDirectory(dst);
                    using (var archive = ZipFile.OpenRead(tmpZip))
                    {
                        foreach (var entry in archive.Entries)
                        {
                            string target = Path.GetFullPath(Path.Combine(dst, entry.FullName));
                            if (!target.StartsWith(Path.GetFullPath(dst), StringComparison.OrdinalIgnoreCase))
                                continue;   // 防 zip 路径穿越
                            if (string.IsNullOrEmpty(entry.Name))
                            {
                                if (!Directory.Exists(target)) Directory.CreateDirectory(target);
                                continue;
                            }
                            string tdir = Path.GetDirectoryName(target);
                            if (!string.IsNullOrEmpty(tdir) && !Directory.Exists(tdir)) Directory.CreateDirectory(tdir);
                            if (File.Exists(target))
                            {
                                try { File.Delete(target); } catch { }   // 被占用则跳过(旧文件同名多半一致)
                            }
                            entry.ExtractToFile(target, false);
                        }
                    }

                    if (FindExe() == null)
                        throw new Exception("解压完成但未找到 Umi-OCR 主程序,压缩包内容可能不完整");

                    try { File.Delete(tmpZip); } catch { }   // 验证通过后删 zip
                }
                catch
                {
                    try { if (File.Exists(tmpZip)) File.Delete(tmpZip); } catch { }
                    throw;
                }
            });
        }

        /// <summary>确保本体已安装且 HTTP 服务在线:未安装抛错引导下载;未运行则静默启动(--hide 主窗口进托盘)并等端口就绪(≤15秒)。
        /// 返回 false=已安装但启动超时(调用方提示)。</summary>
        public static bool EnsureRunning()
        {
            if (!IsInstalled)
                throw new Exception("尚未安装 Umi-OCR 本体。\r\n请到 OCR 设置窗口选择 Umi-OCR 引擎,点\"下载 OCR 本体\"。");
            if (IsHttpUp()) return true;

            string exe = FindExe();
            // --hide:官方参数,启动后主窗口隐藏到托盘,后台提供 HTTP 服务(用户看到零弹窗)
            _startedProc = Process.Start(new ProcessStartInfo
            {
                FileName = exe,
                Arguments = "--hide",
                WorkingDirectory = Path.GetDirectoryName(exe),
                UseShellExecute = false,
                CreateNoWindow = true,
                WindowStyle = ProcessWindowStyle.Hidden
            });
            for (int i = 0; i < 30; i++)   // 30 × 500ms = 最多 15 秒
            {
                if (IsHttpUp()) return true;
                System.Threading.Thread.Sleep(500);
            }
            return false;
        }

        /// <summary>停止由本程序启动的实例(PdfViewerWindow 关闭时调用;用户自己开的不动)。
        /// --hide 启动的窗口在托盘,CloseMainWindow 不可靠:先发官方 --quit 指令优雅退出,超时再 Kill。</summary>
        public static void StopIfStartedByUs()
        {
            var p = _startedProc;
            _startedProc = null;
            if (p == null) return;
            try
            {
                if (!p.HasExited)
                {
                    // 官方退出指令:向已运行实例发 --quit(依赖其 HTTP 通道)
                    try
                    {
                        Process.Start(new ProcessStartInfo
                        {
                            FileName = FindExe(),
                            Arguments = "--quit",
                            UseShellExecute = false,
                            CreateNoWindow = true
                        });
                    }
                    catch { }
                    if (!p.WaitForExit(3000)) p.Kill();
                }
            }
            catch { }
            finally { try { p.Dispose(); } catch { } }
        }
    }
}