208 lines
6.9 KiB
Python
208 lines
6.9 KiB
Python
# -------------------------------
|
|
# Difficult Rocket
|
|
# Copyright © 2020-2023 by shenjackyuanjie 3695888@qq.com
|
|
# All rights reserved
|
|
# -------------------------------
|
|
|
|
"""
|
|
writen by shenjackyuanjie
|
|
mail: 3695888@qq.com
|
|
github: @shenjackyuanjie
|
|
gitee: @shenjackyuanjie
|
|
"""
|
|
|
|
import sys
|
|
import time
|
|
import logging
|
|
import traceback
|
|
import importlib
|
|
import importlib.util
|
|
import logging.config
|
|
import multiprocessing
|
|
|
|
from io import StringIO
|
|
from pathlib import Path
|
|
from typing import TYPE_CHECKING, List, Optional, Dict, TypeVar
|
|
|
|
if TYPE_CHECKING:
|
|
from Difficult_Rocket.api.mod import ModInfo
|
|
else:
|
|
ModInfo = TypeVar('ModInfo')
|
|
from Difficult_Rocket.utils import tools
|
|
from Difficult_Rocket.api.types import Options
|
|
from Difficult_Rocket.utils.translate import tr
|
|
from Difficult_Rocket.runtime import DR_runtime
|
|
from Difficult_Rocket.utils.thread import new_thread
|
|
from Difficult_Rocket.crash import write_info_to_cache
|
|
from Difficult_Rocket import client, server, DR_status
|
|
|
|
|
|
class Console(Options):
|
|
name = 'python stdin console'
|
|
|
|
running: bool = False
|
|
caches: List[str] = []
|
|
|
|
@new_thread('python console', daemon=True, log_thread=True)
|
|
def main(self):
|
|
while self.running:
|
|
try:
|
|
get_str = input('>>>')
|
|
except (EOFError, KeyboardInterrupt):
|
|
get_str = 'stop'
|
|
self.caches.append(get_str)
|
|
if get_str == 'stop':
|
|
self.running = False
|
|
break
|
|
|
|
def start(self):
|
|
self.running = True
|
|
self.main()
|
|
|
|
def stop(self):
|
|
self.running = False
|
|
|
|
def get_command(self) -> Optional[str]:
|
|
return self.caches.pop(0) if self.caches else None
|
|
|
|
def new_command(self) -> None:
|
|
return None
|
|
|
|
|
|
class Game(Options):
|
|
name = 'MainGame'
|
|
|
|
client: client.Client
|
|
server: server.Server
|
|
console: Console
|
|
console_class: Console = Console
|
|
|
|
main_config: Dict
|
|
logging_config: Dict
|
|
logger: logging.Logger
|
|
|
|
mod_module: List["ModInfo"] = []
|
|
|
|
def init_logger(self) -> None:
|
|
log_path = self.logging_config['handlers']['file']['filename']
|
|
log_path = f"logs/{log_path.format(time.strftime('%Y-%m-%d %H-%M-%S' , time.gmtime(DR_runtime.start_time_ns / 1000_000_000)))}"
|
|
mkdir = False
|
|
if not Path('logs/').is_dir():
|
|
Path('logs/').mkdir()
|
|
mkdir = True
|
|
self.logging_config['handlers']['file']['filename'] = log_path
|
|
logging.config.dictConfig(self.logging_config)
|
|
self.logger = logging.getLogger('main')
|
|
if mkdir:
|
|
self.logger.info(tr().main.logger.mkdir())
|
|
|
|
def init_mods(self) -> None:
|
|
"""验证/加载 mod"""
|
|
self.mod_module = []
|
|
mods = []
|
|
mod_path = Path(DR_runtime.mod_path)
|
|
if not mod_path.exists():
|
|
self.logger.info(tr().main.mod.find.faild.no_mod_folder())
|
|
return
|
|
# 寻找有效 mod
|
|
paths = mod_path.iterdir()
|
|
sys.path.append(DR_runtime.mod_path)
|
|
for mod_path in paths:
|
|
try:
|
|
if mod_path.name == '__pycache__':
|
|
continue
|
|
self.logger.info(tr().main.mod.find.start().format(mod_path))
|
|
if mod_path.is_dir():
|
|
if importlib.util.find_spec(mod_path.name) is not None:
|
|
mods.append(mod_path.name)
|
|
else:
|
|
self.logger.warning(tr().main.mod.load.faild.info().format(mod_path.name, tr().main.mod.find.faild.no_spec()))
|
|
elif mod_path.suffix in ('.pyz', '.zip', '.pyd', '.py'):
|
|
if importlib.util.find_spec(mod_path.name) is not None:
|
|
mods.append(mod_path.name)
|
|
except ImportError as e:
|
|
self.logger.warning(tr().main.mod.find.faild().format(mod_path, e))
|
|
self.logger.info(tr().main.mod.find.done())
|
|
# 加载有效 mod
|
|
module = []
|
|
for mod in mods:
|
|
try:
|
|
self.logger.info(tr().main.mod.load.start().format(mod))
|
|
mod_module = importlib.import_module(mod)
|
|
if not hasattr(mod_module, "mod_class"):
|
|
self.logger.warning(tr().main.mod.load.faild.info().format(mod, tr().main.mod.load.faild.no_mod_class()))
|
|
del mod_module # 释放内存
|
|
continue
|
|
mod_class: type(ModInfo) = mod_module.mod_class
|
|
mod_class = mod_class()
|
|
module.append(mod_class)
|
|
self.logger.info(tr().main.mod.load.info().format(mod_class.mod_id, mod_class.version))
|
|
except ImportError as e:
|
|
self.logger.warning(tr().main.mod.load.faild.info().format(mod, e))
|
|
self.logger.info(tr().main.mod.load.done())
|
|
self.mod_module = module
|
|
mod_list = []
|
|
for mod in module:
|
|
mod_list.append((mod.mod_id, mod.version))
|
|
# 调用 on_load
|
|
self.dispatch_event('on_load', game=self)
|
|
DR_runtime.DR_Mod_List = mod_list
|
|
|
|
def init_console(self) -> None:
|
|
self.console = self.console_class()
|
|
self.console.start()
|
|
|
|
def start(self):
|
|
self.server.run()
|
|
if DR_status.use_multiprocess:
|
|
try:
|
|
game_process = multiprocessing.Process(target=self.client.start, name='pyglet app')
|
|
game_process.start()
|
|
game_process.join()
|
|
except Exception:
|
|
return -1
|
|
else:
|
|
return 1
|
|
else:
|
|
self.client.start()
|
|
|
|
def dispatch_event(self, event_name: str, *args, **kwargs) -> None:
|
|
"""向 mod 分发事件"""
|
|
for mod in self.mod_module:
|
|
if hasattr(mod, event_name):
|
|
try:
|
|
getattr(mod, event_name)(*args, **kwargs)
|
|
except Exception:
|
|
error = traceback.format_exc()
|
|
self.logger.error(tr().main.mod.event.error().format(event_name, error, mod.mod_id))
|
|
|
|
def log_env(self) -> None:
|
|
cache_steam = StringIO()
|
|
write_info_to_cache(cache_steam)
|
|
text = cache_steam.getvalue()
|
|
self.logger.info(text)
|
|
self.flush_option()
|
|
config_cache = self.logging_config.copy()
|
|
self.logging_config = {"logging_config": "too long to show"}
|
|
self.logger.info(f"\n{self.as_markdown()}")
|
|
self.logging_config = config_cache
|
|
|
|
def setup(self) -> None:
|
|
self.client = client.Client(game=self, net_mode='local')
|
|
self.server = server.Server(net_mode='local')
|
|
|
|
def init(self, **kwargs) -> bool:
|
|
self.load_file()
|
|
self.setup()
|
|
self.log_env()
|
|
return True
|
|
|
|
def load_file(self) -> bool:
|
|
"""加载文件"""
|
|
self.logging_config = tools.load_file('configs/logger.toml')
|
|
self.init_logger()
|
|
self.init_mods()
|
|
self.init_console()
|
|
return True
|
|
|