From 46faa5c01c9b6a26d9ddef88e0b9ca825cb2ecbd Mon Sep 17 00:00:00 2001
From: cardosofede <federico.cardoso.e@gmail.com>
Date: Wed, 23 Aug 2023 02:27:24 +0200
Subject: [PATCH] (feat) create launch bot

---
 ui_components/launch_bot_card.py | 92 ++++++++++++++++++++++++++++++++
 1 file changed, 92 insertions(+)
 create mode 100644 ui_components/launch_bot_card.py

diff --git a/ui_components/launch_bot_card.py b/ui_components/launch_bot_card.py
new file mode 100644
index 0000000..8e0a8e6
--- /dev/null
+++ b/ui_components/launch_bot_card.py
@@ -0,0 +1,92 @@
+import os
+import time
+
+from docker_manager import DockerManager
+import streamlit as st
+from streamlit_elements import mui, sync, lazy
+
+import constants
+from utils.os_utils import get_directories_from_directory
+from .dashboard import Dashboard
+
+
+class LaunchBotCard(Dashboard.Item):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.is_master_bot_running = False
+        self._bot_name = None
+        self._image_name = "hummingbot/hummingbot:latest"
+        self._base_bot_config = "master_bot_conf"
+
+    def _set_bot_name(self, event):
+        self._bot_name = event.target.value
+
+    def _set_image_name(self, event):
+        self._image_name = event.target.value
+
+    def _set_base_bot_config(self, event):
+        self._base_bot_config = event.target.value
+
+    def launch_new_bot(self):
+        if self._bot_name and self._image_name:
+            bot_name = f"hummingbot-{self._bot_name}"
+            DockerManager().create_hummingbot_instance(instance_name=bot_name,
+                                                       base_conf_folder=f"{constants.HUMMINGBOT_TEMPLATES}/{self._base_bot_config}/.",
+                                                       target_conf_folder=f"{constants.BOTS_FOLDER}/{bot_name}/.",
+                                                       image=self._image_name)
+            with st.spinner('Starting Master Bot Conf... This process may take a few seconds'):
+                time.sleep(3)
+        else:
+            st.warning("You need to define the bot name and image in order to create one.")
+
+    def __call__(self):
+        active_containers = DockerManager.get_active_containers()
+        self.is_master_bot_running = "hummingbot-master_bot_conf" in active_containers
+        password_file_path = os.path.join(constants.HUMMINGBOT_TEMPLATES + "/master_bot_conf/conf",
+                                          '.password_verification')
+        is_master_password_set = os.path.isfile(password_file_path)
+        with mui.Paper(key=self._key,
+                       sx={"display": "flex", "flexDirection": "column", "borderRadius": 3, "overflow": "hidden"},
+                       elevation=1):
+            with self.title_bar(padding="10px 15px 10px 15px", dark_switcher=False):
+                mui.Typography("🚀 Create Instance", variant="h4")
+            with mui.Grid(container=True, spacing=2, sx={"padding": "10px 15px 10px 15px"}):
+                if not is_master_password_set:
+                    with mui.Grid(item=True, xs=8):
+                        base_warning = "You need to set a master password in order to use the dashboard."
+                        if self.is_master_bot_running:
+                            mui.Alert(f"{base_warning} Master Bot Conf is running, run in a terminal docker attach "
+                                      f"hummingbot-master_bot_conf and set the master password.", severity="success")
+                        else:
+                            mui.Alert(f"{base_warning} Master Bot Conf is not running, press start, attach the instance"
+                                      f" in your terminal and introduce the password to continue.", severity="error")
+                else:
+                    with mui.Grid(item=True, xs=5):
+                        mui.TextField(label="Bot Name", variant="outlined", onChange=lazy(self._set_bot_name),
+                                      sx={"width": "100%"})
+                    with mui.Grid(item=True, xs=5):
+                        mui.TextField(label="Docker Image", defaultValue="hummingbot/hummingbot:latest",
+                                      variant="outlined", onChange=lazy(self._set_image_name),
+                                      sx={"width": "100%"})
+
+                with mui.Grid(item=True, xs=2):
+                    button_text = "Stop" if self.is_master_bot_running else "Start"
+                    color = "error" if self.is_master_bot_running else "success"
+                    icon = mui.icon.Stop if self.is_master_bot_running else mui.icon.PlayCircle
+                    with mui.Button(onClick=self.launch_new_bot,
+                                    color=color,
+                                    variant="outlined",
+                                    sx={"width": "100%", "height": "100%"}):
+                        icon()
+                        mui.Typography(button_text)
+
+                with mui.Grid(item=True, xs=2):
+                    master_configs = [conf.split("/")[-2] for conf in
+                                      get_directories_from_directory(constants.HUMMINGBOT_TEMPLATES) if
+                                      "bot_conf" in conf]
+                    with mui.FormControl(variant="standard", sx={"width": "100%"}):
+                        mui.FormHelperText("Base Bot Config")
+                        with mui.Select(label="Base Bot Config", defaultValue=master_configs[0],
+                                        variant="standard", onChange=lazy(self._set_base_bot_config)):
+                            for master_config in master_configs:
+                                mui.MenuItem(master_config, value=master_config)