#!/usr/bin/env python3
# -*- coding: utf-8 -*-

from __future__ import print_function

import os
import sys
import shutil
import tempfile
import argparse
import json
import zipfile
import tarfile
import subprocess
import time
import hashlib

try:
    import paramiko
except ImportError:
    print("错误: 请先安装 paramiko")
    print("  pip install paramiko")
    sys.exit(1)

try:
    import requests
    import urllib3
    urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
except ImportError:
    print("错误: 请先安装 requests")
    print("  pip install requests")
    sys.exit(1)


# ==================== 配置 ====================

# 端口统一从环境变量 DEPLOY_PORT 读取，未设置时使用此默认值（需与 Nginx proxy_pass、ecosystem.config.cjs 一致）
DEPLOY_PM2_APP = "soul"
DEFAULT_DEPLOY_PORT = 3006
DEPLOY_PROJECT_PATH = "/www/wwwroot/soul"
DEPLOY_SITE_URL = "https://soul.quwanzhi.com"
# SSH 端口（支持环境变量 DEPLOY_SSH_PORT，未设置时默认为 22022）
DEFAULT_SSH_PORT = int(os.environ.get("DEPLOY_SSH_PORT", "22022"))

def get_cfg():
    """获取基础部署配置（deploy 模式与 devlop 共用 SSH/宝塔）"""
    return {
        "host": os.environ.get("DEPLOY_HOST", "42.194.232.22"),
        "user": os.environ.get("DEPLOY_USER", "root"),
        "password": os.environ.get("DEPLOY_PASSWORD", "Zhiqun1984"),
        "ssh_key": os.environ.get("DEPLOY_SSH_KEY", ""),
        "project_path": os.environ.get("DEPLOY_PROJECT_PATH", DEPLOY_PROJECT_PATH),
        "panel_url": os.environ.get("BAOTA_PANEL_URL", "https://42.194.232.22:9988"),
        "api_key": os.environ.get("BAOTA_API_KEY", "hsAWqFSi0GOCrunhmYdkxy92tBXfqYjd"),
        "pm2_name": os.environ.get("DEPLOY_PM2_APP", DEPLOY_PM2_APP),
        "site_url": os.environ.get("DEPLOY_SITE_URL", DEPLOY_SITE_URL),
        "port": int(os.environ.get("DEPLOY_PORT", str(DEFAULT_DEPLOY_PORT))),
        "node_version": os.environ.get("DEPLOY_NODE_VERSION", "v22.14.0"),
        "node_path": os.environ.get("DEPLOY_NODE_PATH", "/www/server/nodejs/v22.14.0/bin"),
    }


def get_cfg_devlop():
    """devlop 模式配置：在基础配置上增加 base_path / dist / dist2。
    实际运行目录为 dist_path（切换后新版本在 dist），宝塔 PM2 项目路径必须指向 dist_path，
    否则会从错误目录启动导致 .next/static 等静态资源 404。"""
    cfg = get_cfg().copy()
    cfg["base_path"] = os.environ.get("DEVOP_BASE_PATH", "/www/wwwroot/soul")
    cfg["dist_path"] = cfg["base_path"] + "/dist"
    cfg["dist2_path"] = cfg["base_path"] + "/dist2"
    return cfg


# ==================== 宝塔 API ====================

def _get_sign(api_key):
    now_time = int(time.time())
    sign_str = str(now_time) + hashlib.md5(api_key.encode("utf-8")).hexdigest()
    request_token = hashlib.md5(sign_str.encode("utf-8")).hexdigest()
    return now_time, request_token


def _baota_request(panel_url, api_key, path, data=None):
    req_time, req_token = _get_sign(api_key)
    payload = {"request_time": req_time, "request_token": req_token}
    if data:
        payload.update(data)
    url = panel_url.rstrip("/") + "/" + path.lstrip("/")
    try:
        r = requests.post(url, data=payload, verify=False, timeout=30)
        return r.json() if r.text else {}
    except Exception as e:
        print("  API 请求失败: %s" % str(e))
        return None


def get_node_project_list(panel_url, api_key):
    for path in ["/project/nodejs/get_project_list", "/plugin?action=a&name=nodejs&s=get_project_list"]:
        result = _baota_request(panel_url, api_key, path)
        if result and (result.get("status") is True or "data" in result):
            return result.get("data", [])
    return None


def get_node_project_status(panel_url, api_key, pm2_name):
    projects = get_node_project_list(panel_url, api_key)
    if projects:
        for p in projects:
            if p.get("name") == pm2_name:
                return p
    return None


def start_node_project(panel_url, api_key, pm2_name):
    for path in ["/project/nodejs/start_project", "/plugin?action=a&name=nodejs&s=start_project"]:
        result = _baota_request(panel_url, api_key, path, {"project_name": pm2_name})
        if result and (result.get("status") is True or result.get("msg") or "成功" in str(result)):
            print("  [成功] 启动成功: %s" % pm2_name)
            return True
    return False


def stop_node_project(panel_url, api_key, pm2_name):
    for path in ["/project/nodejs/stop_project", "/plugin?action=a&name=nodejs&s=stop_project"]:
        result = _baota_request(panel_url, api_key, path, {"project_name": pm2_name})
        if result and (result.get("status") is True or result.get("msg") or "成功" in str(result)):
            print("  [成功] 停止成功: %s" % pm2_name)
            return True
    return False


def restart_node_project(panel_url, api_key, pm2_name):
    project_status = get_node_project_status(panel_url, api_key, pm2_name)
    if project_status:
        print("  项目状态: %s" % project_status.get("status", "未知"))
    for path in ["/project/nodejs/restart_project", "/plugin?action=a&name=nodejs&s=restart_project"]:
        result = _baota_request(panel_url, api_key, path, {"project_name": pm2_name})
        if result and (result.get("status") is True or result.get("msg") or "成功" in str(result)):
            print("  [成功] 重启成功: %s" % pm2_name)
            return True
        if result and "msg" in result:
            print("  API 返回: %s" % result.get("msg"))
    print("  [警告] 重启失败，请检查宝塔 Node 插件是否安装、API 密钥是否正确")
    return False


def add_or_update_node_project(panel_url, api_key, pm2_name, project_path, port=None, node_path=None):
    if port is None:
        port = int(os.environ.get("DEPLOY_PORT", str(DEFAULT_DEPLOY_PORT)))
    port_env = "PORT=%d " % port
    run_cmd = port_env + ("%s/node server.js" % node_path if node_path else "node server.js")
    payload = {"name": pm2_name, "path": project_path, "run_cmd": run_cmd, "port": str(port)}
    for path in ["/project/nodejs/add_project", "/plugin?action=a&name=nodejs&s=add_project"]:
        result = _baota_request(panel_url, api_key, path, payload)
        if result and result.get("status") is True:
            print("  [成功] 项目配置已更新: %s" % pm2_name)
            return True
        if result and "msg" in result:
            print("  API 返回: %s" % result.get("msg"))
    return False


# ==================== 本地构建 ====================

def run_build(root):
    """执行本地 pnpm build"""
    use_shell = sys.platform == "win32"
    standalone = os.path.join(root, ".next", "standalone")
    server_js = os.path.join(standalone, "server.js")

    try:
        r = subprocess.run(
            ["pnpm", "build"],
            cwd=root,
            shell=use_shell,
            timeout=600,
            capture_output=True,
            text=True,
            encoding="utf-8",
            errors="replace",
        )
        stdout_text = r.stdout or ""
        stderr_text = r.stderr or ""
        combined = stdout_text + stderr_text
        is_windows_symlink_error = (
            sys.platform == "win32"
            and r.returncode != 0
            and ("EPERM" in combined or "symlink" in combined.lower() or "operation not permitted" in combined.lower() or "errno: -4048" in combined)
        )

        if r.returncode != 0:
            if is_windows_symlink_error:
                print("  [警告] Windows 符号链接权限错误（EPERM）")
                print("  解决方案：开启开发者模式 / 以管理员运行 / 或使用 --no-build")
                if os.path.isdir(standalone) and os.path.isfile(server_js):
                    print("  [成功] standalone 输出可用，继续部署")
                    return True
                return False
            print("  [失败] 构建失败，退出码:", r.returncode)
            for line in (stdout_text.strip().split("\n") or [])[-10:]:
                print("    " + line)
            return False
    except subprocess.TimeoutExpired:
        print("  [失败] 构建超时（超过10分钟）")
        return False
    except FileNotFoundError:
        print("  [失败] 未找到 pnpm，请安装: npm install -g pnpm")
        return False
    except Exception as e:
        print("  [失败] 构建异常:", str(e))
        if os.path.isdir(standalone) and os.path.isfile(server_js):
            print("  [提示] 可尝试使用 --no-build 跳过构建")
        return False

    if not os.path.isdir(standalone) or not os.path.isfile(server_js):
        print("  [失败] 未找到 .next/standalone 或 server.js")
        return False
    print("  [成功] 构建完成")
    return True


def clean_standalone_before_build(root, retries=3, delay=2):
    """构建前删除 .next/standalone，避免 Windows EBUSY"""
    standalone = os.path.join(root, ".next", "standalone")
    if not os.path.isdir(standalone):
        return True
    for attempt in range(1, retries + 1):
        try:
            shutil.rmtree(standalone)
            print("  [清理] 已删除 .next/standalone（第 %d 次尝试）" % attempt)
            return True
        except (OSError, PermissionError):
            if attempt < retries:
                print("  [清理] 被占用，%ds 后重试 (%d/%d) ..." % (delay, attempt, retries))
                time.sleep(delay)
            else:
                print("  [失败] 无法删除 .next/standalone，可改用 --no-build")
                return False
    return False


# ==================== 打包（deploy 模式：tar.gz） ====================

def _copy_with_dereference(src, dst):
    if os.path.islink(src):
        link_target = os.readlink(src)
        real_path = link_target if os.path.isabs(link_target) else os.path.join(os.path.dirname(src), link_target)
        if os.path.exists(real_path):
            if os.path.isdir(real_path):
                shutil.copytree(real_path, dst, symlinks=False, dirs_exist_ok=True)
            else:
                shutil.copy2(real_path, dst)
        else:
            shutil.copy2(src, dst, follow_symlinks=False)
    elif os.path.isdir(src):
        if os.path.exists(dst):
            shutil.rmtree(dst)
        shutil.copytree(src, dst, symlinks=False, dirs_exist_ok=True)
    else:
        shutil.copy2(src, dst)


def pack_standalone_tar(root):
    """打包 standalone 为 tar.gz（deploy 模式用）"""
    print("[2/4] 打包 standalone ...")
    standalone = os.path.join(root, ".next", "standalone")
    static_src = os.path.join(root, ".next", "static")
    public_src = os.path.join(root, "public")
    ecosystem_src = os.path.join(root, "ecosystem.config.cjs")

    if not os.path.isdir(standalone) or not os.path.isdir(static_src):
        print("  [失败] 未找到 .next/standalone 或 .next/static")
        return None
    chunks_dir = os.path.join(static_src, "chunks")
    if not os.path.isdir(chunks_dir):
        print("  [失败] .next/static/chunks 不存在，请先完整执行 pnpm build（本地 pnpm start 能正常打开页面后再部署）")
        return None

    staging = tempfile.mkdtemp(prefix="soul_deploy_")
    try:
        for name in os.listdir(standalone):
            _copy_with_dereference(os.path.join(standalone, name), os.path.join(staging, name))
        node_modules_dst = os.path.join(staging, "node_modules")
        pnpm_dir = os.path.join(node_modules_dst, ".pnpm")
        if os.path.isdir(pnpm_dir):
            for dep in ["styled-jsx"]:
                dep_in_root = os.path.join(node_modules_dst, dep)
                if not os.path.exists(dep_in_root):
                    for pnpm_pkg in os.listdir(pnpm_dir):
                        if pnpm_pkg.startswith(dep + "@"):
                            src_dep = os.path.join(pnpm_dir, pnpm_pkg, "node_modules", dep)
                            if os.path.isdir(src_dep):
                                shutil.copytree(src_dep, dep_in_root, symlinks=False, dirs_exist_ok=True)
                                break
        static_dst = os.path.join(staging, ".next", "static")
        if os.path.exists(static_dst):
            shutil.rmtree(static_dst)
        os.makedirs(os.path.dirname(static_dst), exist_ok=True)
        shutil.copytree(static_src, static_dst)
        # 同步构建索引（与 start-standalone.js 一致），避免宝塔上 server 用错导致页面空白/404
        next_root = os.path.join(root, ".next")
        next_staging = os.path.join(staging, ".next")
        index_files = [
            "BUILD_ID",
            "build-manifest.json",
            "app-path-routes-manifest.json",
            "routes-manifest.json",
            "prerender-manifest.json",
            "required-server-files.json",
            "fallback-build-manifest.json",
        ]
        for name in index_files:
            src = os.path.join(next_root, name)
            if os.path.isfile(src):
                shutil.copy2(src, os.path.join(next_staging, name))
        print("  [已同步] 构建索引: BUILD_ID, build-manifest, routes-manifest 等")
        if os.path.isdir(public_src):
            shutil.copytree(public_src, os.path.join(staging, "public"), dirs_exist_ok=True)
        if os.path.isfile(ecosystem_src):
            shutil.copy2(ecosystem_src, os.path.join(staging, "ecosystem.config.cjs"))
        pkg_json = os.path.join(staging, "package.json")
        if os.path.isfile(pkg_json):
            try:
                with open(pkg_json, "r", encoding="utf-8") as f:
                    data = json.load(f)
                data.setdefault("scripts", {})["start"] = "node server.js"
                with open(pkg_json, "w", encoding="utf-8") as f:
                    json.dump(data, f, indent=2, ensure_ascii=False)
            except Exception:
                pass
        tarball = os.path.join(tempfile.gettempdir(), "soul_deploy.tar.gz")
        with tarfile.open(tarball, "w:gz") as tf:
            for name in os.listdir(staging):
                tf.add(os.path.join(staging, name), arcname=name)
        print("  [成功] 打包完成: %s (%.2f MB)" % (tarball, os.path.getsize(tarball) / 1024 / 1024))
        return tarball
    except Exception as e:
        print("  [失败] 打包异常:", str(e))
        return None
    finally:
        shutil.rmtree(staging, ignore_errors=True)


# ==================== Node 环境检查 & SSH 上传（deploy 模式） ====================

def check_node_environments(cfg):
    print("[检查] Node 环境 ...")
    client = paramiko.SSHClient()
    client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    try:
        if cfg.get("ssh_key"):
            client.connect(cfg["host"], port=DEFAULT_SSH_PORT, username=cfg["user"], key_filename=cfg["ssh_key"], timeout=15)
        else:
            client.connect(cfg["host"], port=DEFAULT_SSH_PORT, username=cfg["user"], password=cfg["password"], timeout=15)
        stdin, stdout, stderr = client.exec_command("which node && node -v", timeout=10)
        print("  默认 Node: %s" % (stdout.read().decode("utf-8", errors="replace").strip() or "未找到"))
        node_path = cfg.get("node_path", "/www/server/nodejs/v22.14.0/bin")
        stdin, stdout, stderr = client.exec_command("%s/node -v 2>/dev/null" % node_path, timeout=5)
        print("  配置 Node: %s" % (stdout.read().decode("utf-8", errors="replace").strip() or "不可用"))
        return True
    except Exception as e:
        print("  [警告] %s" % str(e))
        return False
    finally:
        client.close()


def upload_and_extract(cfg, tarball_path):
    """SSH 上传 tar.gz 并解压到 project_path（deploy 模式）"""
    print("[3/4] SSH 上传并解压 ...")
    if not cfg.get("password") and not cfg.get("ssh_key"):
        print("  [失败] 请设置 DEPLOY_PASSWORD 或 DEPLOY_SSH_KEY")
        return False
    client = paramiko.SSHClient()
    client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    try:
        if cfg.get("ssh_key") and os.path.isfile(cfg["ssh_key"]):
            client.connect(cfg["host"], port=DEFAULT_SSH_PORT, username=cfg["user"], key_filename=cfg["ssh_key"], timeout=15)
        else:
            client.connect(cfg["host"], port=DEFAULT_SSH_PORT, username=cfg["user"], password=cfg["password"], timeout=15)
        sftp = client.open_sftp()
        remote_tar = "/tmp/soul_deploy.tar.gz"
        remote_script = "/tmp/soul_deploy_extract.sh"
        sftp.put(tarball_path, remote_tar)
        node_path = cfg.get("node_path", "/www/server/nodejs/v22.14.0/bin")
        project_path = cfg["project_path"]
        script_content = """#!/bin/bash
export PATH=%s:$PATH
cd %s
rm -rf .next public ecosystem.config.cjs server.js package.json 2>/dev/null
tar -xzf %s
rm -f %s
echo OK
""" % (node_path, project_path, remote_tar, remote_tar)
        with sftp.open(remote_script, "w") as f:
            f.write(script_content)
        sftp.close()
        client.exec_command("chmod +x %s" % remote_script, timeout=10)
        stdin, stdout, stderr = client.exec_command("bash %s" % remote_script, timeout=120)
        out = stdout.read().decode("utf-8", errors="replace").strip()
        exit_status = stdout.channel.recv_exit_status()
        if exit_status != 0 or "OK" not in out:
            print("  [失败] 解压失败，退出码:", exit_status)
            return False
        print("  [成功] 解压完成: %s" % project_path)
        return True
    except Exception as e:
        print("  [失败] SSH 错误:", str(e))
        return False
    finally:
        client.close()


def deploy_via_baota_api(cfg):
    """宝塔 API 重启 Node 项目（deploy 模式）"""
    print("[4/4] 宝塔 API 管理 Node 项目 ...")
    panel_url, api_key, pm2_name = cfg["panel_url"], cfg["api_key"], cfg["pm2_name"]
    project_path = cfg["project_path"]
    node_path = cfg.get("node_path", "/www/server/nodejs/v22.14.0/bin")
    port = cfg["port"]

    if not get_node_project_status(panel_url, api_key, pm2_name):
        add_or_update_node_project(panel_url, api_key, pm2_name, project_path, port, node_path)
    stop_node_project(panel_url, api_key, pm2_name)
    time.sleep(2)
    ok = restart_node_project(panel_url, api_key, pm2_name)
    if not ok:
        ok = start_node_project(panel_url, api_key, pm2_name)
    if not ok:
        print("  请到宝塔 Node 项目手动重启 %s，路径: %s" % (pm2_name, project_path))
    return ok


# ==================== 打包（devlop 模式：zip） ====================

ZIP_EXCLUDE_DIRS = {".cache", "__pycache__", ".git", "node_modules", "cache", "test", "tests", "coverage", ".nyc_output", ".turbo", "开发文档"}
ZIP_EXCLUDE_FILE_NAMES = {".DS_Store", "Thumbs.db"}
ZIP_EXCLUDE_FILE_SUFFIXES = (".log", ".map")


def _should_exclude_from_zip(arcname, is_file=True):
    parts = arcname.replace("\\", "/").split("/")
    for part in parts:
        if part in ZIP_EXCLUDE_DIRS:
            return True
    if is_file and parts:
        name = parts[-1]
        if name in ZIP_EXCLUDE_FILE_NAMES or any(name.endswith(s) for s in ZIP_EXCLUDE_FILE_SUFFIXES):
            return True
    return False


def pack_standalone_zip(root):
    """打包 standalone 为 zip（devlop 模式用）"""
    print("[2/7] 打包 standalone 为 zip ...")
    standalone = os.path.join(root, ".next", "standalone")
    static_src = os.path.join(root, ".next", "static")
    public_src = os.path.join(root, "public")
    ecosystem_src = os.path.join(root, "ecosystem.config.cjs")

    if not os.path.isdir(standalone) or not os.path.isdir(static_src):
        print("  [失败] 未找到 .next/standalone 或 .next/static")
        return None
    chunks_dir = os.path.join(static_src, "chunks")
    if not os.path.isdir(chunks_dir):
        print("  [失败] .next/static/chunks 不存在，请先完整执行 pnpm build（本地 pnpm start 能正常打开页面后再部署）")
        return None

    staging = tempfile.mkdtemp(prefix="soul_devlop_")
    try:
        for name in os.listdir(standalone):
            _copy_with_dereference(os.path.join(standalone, name), os.path.join(staging, name))
        node_modules_dst = os.path.join(staging, "node_modules")
        pnpm_dir = os.path.join(node_modules_dst, ".pnpm")
        if os.path.isdir(pnpm_dir):
            for dep in ["styled-jsx"]:
                dep_in_root = os.path.join(node_modules_dst, dep)
                if not os.path.exists(dep_in_root):
                    for pnpm_pkg in os.listdir(pnpm_dir):
                        if pnpm_pkg.startswith(dep + "@"):
                            src_dep = os.path.join(pnpm_dir, pnpm_pkg, "node_modules", dep)
                            if os.path.isdir(src_dep):
                                shutil.copytree(src_dep, dep_in_root, symlinks=False, dirs_exist_ok=True)
                                break
        os.makedirs(os.path.join(staging, ".next"), exist_ok=True)
        shutil.copytree(static_src, os.path.join(staging, ".next", "static"), dirs_exist_ok=True)
        # 同步构建索引（与 start-standalone.js 一致），避免宝塔上 server 用错导致页面空白/404
        next_root = os.path.join(root, ".next")
        next_staging = os.path.join(staging, ".next")
        index_files = [
            "BUILD_ID",
            "build-manifest.json",
            "app-path-routes-manifest.json",
            "routes-manifest.json",
            "prerender-manifest.json",
            "required-server-files.json",
            "fallback-build-manifest.json",
        ]
        for name in index_files:
            src = os.path.join(next_root, name)
            if os.path.isfile(src):
                shutil.copy2(src, os.path.join(next_staging, name))
        print("  [已同步] 构建索引: BUILD_ID, build-manifest, routes-manifest 等")
        if os.path.isdir(public_src):
            shutil.copytree(public_src, os.path.join(staging, "public"), dirs_exist_ok=True)
        if os.path.isfile(ecosystem_src):
            shutil.copy2(ecosystem_src, os.path.join(staging, "ecosystem.config.cjs"))
        pkg_json = os.path.join(staging, "package.json")
        if os.path.isfile(pkg_json):
            try:
                with open(pkg_json, "r", encoding="utf-8") as f:
                    data = json.load(f)
                data.setdefault("scripts", {})["start"] = "node server.js"
                with open(pkg_json, "w", encoding="utf-8") as f:
                    json.dump(data, f, indent=2, ensure_ascii=False)
            except Exception:
                pass
        server_js = os.path.join(staging, "server.js")
        if os.path.isfile(server_js):
            try:
                deploy_port = int(os.environ.get("DEPLOY_PORT", str(DEFAULT_DEPLOY_PORT)))
                with open(server_js, "r", encoding="utf-8") as f:
                    c = f.read()
                if "|| 3000" in c:
                    with open(server_js, "w", encoding="utf-8") as f:
                        f.write(c.replace("|| 3000", "|| %d" % deploy_port))
            except Exception:
                pass
        zip_path = os.path.join(tempfile.gettempdir(), "soul_devlop.zip")
        with zipfile.ZipFile(zip_path, "w", zipfile.ZIP_DEFLATED) as zf:
            for name in os.listdir(staging):
                path = os.path.join(staging, name)
                if os.path.isfile(path):
                    if not _should_exclude_from_zip(name):
                        zf.write(path, name)
                else:
                    for dirpath, dirs, filenames in os.walk(path):
                        dirs[:] = [d for d in dirs if not _should_exclude_from_zip(os.path.join(name, os.path.relpath(os.path.join(dirpath, d), path)), is_file=False)]
                        for f in filenames:
                            full = os.path.join(dirpath, f)
                            arcname = os.path.join(name, os.path.relpath(full, path))
                            if not _should_exclude_from_zip(arcname):
                                zf.write(full, arcname)
        print("  [成功] 打包完成: %s (%.2f MB)" % (zip_path, os.path.getsize(zip_path) / 1024 / 1024))
        return zip_path
    except Exception as e:
        print("  [失败] 打包异常:", str(e))
        return None
    finally:
        shutil.rmtree(staging, ignore_errors=True)


def upload_zip_and_extract_to_dist2(cfg, zip_path):
    """上传 zip 并解压到 dist2（devlop 模式）"""
    print("[3/7] SSH 上传 zip 并解压到 dist2 ...")
    sys.stdout.flush()
    if not cfg.get("password") and not cfg.get("ssh_key"):
        print("  [失败] 请设置 DEPLOY_PASSWORD 或 DEPLOY_SSH_KEY")
        return False
    zip_size_mb = os.path.getsize(zip_path) / (1024 * 1024)
    client = paramiko.SSHClient()
    client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    try:
        print("  正在连接 %s@%s:%s ..." % (cfg["user"], cfg["host"], DEFAULT_SSH_PORT))
        sys.stdout.flush()
        if cfg.get("ssh_key") and os.path.isfile(cfg["ssh_key"]):
            client.connect(cfg["host"], port=DEFAULT_SSH_PORT, username=cfg["user"], key_filename=cfg["ssh_key"], timeout=30, banner_timeout=30)
        else:
            client.connect(cfg["host"], port=DEFAULT_SSH_PORT, username=cfg["user"], password=cfg["password"], timeout=30, banner_timeout=30)
        print("  [OK] SSH 已连接，正在上传 zip（%.1f MB）..." % zip_size_mb)
        sys.stdout.flush()
        remote_zip = cfg["base_path"].rstrip("/") + "/soul_devlop.zip"
        sftp = client.open_sftp()
        # 上传进度：每 5MB 打印一次
        chunk_mb = 5.0
        last_reported = [0]

        def _progress(transferred, total):
            if total and total > 0:
                now_mb = transferred / (1024 * 1024)
                if now_mb - last_reported[0] >= chunk_mb or transferred >= total:
                    last_reported[0] = now_mb
                    print("\r  上传进度: %.1f / %.1f MB" % (now_mb, total / (1024 * 1024)), end="")
                    sys.stdout.flush()

        sftp.put(zip_path, remote_zip, callback=_progress)
        if zip_size_mb >= chunk_mb:
            print("")
        print("  [OK] zip 已上传，正在服务器解压（约 1–3 分钟）...")
        sys.stdout.flush()
        sftp.close()
        dist2 = cfg["dist2_path"]
        cmd = "rm -rf %s && mkdir -p %s && unzip -o -q %s -d %s && rm -f %s && echo OK" % (dist2, dist2, remote_zip, dist2, remote_zip)
        stdin, stdout, stderr = client.exec_command(cmd, timeout=300)
        out = stdout.read().decode("utf-8", errors="replace").strip()
        err = stderr.read().decode("utf-8", errors="replace").strip()
        if err:
            print("  服务器 stderr: %s" % err[:500])
        exit_status = stdout.channel.recv_exit_status()
        if exit_status != 0 or "OK" not in out:
            print("  [失败] 解压失败，退出码: %s" % exit_status)
            if out:
                print("  stdout: %s" % out[:300])
            return False
        print("  [成功] 已解压到: %s" % dist2)
        return True
    except Exception as e:
        print("  [失败] SSH 错误: %s" % str(e))
        import traceback
        traceback.print_exc()
        return False
    finally:
        client.close()


def run_pnpm_install_in_dist2(cfg):
    """服务器 dist2 内执行 pnpm install，阻塞等待完成后再返回（改目录前必须完成）"""
    print("[4/7] 服务器 dist2 内执行 pnpm install（等待完成后再切换目录）...")
    sys.stdout.flush()
    client = paramiko.SSHClient()
    client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    try:
        if cfg.get("ssh_key") and os.path.isfile(cfg["ssh_key"]):
            client.connect(cfg["host"], port=DEFAULT_SSH_PORT, username=cfg["user"], key_filename=cfg["ssh_key"], timeout=15)
        else:
            client.connect(cfg["host"], port=DEFAULT_SSH_PORT, username=cfg["user"], password=cfg["password"], timeout=15)
        stdin, stdout, stderr = client.exec_command("bash -lc 'which pnpm'", timeout=10)
        pnpm_path = stdout.read().decode("utf-8", errors="replace").strip()
        if not pnpm_path:
            return False, "未找到 pnpm，请服务器安装: npm install -g pnpm"
        cmd = "bash -lc 'cd %s && %s install'" % (cfg["dist2_path"], pnpm_path)
        stdin, stdout, stderr = client.exec_command(cmd, timeout=300)
        out = stdout.read().decode("utf-8", errors="replace").strip()
        err = stderr.read().decode("utf-8", errors="replace").strip()
        if stdout.channel.recv_exit_status() != 0:
            return False, "pnpm install 失败\n" + (err or out)
        print("  [成功] dist2 内 pnpm install 已执行完成，可安全切换目录")
        return True, None
    except Exception as e:
        return False, str(e)
    finally:
        client.close()


def remote_swap_dist_and_restart(cfg):
    """暂停 → dist→dist1, dist2→dist → 删除 dist1 → 更新 PM2 项目路径 → 重启（devlop 模式）"""
    print("[5/7] 宝塔 API 暂停 Node 项目 ...")
    stop_node_project(cfg["panel_url"], cfg["api_key"], cfg["pm2_name"])
    time.sleep(2)
    print("[6/7] 服务器切换目录: dist→dist1, dist2→dist ...")
    client = paramiko.SSHClient()
    client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    try:
        if cfg.get("ssh_key") and os.path.isfile(cfg["ssh_key"]):
            client.connect(cfg["host"], port=DEFAULT_SSH_PORT, username=cfg["user"], key_filename=cfg["ssh_key"], timeout=15)
        else:
            client.connect(cfg["host"], port=DEFAULT_SSH_PORT, username=cfg["user"], password=cfg["password"], timeout=15)
        cmd = "cd %s && mv dist dist1 2>/dev/null; mv dist2 dist && rm -rf dist1 && echo OK" % cfg["base_path"]
        stdin, stdout, stderr = client.exec_command(cmd, timeout=60)
        out = stdout.read().decode("utf-8", errors="replace").strip()
        if stdout.channel.recv_exit_status() != 0 or "OK" not in out:
            print("  [失败] 切换失败")
            return False
        print("  [成功] 新版本位于 %s" % cfg["dist_path"])
    finally:
        client.close()
    # 关键：devlop 实际运行目录是 dist_path，必须让宝塔 PM2 从该目录启动，否则会从错误目录跑导致静态资源 404
    print("[7/7] 更新宝塔 Node 项目路径并重启 ...")
    add_or_update_node_project(
        cfg["panel_url"], cfg["api_key"], cfg["pm2_name"],
        cfg["dist_path"],  # 使用 dist_path，不是 project_path
        port=cfg["port"],
        node_path=cfg.get("node_path"),
    )
    if not start_node_project(cfg["panel_url"], cfg["api_key"], cfg["pm2_name"]):
        print("  [警告] 请到宝塔手动启动 %s，并确认项目路径为: %s" % (cfg["pm2_name"], cfg["dist_path"]))
        return False
    return True


# ==================== 主函数 ====================

def main():
    parser = argparse.ArgumentParser(description="Soul 创业派对 - 统一部署脚本", formatter_class=argparse.RawDescriptionHelpFormatter, epilog=__doc__)
    parser.add_argument("--mode", choices=["devlop", "deploy"], default="devlop", help="devlop=dist切换(默认), deploy=直接覆盖")
    parser.add_argument("--no-build", action="store_true", help="跳过本地构建")
    parser.add_argument("--no-upload", action="store_true", help="仅 deploy 模式：跳过 SSH 上传")
    parser.add_argument("--no-api", action="store_true", help="仅 deploy 模式：上传后不调宝塔 API")
    args = parser.parse_args()

    script_dir = os.path.dirname(os.path.abspath(__file__))
    # 支持 devlop.py 在项目根或 scripts/ 下：以含 package.json 的目录为 root
    if os.path.isfile(os.path.join(script_dir, "package.json")):
        root = script_dir
    else:
        root = os.path.dirname(script_dir)

    if args.mode == "devlop":
        cfg = get_cfg_devlop()
        print("=" * 60)
        print("  Soul 自动部署（dist 切换）")
        print("=" * 60)
        print("  服务器: %s@%s  目录: %s  Node: %s" % (cfg["user"], cfg["host"], cfg["base_path"], cfg["pm2_name"]))
        print("=" * 60)
        if not args.no_build:
            print("[1/7] 本地构建 pnpm build ...")
            if sys.platform == "win32" and not clean_standalone_before_build(root):
                return 1
            if not run_build(root):
                return 1
        elif not os.path.isfile(os.path.join(root, ".next", "standalone", "server.js")):
            print("[错误] 未找到 .next/standalone/server.js")
            return 1
        else:
            print("[1/7] 跳过本地构建")
        zip_path = pack_standalone_zip(root)
        if not zip_path:
            return 1
        if not upload_zip_and_extract_to_dist2(cfg, zip_path):
            return 1
        try:
            os.remove(zip_path)
        except Exception:
            pass
        # 必须在 dist2 内 pnpm install 执行完成后再切换目录
        ok, err = run_pnpm_install_in_dist2(cfg)
        if not ok:
            print("  [失败] %s" % (err or "pnpm install 失败"))
            return 1
        # install 已完成，再执行 dist→dist1、dist2→dist 切换
        if not remote_swap_dist_and_restart(cfg):
            return 1
        print("")
        print("  部署完成！运行目录: %s" % cfg["dist_path"])
        return 0

    # deploy 模式
    cfg = get_cfg()
    print("=" * 60)
    print("  Soul 一键部署（直接覆盖）")
    print("=" * 60)
    print("  服务器: %s@%s  项目路径: %s  PM2: %s" % (cfg["user"], cfg["host"], cfg["project_path"], cfg["pm2_name"]))
    print("=" * 60)
    if not args.no_upload:
        check_node_environments(cfg)
    if not args.no_build:
        print("[1/4] 本地构建 ...")
        if not run_build(root):
            return 1
    elif not os.path.isfile(os.path.join(root, ".next", "standalone", "server.js")):
        print("[错误] 未找到 .next/standalone/server.js")
        return 1
    else:
        print("[1/4] 跳过本地构建")
    tarball = pack_standalone_tar(root)
    if not tarball:
        return 1
    if not args.no_upload:
        if not upload_and_extract(cfg, tarball):
            return 1
        try:
            os.remove(tarball)
        except Exception:
            pass
    else:
        print("  压缩包: %s" % tarball)
    if not args.no_api and not args.no_upload:
        deploy_via_baota_api(cfg)
    print("")
    print("  部署完成！站点: %s" % cfg["site_url"])
    return 0


if __name__ == "__main__":
    sys.exit(main())