import shutil
import os
import time
from loguru import logger


def export_data_folder(source_folder="data", output_folder="backup"):
    """
    将指定文件夹压缩为 zip 包
    :param source_folder: 要压缩的文件夹路径 (默认 'data')
    :param output_folder: 压缩包存放的文件夹路径 (默认 'backup')
    """
    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 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 文件。"
        )