#!/usr/bin/python3 -u # Imports from standard library import enum import os import signal import sys import json import warnings from pathlib import Path sys.path.insert(1, str(Path(__file__).parents[1] / 'shared')) # imports from shared import osc_paths.ray as r # import subprocess and osc_server (local file) conditionnally # in order to answer faster in many cases. class OperationType(enum.Enum): NULL = 0 CONTROL = 1 SERVER = 2 SESSION = 3 CLIENT = 4 TRASHED_CLIENT = 5 ALL = 6 'used only for help-all message' class Daemon: net_daemon_id = 0 root = "" session_path = "" pid = 0 port = 0 user = "" not_default = False has_gui = 0 has_local_gui = 0 def __init__(self): self.local_gui_pids = [] CONTROL_OPERATIONS = ( 'start', 'start_new', 'start_new_hidden', 'stop', 'list_daemons', 'get_root', 'get_port', 'get_port_gui_free', 'get_pid', 'get_session_path', 'has_local_gui', 'has_gui') SERVER_OPERATIONS = ( 'quit', 'change_root', 'set_terminal_command', 'list_session_templates', 'list_user_client_templates', 'list_factory_client_templates', 'remove_client_template', 'list_sessions', 'new_session', 'open_session', 'open_session_off', 'save_session_template', 'rename_session', 'set_options', 'has_option', 'script_info', 'hide_script_info', 'script_user_action', 'auto_export_custom_names', 'export_custom_names', 'import_pretty_names', 'clear_pretty_names') def signal_handler(sig, frame): if sig in (signal.SIGINT, signal.SIGTERM): global terminate terminate = True def add_self_bin_to_path(): # Add raysession/src/bin to $PATH to can use ray executables after make # Warning, will works only if link to this file is in RaySession/*/*/*.py this_path = os.path.realpath(os.path.dirname(os.path.realpath(__file__))) bin_path = "%s/bin" % os.path.dirname(this_path) if not os.environ['PATH'].startswith("%s:" % bin_path): os.environ['PATH'] = "%s:%s" % (bin_path, os.environ['PATH']) def pid_exists(pid: int | str) -> bool: if isinstance(pid, str): pid = int(pid) try: os.kill(pid, 0) except OSError: return False else: return True def pid_is_stopped(pid: int) -> bool: proc_file_path = f'/proc/{pid}/status' if os.path.exists(proc_file_path): proc_file = open(proc_file_path) for line in proc_file.readlines(): if line.startswith('State: '): value = line.replace('State: ', '', 1) if value and value[0] == 'T': return True return False return True def get_daemon_list() -> list[Daemon]: try: with open('/tmp/RaySession/multi-daemon.json') as f: json_list = json.load(f) except: return list[Daemon]() if not isinstance(json_list, list): return list[Daemon]() l_daemon_list = list[Daemon]() for dmn in json_list: if not isinstance(dmn, dict): continue l_daemon = Daemon() for key, value in dmn.items(): match key: case 'root': l_daemon.root = str(value) case 'session_path': l_daemon.session_path = str(value) case 'user': l_daemon.user = str(value) case 'not_default': l_daemon.not_default = bool(value) case 'net_daemon_id': if isinstance(value, int): l_daemon.net_daemon_id = int(value) case 'pid': if isinstance(value, int) and pid_exists(value): l_daemon.pid = value case 'port': if isinstance(value, int): l_daemon.port = value case 'has_gui': if isinstance(value, int): l_daemon.has_gui = bool(value == 1) l_daemon.has_local_gui = bool(value == 3) case 'local_gui_pids': if isinstance(value, list): for pid in value: if isinstance(pid, int): l_daemon.local_gui_pids.append(pid) if not (l_daemon.net_daemon_id and l_daemon.pid and l_daemon.port): continue l_daemon_list.append(l_daemon) return l_daemon_list def print_help(stdout=False, category=OperationType.NULL): lang_file = "help_en_US" if os.getenv('LANG', '').startswith('fr_'): lang_file = "help_fr_FR" help_path = Path(__file__).parent / lang_file try: with open(help_path, 'r') as help_file: full_message = help_file.read() except: sys.stderr.write('error: help_file %s is missing\n' % help_path) sys.exit(101) message = '' stars = 0 if category is OperationType.ALL: message = full_message else: for line in full_message.split('\n'): if line.startswith('* '): stars += 1 if (stars == 0 or (stars == 1 and category is OperationType.CONTROL) or (stars == 2 and category is OperationType.SERVER) or (stars == 3 and category is OperationType.SESSION) or (stars >= 4 and category is OperationType.CLIENT)): message += "%s\n" % line if stdout: sys.stdout.write(message) else: sys.stderr.write(message) def auto_type_string(string: str): if string.isdigit(): return int(string) if string.replace('.', '', 1).isdigit(): return float(string) return string # if __name__ == '__main__': if True: warnings.filterwarnings("ignore", category=DeprecationWarning) add_self_bin_to_path() if len(sys.argv) <= 1: print_help() sys.exit(100) terminate = False operation_type = OperationType.NULL client_id = '' args = sys.argv[1:] wanted_port = 0 detach = False dport = os.getenv('RAY_CONTROL_PORT') if dport and dport.isdigit(): wanted_port = int(dport) while args and args[0].startswith('--'): option = args.pop(0) match option: case opt if opt.startswith('--help'): match opt: case '--help': print_help(True, OperationType.NULL) case '--help-all': print_help(True, OperationType.ALL) case '--help-control': print_help(True, OperationType.CONTROL) case '--help-server': print_help(True, OperationType.SERVER) case '--help-session': print_help(True, OperationType.SESSION) case '--help-client' | '--help-clients': print_help(True, OperationType.CLIENT) case _: print_help() sys.exit(100) sys.exit(0) case '--port': if not args: print_help() sys.exit(100) port = args.pop(0) if not port.isdigit(): sys.stderr.write('Invalid value for port: %s . Use digits !' % port) sys.exit(100) wanted_port = int(port) case '--detach': detach = True case _: print_help() sys.exit(100) operation = args.pop(0) if operation in ('client', 'trashed_client'): if len(args) < 2: print_help(False, OperationType.CLIENT) sys.exit(100) operation_type = OperationType.CLIENT if operation == 'trashed_client': operation_type = OperationType.TRASHED_CLIENT client_id = args.pop(0) operation = args.pop(0) if operation_type is OperationType.NULL: if operation in CONTROL_OPERATIONS: operation_type = OperationType.CONTROL elif operation in SERVER_OPERATIONS: operation_type = OperationType.SERVER else: operation_type = OperationType.SESSION arg_list = [auto_type_string(s) for s in args] if operation_type in (OperationType.CLIENT, OperationType.TRASHED_CLIENT): arg_list.insert(0, client_id) if operation in ('new_session', 'open_session', 'change_root', 'save_as_template', 'take_snapshot', 'duplicate', 'open_snapshot', 'rename', 'add_executable', 'add_client_template', 'script_info'): if not arg_list: sys.stderr.write('operation %s needs argument(s).\n' % operation) sys.exit(100) exit_code = 0 daemon_announced = False daemon_list = get_daemon_list() daemon_port = 0 daemon_started = True for daemon in daemon_list: if ((daemon.user == os.environ['USER'] and not wanted_port and not daemon.not_default) or (wanted_port == daemon.port)): daemon_port = daemon.port break else: daemon_started = False if operation_type is OperationType.CONTROL: match operation: # global operations case 'start': if daemon_started: sys.stderr.write('server already started.\n') sys.exit(0) case 'start_new' | 'start_new_hidden': pass case 'stop': if not daemon_started: sys.stderr.write('No server started.\n') sys.exit(0) case 'list_daemons': for daemon in daemon_list: if daemon.not_default: continue sys.stdout.write('%s\n' % str(daemon.port)) sys.exit(0) case _ if not daemon_started: sys.stderr.write( 'No server started. So impossible to %s\n' % operation) sys.exit(100) # operations needing a running daemon case 'get_pid': for daemon in daemon_list: if daemon.port == daemon_port: sys.stdout.write('%s\n' % str(daemon.pid)) sys.exit(0) case 'get_port': sys.stdout.write("%s\n" % str(daemon_port)) sys.exit(0) case 'get_port_gui_free': wanted_session_root = '' if args: wanted_session_root = args[0] for daemon in daemon_list: if (daemon.user == os.environ['USER'] and (daemon.root == wanted_session_root or not wanted_session_root) and not daemon.not_default): if not daemon.has_local_gui: sys.stdout.write('%s\n' % daemon.port) break for pid in daemon.local_gui_pids: if pid == 0: # This means we don't know the pid of the local GUI # So consider this daemon has already a GUI break if pid_exists(pid) and not pid_is_stopped(pid): break else: sys.stdout.write('%s\n' % daemon.port) break sys.exit(0) case 'get_root': for daemon in daemon_list: if daemon.port == daemon_port: sys.stdout.write('%s\n' % daemon.root) sys.exit(0) case 'get_session_path': for daemon in daemon_list: if daemon.port == daemon_port: if not daemon.session_path: sys.exit(1) sys.stdout.write('%s\n' % daemon.session_path) sys.exit(0) case 'has_local_gui': for daemon in daemon_list: if daemon.port == daemon_port: if daemon.has_local_gui: sys.exit(0) sys.exit(1) case 'has_gui': for daemon in daemon_list: if daemon.port == daemon_port: if daemon.has_gui: sys.exit(0) sys.exit(1) elif not daemon_started: at_port = '' if daemon_port: at_port = "at port %i" % daemon_port match operation_type: case OperationType.SERVER: if operation == 'quit': sys.stderr.write(f'No server {at_port} to quit !\n') sys.exit(0) case OperationType.SESSION: sys.stderr.write( f"No server started {at_port}. " f"So no session to {operation}\n") sys.exit(100) case OperationType.CLIENT: sys.stderr.write( f"No server started {at_port}. " f"So no client to {operation}\n") sys.exit(100) case OperationType.TRASHED_CLIENT: sys.stderr.write( f"No server started {at_port}. " f"So no trashed client to {operation}\n") sys.exit(100) case _: print_help() sys.exit(100) osc_order_path = '/ray/' match operation_type: case OperationType.CLIENT: osc_order_path += 'client/' case OperationType.TRASHED_CLIENT: osc_order_path += 'trashed_client/' case OperationType.SERVER: osc_order_path += 'server/' case OperationType.SESSION: osc_order_path += 'session/' osc_order_path += operation if operation_type is OperationType.CONTROL and operation == 'stop': osc_order_path = r.server.QUIT from osc_server import OscServer # see top of the file server = OscServer(detach) server.set_order_path_args(osc_order_path, arg_list) daemon_process = None if (daemon_started and not (operation_type is OperationType.CONTROL and operation in ('start_new', 'start_new_hidden'))): if (operation_type is OperationType.CONTROL and operation == 'stop'): daemon_port_list = list[int]() if wanted_port: daemon_port_list.append(wanted_port) else: for daemon in daemon_list: if (daemon.user == os.getenv('USER') and not daemon.not_default): daemon_port_list.append(daemon.port) server.stop_daemons(daemon_port_list) else: server.set_daemon_address(daemon_port) server.send_order_message() if detach: sys.exit(0) else: session_root = "%s/Ray Sessions" % os.getenv('HOME') try: settings_file = open( "%s/.config/RaySession/RaySession.conf" % os.getenv('HOME'), 'r') contents = settings_file.read() for line in contents.split('\n'): if line.startswith('default_session_root='): session_root = line.partition('=')[2] break except: pass # start a daemon because no one is running import subprocess # see top of the file process_args = ['ray-daemon', '--control-url', str(server.url), '--session-root', session_root] if wanted_port: process_args.append('--osc-port') process_args.append(str(wanted_port)) if (operation_type is OperationType.CONTROL and operation == 'start_new_hidden'): process_args.append('--hidden') process_args.append('--no-options') daemon_process = subprocess.Popen( process_args, -1, None, None, subprocess.DEVNULL, subprocess.DEVNULL) server.wait_for_start() if (operation_type is OperationType.CONTROL and operation in ('start', 'start_new', 'start_new_hidden')): server.wait_for_start_only() #connect SIGINT and SIGTERM signal.signal(signal.SIGINT, signal_handler) signal.signal(signal.SIGTERM, signal_handler) exit_code = -1 while True: server.recv(50) if terminate: break exit_code = server.final_error() if exit_code >= 0: break if server.is_waiting_start_for_a_long(): exit_code = 103 break if daemon_process and daemon_process.poll() is not None: sys.stderr.write('daemon terminates, sorry\n') exit_code = 104 break if (operation_type is OperationType.CONTROL and operation in ('start_new', 'start_new_hidden') and exit_code == 0): daemon_port = server.get_daemon_port() if daemon_port: sys.stdout.write("%i\n" % daemon_port) server.disannounce_to_daemon() sys.exit(exit_code)