import os
import difflib

USER_HOME = os.path.expanduser("~")
LEDE_DIR = os.path.join("/", "home", "lede")  # 修正此处路径
CONFIG_FILE = os.path.join(LEDE_DIR, ".config")
ORIGINAL_FILE = "d1.config"
DIFF_FILE = "dX-diff.config"
SUMMARY_FILE = "multi-device-diff.config"
COUNT_FILE = "count.txt"

def read_file(file_path):
    with open(file_path, 'r') as file:
        return file.readlines()

def write_file(file_path, lines):
    with open(file_path, 'w') as file:
        file.writelines(lines)

def append_to_file(file_path, lines):
    with open(file_path, 'a') as file:
        file.writelines(lines)

def read_count():
    if os.path.isfile(COUNT_FILE):
        with open(COUNT_FILE, 'r') as file:
            return int(file.read().strip())
    return 0

def write_count(count):
    with open(COUNT_FILE, 'w') as file:
        file.write(str(count))

def compare_configs():
    current_config = read_file(CONFIG_FILE)
    original_config = read_file(ORIGINAL_FILE)

    diff = difflib.unified_diff(original_config, current_config, fromfile=ORIGINAL_FILE, tofile=CONFIG_FILE)
    diff_lines = list(diff)

    if diff_lines:
        append_to_file(DIFF_FILE, "\n\n")
        append_to_file(DIFF_FILE, diff_lines)

        count = read_count() + 1
        write_count(count)

        print(f"已追加差异到 {DIFF_FILE}。当前总机型次数：{count}")
    else:
        print("没有检测到差异")

def summarize_diffs():
    with open(DIFF_FILE, 'r') as diff_file:
        diff_lines = diff_file.readlines()

    summary_lines = []
    for line in diff_lines:
        if (line.startswith('-# CONFIG_TARGET') or line.startswith('-CONFIG_TARGET') or line.startswith('+CONFIG_TARGET') or line.startswith('+# CONFIG_TARGET') or
            line.startswith('-# CONFIG_DEFAULT') or line.startswith('-CONFIG_DEFAULT') or line.startswith('+CONFIG_DEFAULT') or line.startswith('+# CONFIG_DEFAULT')):
            summary_lines.append(line)

    write_file(SUMMARY_FILE, summary_lines)
    print(f"multi-device-diff: 已生成总结文件 {SUMMARY_FILE}")

def main():
    if not os.path.isdir(LEDE_DIR):
        print(f"错误：{LEDE_DIR} 目录不存在。请确保 LEDE 源码仓库在 /home 目录下。")
        return

    if not os.path.isfile(ORIGINAL_FILE):
        write_file(ORIGINAL_FILE, read_file(CONFIG_FILE))
        write_count(1)
        print(f"multi-device-diff: 初始配置文件已保存为 {ORIGINAL_FILE}。当前追加的总机型次数：1")

    while True:
        user_input = input("multi-device-diff: 输入 [1] 进行配置比较，输入 [2] 结束脚本：")
        if user_input == '1':
            compare_configs()
        elif user_input == '2':
            summary_input = input("multi-device-diff: 是否处理总结归纳？输入 [1] 处理，输入 [2] 不处理: ")
            if summary_input == '1':
                summarize_diffs()
            print("multi-device-diff: 脚本结束")
            break
        else:
            print("multi-device-diff: 无效输入，请输入 [1] 或 [2]")

if __name__ == "__main__":
    main()