import os
import shutil
import logging
from pathlib import Path
from nep_auto.utils.runner import CommandRunner


class BaseModule:
    def __init__(self, driver, iter_id):
        """
        :param driver: NEPDriver 实例，包含所有配置
        :param iter_id: 当前轮次 (int)
        """
        self.driver = driver
        self.config_sys = driver.config_sys
        self.config_param = driver.config_param
        self.machine_config = driver.config_machine['systems'][driver.config_machine['current_system']]

        self.iter_id = iter_id
        self.iter_name = f"iter_{iter_id:03d}"
        self.logger = logging.getLogger("NEP_Auto")

        # 初始化运行器
        self.runner = CommandRunner(self.machine_config)

        # 定义路径
        self.root = Path(driver.root) / "workspace"
        self.iter_dir = self.root / self.iter_name
        self.output_dir = self.iter_dir / "05.output"  # 公共输出区

    def get_work_dir(self):
        """需由子类实现：返回当前模块的具体工作目录"""
        raise NotImplementedError

    def initialize(self):
        """通用初始化：创建目录，复制通用文件"""
        work_dir = self.get_work_dir()
        if not work_dir.exists():
            work_dir.mkdir(parents=True, exist_ok=True)
            self.logger.debug(f"📁 Created dir: {work_dir}")

        # 确保公共输出目录存在
        if not self.output_dir.exists():
            self.output_dir.mkdir(parents=True, exist_ok=True)

    def run(self):
        """核心逻辑入口，子类必须实现"""
        raise NotImplementedError

    def check_done(self):
        """检查任务是否完成，子类必须实现"""
        raise NotImplementedError

    # --- 通用工具方法 ---

    def copy_template(self, template_name, target_name=None):
        """从 template 目录复制文件"""
        if target_name is None:
            target_name = template_name

        # 根据模块类型寻找模板目录 (需要在子类定义 self.template_subdir)
        src = Path("template") / getattr(self, "template_subdir", "common") / template_name
        dst = self.get_work_dir() / target_name

        if src.exists():
            shutil.copy(src, dst)
            # self.logger.debug(f"📄 Copied {template_name} -> {dst}")
        else:
            self.logger.warning(f"⚠️ Template not found: {src}")

    def link_file(self, src_path, dst_name):
        """创建软链接"""
        src = Path(src_path).resolve()
        dst = self.get_work_dir() / dst_name
        if dst.exists():
            dst.unlink()
        os.symlink(src, dst)