dumper.py

from pymem import *
from pymem.process import module_from_name

from loguru import logger

from roleskin import RoleSkin
from manager import InstanceManager
from utils import *


class Dumper:
    def __init__(self, types: list[str]):
        self.proc_name = "MiniGameApp.exe"
        self.dump_types = types
        self.pm = None
        self.libiworld = None
        self.instance_mgr = None
        self.init_success = self._initialize_process()

    @logger.catch
    def _initialize_process(self) -> bool:
        try:
            self.pm = Pymem(self.proc_name)
            self.libiworld = module_from_name(self.pm.process_handle, "libiworld.dll").lpBaseOfDll
            self.instance_mgr = InstanceManager(self.pm)
            logger.success("初始化进程信息成功!")
            return True
        except pymem.exception.ProcessNotFound:
            logger.error(f"未找到进程<{self.proc_name}>, 请检查是否启动游戏!")
        except Exception as e:
            logger.error(f"初始化进程信息失败: {e}")
        return False

    def is_init_success(self) -> bool:
        return self.init_success

    @logger.catch
    def dump(self):
        if not self.is_init_success():
            logger.error("初始化失败，无法执行 dump 操作。")
            return
        
        create_folder("export")
        
        try:
            roleskin_instance = self.instance_mgr.get("RoleSkin")
            roleskin = RoleSkin(self.pm, roleskin_instance)
            roleskin.export_json("export/RoleSkin.json")
            
            logger.success("Dump 成功!")
        except Exception as e:
            logger.error(f"Dump 失败: {e}")

    def close(self):
        if self.pm:
            self.pm.close_process()
            logger.info("程序已退出!")