import re import os from pathlib import Path from textual.app import App, ComposeResult from textual.containers import Container, Horizontal, Vertical, VerticalScroll from textual.widgets import ( Header, Footer, Button, Label, Input, ListItem, ListView, Rule, RadioButton, RadioSet, ) from textual.screen import Screen, ModalScreen from src.core.configuration import ( get_available_configs, create_new_config, load_config, save_config, ) # ────────────────────────────────────────────────────────────────────────────── # Modal: create a new ReaperFiles-* config folder # ────────────────────────────────────────────────────────────────────────────── class NewConfigModal(ModalScreen[str]): """Modal to enter a name for a new configuration.""" DEFAULT_CSS = """ NewConfigModal { align: center middle; } #new_config_dialog { width: 50; height: auto; border: thick $background 80%; background: $surface; padding: 1 2; } #new_config_title { text-style: bold; margin-bottom: 1; } #new_config_buttons { height: auto; margin-top: 1; } #new_config_buttons Button { width: 1fr; margin: 0 1; } """ def compose(self) -> ComposeResult: with Vertical(id="new_config_dialog"): yield Label("Enter new configuration name:", id="new_config_title") yield Input(placeholder="e.g. MyServer", id="new_config_input") with Horizontal(id="new_config_buttons"): yield Button("Create", variant="success", id="btn_create") yield Button("Cancel", variant="primary", id="btn_cancel") def _get_sanitized_name(self) -> str: raw = self.query_one("#new_config_input", Input).value.strip() return re.sub(r"[^a-zA-Z0-9_-]+", "-", raw).strip("-") def on_button_pressed(self, event: Button.Pressed) -> None: if event.button.id == "btn_create": name = self._get_sanitized_name() if name: self.dismiss(name) elif event.button.id == "btn_cancel": self.dismiss(None) def on_key(self, event) -> None: if event.key == "enter": name = self._get_sanitized_name() if name: self.dismiss(name) elif event.key == "escape": self.dismiss(None) # ────────────────────────────────────────────────────────────────────────────── # Screen 1: pick (or create) a ReaperFiles-* config # ────────────────────────────────────────────────────────────────────────────── class ConfigSelectionScreen(Screen): """Screen to select or create a Reaper configuration.""" DEFAULT_CSS = """ ConfigSelectionScreen { align: center middle; } #config_sel_container { width: 60; height: auto; border: solid green; padding: 1 2; } #config_sel_title { text-style: bold; color: green; margin-bottom: 1; content-align: center middle; width: 100%; } #config_list_container { height: 10; max-height: 20; border: solid $primary; margin-bottom: 1; } #config_sel_actions { height: auto; margin-top: 1; } #config_sel_actions Button { width: 1fr; margin: 0 1; } """ def compose(self) -> ComposeResult: yield Header(show_clock=True) with Container(id="config_sel_container"): yield Label("Disco Reaper — Select Configuration", id="config_sel_title") with VerticalScroll(id="config_list_container"): yield ListView(id="config_list") with Horizontal(id="config_sel_actions"): yield Button("New Config", id="btn_new_config", variant="success") yield Button("Exit", id="btn_exit", variant="error") yield Footer() def on_mount(self) -> None: self.refresh_configs() def on_screen_resume(self) -> None: self.refresh_configs() def refresh_configs(self) -> None: configs = get_available_configs() lv = self.query_one("#config_list", ListView) lv.clear() for c in configs: lv.append(ListItem(Label(f"ReaperFiles-{c}"), name=c)) def on_list_view_selected(self, event: ListView.Selected) -> None: cfg_name = event.item.name cfg_path = Path(f"ReaperFiles-{cfg_name}") / "config.yaml" from src.ui.mode_screen import ModeScreen self.app.push_screen(ModeScreen(cfg_name, cfg_path)) def on_button_pressed(self, event: Button.Pressed) -> None: if event.button.id == "btn_new_config": def cb(name: str | None): if name: create_new_config(name) self.refresh_configs() # Immediately open the ConfigScreen for the new config cfg_path = Path(f"ReaperFiles-{name}") / "config.yaml" def on_config_saved(saved: bool = False): if saved: self.refresh_configs() # Navigate into the ModeScreen from src.ui.mode_screen import ModeScreen self.app.push_screen(ModeScreen(name, cfg_path)) self.app.push_screen(ConfigScreen(name, cfg_path), on_config_saved) self.app.push_screen(NewConfigModal(), cb) elif event.button.id == "btn_exit": self.app.exit() # ────────────────────────────────────────────────────────────────────────────── # Screen 2: edit config + pick mode + start # ────────────────────────────────────────────────────────────────────────────── _MODE_MAP = { "radio_direct": "direct_transfer", "radio_backup": "backup_transfer", "radio_bkonly": "backup_only", } _MODE_LABELS = { "direct_transfer": "radio_direct", "backup_transfer": "radio_backup", "backup_only": "radio_bkonly", } _PLAT_MAP = { "radio_fluxer": "fluxer", "radio_stoat": "stoat", } _PLAT_LABELS = { "fluxer": "radio_fluxer", "stoat": "radio_stoat", } class ConfigScreen(Screen): """Configuration screen — Discord config, tool mode, and target platform.""" DEFAULT_CSS = """ ConfigScreen { align: center middle; } #cfg_outer { width: 100%; height: 100%; align: center top; } #cfg_container { width: 80%; height: 100%; layout: vertical; border: solid green; padding: 1 2; margin: 2 0; } #cfg_scroll { width: 100%; height: 1fr; margin-bottom: 1; } #cfg_title { text-style: bold; color: green; margin-bottom: 1; content-align: center middle; width: 100%; } .section_title { text-style: bold; color: cyan; margin-top: 1; margin-bottom: 0; } .field_label { margin-top: 1; } #cfg_container Input { margin-bottom: 0; } #mode_radio, #plat_radio { height: auto; margin: 0 0 0 2; } #target_section { height: auto; } #cfg_actions { height: auto; margin-top: 1; margin-bottom: 0; dock: bottom; } #cfg_actions Button { width: 1fr; margin: 0 1; } """ BINDINGS = [("escape", "go_back", "Back")] def __init__(self, cfg_name: str, cfg_path: Path, *args, **kwargs): super().__init__(*args, **kwargs) self.cfg_name = cfg_name self.cfg_path = cfg_path self.config = load_config(cfg_path) def compose(self) -> ComposeResult: yield Header(show_clock=True) with Container(id="cfg_outer"): with Container(id="cfg_container"): yield Label(f"Configuration — {self.cfg_name}", id="cfg_title") with VerticalScroll(id="cfg_scroll"): # ── Discord ────────────────────────────────────────────── yield Label("Discord", classes="section_title") yield Label("Bot Token:", classes="field_label") yield Input( value=self.config.discord_bot_token or "", id="inp_discord_token", password=True, ) yield Label("Server ID:", classes="field_label") yield Input( value=self.config.discord_server_id or "", id="inp_discord_server", ) # ── Reaper Mode ────────────────────────────────────────── yield Label("Reaper Mode", classes="section_title") cur_mode = self.config.tool_mode or "backup_only" with RadioSet(id="mode_radio"): yield RadioButton( "Shuttle Transfer (direct migration)", id="radio_direct", value=(cur_mode == "direct_transfer"), ) yield RadioButton( "Backup & Migrate (backup first, then migrate)", id="radio_backup", value=(cur_mode == "backup_transfer"), ) yield RadioButton( "Backup Only (local backup, no migration)", id="radio_bkonly", value=(cur_mode == "backup_only"), ) # ── Target Platform (hidden for backup_only) ───────────── with Vertical(id="target_section"): yield Label("Target Platform", classes="section_title") cur_plat = self.config.target_platform or "none" with RadioSet(id="plat_radio"): yield RadioButton( "Fluxer", id="radio_fluxer", value=(cur_plat == "fluxer"), ) yield RadioButton( "Stoat", id="radio_stoat", value=(cur_plat == "stoat"), ) yield Label("Bot Token:", classes="field_label") yield Input( value=self.config.target_bot_token or "", id="inp_target_token", password=True, ) yield Label("Community / Server ID:", classes="field_label") yield Input( value=self.config.target_server_id or "", id="inp_target_server", ) yield Label("Target API URL:", classes="field_label") yield Input( value=self.config.target_api_url or "default", id="inp_target_api", ) yield Rule() with Horizontal(id="cfg_actions"): yield Button("Save Configuration", variant="success", id="btn_save") yield Button("Back", id="btn_back") yield Footer() def on_mount(self) -> None: self._toggle_target_section() # ── show / hide the target platform section ────────────────────────── def _get_selected_mode(self) -> str: for rb in self.query("#mode_radio RadioButton"): if rb.value: return _MODE_MAP.get(rb.id, "backup_only") return "backup_only" def _get_selected_platform(self) -> str: for rb in self.query("#plat_radio RadioButton"): if rb.value: return _PLAT_MAP.get(rb.id, "fluxer") return "fluxer" def _toggle_target_section(self) -> None: section = self.query_one("#target_section") section.display = self._get_selected_mode() != "backup_only" def on_radio_set_changed(self, event: RadioSet.Changed) -> None: if event.radio_set.id == "mode_radio": self._toggle_target_section() # ── save / start ───────────────────────────────────────────────────── def _collect_and_save(self) -> None: self.config.discord_bot_token = self.query_one("#inp_discord_token", Input).value.strip() or self.config.discord_bot_token self.config.discord_server_id = self.query_one("#inp_discord_server", Input).value.strip() or self.config.discord_server_id self.config.tool_mode = self._get_selected_mode() if self.config.tool_mode != "backup_only": self.config.target_platform = self._get_selected_platform() self.config.target_bot_token = self.query_one("#inp_target_token", Input).value.strip() or self.config.target_bot_token self.config.target_server_id = self.query_one("#inp_target_server", Input).value.strip() or self.config.target_server_id self.config.target_api_url = self.query_one("#inp_target_api", Input).value.strip() or self.config.target_api_url else: self.config.target_platform = "none" save_config(self.config, self.cfg_path) def _launch_mode(self) -> None: pass # No longer needed def action_go_back(self) -> None: self.app.pop_screen() def on_button_pressed(self, event: Button.Pressed) -> None: if event.button.id == "btn_back": self.app.pop_screen() elif event.button.id == "btn_save": self._collect_and_save() self.notify("Configuration saved.", severity="information") self.dismiss(True) # ────────────────────────────────────────────────────────────────────────────── # App # ────────────────────────────────────────────────────────────────────────────── class ReaperApp(App): def on_mount(self) -> None: self.push_screen(ConfigSelectionScreen()) self.theme = "dracula" def run_disco_reaper_tui(): app = ReaperApp() app.run()