Merge pull request #221 from yungwine/mytonctrl2_dev

node args

Author: igroman787

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, set_node_arg
 from mytoninstaller.utils import GetInitBlock
 from mytoncore.utils import dict2b64, str2bool, b642dict
 
@@ -62,6 +63,7 @@ def inject_globals(func):
 	console.name = "MyTonInstaller"
 	console.color = console.RED
 	console.AddItem("status", inject_globals(Status), "Print TON component status")
+	console.AddItem("set_node_argument", inject_globals(set_node_argument), "Set node argument")
 	console.AddItem("enable", inject_globals(Enable), "Enable some function")
 	console.AddItem("update", inject_globals(Enable), "Update some function: 'JR' - jsonrpc.  Example: 'update JR'") 
 	console.AddItem("plsc", inject_globals(PrintLiteServerConfig), "Print lite-server config")
@@ -111,16 +113,50 @@ def Status(local, args):
 	liteserver_key = keys_dir + "liteserver"
 	liteserver_pubkey = liteserver_key + ".pub"
 
+	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
+
+
+def restart_node():
+	exit_code = run_as_root(["systemctl", "daemon-reload"])
+	if not exit_code:
+		raise Exception(f"`systemctl daemon-reload` failed with exit code {exit_code}")
+	exit_code = run_as_root(["systemctl", "restart", "validator"])
+	if not exit_code:
+		raise Exception(f"`systemctl restart validator` failed with exit code {exit_code}")
+#end define
 
-	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)
+def set_node_argument(local, args):
+	if len(args) < 1:
+		color_print("{red}Bad args. Usage:{endc} set_node_argument <arg-name> [arg-value] [-d (to delete)]")
+		return
+	arg_name = args[0]
+	if len(args) == 1:
+		set_node_arg(arg_name)
+	else:
+		arg_value = args[1]
+		if arg_value == "-d":
+			set_node_arg(arg_name, None)
+		else:
+			set_node_arg(arg_name, arg_value)
+	restart_node()
+	color_print("set_node_argument - {green}OK{endc}")
 #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)