import os, yaml
from datetime import datetime
from utils.logger import logger


def load_config():
    """加载 YAML 配置文件并解析其内容"""
    try:
        with open(CONFIG_PATH, 'r', encoding='utf-8') as f:
            raw_config = yaml.safe_load(f)

        # 全局配置
        global_config = raw_config.get('global', {})
        config = {
            # vCenter/ESXi节点列表
            "MANAGEMENT_NODES": raw_config.get('management_nodes', []),
            "SNAPSHOT_RETENTION_DAYS": int(global_config.get('snapshot_retention_days', 15)),
            "EXCEL_OUTPUT_PATH": global_config.get('excel_output_path', os.path.join(DATA_DIR, f'vm_snapshots_report-{datetime.now().strftime('%Y-%m-%d')}.xlsx')),
            "YAML_OUTPUT_PATH": global_config.get('yaml_output_path', os.path.join(DATA_DIR, f'old_snapshots-{datetime.now().strftime('%Y-%m-%d')}.yaml')),
            "DISABLE_SSL_VERIFY": global_config.get('disable_ssl_verify', True),
        }

        # 验证配置
        if not config["MANAGEMENT_NODES"]:
            raise ValueError("未配置任何管理节点（vCenter 或 ESXi），只少要有一台 management_nodes 节点。")

        # 检查每个节点的必填字段
        required_fields = ['type', 'name', 'host', 'user', 'password', 'max_delete_concurrent']
        for node in config["MANAGEMENT_NODES"]:
            missing = [f for f in required_fields if f not in node]
            if missing:
                raise ValueError(f"节点 {node.get('name', '未知')} 缺少配置字段: {missing}")
            # 验证类型合法性
            if node['type'] not in ['vcenter', 'esxi']:
                raise ValueError(f"节点 {node['name']} 类型错误（仅支持 vcenter 或 esxi）")

        logger.info(f"✅ 成功加载配置，共 {len(config['MANAGEMENT_NODES'])} 个管理节点")
        return config
    except Exception as e:
        logger.error(f"❌ 加载配置失败: {str(e)}")
        raise


BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))  # 获取项目根目录
DATA_DIR = os.path.join(BASE_DIR, 'output')  # 获取导出数据的文件根目录
os.makedirs(DATA_DIR, exist_ok=True)  # 自动创建目录

# 配置文件路径
CONFIG_PATH = os.path.join(os.path.dirname(__file__), 'config.yaml')
# 加载配置并导出全局变量
config = load_config()  # 模块导入时立即执行
MANAGEMENT_NODES = config["MANAGEMENT_NODES"]
SNAPSHOT_RETENTION_DAYS = config["SNAPSHOT_RETENTION_DAYS"]
EXCEL_OUTPUT_PATH = config["EXCEL_OUTPUT_PATH"]
YAML_OUTPUT_PATH = config["YAML_OUTPUT_PATH"]
DISABLE_SSL_VERIFY = config["DISABLE_SSL_VERIFY"]


if __name__ == "__main__":
    # 打印全局配置
    print("\n【全局配置】")
    print(f"  快照保留天数: {config['SNAPSHOT_RETENTION_DAYS']}")
    print(f"  Excel输出路径: {EXCEL_OUTPUT_PATH}")

    # 打印管理节点
    nodes = config['MANAGEMENT_NODES']
    print(f"\n【管理节点】共 {len(nodes)} 个")

    for i, node in enumerate(nodes, 1):
        print(f"\n  节点[{i}]:")
        print(f"    地址: {node.get('host')}")
        print(f"    用户: {node.get('user')}")
        print(f"    密码: {'*' * len(node.get('password', ''))}")
        # print(f"    密码: {node.get('password', '')}")  # 直接打印出密码