#!/usr/bin/env python
# -*- coding: UTF-8 -*-

import os, sys
import shutil
import re

def find_config_item(item):
    current_directory = os.getcwd()
    file_path = os.path.join(current_directory, 'b_config.h')
    if os.path.exists(file_path):
        with open(file_path, 'r', encoding='utf-8') as file:
            content = file.read()
            if item in content:
                return 1
            else:
                return 0
    else:
        return 0
    
def file_add_bconfig_limit(file_path, pat):
    # 读取原始文件内容
    with open(file_path, 'r', encoding='utf-8') as file:
        original_content = file.read()

    # 定义要添加的内容
    header_content = f'#include "b_config.h" \r\n#if (defined({pat}) && {pat} == 1)\n'
    footer_content = '\n#endif\r\n'

    # 创建新的文件内容
    new_content = header_content + original_content + footer_content

    # 将新的内容写回文件
    with open(file_path, 'w', encoding='utf-8') as file:
        file.write(new_content)

def find_files(dirs, des):
    source_file = []
    source_file_des = []
    for d in dirs:
        files = os.listdir(d)
        for file_name in files:
            # 检查文件扩展名是否为.c或.h
            if file_name.endswith('.c') or file_name.endswith('.h') or file_name.endswith('.inc'):
                # 构建源文件和目标文件的完整路径
                source_file.append(os.path.join(d, file_name))
                source_file_des.append(os.path.join(des, file_name))
    return source_file,source_file_des

def clear_directory(directory):
    # 获取目录中的所有文件和子目录
    for filename in os.listdir(directory):
        file_path = os.path.join(directory, filename)
        # 判断是否是文件
        if os.path.isfile(file_path):
            # 删除文件
            os.remove(file_path)
        # 如果是子目录，则递归清空子目录中的文件
        elif os.path.isdir(file_path):
            clear_directory(file_path)

#将文件里面的相对路径，去掉相对路径部分字符串
def process_includes(file_path):
    with open(file_path, 'r', encoding="utf-8") as f:
        content = f.read()
    # 使用正则表达式匹配所有#include语句，并去掉路径部分
    pattern = r'#include\s+"(\.\./)*(\S*/)*(.+?)"'
    content = re.sub(pattern, r'#include "\3"', content)
    pattern = r'#include\s+<(\.\./)*(\S*/)*(.+?)>'
    content = re.sub(pattern, r'#include "\3"', content)
    with open(file_path, 'w', encoding="utf-8") as f:
        f.write(content)

def remove_relative_paths_in_includes(directory):
    """
    遍历指定目录下的所有.c和.h文件，移除#include语句中的相对路径部分。
    """
    # 遍历目录下的所有文件
    for root, dirs, files in os.walk(directory):
        for file in files:
            if file.endswith('.c') or file.endswith('.h'):
                file_path = os.path.join(root, file)
                process_includes(file_path)

def copy_files_R(src_dir, des_dir):
    try:
        shutil.copytree(src_dir, des_dir)
        print("目录复制成功！")
    except FileExistsError:
        print("目标目录已存在！")
    except:
        print("目录复制失败！")

def insert_include_line(file_path, search_line, new_line):
    """
    在指定文件中搜索特定行，并在其下一行插入新的文本行。
    
    :param file_path: 文件的路径。
    :param search_line: 要搜索的行。
    :param new_line: 要插入的新行。
    """
    # 读取文件内容
    with open(file_path, 'r', encoding='utf-8') as file:
        lines = file.readlines()
    
    # 查找指定行并在其后插入新行
    for i, line in enumerate(lines):
        if search_line in line:
            # 在找到的行后面插入新行，注意i+1表示下一行的位置
            lines.insert(i + 1, new_line + '\n')
            break  # 如果只需要替换第一个匹配项，找到后即可退出循环
    
    # 将修改后的内容写回文件
    with open(file_path, 'w', encoding='utf-8') as file:
        file.writelines(lines)

#在文件里相对路径前增加一个路径
def lwip_replace_includes(directory, replacement):
    for root, dirs, files in os.walk(directory):
        for file in files:
            if file.endswith(".c") or file.endswith(".h"):
                file_path = os.path.join(root, file)
                with open(file_path, 'r') as f:
                    content = f.read()
                content = content.replace('#include "lwip/', '#include "{}lwip/'.format(replacement))
                content = content.replace('#include "netif/', '#include "{}netif/'.format(replacement))
                with open(file_path, 'w') as f:
                    f.write(content)

#在文件里相对路径前增加一个路径
def mbedtls_replace_includes(directory, replacement):
    for root, dirs, files in os.walk(directory):
        for file in files:
            if file.endswith(".c") or file.endswith(".h"):
                file_path = os.path.join(root, file)
                with open(file_path, 'r') as f:
                    content = f.read()
                content = content.replace('#include "mbedtls/', '#include "{}mbedtls/'.format(replacement))
                content = content.replace('#include "psa/', '#include "{}psa/'.format(replacement))

                content = content.replace('#include <mbedtls/', '#include <{}mbedtls/'.format(replacement))
                content = content.replace('#include <psa/', '#include <{}psa/'.format(replacement))
                with open(file_path, 'w') as f:
                    f.write(content)

# 拷贝文件到目录
def copy_file_to_directory(file, des_dir):
    shutil.copy(file, des_dir)

# 拷贝目录下文件到dst_dir
def copy_files(src_dir, dst_dir):
    """
    复制src_dir目录下的所有文件到dst_dir目录。
    :param src_dir: 源目录
    :param dst_dir: 目标目录
    """
    # 检查源目录是否存在
    if not os.path.exists(src_dir):
        print(f"源目录 {src_dir} 不存在。")
        return
    
    # 如果目标目录不存在，则创建它
    if not os.path.exists(dst_dir):
        os.makedirs(dst_dir)
    
    # 遍历源目录中的所有文件和目录
    for item in os.listdir(src_dir):
        src_path = os.path.join(src_dir, item)
        dst_path = os.path.join(dst_dir, item)
        
        # 检查当前项是否为文件
        if os.path.isfile(src_path):
            # 复制文件
            shutil.copy2(src_path, dst_path)
            print(f"文件 {src_path} 已复制到 {dst_path}")

#---------------------------------------------------------------------
#---------------------------------------------------------------------
#---------------------------------------------------------------------
# arm-2d
def cp_arm_2d_file(bos_dir):
    arm_2d_dir = bos_dir + "/thirdparty/arm-2d/"
    tmp_dir = arm_2d_dir + "bos_arm-2d/"
    if not os.path.exists(tmp_dir):
        os.makedirs(tmp_dir)
    else:
        print("bos_arm-2d exist !")
        return
    arm_2d_file_dir = [arm_2d_dir + "Helper/Source/", arm_2d_dir + "Helper/Include/",
                       arm_2d_dir + "Library/Source/", arm_2d_dir + "Library/Include/", 
                       arm_2d_dir + "examples/[template][babyos]/",
                       arm_2d_dir + "examples/common/controls/",
                       arm_2d_dir + "examples/common/asset/"]
    arm_2d_files,tmp_dir_files = find_files(arm_2d_file_dir, tmp_dir)
    for i in range(len(arm_2d_files)):
        shutil.copy2(arm_2d_files[i], tmp_dir_files[i])
        process_includes(tmp_dir_files[i])

# lwip
def cp_lwip_file(bos_dir):
    lwip_dir = bos_dir + "/thirdparty/lwip/"
    tmp_dir = lwip_dir + "bos_lwip/"
    if not os.path.exists(tmp_dir):
        os.makedirs(tmp_dir)
    else:
        print("bos_lwip exist !")
        return
    
    lwip_c_file_dir = [lwip_dir + "src/core/", lwip_dir + "src/core/ipv4/", lwip_dir + "test/bos/arch/"]
    lwip_c_files,tmp_dir_files = find_files(lwip_c_file_dir, tmp_dir)
    for i in range(len(lwip_c_files)):
        shutil.copy2(lwip_c_files[i], tmp_dir_files[i])
    copy_files_R(lwip_dir + "src/include/", tmp_dir + "include/")
    copy_files_R(lwip_dir + "test/bos/arch/", tmp_dir + "include/lwip/arch/")
    copy_file_to_directory(lwip_dir + "test/bos/lwipopts.h", tmp_dir + "include/lwip/")
    copy_file_to_directory(lwip_dir + "test/bos/netif/ethernet.c", tmp_dir)
    lwip_replace_includes(tmp_dir, "thirdparty/lwip/bos_lwip/include/")

# mbedtls
def cp_mbedtls_file(bos_dir):
    mbedtls_dir = bos_dir + "/thirdparty/mbedtls/"
    tmp_dir = mbedtls_dir + "bos_mbedtls/"
    if not os.path.exists(tmp_dir):
        os.makedirs(tmp_dir)
    else:
        print("bos_mbedtls exist !")
        return
    copy_files(mbedtls_dir + "mbedtls-3.6.0/library/", tmp_dir)
    copy_files_R(mbedtls_dir + "mbedtls-3.6.0/include/psa/", tmp_dir + "psa/")
    copy_files_R(mbedtls_dir + "mbedtls-3.6.0/include/mbedtls/", tmp_dir + "mbedtls/")
    mbedtls_replace_includes(tmp_dir + "psa/", "thirdparty/mbedtls/bos_mbedtls/")
    mbedtls_replace_includes(tmp_dir + "mbedtls/", "thirdparty/mbedtls/bos_mbedtls/")
    mbedtls_replace_includes(tmp_dir, "thirdparty/mbedtls/bos_mbedtls/")
    insert_include_line(tmp_dir + "mbedtls/build_info.h", '#define MBEDTLS_BUILD_INFO_H', '#include "b_config.h"')

#---------------------------------------------------------------------
#---------------------------------------------------------------------
def copy_specific_files(source_dir, destination_dir, extensions, del_include_path):
    for root, dirs, files in os.walk(source_dir):
        for file in files:
            if file.endswith(tuple(extensions)):
                source_file = os.path.join(root, file)
                destination_file = os.path.join(destination_dir, file)
                shutil.copy2(source_file, destination_file)
                if del_include_path:
                    process_includes(destination_file)

def replace_line_with_pattern(file_path, pattern_to_replace, replacement_line):
    with open(file_path, 'r', encoding="utf-8") as file:
        lines = file.readlines()
    with open(file_path, 'w', encoding="utf-8") as file:
        for line in lines:
            if pattern_to_replace in line:
                file.write(replacement_line + '\n')
            else:
                file.write(line)

def replace_line_with_pattern_plus(file_path, pattern_to_replace, replacement_line):
    # 读取文件内容
    with open(file_path, 'r', encoding="utf-8") as file:
        file_content = file.read()
    # 使用正则表达式进行替换
    file_content = re.sub(pattern_to_replace, replacement_line, file_content)
    # 写入替换后的内容
    with open(file_path, 'w', encoding="utf-8") as file:
        file.write(file_content)
                
# tinyusb
def cp_tinyusb_file(bos_dir):
    tinyusb_dir = bos_dir + "/thirdparty/tinyusb/"
    tmp_dir = tinyusb_dir + "bos_tinyusb/"
    if not os.path.exists(tmp_dir):
        os.makedirs(tmp_dir)
    else:
        print("bos_tinyusb exist !")
        return
    extensions = (".c", ".h")
    copy_specific_files(tinyusb_dir + "tinyusb/src/", tmp_dir, extensions, True)
    replace_line_with_pattern(tmp_dir + 'tusb_option.h','#define _TUSB_OPTION_H_', '#define _TUSB_OPTION_H_\r\n#include "b_config.h"')
    pattern = r'#\s*define\s+tu_printf\s+printf'
    replace_line_with_pattern_plus(tmp_dir + 'tusb_debug.h', pattern, '#include "utils/inc/b_util_log.h"\r\n  #define tu_printf  b_log')

# lua
# ----------------------------------------------------------------------
def delete_specific_files(s_dir, del_files):
    for file in del_files:
        try:
            file_path = os.path.join(s_dir, file)
            os.remove(file_path)
            print(f"Deleted {file_path}")
        except FileNotFoundError:
            print(f"{file} not found in {s_dir}, skipping.")
#-----------------------------------------------------------------------
def cp_lua_file(bos_dir):
    lua_dir = bos_dir + "/thirdparty/lua/"
    tmp_dir = lua_dir + "bos_lua/"
    if not os.path.exists(tmp_dir):
        os.makedirs(tmp_dir)
    else:
        print("bos_lua exist !")
        return
    extensions = (".c", ".h")
    copy_specific_files(lua_dir + "lua-5.4.7/", tmp_dir, extensions, False)
    delete_specific_files(tmp_dir, ['lua.c', 'luac.c', 'loslib.c', 'liolib.c', 'lcorolib.c'])
    replace_line_with_pattern(tmp_dir + "lauxlib.c", "free(ptr);", "bFree(ptr);")
    replace_line_with_pattern(tmp_dir + "lauxlib.c", "return realloc(ptr, nsize);", "return bRealloc(ptr, nsize);")
    replace_line_with_pattern(tmp_dir + "lauxlib.c", '#include "lauxlib.h"', '#include "lauxlib.h"\r\n#include "utils/inc/b_util_memp.h"')
    replace_line_with_pattern(tmp_dir + "linit.c", '{LUA_OSLIBNAME, luaopen_os},', '//{LUA_OSLIBNAME, luaopen_os},')
    replace_line_with_pattern(tmp_dir + "linit.c", '{LUA_IOLIBNAME, luaopen_io},', '//{LUA_IOLIBNAME, luaopen_io},')
    replace_line_with_pattern(tmp_dir + "linit.c", '{LUA_COLIBNAME, luaopen_coroutine},', '//{LUA_COLIBNAME, luaopen_coroutine},')
    replace_line_with_pattern(tmp_dir + "luaconf.h", '#include <stddef.h>', '#include <stddef.h>\r\n#include "b_config.h"')
    replace_line_with_pattern(tmp_dir + "linit.c", '{LUA_MATHLIBNAME, luaopen_math},', '#if (defined(_LUA_MATH_ENABLE) && (_LUA_MATH_ENABLE == 1))\n{LUA_MATHLIBNAME, luaopen_math},\n#endif')
    replace_line_with_pattern(tmp_dir + "linit.c", '{LUA_UTF8LIBNAME, luaopen_utf8},', '#if (defined(_LUA_UTF8_ENABLE) && (_LUA_UTF8_ENABLE == 1))\n{LUA_UTF8LIBNAME, luaopen_utf8},\n#endif')
    file_add_bconfig_limit(tmp_dir + "lmathlib.c", '_LUA_MATH_ENABLE')
    file_add_bconfig_limit(tmp_dir + "lutf8lib.c", '_LUA_UTF8_ENABLE')
#---------------------------------------------------------------------
def cp_thirdparty_file(bos_dir):
    cp_arm_2d_file(bos_dir)
    cp_lwip_file(bos_dir)
    cp_tinyusb_file(bos_dir)
    cp_mbedtls_file(bos_dir)
    cp_lua_file(bos_dir)