add --only-node flag

Author: yungwine

modules/backups.py

@@ -27,6 +27,11 @@ def create_tmp_ton_dir(self):
         self.create_keyring(dir_name_db)
         return dir_name
 
+    @staticmethod
+    def run_create_backup(args):
+        backup_script_path = pkg_resources.resource_filename('mytonctrl', 'scripts/create_backup.sh')
+        return subprocess.run(["bash", backup_script_path] + args, timeout=5)
+
     def create_backup(self, args):
         if len(args) > 1:
             color_print("{red}Bad args. Usage:{endc} create_backup [filename]")
@@ -35,8 +40,7 @@ def create_backup(self, args):
         command_args = ["-m", self.ton.local.buffer.my_work_dir, "-t", tmp_dir]
         if len(args) == 1:
             command_args += ["-d", args[0]]
-        backup_script_path = pkg_resources.resource_filename('mytonctrl', 'scripts/create_backup.sh')
-        process = subprocess.run(["bash", backup_script_path] + command_args, timeout=5)
+        process = self.run_create_backup(command_args)
 
         if process.returncode == 0:
             color_print("create_backup - {green}OK{endc}")

mytoninstaller/mytoninstaller.py

@@ -30,7 +30,7 @@
 	enable_ls_proxy,
 	enable_ton_storage,
 	enable_ton_storage_provider,
-	EnableMode, ConfigureFromBackup
+	EnableMode, ConfigureFromBackup, ConfigureOnlyNode
 )
 from mytoninstaller.config import (
 	CreateLocalConfig,
@@ -280,7 +280,8 @@ def General(local, console):
 		ox = sys.argv.index("--only-mtc")
 		local.buffer.only_mtc = str2bool(sys.argv[ox+1])
 	if "--only-node" in sys.argv:
-		pass
+		ox = sys.argv.index("--only-node")
+		local.buffer.only_node = str2bool(sys.argv[ox+1])
 	if "--backup" in sys.argv:
 		bx = sys.argv.index("--backup")
 		backup = sys.argv[bx+1]
@@ -297,6 +298,7 @@ def General(local, console):
 	CreateSymlinks(local)
 	EnableMode(local)
 	ConfigureFromBackup(local)
+	ConfigureOnlyNode(local)
 #end define
 
 

mytoninstaller/settings.py

@@ -960,3 +960,20 @@ def ConfigureFromBackup(local):
 		args = ["chown", '-R', user + ':' + user, local.buffer.keys_dir]
 		subprocess.run(args)
 		set_external_ip(local, node_ip)
+
+
+def ConfigureOnlyNode(local):
+	if not local.buffer.only_node:
+		return
+	from modules.backups import BackupModule
+	mconfig_path = local.buffer.mconfig_path
+	mconfig_dir = get_dir_from_path(mconfig_path)
+	local.add_log("start ConfigureOnlyNode function", "info")
+
+	process = BackupModule.run_create_backup(["-m", mconfig_dir, ])
+	if process.returncode != 0:
+		local.add_log("Backup creation failed", "error")
+		return
+	local.add_log("Backup successfully created. Use this file on the controller server with `--only-mtc` flag on installation.", "info")
+
+	stop_service(local, 'mytoncore')

scripts/install.sh

@@ -32,6 +32,9 @@ show_help_and_exit() {
     echo ' -n  NETWORK      Specify the network (mainnet or testnet)'
     echo ' -v  VERSION      Specify the ton node version (commit, branch, or tag)'
     echo ' -u  USER         Specify the user to be used for MyTonCtrl installation'
+    echo ' -p  PATH         Provide backup file for MyTonCtrl installation'
+    echo ' -o               Install only MyTonCtrl. Must be used with -p'
+    echo ' -l               Install only TON node'
     echo ' -h               Show this help'
     exit
 }
@@ -46,11 +49,12 @@ telemetry=true
 ignore=false
 dump=false
 only_mtc=false
+only_node=false
 backup=none
 cpu_required=16
 mem_required=64000000  # 64GB in KB
 
-while getopts ":c:tidoa:r:b:m:n:v:u:p:h" flag; do
+while getopts ":c:tidola:r:b:m:n:v:u:p:h" flag; do
     case "${flag}" in
         c) config=${OPTARG};;
         t) telemetry=false;;
@@ -64,6 +68,7 @@ while getopts ":c:tidoa:r:b:m:n:v:u:p:h" flag; do
         v) ton_node_version=${OPTARG};;
         u) user=${OPTARG};;
         o) only_mtc=true;;
+        l) only_node=true;;
         p) backup=${OPTARG};;
         h) show_help_and_exit;;
         *)
@@ -154,7 +159,7 @@ if [ "${user}" = "" ]; then  # no user
     fi
 fi
 echo "User: $user"
-python3 -m mytoninstaller -u ${user} -t ${telemetry} --dump ${dump} -m ${mode} --only-mtc ${only_mtc} --backup ${backup}
+python3 -m mytoninstaller -u ${user} -t ${telemetry} --dump ${dump} -m ${mode} --only-mtc ${only_mtc} --backup ${backup} --only-node ${only_node}
 
 # set migrate version
 migrate_version=1