restore & save seed in dev mode

Author: reliveyy

images/utils/launcher/check_wallets.py

@@ -18,30 +18,6 @@
 # from .table import ServiceTable
 #
 #
-# class CFHeaderState:
-#     def __init__(self):
-#         self.current = 0
-#         self.total = 0
-#         self.ready = False
-#
-#
-# class Action:
-#     def __init__(self, node_manager: NodeManager):
-#         self.logger = logging.getLogger("launcher.CheckWalletsAction")
-#         self.node_manager = node_manager
-#         self.lnd_cfheaders = {}
-#
-#     @property
-#     def shell(self):
-#         return self.node_manager.shell
-#
-#     @property
-#     def config(self):
-#         return self.node_manager.config
-#
-#     @property
-#     def network(self) -> XudNetwork:
-#         return self.config.network
 #
 #     def lnd_has_unlock_log_line(self, c):
 #         pass
@@ -142,18 +118,6 @@
 #             except Exception as e:
 #                 raise RuntimeError("Failed to restart xud") from e
 #
-#     @staticmethod
-#     def _get_percentage(current, total):
-#         if total == 0:
-#             return "0.00%% (%d/%d)" % (current, total)
-#         if current >= total:
-#             return "100.00%% (%d/%d)" % (current, total)
-#         p = current / total * 100
-#         if p > 0.005:
-#             p = p - 0.005
-#         else:
-#             p = 0
-#         return "%.2f%% (%d/%d)" % (p, current, total)
 #
 #     def _print_lnd_cfheaders(self, erase_last_line=True):
 #         services = {}
@@ -301,23 +265,6 @@
 #         if self.node_manager.newly_installed:
 #             print()
 #
-#     def xucli_create_wrapper(self, xud):
-#         counter = 0
-#         ok = False
-#         while counter < 3:
-#             try:
-#                 xud.cli("create", self.shell)
-#                 while True:
-#                     confirmed = self.shell.confirm("YOU WILL NOT BE ABLE TO DISPLAY YOUR XUD SEED AGAIN. Press ENTER to continue...")
-#                     if confirmed:
-#                         break
-#                 ok = True
-#                 break
-#             except (PasswordNotMatch, InvalidPassword):
-#                 counter += 1
-#                 continue
-#         if not ok:
-#             raise Exception("Failed to create wallets")
 #
 #     def xucli_restore_wrapper(self, xud):
 #         counter = 0
@@ -337,81 +284,6 @@
 #         if not ok:
 #             raise Exception("Failed to restore wallets")
 #
-#     def check_backup_dir(self, backup_dir):
-#         assert not backup_dir.startswith("/mnt/hostfs")
-#
-#         hostfs_dir = get_hostfs_file(backup_dir)
-#
-#         if not os.path.exists(hostfs_dir):
-#             return False, "not existed"
-#
-#         if not os.path.isdir(hostfs_dir):
-#             return False, "not a directory"
-#
-#         if not os.access(hostfs_dir, os.R_OK):
-#             return False, "not readable"
-#
-#         if not os.access(hostfs_dir, os.W_OK):
-#             return False, "not writable"
-#
-#         return True, None
-#
-#     def check_restore_dir(self, restore_dir):
-#         return self.check_backup_dir(restore_dir)
-#
-#     def check_restore_dir_files(self, restore_dir):
-#         files = os.listdir(get_hostfs_file(restore_dir))
-#         contents = []
-#         if "xud" in files:
-#             contents.append("xud")
-#         if "lnd-BTC" in files:
-#             contents.append("lndbtc")
-#         if "lnd-LTC" in files:
-#             contents.append("lndltc")
-#         return contents
-#
-#     def setup_backup_dir(self):
-#         if self.config.backup_dir:
-#             return
-#
-#         backup_dir = None
-#
-#         while True:
-#             reply = self.shell.input("Enter path to backup location: ")
-#             reply = reply.strip()
-#             if len(reply) == 0:
-#                 continue
-#
-#             backup_dir = normalize_path(reply)
-#
-#             print("Checking backup location... ", end="")
-#             sys.stdout.flush()
-#             ok, reason = self.check_backup_dir(backup_dir)
-#             if ok:
-#                 print("OK.")
-#                 break
-#             else:
-#                 print(f"Failed. ", end="")
-#                 self.logger.debug(f"Failed to check backup dir {backup_dir}: {reason}")
-#                 sys.stdout.flush()
-#                 r = self.shell.no_or_yes("Retry?")
-#                 if r == "no":
-#                     self.node_manager.down()
-#                     raise FatalError("Backup directory not available")
-#
-#         self.config.backup_dir = backup_dir
-#
-#     def is_backup_available(self):
-#         if self.config.backup_dir is None:
-#             return False
-#
-#         ok, reason = self.check_backup_dir(self.config.backup_dir)
-#
-#         if not ok:
-#             return False
-#
-#         return True
-#
 #     def setup_restore_dir(self) -> None:
 #         """This function will try to interactively setting up restore_dir. And
 #         store it in self._config.restore_dir
@@ -463,15 +335,6 @@
 #
 #         self.config.restore_dir = restore_dir
 #
-#     def _get_service(self, name: str) -> Node:
-#         return self.node_manager.get_node(name)
-#
-#     def _create_wallets(self):
-#         try:
-#             self.xucli_create_wrapper(xud)
-#             break
-#         except:
-#             pass
 #
 #     def _restore_wallets(self):
 #         self.setup_restore_dir()
@@ -493,8 +356,6 @@
 #
 #         self.config.restore_dir = None
 #
-#     def _has_wallets(self) -> bool:
-#         pass
 #
 #     def _setup_wallets(self) -> None:
 #         xud = self._get_service("xud")

images/utils/launcher/config/config.py

@@ -36,6 +36,9 @@ def __init__(self):
         self.network = os.environ["NETWORK"]
 
         self.network_dir = os.path.join("/root", self.network)
+        self.host_network_dir = os.environ["NETWORK_DIR"]
+        self.host_home = os.environ["HOST_HOME"]
+        self.host_pwd = os.environ["HOST_PWD"]
         self.backup_dir = None
         self.restore_dir = None
 
@@ -1030,20 +1033,38 @@ def expand_vars(self, value):
     def logs_dir(self) -> str:
         return os.path.join(self.network_dir, "logs")
 
+    @property
+    def host_logs_dir(self) -> str:
+        return os.path.join(self.host_network_dir, "logs")
+
     @property
     def data_dir(self) -> str:
         return os.path.join(self.network_dir, "data")
 
+    @property
+    def host_data_dir(self) -> str:
+        return os.path.join(self.host_network_dir, "data")
+
     @property
     def logfile(self) -> str:
         filename = f"{self.network}.log"
         return os.path.join(self.logs_dir, filename)
 
+    @property
+    def host_logfile(self) -> str:
+        filename = f"{self.network}.log"
+        return os.path.join(self.host_logs_dir, filename)
+
     @property
     def dumpfile(self) -> str:
         filename = f"config.sh"
         return os.path.join(self.logs_dir, filename)
 
+    @property
+    def host_dumpfile(self) -> str:
+        filename = f"config.sh"
+        return os.path.join(self.host_logs_dir, filename)
+
     def dump(self) -> None:
         """Dump xud-docker configurations as bash key-value file in logs_dir"""
         prefix = "XUD_DOCKER"
@@ -1134,5 +1155,6 @@ def dump_node_attr(node: str, attr: str) -> None:
                     # dump_node_attr(node, "cex_api_secret")
                     dump_node_attr(node, "margin")
 
-    def expand_path(self, value) -> str:
-        pass
+    @property
+    def dev_mode(self) -> bool:
+        return self.args.dev

images/utils/launcher/node/__init__.py

@@ -211,7 +211,7 @@ def create_docker_network(self):
 
     def get_service(self, name: str) -> Node:
         try:
-            return self.valid_nodes[name]
+            return self.nodes[name]
         except KeyError as e:
             raise ServiceNotFound(name) from e
 
@@ -296,7 +296,8 @@ def update(self) -> None:
             reply = yes_or_no(
                 "A new version is available. Would you like to upgrade (Warning: this may restart your environment and cancel all open orders)?")
             if reply == "yes":
-                self.image_manager.update_images()
+                for img in pulls:
+                    img.pull()
 
         b1 = len(pulls) == 0
         b2 = functools.reduce(lambda r, item: r and item == "NONE", containers.values(), True)

images/utils/launcher/node/base.py

@@ -264,22 +264,24 @@ def exec(self, command: str) -> Tuple[int, str]:
         exit_code, output = self.container.exec_run(command)
         return exit_code, output.decode()
 
-    def cli(self, command: str, exception=False) -> None:
+    def cli(self, command: str, exception=False, parse_output=None) -> None:
         if self.mode != "native":
             return
 
         try:
             full_cmd = "%s %s" % (self._cli, command)
+            logger.debug("[Execute] %s (interactive)", full_cmd)
             # FIXME use blocking docker client here
             output = exec_command(self.client.api, self.container_name, full_cmd)
-            logger.debug("[Execute] %s (interactive)\n%s", full_cmd, output)
             try:
                 self.extract_exception(command, output)
             except KeyboardInterrupt:
                 raise
             except:
                 if exception:
                     raise
+            if parse_output:
+                parse_output(output)
         except docker.errors.NotFound:
             # FIXME use self.container
             raise ContainerNotFound(self.name)

images/utils/launcher/node/image.py

@@ -280,6 +280,12 @@ def get_update_action(self) -> str:
         logger.info("Image %s: status=%s, action=%s", self.name, status, action)
         return action
 
+    def pull(self):
+        print("Pulling %s..." % self.pull_image)
+        repo, tag = self.pull_image.split(":")
+        output = self.client.api.pull(repo, tag=tag, stream=True, decode=True)
+        print_status(output)
+
 
 class ImageManager:
     config: Config
@@ -360,13 +366,3 @@ def check_for_updates(self) -> Dict[Image, str]:
                 print("WARNING: %s" % e)
 
         return result
-
-    def update_images(self) -> None:
-        for image in self.images.values():
-            status = image.status
-            pull_image = image.pull_image
-            if status in ["LOCAL_MISSING", "LOCAL_OUTDATED"]:
-                print("Pulling %s..." % pull_image)
-                repo, tag = pull_image.split(":")
-                output = self.client.api.pull(repo, tag=tag, stream=True, decode=True)
-                print_status(output)