Initial commit

menus/new_tab.py

@@ -0,0 +1,77 @@
+import arcade, arcade.gui, os, json
+
+from utils.constants import button_style
+from utils.preload import button_texture, button_hovered_texture
+from utils.utils import UIFocusTextureButton
+
+from arcade.gui.experimental.focus import UIFocusGroup
+
+class NewTab(arcade.gui.UIView):
+    def __init__(self, pypresence_client, current_mode, current_music_name, current_length, current_music_player, queue, loaded_sounds, shuffle):
+        super().__init__()
+
+        self.current_mode = current_mode
+        self.current_music_name = current_music_name
+        self.current_length = current_length
+        self.current_music_player = current_music_player
+        self.queue = queue
+        self.loaded_sounds = loaded_sounds
+        self.shuffle = shuffle
+
+        with open("settings.json", "r") as file:
+            self.settings_dict = json.load(file)
+
+        self.tab_options = self.settings_dict.get("tab_options", ["~/Music", "~/Downloads"])
+        self.playlists = self.settings_dict.get("playlists", {})
+
+        self.pypresence_client = pypresence_client
+        self.pypresence_client.update(state="Adding new tab", start=self.pypresence_client.start_time)
+
+    def on_show_view(self):
+        super().on_show_view()
+
+        self.anchor = self.add_widget(UIFocusGroup(size_hint=(1, 1)))
+        self.box = self.anchor.add(arcade.gui.UIBoxLayout(space_between=10), anchor_x="center", anchor_y="center")
+
+        if self.current_mode == "files":
+            self.new_tab_label = self.box.add(arcade.gui.UILabel(text="New Tab Path:", font_name="Protest Strike", font_size=32))
+            self.new_tab_input = self.box.add(arcade.gui.UIInputText(font_name="Protest Strike", font_size=32, width=self.window.width / 2, height=self.window.height / 10))
+            self.new_tab_button = self.box.add(UIFocusTextureButton(texture=button_texture, texture_hovered=button_hovered_texture, text='Add new tab', style=button_style, width=self.window.width / 2, height=self.window.height / 10))
+            self.new_tab_button.on_click = lambda event: self.add_tab()
+        elif self.current_mode == "playlist":
+            self.new_tab_label = self.box.add(arcade.gui.UILabel(text="New Playlist Name:", font_name="Protest Strike", font_size=32))
+            self.new_tab_input = self.box.add(arcade.gui.UIInputText(font_name="Protest Strike", font_size=32, width=self.window.width / 2, height=self.window.height / 10))
+            self.new_tab_button = self.box.add(UIFocusTextureButton(texture=button_texture, texture_hovered=button_hovered_texture, text='Add new Playlist', style=button_style, width=self.window.width / 2, height=self.window.height / 10))
+            self.new_tab_button.on_click = lambda event: self.add_tab()
+
+        self.back_button = self.anchor.add(UIFocusTextureButton(texture=button_texture, texture_hovered=button_hovered_texture, text='<--', style=button_style, width=100, height=50), anchor_x="left", anchor_y="top", align_x=5, align_y=-5)
+        self.back_button.on_click = lambda event: self.main_exit()
+
+        self.anchor.detect_focusable_widgets()
+
+    def add_tab(self):
+        if self.current_mode == "files":
+            tab_path = self.new_tab_input.text
+
+            if not tab_path:
+                return
+
+            if not os.path.exists(os.path.expanduser(tab_path)) or not os.path.isdir(os.path.expanduser(tab_path)) or not os.path.isabs(os.path.expanduser(tab_path)):
+                return
+
+            if tab_path in self.tab_options or os.path.expanduser(tab_path) in self.tab_options:
+                return
+
+            self.tab_options.append(tab_path)
+            self.settings_dict["tab_options"] = self.tab_options
+
+        elif self.current_mode == "playlist":
+            self.playlists[self.new_tab_input.text] = []
+            self.settings_dict["playlists"] = self.playlists
+
+        with open("settings.json", "w") as file:
+            file.write(json.dumps(self.settings_dict))
+
+    def main_exit(self):
+        from menus.main import Main
+        self.window.show_view(Main(self.pypresence_client, self.current_mode, self.current_music_name, self.current_length, self.current_music_player, self.queue, self.loaded_sounds, self.shuffle))