update installer status

Author: yungwine

mytoninstaller/mytoninstaller.py

@@ -11,6 +11,7 @@
 from mypyconsole.mypyconsole import MyPyConsole
 
 from mytoninstaller.config import GetLiteServerConfig, get_ls_proxy_config
+from mytoninstaller.node_args import get_node_args
 from mytoninstaller.utils import GetInitBlock
 from mytoncore.utils import dict2b64, str2bool, b642dict
 
@@ -111,16 +112,22 @@ def Status(local, args):
 	liteserver_key = keys_dir + "liteserver"
 	liteserver_pubkey = liteserver_key + ".pub"
 
-
-	fnStatus = os.path.isfile(local.buffer.vconfig_path)
-	mtcStatus = os.path.isfile(local.buffer.mconfig_path)
-	vcStatus = os.path.isfile(server_key) or os.path.isfile(client_key)
-	lsStatus = os.path.isfile(liteserver_pubkey)
-
-	print("Full node status:", fnStatus)
-	print("Mytoncore status:", mtcStatus)
-	print("V.console status:", vcStatus)
-	print("Liteserver status:", lsStatus)
+	statuses = {
+		'Full node status': os.path.isfile(local.buffer.vconfig_path),
+		'Mytoncore status': os.path.isfile(local.buffer.mconfig_path),
+		'V.console status': os.path.isfile(server_key) or os.path.isfile(client_key),
+		'Liteserver status': os.path.isfile(liteserver_pubkey)
+	}
+
+	color_print("{cyan}===[ Services status ]==={endc}")
+	for item in statuses.items():
+		status = '{green}enabled{endc}' if item[1] else '{red}disabled{endc}'
+		color_print(f"{item[0]}: {status}")
+
+	node_args = get_node_args()
+	color_print("{cyan}===[ Node arguments ]==={endc}")
+	for key, value in node_args.items():
+		print(f"{key}: {value}")
 #end define
 
 

mytoninstaller/node_args.py

@@ -0,0 +1,51 @@
+
+
+def get_validator_service():
+    path = '/etc/systemd/system/validator.service'
+    with open(path, 'r') as f:
+        return f.read()
+
+
+def get_node_start_command():
+    service = get_validator_service()
+    for line in service.split('\n'):
+        if 'ExecStart' in line:
+            return line.split('=')[1].strip()
+
+
+def get_node_args(command: str = None):
+    if command is None:
+        command = get_node_start_command()
+    result = {}
+    key = ''
+    for c in command.split(' ')[1:]:
+        if c.startswith('--') or c.startswith('-'):
+            if key:
+                result[key] = ''
+            key = c
+        elif key:
+            result[key] = c
+            key = ''
+    return result
+
+
+def set_node_arg(arg_name: str, arg_value: str = ''):
+    """
+    :param arg_name:
+    :param arg_value: arg value. if None, remove the arg; if empty string, argument is set without value
+    :return:
+    """
+    assert arg_name.startswith('-'), 'arg_name must start with "-" or "--"'
+    service = get_validator_service()
+    command = get_node_start_command()
+    if command is None:
+        raise Exception('Cannot find node start command in service file')
+    args = get_node_args(command)
+    if arg_value is None:
+        args.pop(arg_name, None)
+    else:
+        args[arg_name] = arg_value
+    new_command = command.split(' ')[0] + ' ' + ' '.join([f'{k} {v}' for k, v in args.items()])
+    new_service = service.replace(command, new_command)
+    with open('/etc/systemd/system/validator.service', 'w') as f:
+        f.write(new_service)