# src/utils.py
import yaml
import logging
import os
import sys


def load_yaml(path):
    """加载 YAML 配置文件"""
    if not os.path.exists(path):
        logging.error(f"Config file not found: {path}")
        sys.exit(1)
    with open(path, 'r') as f:
        return yaml.safe_load(f)


def setup_logger(work_dir, log_file="autonep.log"):
    """配置日志：同时输出到文件和控制台"""
    logger = logging.getLogger()
    logger.setLevel(logging.INFO)

    # 清楚之前的 handler 防止重复
    if logger.hasHandlers():
        logger.handlers.clear()

    # 文件 Handler
    file_handler = logging.FileHandler(os.path.join(work_dir, log_file))
    file_handler.setFormatter(logging.Formatter('%(asctime)s - %(levelname)s - %(message)s'))
    logger.addHandler(file_handler)

    # 控制台 Handler
    console_handler = logging.StreamHandler()
    console_handler.setFormatter(logging.Formatter('%(message)s'))  # 控制台只看消息，简洁点
    logger.addHandler(console_handler)

    return logger


class Notifier:
    """(预留) 通知模块"""

    def __init__(self, url=None):
        self.url = url

    def send(self, title, msg, priority=5):
        # 暂时只打印日志，不实际发送
        logging.info(f"[[Notification]] {title}: {msg}")