# -*- coding: utf-8 -*-
"""重新生成部署包——在主项目目录下运行：python _mkpkg.py [--public]

- 默认：生成 owner 私有部署包（含 data/site.db、data/secret.key、cookie.txt，
  供 owner 自己的服务器迁移使用，严禁对外分发）
- --public：生成可对外分发的干净包（剔除全部敏感数据——数据库/密钥/网易云 Cookie/
  日志），包内另附 _pkg_ok.txt 标记

源/目标路径基于脚本自身位置推导，不再写死原作者桌面绝对路径。
"""
import glob
import os
import shutil
import sys

SRC = os.path.dirname(os.path.abspath(__file__))
DST = os.path.join(os.path.dirname(SRC), 'NeteaseUrl-部署包')

public = '--public' in sys.argv[1:]

ignore = shutil.ignore_patterns(
    'venv*', '__pycache__', 'downloads', 'release*', '*.log', 'music_api.log*',
    '_smoke_test_downloader.py', 'backups', '云村小助手-网站开关.bat',
    '_audit_work*',
    # .zcode 为 AI 工具工作目录（内部规划/会话文件），不得进任何分发包
    '.zcode*',
    # Cookie 备份文件（music_api.qr_client._backup_cookie 产生，含真实 MUSIC_U）：
    # 'cookie.*.txt' 匹配 cookie.prewrite/backup/logout.*.txt 但不匹配 cookie.txt 本身
    'cookie.*.txt', 'user_cookie.*.txt')

if os.path.exists(DST):
    shutil.rmtree(DST)
shutil.copytree(SRC, DST, ignore=ignore)

# 双保险：显式删除目标里可能漏网的敏感文件（万一 ignore 模式被绕过）。
# 含轮转日志 music_api.log.1/2/3——fnmatch 的 *.log 不匹配它们，曾会随包外泄（含 IP/UA）
for _pat in ('cookie.*.txt', 'user_cookie.*.txt', 'music_api.log*'):
    for _p in glob.glob(os.path.join(DST, _pat)):
        try:
            os.remove(_p)
        except OSError:
            pass

if public:
    # 公开包必须剔除的敏感数据：站点库/密钥、网易云 Cookie、运行日志，以及私有包标记文件
    for rel in ('data', 'cookie.txt', 'user_cookie.txt', 'music_api.log',
                '_pkg_ok.txt',):
        p = os.path.join(DST, rel)
        try:
            if os.path.isdir(p):
                shutil.rmtree(p)
            elif os.path.exists(p):
                os.remove(p)
        except OSError:
            pass
    # 公开包同样剔除轮转日志与 Cookie 备份（双保险与上面同集）
    for _pat in ('music_api.log*', 'cookie.*.txt', 'user_cookie.*.txt'):
        for _p in glob.glob(os.path.join(DST, _pat)):
            try:
                os.remove(_p)
            except OSError:
                pass
    marker = 'public-clean package (no secrets)'
else:
    # 私有包严禁携带《README_公开包说明.txt》——它宣称
    # "不含任何敏感数据/由 --public 生成"，而私有包含 cookie.txt/site.db/secret.key，
    # 接手运维按该 README 认知会整包外发造成泄露。
    # 私有包写入显式警告文件替代。
    _p = os.path.join(DST, 'README_公开包说明.txt')
    if os.path.exists(_p):
        try:
            os.remove(_p)
        except OSError:
            pass
    with open(os.path.join(DST, 'README_私有包警告.txt'), 'w', encoding='utf-8') as _f:
        _f.write(
            '⚠️ 私有部署包——严禁外传、严禁上传网盘/代码仓库/发给任何人。\n'
            '本包包含全部敏感数据：cookie.txt（网易云 SVIP 登录态）、data/site.db（用户库）、\n'
            'data/secret.key（会话与音频签名根）。\n'
            '一旦泄露，请立即轮换 data/secret.key（全部存量会话与音频直链随即失效）、\n'
            '吊销 cookie.txt 中的网易云登录态，并核查 data/site.db 是否已被第三方获取。\n'
            '（本文件由 _mkpkg.py 私有模式自动生成；对外分发包由 --public 生成并剔除全部敏感物。）\n')
    marker = None

n = 0
sz = 0
for r, _, fs in os.walk(DST):
    for f in fs:
        n += 1
        sz += os.path.getsize(os.path.join(r, f))
with open(os.path.join(SRC, '_pkg_ok.txt'), 'w', encoding='utf-8') as f:
    f.write(marker if marker else f'{n} files, {sz // 1024} KB')
shutil.copy2(os.path.join(SRC, '_pkg_ok.txt'), os.path.join(DST, '_pkg_ok.txt'))
print('ok:', 'public' if public else 'private', '-', n, 'files')