import os
import re
import argparse
import shutil
from datetime import datetime

# ====================== 【配置区】 ======================
LOG_FILE = "./rename_log.txt"

# 需要完全跳过的目录名称（basename，大小写敏感）
SKIP_DIRS = [
    "@eaDir",          # 群晖隐含目录
]

# 需要跳过的隐含文件名称（完整文件名匹配）
SKIP_FILES = {
    ".DS_Store",
    "Thumbs.db",
    "desktop.ini",
}

# 文件清洗规则
RULES_FILE = [
    (r"^[A-Za-z] ", ""),
    (r"\[.*?\]", ""),
    (r"【.*?】", ""),
    (r"丨", ""),
    (r"(?<!\w)(2160p|1080p|720p|HDR|DV|DoVi|HEVC|x265|x264|DDP|AAC|Atmos|TrueHD|BD|BluRay|BLURAY|WEBRip|WEB-DL|HD|4K|中英双语|双语字幕|双语|中文|国语|粤语|台语|韩语|中英双字|国语中字|繁体中字|简体中字|简繁中字|双语中字|官方中字|特效中字)(?!\w)", "", re.IGNORECASE),
    (r"\.+", ""),
    (r"\s", ""),
    (r"[^\u4e00-\u9fa5a-zA-Z0-9().:+\-]", ""),
]

# 文件夹清洗规则
RULES_DIR = [
    (r"【.*?】", ""),             # 删除中文全角【】以及内部内容
    (r"^\[(.*)\]$", r"\1"),       # 整串被[]包裹，只剥外层
    (r"\]\[", ""),                # 清除中间碎片 ][
    (r"^[A-Za-z] ", ""),
    (r"丨", ""),
    (r"(?<!\w)(2160p|1080p|720p|HDR|DV|DoVi|HEVC|x265|x264|DDP|AAC|Atmos|TrueHD|BD|BluRay|BLURAY|WEBRip|WEB-DL|HD|4K|中英双语|双语字幕|双语|中文|国语|粤语|台语|韩语|中英双字|国语中字|繁体中字|简体中字|简繁中字|双语中字|官方中字|特效中字)(?!\w)", "", re.IGNORECASE),
    (r"\.+", ""),
    (r"\s", ""),
    (r"\(\)", ""),                # 清除英文空括号 ()
    (r"（）", ""),                # 清除中文全角空括号 （）
    (r"[^\u4e00-\u9fa5a-zA-Z0-9().:+\-[\]]", ""), # 允许文件夹内部保留[]
]

#完整全部媒体后缀（不带--strm-mode时使用）
FULL_ALLOW_EXT = {".mkv", ".mp4", ".mov", ".srt", ".ass", ".sub", ".strm", ".flv", ".avi", ".wmv", ".ts", ".rmvb", ".webm", ".mp3", ".flac", ".aac", ".wav", ".ogg", ".m4a", ".wma", ".alac", ".rm", ".iso"}
# strm模式：仅strm+字幕
STRM_ONLY_EXT = {".strm", ".srt", ".ass", ".sub"}

MAX_FILENAME_LEN = 245   # mac文件名最大安全长度，预留扩展名位置
# =======================================================

def full2half(s):
    result = []
    for char in s:
        code = ord(char)
        if code == 12288:
            char = chr(32)
        elif 65281 <= code <= 65374:
            char = chr(code - 65248)
        result.append(char)
    return "".join(result)

def clean_file_name(name: str) -> str:
    name_no_ext, ext = os.path.splitext(name)
    name_no_ext = full2half(name_no_ext)
    for item in RULES_FILE:
        if len(item) == 3:
            pat, repl, flag = item
            name_no_ext = re.sub(pat, repl, name_no_ext, flags=flag)
        else:
            pat, repl = item
            name_no_ext = re.sub(pat, repl, name_no_ext)
    name_no_ext = name_no_ext.strip()
    if len(name_no_ext) > MAX_FILENAME_LEN:
        name_no_ext = name_no_ext[:MAX_FILENAME_LEN]
    return name_no_ext + ext

def clean_dir_name(name: str) -> str:
    name = full2half(name)
    year_match = re.search(r"^\[(\d{4})\]$", name)
    year_str = ""
    if year_match:
        year_str = year_match.group(1)

    for item in RULES_DIR:
        if len(item) == 3:
            pat, repl, flag = item
            name = re.sub(pat, repl, name, flags=flag)
        else:
            pat, repl = item
            name = re.sub(pat, repl, name)
    name = name.strip()
    if year_str and name:
        name = f"{name}({year_str})"
    if len(name) > MAX_FILENAME_LEN:
        name = name[:MAX_FILENAME_LEN]
    return name

def get_quality_tag(filename:str):
    m = re.search(r"(?<!\w)(2160p|1080p|720p)(?!\w)", filename, re.IGNORECASE)
    if m:
        return m.group(1).lower()
    return None

def resolve_conflict(dst_dir:str, base_name:str, ext:str, orig_filename:str):
    candidate = base_name + ext
    dst_path = os.path.join(dst_dir, candidate)
    if not os.path.exists(dst_path):
        return candidate
    qual = get_quality_tag(orig_filename)
    if qual:
        candidate = f"{base_name}({qual}){ext}"
        dst_path = os.path.join(dst_dir, candidate)
        if not os.path.exists(dst_path):
            return candidate
    idx = 1
    while True:
        candidate = f"{base_name}({idx}){ext}"
        dst_path = os.path.join(dst_dir, candidate)
        if not os.path.exists(dst_path):
            return candidate
        idx +=1

def write_log(msg: str, log_file: str):
    now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
    line = f"[{now}] {msg}"
    print(line)
    try:
        with open(log_file, "a", encoding="utf-8") as f:
            f.write(line + "\n")
            f.flush()
    except Exception:
        #日志写入失败降级控制台输出，不打断程序
        print(f"[LOG-WRITE-FAILED] {line}")

def main():
    parser = argparse.ArgumentParser(description="媒体文件名镜像清洗工具，源目录只读，生成镜像副本")
    parser.add_argument("--dir", required=True, help="原始源媒体根目录")
    parser.add_argument("--output", required=True, help="输出镜像根目录")
    parser.add_argument("--apply", action="store_true", help="加上此参数才执行真实复制生成镜像")
    parser.add_argument("--dry-run", action="store_true", help="预览模式（默认就是预览，该参数为显式简写）")
    parser.add_argument("--strm-mode", action="store_true", help="开启strm模式：仅处理 .strm .srt .ass .sub；过滤mkv/mp4等视频文件")
    parser.add_argument("--log", default="./rename_log.txt", help="日志文件路径，默认 ./rename_log.txt")
    args = parser.parse_args()

    target_dir = args.dir
    out_dir = args.output
    log_path = args.log

    DRY_RUN = True
    if args.apply:
        DRY_RUN = False
    if args.dry_run:
        DRY_RUN = True

    if args.strm_mode:
        ALLOW_EXT = STRM_ONLY_EXT
        mode_info = "STRM模式【仅strm+字幕】"
    else:
        ALLOW_EXT = FULL_ALLOW_EXT
        mode_info = "完整媒体模式【全部媒体后缀】"

    if not os.path.isdir(target_dir):
        print(f"错误：源目录不存在 -> {target_dir}")
        return

    #初始化日志
    try:
        with open(log_path, "w", encoding="utf-8") as f:
            f.flush()
    except Exception as e:
        print(f"警告：无法初始化日志文件 {log_path}，后续仅控制台输出日志，异常：{e}")

    write_log("===== 文件名+文件夹清洗【修复参数符号，过滤BD/HD/4K/双语/国语等标记】 =====", log_path)
    write_log(f"源目录：{target_dir}", log_path)
    write_log(f"输出镜像目录：{out_dir}", log_path)
    write_log(f"运行模式：{mode_info}", log_path)
    write_log(f"预览模式 DRY_RUN = {DRY_RUN}", log_path)
    if SKIP_DIRS:
        write_log(f"配置跳过目录名列表：{SKIP_DIRS}", log_path)
    write_log(f"过滤隐含系统文件：{SKIP_FILES}", log_path)

    stat = {
        "scan_dir_total":0,
        "scan_file_total":0,
        "media_file_total":0,
        "file_success":0,
        "file_fail":0,
        "dir_success":0,
        "dir_fail":0
    }

    try:
        for root, dirs, files in os.walk(target_dir, topdown=False):
            #过滤跳过目录：@eaDir、配置SKIP_DIRS
            remove_idx = []
            for idx, dname in enumerate(dirs):
                if dname in SKIP_DIRS:
                    remove_idx.append(idx)
            for idx in reversed(remove_idx):
                del dirs[idx]

            stat["scan_dir_total"] += 1
            rel = os.path.relpath(root, target_dir)
            if rel == ".":
                dst_current_dir = out_dir
            else:
                src_basename = os.path.basename(root)
                new_dir_name = clean_dir_name(src_basename)
                if len(new_dir_name) == 0:
                    write_log(f"【警告】文件夹清洗后名称为空，使用原名保留：{src_basename}", log_path)
                    new_dir_name = src_basename
                dst_current_dir = os.path.join(out_dir, rel.replace(src_basename, new_dir_name))

            # 创建输出目录
            if not DRY_RUN:
                try:
                    os.makedirs(dst_current_dir, exist_ok=True)
                    stat["dir_success"] +=1
                except OSError as e:
                    stat["dir_fail"] +=1
                    write_log(f"【目录创建失败】{dst_current_dir} | {e}", log_path)

            # 处理本目录全部文件
            for fname in files:
                if fname in SKIP_FILES:
                    continue
                stat["scan_file_total"] +=1
                ext = os.path.splitext(fname)[1].lower()
                if ext in ALLOW_EXT:
                    stat["media_file_total"] += 1
                    new_name = clean_file_name(fname)
                    old_path = os.path.join(root, fname)
                    base, ext_out = os.path.splitext(new_name)
                    final_fname = resolve_conflict(dst_current_dir, base, ext_out, fname)
                    dst_file = os.path.join(dst_current_dir, final_fname)

                    if final_fname != fname:
                        write_log(f"待修改文件-> {fname} → {final_fname}", log_path)

                    if not DRY_RUN:
                        try:
                            shutil.copy2(old_path, dst_file)
                            stat["file_success"] +=1
                        except OSError as e:
                            stat["file_fail"] +=1
                            write_log(f"【文件复制失败】src:{old_path} dst:{dst_file} | 错误：{e}", log_path)

            # 处理子文件夹，只打印名称发生变更的文件夹
            for dname in dirs:
                new_name = clean_dir_name(dname)
                if len(new_name) == 0:
                    write_log(f"【警告】文件夹清洗后名称为空，跳过改名：{dname}", log_path)
                    continue
                if new_name != dname:
                    write_log(f"待修改文件夹-> {dname} → {new_name}", log_path)

    except OSError as walk_err:
        write_log(f"【严重】遍历目录发生IO异常，部分目录未扫描：{walk_err}", log_path)

    write_log("----------任务统计汇总----------", log_path)
    write_log(f"总扫描目录：{stat['scan_dir_total']}", log_path)
    write_log(f"总扫描文件：{stat['scan_file_total']}，媒体文件：{stat['media_file_total']}", log_path)
    write_log(f"目录：成功创建 {stat['dir_success']}，失败 {stat['dir_fail']}", log_path)
    write_log(f"文件：成功复制 {stat['file_success']}，失败 {stat['file_fail']}", log_path)
    write_log("============================================================", log_path)

if __name__ == "__main__":
    main()