utim-cli 1.0.0__py3-none-any.whl

utim_cli/share_tui.py

@@ -0,0 +1,554 @@
+import os
+import subprocess
+from prompt_toolkit import Application
+from prompt_toolkit.layout import Layout, HSplit, Window
+from prompt_toolkit.layout.controls import FormattedTextControl
+from prompt_toolkit.widgets import TextArea, Frame
+from prompt_toolkit.key_binding import KeyBindings
+from prompt_toolkit.styles import Style as PTStyle
+from prompt_toolkit.filters import Condition
+from prompt_toolkit.application import run_in_terminal
+
+from rich.console import Console
+from rich.panel import Panel
+from rich.text import Text
+
+from utim_cli.share import ShareManager, EXCLUDE_OPTIONS, EXPIRY_OPTIONS, ShareRecord
+
+# Initialize rich console with standard styling matching terminal-ui-design guide
+theme_console = Console()
+
+def copy_to_clipboard(text: str) -> bool:
+    """Helper to copy text to clipboard using clip.exe on Windows."""
+    try:
+        process = subprocess.Popen("clip", stdin=subprocess.PIPE, shell=True)
+        process.communicate(input=text.encode('utf-8'))
+        return process.returncode == 0
+    except Exception:
+        return False
+
+def _run_checkbox_dialog(items, title="", legend=""):
+    """
+    Checklist dialog using prompt_toolkit.
+    Allows toggling options with SPACE and accepting with ENTER.
+    """
+    sel = [0]
+    selected_keys = set(item["key"] for item in items)  # Pre-select all by default
+    N = len(items)
+    act = [None]
+
+    def content():
+        out = []
+        out.append(('bold #42bcf5', f'\n  {title}\n'))
+        out.append(('class:dim',    f'  {legend}\n\n'))
+
+        for i, item in enumerate(items):
+            is_selected = item["key"] in selected_keys
+            highlighted = i == sel[0]
+            
+            bg = 'bg:#1a3a2a bold #cdd6f4' if highlighted else ''
+            check = ' [x] ' if is_selected else ' [ ] '
+            check_style = 'bold #a6e3a1' if is_selected else 'dim #f38ba8'
+            if highlighted:
+                check_style = 'bold bg:#1a3a2a'
+                
+            out.extend([
+                (bg or check_style, check),
+                (bg or 'bold #cdd6f4', f"{item['name'].ljust(20)}"),
+                (bg or 'dim', f"— {item['desc']}\n")
+            ])
+        out.append(('', '\n'))
+        return out
+
+    kb = KeyBindings()
+
+    @kb.add('up')
+    @kb.add('k')
+    def _up(e):
+        sel[0] = (sel[0] - 1) % N
+        e.app.invalidate()
+
+    @kb.add('down')
+    @kb.add('j')
+    def _dn(e):
+        sel[0] = (sel[0] + 1) % N
+        e.app.invalidate()
+
+    @kb.add('space')
+    def _toggle(e):
+        key = items[sel[0]]["key"]
+        if key in selected_keys:
+            selected_keys.remove(key)
+        else:
+            selected_keys.add(key)
+        e.app.invalidate()
+
+    @kb.add('enter')
+    def _enter(e):
+        act[0] = 'select'
+        e.app.exit()
+
+    @kb.add('escape')
+    @kb.add('q')
+    @kb.add('c-c')
+    def _quit(e):
+        e.app.exit()
+
+    dialog_app = Application(
+        layout=Layout(Window(
+            FormattedTextControl(content),
+            wrap_lines=False,
+        )),
+        key_bindings=kb,
+        full_screen=True,
+        style=PTStyle.from_dict({'dim': '#555577', '': 'bg:#0d0d16 fg:#cdd6f4'}),
+        mouse_support=False,
+    )
+    dialog_app.run()
+    
+    if act[0] == 'select':
+        return list(selected_keys)
+    return None
+
+def _run_expiry_dialog(options, title="", legend=""):
+    """Simple selection list for picking the link expiry duration."""
+    sel = [0]
+    N = len(options)
+    act = [None]
+
+    def content():
+        out = []
+        out.append(('bold #42bcf5', f'\n  {title}\n'))
+        out.append(('class:dim',    f'  {legend}\n\n'))
+
+        for i, opt in enumerate(options):
+            highlighted = i == sel[0]
+            bg = 'bg:#1a3a2a bold #cdd6f4' if highlighted else ''
+            bullet = '➔ ' if highlighted else '• '
+            bullet_style = 'bold #42bcf5' if highlighted else 'dim'
+            
+            out.extend([
+                (bg or bullet_style, f"  {bullet}"),
+                (bg or 'bold #cdd6f4', f"{opt['label']}\n")
+            ])
+        out.append(('', '\n'))
+        return out
+
+    kb = KeyBindings()
+
+    @kb.add('up')
+    @kb.add('k')
+    def _up(e):
+        sel[0] = (sel[0] - 1) % N
+        e.app.invalidate()
+
+    @kb.add('down')
+    @kb.add('j')
+    def _dn(e):
+        sel[0] = (sel[0] + 1) % N
+        e.app.invalidate()
+
+    @kb.add('enter')
+    def _enter(e):
+        act[0] = 'select'
+        e.app.exit()
+
+    @kb.add('escape')
+    @kb.add('q')
+    @kb.add('c-c')
+    def _quit(e):
+        e.app.exit()
+
+    dialog_app = Application(
+        layout=Layout(Window(
+            FormattedTextControl(content),
+            wrap_lines=False,
+        )),
+        key_bindings=kb,
+        full_screen=True,
+        style=PTStyle.from_dict({'dim': '#555577', '': 'bg:#0d0d16 fg:#cdd6f4'}),
+        mouse_support=False,
+    )
+    dialog_app.run()
+    
+    if act[0] == 'select':
+        return options[sel[0]]
+    return None
+
+def _run_shares_dashboard(manager: ShareManager):
+    """
+    Searchable dashboard showing existing shares, remaining time, delete options,
+    and a top option for creating a new share.
+    """
+    current_raw_rows = []
+    filtered_rows = []
+    N_filtered = [0]
+    sel = [0]
+    act = [None]
+    viewport_start = [0]
+
+    def update_filtered_rows(query):
+        nonlocal filtered_rows
+        q = query.strip().lower()
+        if not q:
+            filtered_rows = list(enumerate(current_raw_rows))
+        else:
+            filtered_rows = []
+            for idx, row in enumerate(current_raw_rows):
+                if row["type"] == "new":
+                    filtered_rows.append((idx, row))
+                else:
+                    rec = row["record"]
+                    if (q in rec.name.lower() or 
+                        q in rec.id.lower() or 
+                        q in rec.link.lower() or 
+                        any(q in excl.lower() for excl in rec.excluded)):
+                        filtered_rows.append((idx, row))
+        N_filtered[0] = len(filtered_rows)
+        if N_filtered[0] > 0:
+            sel[0] = min(sel[0], N_filtered[0] - 1)
+        else:
+            sel[0] = 0
+        viewport_start[0] = min(viewport_start[0], sel[0])
+
+    def update_rows():
+        records = manager.get_all()
+        raw_rows = [{"type": "new"}]
+        for r in records:
+            raw_rows.append({"type": "record", "record": r})
+        nonlocal current_raw_rows
+        current_raw_rows = raw_rows
+        update_filtered_rows(search_field.text)
+
+    def get_row_height(row):
+        return 3 if row["type"] == "new" else 4
+
+    def content():
+        import shutil
+        term_h = shutil.get_terminal_size().lines
+        
+        # 10 lines budget for frame, search, legend, spacing
+        available_height = max(5, term_h - 10)
+        
+        if N_filtered[0] == 0:
+            return [
+                ('', '\n'),
+                ('bold #f38ba8', '    ⚠️  No matching shared items found.\n'),
+            ]
+
+        matching_heights = [get_row_height(row) for _, row in filtered_rows]
+
+        # Slide viewport so that sel[0] is visible
+        if sel[0] < viewport_start[0]:
+            viewport_start[0] = sel[0]
+        else:
+            current_height = 0
+            for idx in range(viewport_start[0], sel[0] + 1):
+                current_height += matching_heights[idx]
+            while current_height > available_height and viewport_start[0] < sel[0]:
+                current_height -= matching_heights[viewport_start[0]]
+                viewport_start[0] += 1
+
+        out = []
+        if viewport_start[0] > 0:
+            out.append(('dim yellow', f'    ▲ ... ({viewport_start[0]} more above) ... ▲\n\n'))
+            available_height -= 2
+
+        rendered_height = 0
+        end_idx = viewport_start[0]
+        while end_idx < N_filtered[0] and rendered_height + matching_heights[end_idx] <= available_height:
+            rendered_height += matching_heights[end_idx]
+            end_idx += 1
+
+        if end_idx <= sel[0]:
+            end_idx = sel[0] + 1
+
+        for i in range(viewport_start[0], end_idx):
+            orig_idx, row = filtered_rows[i]
+            highlighted = i == sel[0]
+            bg = 'bg:#1a3a2a bold #cdd6f4' if highlighted else ''
+            
+            if row["type"] == "new":
+                out.extend([
+                    (bg or 'bold #a6e3a1', "  [+] Share New Link\n"),
+                    (bg or 'dim', "      Compress workspace and generate a new share link\n\n")
+                ])
+            else:
+                rec = row["record"]
+                status = rec.time_remaining()
+                status_style = 'bold #a6e3a1' if status != 'Expired' else 'bold #f38ba8'
+                if highlighted:
+                    status_style = 'bold bg:#1a3a2a'
+                
+                excl_part = f"Excluded: {', '.join(rec.excluded) if rec.excluded else 'none'}"
+                
+                out.extend([
+                    (bg or 'bold #cdd6f4', f"  📁  {rec.name} "),
+                    (bg or 'dim', f"({rec.id})"),
+                    (bg or status_style, f" [{status}]".rjust(15) + "\n"),
+                    (bg or 'dim cyan', f"      {rec.link}\n"),
+                    (bg or 'dim', f"      {excl_part}\n\n")
+                ])
+
+        remaining = N_filtered[0] - end_idx
+        if remaining > 0:
+            out.append(('dim yellow', f'    ▼ ... ({remaining} more below) ... ▼\n'))
+
+        return out
+
+    kb = KeyBindings()
+
+    # Tab navigation to shift focus
+    @kb.add('tab')
+    def _tab(e):
+        if e.app.layout.has_focus(search_field):
+            e.app.layout.focus(list_control)
+        else:
+            e.app.layout.focus(search_field)
+
+    @kb.add('s-tab')
+    def _stab(e):
+        if e.app.layout.has_focus(search_field):
+            e.app.layout.focus(list_control)
+        else:
+            e.app.layout.focus(search_field)
+
+    # Search mode navigation overrides
+    @kb.add('up', filter=Condition(lambda: dialog_app.layout.has_focus(search_field)))
+    def _up_search(e):
+        if N_filtered[0] > 0:
+            sel[0] = (sel[0] - 1) % N_filtered[0]
+            e.app.invalidate()
+
+    @kb.add('down', filter=Condition(lambda: dialog_app.layout.has_focus(search_field)))
+    def _down_search(e):
+        if N_filtered[0] > 0:
+            sel[0] = (sel[0] + 1) % N_filtered[0]
+            e.app.invalidate()
+
+    # List mode navigation
+    @kb.add('up', filter=Condition(lambda: dialog_app.layout.has_focus(list_control)))
+    @kb.add('k', filter=Condition(lambda: dialog_app.layout.has_focus(list_control)))
+    def _up_list(e):
+        if N_filtered[0] > 0:
+            sel[0] = (sel[0] - 1) % N_filtered[0]
+            e.app.invalidate()
+
+    @kb.add('down', filter=Condition(lambda: dialog_app.layout.has_focus(list_control)))
+    @kb.add('j', filter=Condition(lambda: dialog_app.layout.has_focus(list_control)))
+    def _down_list(e):
+        if N_filtered[0] > 0:
+            sel[0] = (sel[0] + 1) % N_filtered[0]
+            e.app.invalidate()
+
+    @kb.add('d', filter=Condition(lambda: dialog_app.layout.has_focus(list_control)))
+    def _delete_list(e):
+        if N_filtered[0] > 0:
+            orig_idx, row = filtered_rows[sel[0]]
+            if row["type"] == "record":
+                rec = row["record"]
+                manager.delete(rec.id)
+                update_rows()
+                e.app.invalidate()
+
+    @kb.add('enter')
+    def _enter(e):
+        if N_filtered[0] > 0:
+            orig_idx, row = filtered_rows[sel[0]]
+            if row["type"] == "new":
+                act[0] = ("new", None)
+            else:
+                act[0] = ("details", row["record"])
+            e.app.exit()
+
+    @kb.add('escape')
+    @kb.add('c-c')
+    def _quit(e):
+        e.app.exit()
+
+    @kb.add('q', filter=Condition(lambda: dialog_app.layout.has_focus(list_control)))
+    def _quit_list(e):
+        e.app.exit()
+
+    title_window = Window(FormattedTextControl([('bold #42bcf5', f'\n  📤  Workspace Share Manager\n')]), height=2)
+
+    search_field = TextArea(
+        multiline=False,
+        prompt=" 🔍 Search Shares: ",
+        style="class:search-field",
+    )
+    search_field.buffer.on_text_changed += lambda buf: update_filtered_rows(buf.text)
+    search_frame = Frame(search_field, title="Filter Shared Items (Search Name, ID, Link, or Exclusions)")
+
+    list_control = FormattedTextControl(content, focusable=True)
+    list_window = Window(list_control, wrap_lines=False)
+    list_frame = Frame(list_window, title="Shared Archives")
+
+    def get_legend():
+        if dialog_app.layout.has_focus(search_field):
+            return [
+                ('bold #a6e3a1', " MODE: SEARCH  "),
+                ('dim', "—  Type to filter  |  "),
+                ('bold #cba6f7', "TAB"),
+                ('dim', " to switch to list  |  "),
+                ('bold #f38ba8', "ESC"),
+                ('dim', " to close")
+            ]
+        else:
+            return [
+                ('bold #f9e2af', " MODE: SELECT  "),
+                ('dim', "—  "),
+                ('bold #89b4fa', "UP/DOWN / J/K"),
+                ('dim', " to navigate  |  "),
+                ('bold #a6e3a1', "ENTER"),
+                ('dim', " to select  |  "),
+                ('bold #f38ba8', "d"),
+                ('dim', " to delete active link  |  "),
+                ('bold #cba6f7', "TAB"),
+                ('dim', " to search  |  "),
+                ('bold #f38ba8', "ESC/Q"),
+                ('dim', " to close")
+            ]
+
+    legend_control = FormattedTextControl(get_legend)
+    legend_window = Window(legend_control, height=2)
+
+    layout = Layout(HSplit([
+        title_window,
+        search_frame,
+        list_frame,
+        legend_window
+    ]))
+
+    dialog_app = Application(
+        layout=layout,
+        key_bindings=kb,
+        full_screen=True,
+        style=PTStyle.from_dict({
+            'dim': '#555577',
+            '': 'bg:#0d0d16 fg:#cdd6f4',
+            'frame.border': '#89b4fa',
+            'search-field': 'bg:#1e1e2e fg:#cdd6f4'
+        }),
+        mouse_support=False,
+    )
+
+    update_rows()
+    dialog_app.layout.focus(search_field)
+    dialog_app.run()
+
+    return act[0]
+
+def show_share_details(record: ShareRecord):
+    """Render details of a share record and copy the link to the clipboard."""
+    copied = copy_to_clipboard(record.link)
+    clipboard_msg = " [green](Copied to clipboard!)[/green]" if copied else " [red](Failed to copy to clipboard)[/red]"
+    
+    theme_console.print()
+    panel = Panel(
+        Text.from_markup(
+            f"[bold white]📁 Share Package: {record.name}[/bold white]\n\n"
+            f"[dim]ID:[/dim]          {record.id}\n"
+            f"[dim]Created At:[/dim]  {record.created_at}\n"
+            f"[dim]Expires At:[/dim]  {record.expires_at}\n"
+            f"[dim]Time Left:[/dim]   {record.time_remaining()}\n"
+            f"[dim]Link:[/dim]        [bold cyan]{record.link}[/bold cyan]{clipboard_msg}\n"
+            f"[dim]File Path:[/dim]   {record.file_path}\n"
+            f"[dim]Excluded:[/dim]    {', '.join(record.excluded) if record.excluded else 'none'}"
+        ),
+        title="Share Details",
+        border_style="#42bcf5",
+        expand=False,
+        padding=(1, 2)
+    )
+    theme_console.print(panel)
+    theme_console.print("\n[dim]Press Enter to return to Dashboard...[/dim]")
+    input()
+
+def do_create_share(manager: ShareManager, exclude_keys, expiry_hours, chat_messages):
+    """Run zip compression, upload, copy to clipboard, and print status."""
+    theme_console.print()
+    panel = Panel(
+        Text.from_markup(
+            "[bold yellow]⚡ Building Share Package...[/bold yellow]\n"
+            "[dim]Compressing workspace files and formatting chat history...[/dim]"
+        ),
+        border_style="yellow",
+        expand=False,
+        padding=(1, 2)
+    )
+    theme_console.print(panel)
+    
+    try:
+        with theme_console.status("[bold green]Compressing and uploading...", spinner="dots"):
+            rec = manager.create_share(exclude_keys, expiry_hours, chat_messages)
+            
+        copied = copy_to_clipboard(rec.link)
+        clipboard_msg = " [green](Copied to clipboard!)[/green]" if copied else " [red](Failed to copy to clipboard)[/red]"
+        
+        theme_console.print("\n[bold green]✓ Share link created successfully![/bold green]")
+        theme_console.print(f"Link: [bold cyan]{rec.link}[/bold cyan]{clipboard_msg}\n")
+    except Exception as e:
+        theme_console.print(f"\n[bold red]✗ Failed to create share link: {e}[/bold red]\n")
+        
+    theme_console.print("[dim]Press Enter to return...[/dim]")
+    input()
+
+def run_share_flow(orchestrator):
+    """Main orchestrator for the /share flow."""
+    manager = ShareManager()
+    
+    # Extract chat messages from orchestrator
+    chat_messages = getattr(orchestrator, "messages", [])
+    
+    while True:
+        # If there are no shares, go directly to the wizard
+        records = manager.get_all()
+        if not records:
+            # Step 1: select exclusions
+            exclude_keys = _run_checkbox_dialog(
+                EXCLUDE_OPTIONS,
+                title="📤 Step 1: Select items to exclude (space to toggle)",
+                legend="Select which files/folders to omit to save space and exclude secrets"
+            )
+            if exclude_keys is None:
+                return  # user cancelled
+                
+            # Step 2: select expiry
+            expiry_opt = _run_expiry_dialog(
+                EXPIRY_OPTIONS,
+                title="⏰ Step 2: Choose link expiry duration",
+                legend="Select how long the link should remain active before expiring"
+            )
+            if expiry_opt is None:
+                return  # user cancelled
+                
+            # Step 3: create share
+            run_in_terminal(lambda: do_create_share(manager, exclude_keys, expiry_opt["hours"], chat_messages))
+            
+            # Since records are no longer empty, we will loop and the dashboard will show next
+            continue
+            
+        # Otherwise, show the dashboard
+        action_data = _run_shares_dashboard(manager)
+        if not action_data:
+            break  # user exited
+            
+        action, value = action_data
+        if action == "new":
+            # Run the wizard
+            exclude_keys = _run_checkbox_dialog(
+                EXCLUDE_OPTIONS,
+                title="📤 Step 1: Select items to exclude (space to toggle)",
+                legend="Select which files/folders to omit to save space and exclude secrets"
+            )
+            if exclude_keys is not None:
+                expiry_opt = _run_expiry_dialog(
+                    EXPIRY_OPTIONS,
+                    title="⏰ Step 2: Choose link expiry duration",
+                    legend="Select how long the link should remain active before expiring"
+                )
+                if expiry_opt is not None:
+                    run_in_terminal(lambda: do_create_share(manager, exclude_keys, expiry_opt["hours"], chat_messages))
+        elif action == "details":
+            run_in_terminal(lambda: show_share_details(value))