Files
emod-cli/debug_mod/DEBUG_ENV_SCRIPT/modMain.py

129 lines
2.8 KiB
Python
Raw Normal View History

# -*- coding: utf-8 -*-
from .QuModLibs.QuMod import *
from .Game import (
RELOAD_MOD,
INIT_RELOAD_TIME,
RELOAD_ADDON,
RELOAD_WORLD,
RELOAD_SHADERS,
)
from .Config import DEBUG_CONFIG
import threading
import sys
lambda: "By Zero123"
REF = 0
class STD_OUT_WRAPPER(object):
def __init__(self, baseIO):
self.baseIO = baseIO
self.writeLock = threading.Lock()
self._buffer = []
def __getattr__(self, name):
return getattr(self.baseIO, name)
def write(self, data):
with self.writeLock:
parts = data.splitlines(True)
for part in parts:
if part.endswith("\n"):
if self._buffer:
line = "".join(self._buffer) + part
self._buffer = []
else:
line = part
self.baseIO.write("[Python] " + line)
else:
self._buffer.append(part)
def close(self):
return self.baseIO.close()
def writelines(self, lines):
for line in lines:
self.write(line)
def fileno(self):
return self.baseIO.fileno()
stdout = sys.stdout
stderr = sys.stderr
def REST_STDOUT():
sys.stdout = stdout
sys.stderr = stderr
sys.stdout = STD_OUT_WRAPPER(sys.stdout)
sys.stderr = STD_OUT_WRAPPER(sys.stderr)
@PRE_SERVER_LOADER_HOOK
def SERVER_INIT():
global REF
REF += 1
def _DESTROY():
global REF
REF -= 1
if REF != 0:
return
REST_STDOUT()
from .QuModLibs.Systems.Loader.Server import LoaderSystem
LoaderSystem.REG_DESTROY_CALL_FUNC(_DESTROY)
from . import IPCSystem
IPCSystem.ON_SERVER_INIT()
def CLOnKeyPressInGame(args={}):
if args["isDown"] != "0":
return
if args["screenName"] != "hud_screen" and not DEBUG_CONFIG.get(
"reload_key_global", False
):
return
key = args["key"]
if key == str(DEBUG_CONFIG.get("reload_key", "82")):
RELOAD_MOD()
elif key == str(DEBUG_CONFIG.get("reload_world_key", "")):
RELOAD_WORLD()
elif key == str(DEBUG_CONFIG.get("reload_addon_key", "")):
RELOAD_ADDON()
elif key == str(DEBUG_CONFIG.get("reload_shaders_key", "")):
RELOAD_SHADERS()
@PRE_CLIENT_LOADER_HOOK
def CLIENT_INIT():
global REF
REF += 1
from . import IPCSystem
def _DESTROY():
global REF
REF -= 1
IPCSystem.ON_CLIENT_EXIT()
if REF != 0:
return
REST_STDOUT()
from .QuModLibs.Systems.Loader.Client import LoaderSystem
LoaderSystem.REG_DESTROY_CALL_FUNC(_DESTROY)
LoaderSystem.getSystem().nativeStaticListen("OnKeyPressInGame", CLOnKeyPressInGame)
IPCSystem.ON_CLIENT_INIT()
try:
INIT_RELOAD_TIME()
except:
pass