import shutil
import os
import time
from loguru import logger
import zipfile


def export_templates_folder(output_folder="backup"):
    """
    将指定文件夹压缩为 zip 包
    :param source_folder: 要压缩的文件夹路径 (默认 'data')
    :param output_folder: 压缩包存放的文件夹路径 (默认 'backup')
    """
    source_folder = "data"
    try:
        # 1. 检查源文件夹是否存在
        if not os.path.exists(source_folder):
            logger.error(f"备份失败: 找不到源文件夹 '{source_folder}'")
            return

        # 2. 确保输出目录存在
        if not os.path.exists(output_folder):
            os.makedirs(output_folder)
            logger.info(f"创建备份目录: {output_folder}")

        # 3. 生成带时间戳的文件名 (例如: data_backup_20251211_103000)
        timestamp = time.strftime("%Y%m%d_%H%M%S")
        # 注意: make_archive 不需要写后缀 .zip，它会自动加
        base_name = os.path.join(output_folder, f"data_备份_{timestamp}")

        logger.info(f"正在压缩 '{source_folder}' ...")

        # 4. 执行压缩
        # format="zip": 压缩格式
        # root_dir: 要压缩的根目录的上级目录 (通常填 None)
        # base_dir: 要压缩的具体目录
        zip_path = shutil.make_archive(
            base_name=base_name,
            format="zip",
            root_dir=os.path.dirname(
                os.path.abspath(source_folder)
            ),  # 这里为了安全，定位到父级
            base_dir=os.path.basename(
                os.path.abspath(source_folder)
            ),  # 只压缩 data 文件夹本身
        )

        logger.success(f"✅ 备份成功! 文件已保存至:\n{zip_path}")
        return zip_path

    except Exception as e:
        logger.error(f"❌ 备份出错: {str(e)}")
        import traceback

        logger.error(traceback.format_exc())


def export_data(save_dir, root_dir="."):
    """
    导出 data 和 output 两个文件夹到同一个 zip 包中
    :param save_dir: 用户在 GUI 弹窗中选择的保存目录 (例如: D:/Backup)
    :param root_dir: 项目根目录 (用于找到 data 和 output)
    """

    # 1. 定义要打包的目标文件夹
    targets = ["data", "output"]

    # 2. 检查保存目录
    if not os.path.exists(save_dir):
        logger.error(f"保存目录不存在: {save_dir}")
        return

    # 3. 生成压缩包路径
    timestamp = time.strftime("%Y%m%d_%H%M%S")
    zip_filename = f"完整备份_{timestamp}.zip"
    zip_path = os.path.join(save_dir, zip_filename)

    logger.info(f"开始备份，目标文件: {zip_path}")

    try:
        # 4. 创建压缩包 (使用 'w' 写入模式，ZIP_DEFLATED 表示压缩)
        with zipfile.ZipFile(zip_path, "w", zipfile.ZIP_DEFLATED) as zf:

            has_files = False  # 标记是否真的压缩了文件

            for target in targets:
                target_abs_path = os.path.join(root_dir, target)

                # 检查 data 或 output 是否存在
                if not os.path.exists(target_abs_path):
                    logger.warning(f"⚠️ 跳过: 找不到文件夹 '{target}'")
                    continue

                logger.info(f"正在压缩: {target} ...")

                # 5. 遍历文件夹写入 ZIP
                # os.walk 会递归遍历子文件夹
                for root, dirs, files in os.walk(target_abs_path):
                    for file in files:
                        # 获取文件的绝对路径
                        file_abs_path = os.path.join(root, file)

                        # 【关键】计算在压缩包里的相对路径
                        # 例如: D:/Project/data/images/1.jpg -> data/images/1.jpg
                        arcname = os.path.relpath(file_abs_path, root_dir)

                        # 写入压缩包
                        zf.write(file_abs_path, arcname)
                        has_files = True

        if has_files:
            logger.success(f"✅ 备份成功! 文件已保存至:\n{zip_path}")
            return zip_path
        else:
            logger.error("❌ 备份失败: data 和 output 文件夹均为空或不存在。")
            # 如果生成了空文件，建议删除
            if os.path.exists(zip_path):
                os.remove(zip_path)
            return None

    except Exception as e:
        logger.error(f"导出过程出错: {str(e)}")
        import traceback

        logger.error(traceback.format_exc())


def initialize_project(root_dir="."):
    """
    初始化项目：清空 data，重建目录，复制模板
    :param root_dir: 项目根目录
    """
    # 定义路径
    data_dir = os.path.join(root_dir, "data")
    images_dir = os.path.join(data_dir, "images")
    output_dir = os.path.join(root_dir, "output")

    # 模板源文件路径
    template_dir = os.path.join(root_dir, "templates")
    src_excel = os.path.join(template_dir, "names.xlsx")
    dst_excel = os.path.join(data_dir, "names.xlsx")

    logger.info("开始初始化/重置项目...")

    # --- 1. 清空 data 文件夹 ---
    if os.path.exists(data_dir):
        try:
            # 暴力删除整个 data 文件夹及其内容
            shutil.rmtree(data_dir)
            logger.info(f"已清理旧数据: {data_dir}")
        except PermissionError:
            logger.error(
                f"❌ 删除失败！请检查 '{data_dir}' 下的文件（如 Excel）是否正在被打开。"
            )
            return
        except Exception as e:
            logger.error(f"❌ 删除出错: {e}")
            return
    # --- 2. 清空 output_dir 文件夹 ---
    if os.path.exists(output_dir):
        try:
            # 暴力删除整个 output 文件夹及其内容
            shutil.rmtree(output_dir)
            logger.info(f"已清理旧数据: {output_dir}")
        except PermissionError:
            logger.error(f"❌ 删除失败！请检查 '{output_dir}' 下的文件是否正在被打开。")
            return
        except Exception as e:
            logger.error(f"❌ 删除出错: {e}")
            return

    # --- 2. 新建目录结构 ---
    try:
        # makedirs 会自动创建父级目录 (data 和 data/images 都会被创建)
        os.makedirs(images_dir, exist_ok=True)
        os.makedirs(output_dir, exist_ok=True)
        logger.info(f"✅ 已创建目录: {images_dir}")
    except Exception as e:
        logger.error(f"创建目录失败: {e}")
        return

    # --- 3. 复制模板文件 ---
    if os.path.exists(src_excel):
        try:
            shutil.copy(src_excel, dst_excel)
            logger.success(f"✅ 成功从模板恢复: {dst_excel}")
            logger.success("🎉 初始化完成！一切已恢复出厂设置。")
        except Exception as e:
            logger.error(f"复制模板文件失败: {e}")
    else:
        logger.warning(
            f"⚠️ 警告: 模板文件不存在 ({src_excel})，data 文件夹内将没有 Excel 文件。"
        )

def check_file_exists(file_path):
    """
    判断文件是否存在
    """
    if (file_path and isinstance(file_path, str) and os.path.exists(file_path)):
        logger.info(f"✅ 文件存在: {file_path}")
        return True
    else:
        logger.error(f"❌ 文件不存在: {file_path}")
        return False