import os
import shutil

script_path = os.path.abspath(__file__)

# 获取所在目录
script_dir = os.path.dirname(script_path)


def replace_txt(file_path, old_text, new_text):
    # 读取文件内容
    with open(file_path, 'r', encoding='utf-8') as f:
        content = f.read()
    shutil.copy2(file_path, file_path + ".bak")
    # 执行替换
    new_content = content.replace(old_text, new_text)

    # 写回文件
    with open(file_path, 'w', encoding='utf-8') as f:
        f.write(new_content)


for root, dirs, files in os.walk(script_dir, topdown=False):
    dir_name = os.path.basename(root)
    if dir_name == "unittest":
        continue
    for file in files:
        if file != "xmake.lua":
            continue
        replace_txt(os.path.join(root, file), "\tadd_deps(", f"\tset_default(false)\n\tadd_tests(\"default\")\n\tset_group(\"{dir_name}\")\n\tadd_deps(")
        break