userscript_localization_tool.py

import argparse
import os
import json
import re
import threading
from urllib.parse import urlencode
from urllib.request import urlopen
from searcher import search_in_file
import subprocess

# 正则表达式匹配中文字符
chinese_pattern = re.compile(r'[\u4e00-\u9fff]+')

# 全局翻译缓存字典
translation_cache = {
    "复刻": ("Fork", False),
    "问题": ("issues", False),
    # 可以继续添加其他常见的翻译
}


def insert_into_meta(file_path, content):
    with open(file_path, 'r', encoding='utf-8') as file:
        lines = file.readlines()
    if len(lines) < 1:
        lines.append("\n")
    lines.insert(1, content + "\n")

    with open(file_path, 'w', encoding='utf-8') as file:
        file.writelines(lines)
    print(f"区域化已插入 '{file_path}' 的元数据下")


# 删除中文
def remove_zh_cn_lines(file_path):
    try:
        with open(file_path, 'r', encoding='utf-8') as file:
            lines = file.readlines()
        filtered_lines = [
            line for line in lines
            if not line.strip().startswith("// @name:zh-CN") and not line.strip().startswith("// @description:zh-CN")
        ]
        with open(file_path, 'w', encoding='utf-8') as file:
            file.writelines(filtered_lines)
    except FileNotFoundError:
        print("文件未找到，请检查文件路径。")
    except Exception as e:
        print(f"发生错误: {e}")


# 读取文件并查找中文简介
def read_file_to_memory(file_path):
    search_results = search_in_file(file_path, "zh-CN")
    lines = []
    for name_match in search_results.name_matches:
        lines.append("\n".join(search_results.name_matches))
    for description_match in search_results.description_matches:
        lines.append("\n".join(search_results.description_matches))
    remove_zh_cn_lines(file_path)
    return lines


# 排序文本
def sort_userscript_section(file_path):
    with open(file_path, "r", encoding="utf-8") as file:
        content = file.readlines()

    # 找到 // ==UserScript== 和 // ==/UserScript== 的范围
    start_index = None
    end_index = None

    for i, line in enumerate(content):
        if "// ==UserScript==" in line:
            start_index = i
        elif "// ==/UserScript==" in line:
            end_index = i
            break

    # 如果没有找到对应范围，直接返回
    if start_index is None or end_index is None:
        print("未找到有效的 UserScript 区域")
        return

    # 提取范围内的内容
    userscript_section = content[start_index + 1:end_index]

    # 按类别分组
    descriptions = []
    names = []
    others = []

    for line in userscript_section:
        if re.match(r"// @description", line):
            descriptions.append(line.strip())
        elif re.match(r"// @name", line) and not re.match(r"// @namespace", line):
            names.append(line.strip())
        else:
            others.append(line.strip())

    # 按语言后缀排序（按规范排列规则，语言后缀优先）
    def sort_by_suffix(lines):
        return sorted(lines, key=lambda x: re.search(r"[:\-]([a-zA-Z\-]*)", x).group(1) if re.search(r"[:\-]([a-zA-Z\-]*)", x) else "")

    sorted_descriptions = sort_by_suffix(descriptions)
    sorted_names = sort_by_suffix(names)

    # 合并排序结果
    sorted_section = sorted_names + sorted_descriptions + others

    # 替换原始内容中的 UserScript 区域
    content[start_index + 1:end_index] = [line + "\n" for line in sorted_section]

    # 写入文件
    with open(file_path, "w", encoding="utf-8") as file:
        file.writelines(content)


# 翻译函数
def translate_text(text, target_lang):
    text = text.replace("// @name:zh-CN", '').replace("// @description:zh-CN", '')
    if text in translation_cache:
        cached_translation, needs_api_translation = translation_cache[text]
        if not needs_api_translation:
            return cached_translation
    api_url = 'https://translate.googleapis.com/translate_a/single'
    params = {'client': 'gtx', 'dt': 't', 'sl': 'auto', 'tl': target_lang, 'q': text}
    full_url = api_url + '?' + urlencode(params)
    try:
        response = urlopen(full_url)
        data = response.read().decode('utf-8')
        translated_text = ''.join(item[0] for item in json.loads(data.replace("'", "\u2019"))[0])
        return translated_text
    except Exception as e:
        print(f"翻译错误：{e}")
        return None


# 翻译锁，确保多个线程不会同时修改 translations
translation_lock = threading.Lock()


# 用于保存翻译结果的线程函数
def translate_worker(chinese_texts, translations, lang):
    for idx, chinese_text in chinese_texts:
        translated_text = translate_text(chinese_text, lang)
        if translated_text:
            with translation_lock:
                translations[(idx, chinese_text)] = translated_text


# 翻译并返回翻译结果
def translate_and_collect(lines, chinese_texts, lang):
    translations = {}
    threads = []
    chunk_size = len(chinese_texts) // 5 or 1
    for i in range(0, len(chinese_texts), chunk_size):
        chunk = chinese_texts[i:i + chunk_size]
        thread = threading.Thread(target=translate_worker, args=(chunk, translations, lang))
        threads.append(thread)
        thread.start()

    for thread in threads:
        thread.join()

    # 收集翻译结果，保存在字符串中
    translation_output = ""
    for line_number, chinese_text, translated_text in [(ln, ct, translations.get((ln, ct), None)) for ln, ct in chinese_texts if (ln, ct) in translations]:
        if line_number == 0:  # 假设第一行是标题
            translation_output += f'// @name:{lang}    {translated_text}\n'
            if lang == 'en':
                translation_output += f'// @name    {translated_text}\n'
        else:
            translation_output += f'// @description:{lang}    {translated_text}'
            if lang == 'en':
                translation_output += f'\n// @description    {translated_text}'
    return translation_output


# 处理翻译逻辑
def translate_localized(readme_path, target_langs):
    lines = read_file_to_memory(readme_path)
    # 保存整行包含中文文本的位置信息
    chinese_texts = []
    for line_number, line in enumerate(lines):
        if chinese_pattern.search(line):  # 整行匹配中文文本
            chinese_texts.append((line_number, line))

    # 遍历 target_langs 中的语言
    for lang_code in target_langs:
        print(f"开始翻译 {lang_code} ...")
        translation_output = translate_and_collect(lines, chinese_texts, lang_code)
        insert_into_meta(readme_path, translation_output)


def main():
    # 创建命令行参数解析器
    parser = argparse.ArgumentParser(description="UserScript 多语言自动化翻译与优化工具")
    parser.add_argument("file_path", type=str, help="需要处理的 UserScript 文件路径")
    parser.add_argument("--langs", nargs="+", default=['ar', 'bg', 'cs', 'da', 'de', 'el', 'en', 'eo', 'es', 'fi', 'fr', 'he', 'hr', 'hu', 'id', 'it', 'ja', 'ka', 'ko', 'nl', 'nb', 'pl', 'pt-BR', 'ro', 'ru', 'sk', 'sr', 'sv', 'th', 'tr', 'ug', 'uk', 'vi', 'zh-SG', 'zh', 'zh-TW', 'zh-HK', 'zh-CN', 'fr-CA'],
                        help="目标翻译语言列表，默认包含 所有语言")
    args = parser.parse_args()
    file_path = args.file_path
    target_langs = args.langs

    translate_localized(file_path, target_langs)
    sort_userscript_section(file_path)
    subprocess.run(['node', 'utils/single-format-monkey-meta.js', file_path], check=True)
    print("翻译和排序完成！")


if __name__ == "__main__":
    main()