context-forge-cli 0.1.0__py3-none-any.whl

contextforge/tui/widgets/session_detail.py

@@ -0,0 +1,191 @@
+"""SessionDetail widget — right panel showing summary and metadata for selected session."""
+from __future__ import annotations
+
+import json
+from datetime import datetime, timezone
+from pathlib import Path
+
+from textual.app import ComposeResult
+from textual.containers import Vertical
+from textual.widget import Widget
+from textual.widgets import Markdown, Static
+
+TOOL_COLORS = {
+    "claude_code": "cyan",
+    "codex": "green",
+    "altimate_code": "magenta",
+    "claude_desktop": "yellow",
+}
+TOOL_LABELS = {
+    "claude_code": "Claude Code",
+    "codex": "Codex",
+    "altimate_code": "altimate",
+    "claude_desktop": "Desktop",
+}
+
+
+def _meta_line(label: str, value: str, color: str = "cyan") -> str:
+    return f"[bold {color}]{label:<10}[/bold {color}] {value}"
+
+
+def _fmt_tokens(n: int | None) -> str:
+    if not n:
+        return "—"
+    if n >= 1_000_000:
+        return f"{n / 1_000_000:.1f}M"
+    if n >= 1_000:
+        return f"{n // 1_000}k"
+    return str(n)
+
+
+def _token_color(n: int | None) -> str:
+    if not n:
+        return "dim"
+    if n >= 100_000:
+        return "bold red"
+    if n >= 20_000:
+        return "yellow"
+    return "dim"
+
+
+def _fmt_ts(ms: int | None, fmt: str = "%Y-%m-%d %H:%M UTC") -> str:
+    if not ms:
+        return "?"
+    try:
+        dt = datetime.fromtimestamp(ms / 1000, tz=timezone.utc)
+        return dt.strftime(fmt)
+    except Exception:
+        return "?"
+
+
+class SessionDetail(Widget):
+    """Shows title, metadata, and summary for the currently selected session."""
+
+    DEFAULT_CSS = """
+    SessionDetail {
+        width: 2fr;
+        height: 1fr;
+        padding: 1 2;
+        border: tall $primary;
+        border-title-color: $accent;
+        border-title-style: bold;
+        overflow-y: auto;
+    }
+    SessionDetail #detail-placeholder {
+        color: $text-disabled;
+        margin: 4 0;
+        text-align: center;
+    }
+    SessionDetail #detail-title {
+        text-style: bold;
+        color: $accent;
+        margin-bottom: 1;
+    }
+    SessionDetail #detail-meta-block {
+        margin-bottom: 1;
+        border-bottom: solid $primary-background-lighten-1;
+        padding-bottom: 1;
+    }
+    SessionDetail #detail-summary-label {
+        text-style: bold;
+        color: $text-muted;
+        margin-bottom: 1;
+    }
+    SessionDetail Markdown {
+        background: transparent;
+        color: $text;
+    }
+    """
+
+    _current_id: str | None = None
+
+    def compose(self) -> ComposeResult:
+        yield Static("← Select a session", id="detail-placeholder")
+        yield Static("", id="detail-title")
+        with Vertical(id="detail-meta-block"):
+            yield Static("", id="detail-tool")
+            yield Static("", id="detail-cwd")
+            yield Static("", id="detail-tokens")
+            yield Static("", id="detail-created")
+            yield Static("", id="detail-updated")
+            yield Static("", id="detail-status")
+            yield Static("", id="detail-tags")
+            yield Static("", id="detail-id")
+        yield Static("── Summary ──", id="detail-summary-label")
+        yield Markdown("", id="detail-summary-md")
+
+    def on_mount(self) -> None:
+        self.border_title = "Detail"
+        self.query_one("#detail-summary-label").display = False
+        self.query_one("#detail-meta-block").display = False
+        self.query_one("#detail-title").display = False
+
+    def load(self, session_id: str) -> None:
+        """Load and display the given session."""
+        if session_id == self._current_id:
+            return
+        self._current_id = session_id
+
+        db_path: Path | None = getattr(self.app, "db_path", None)
+        if db_path is None:
+            return
+
+        from contextforge.core.db import get_db, get_session
+        db = get_db(db_path)
+        row = get_session(db, session_id)
+        if row is None:
+            return
+
+        tool = row.get("tool", "?")
+        title = row.get("title") or "(no title)"
+        cwd = row.get("cwd") or "?"
+        tokens_int = row.get("token_count")
+        status = row.get("status") or "?"
+        tags_raw = row.get("tags") or "[]"
+        try:
+            tags = ", ".join(json.loads(tags_raw)) or "—"
+        except Exception:
+            tags = "—"
+
+        created_str = _fmt_ts(row.get("created_at"))
+        updated_str = _fmt_ts(row.get("updated_at"))
+
+        summary = row.get("summary") or "*No summary yet* — run: `cf summarize <id>`"
+
+        # Tool label with color
+        tool_color = TOOL_COLORS.get(tool, "white")
+        tool_display = TOOL_LABELS.get(tool, tool)
+        tool_markup = f"[{tool_color}]{tool_display}[/{tool_color}]"
+
+        # Token display with color
+        tok_color = _token_color(tokens_int)
+        tok_str = _fmt_tokens(tokens_int)
+        tok_markup = f"[{tok_color}]{tok_str}[/{tok_color}]"
+
+        # Show everything
+        self.query_one("#detail-placeholder").display = False
+        self.query_one("#detail-title").display = True
+        self.query_one("#detail-meta-block").display = True
+        self.query_one("#detail-summary-label").display = True
+
+        self.query_one("#detail-title", Static).update(f"[bold accent]{title}[/bold accent]")
+        self.query_one("#detail-tool", Static).update(_meta_line("Tool", tool_markup, tool_color))
+        self.query_one("#detail-cwd", Static).update(_meta_line("CWD", f"[dim]{cwd}[/dim]", "dim"))
+        self.query_one("#detail-tokens", Static).update(_meta_line("Tokens", tok_markup, "dim"))
+        self.query_one("#detail-created", Static).update(_meta_line("Created", f"[dim]{created_str}[/dim]", "dim"))
+        self.query_one("#detail-updated", Static).update(_meta_line("Updated", f"[dim]{updated_str}[/dim]", "dim"))
+        self.query_one("#detail-status", Static).update(_meta_line("Status", status, "dim"))
+        self.query_one("#detail-tags", Static).update(_meta_line("Tags", tags, "dim"))
+        self.query_one("#detail-id", Static).update(_meta_line("ID", f"[dim]{session_id}[/dim]", "dim"))
+
+        md = self.query_one("#detail-summary-md", Markdown)
+        self.app.call_later(md.update, summary)
+
+    def clear(self) -> None:
+        self._current_id = None
+        self.query_one("#detail-placeholder").display = True
+        self.query_one("#detail-title").display = False
+        self.query_one("#detail-meta-block").display = False
+        self.query_one("#detail-summary-label").display = False
+        self.query_one("#detail-title", Static).update("")
+        self.app.call_later(self.query_one("#detail-summary-md", Markdown).update, "")

contextforge/tui/widgets/session_table.py

@@ -0,0 +1,277 @@
+"""SessionTable widget — scrollable DataTable of all indexed sessions."""
+from __future__ import annotations
+
+from datetime import datetime, timezone
+from pathlib import Path
+
+from textual.app import ComposeResult
+from textual.containers import Horizontal, Vertical
+from textual.events import Key
+from textual.message import Message
+from textual.reactive import reactive
+from textual.widget import Widget
+from textual.widgets import Button, DataTable, Input, Label
+
+TOOL_EMOJI = {
+    "claude_code": "🔵",
+    "codex": "🟢",
+    "altimate_code": "🟣",
+    "claude_desktop": "🟡",
+}
+
+TOOL_MARKUP = {
+    "claude_code":   "[cyan]🔵 CC[/cyan]",
+    "codex":         "[green]🟢 Codex[/green]",
+    "altimate_code": "[magenta]🟣 Alt[/magenta]",
+    "claude_desktop": "[yellow]🟡 Desktop[/yellow]",
+}
+
+TOOL_DISPLAY = {
+    "claude_code": "CC",
+    "codex": "Codex",
+    "altimate_code": "Alt",
+    "claude_desktop": "Desktop",
+}
+
+_FTOOL_MAP = {
+    "ftool-all":   None,
+    "ftool-cc":    "claude_code",
+    "ftool-codex": "codex",
+    "ftool-alt":   "altimate_code",
+}
+
+
+def _fmt_tokens(n: int | None) -> str:
+    if not n:
+        return ""
+    if n >= 1_000_000:
+        return f"{n / 1_000_000:.1f}M"
+    if n >= 1_000:
+        return f"{n // 1_000}k"
+    return str(n)
+
+
+def _token_markup(n: int | None) -> str:
+    if not n:
+        return "[dim]—[/dim]"
+    raw = _fmt_tokens(n)
+    if n >= 100_000:
+        return f"[bold red]{raw}[/bold red]"
+    if n >= 20_000:
+        return f"[yellow]{raw}[/yellow]"
+    return f"[dim]{raw}[/dim]"
+
+
+class SessionTable(Widget):
+    """Displays all sessions in a navigable DataTable with live filtering."""
+
+    DEFAULT_CSS = """
+    SessionTable {
+        width: 1fr;
+        height: 1fr;
+        border: tall $primary;
+        border-title-color: $accent;
+        border-title-style: bold;
+    }
+    SessionTable > Vertical {
+        height: 1fr;
+    }
+    SessionTable DataTable {
+        height: 1fr;
+    }
+    SessionTable #filter-bar {
+        height: 3;
+        background: $primary-background-darken-1;
+        padding: 0 1;
+        layout: horizontal;
+        align: left middle;
+    }
+    SessionTable #filter-label {
+        width: auto;
+        margin-right: 1;
+        color: $text-muted;
+    }
+    SessionTable #filter-input {
+        width: 28;
+        margin-right: 1;
+    }
+    SessionTable .filter-tool-btn {
+        min-width: 7;
+        height: 1;
+        margin-right: 1;
+        background: $surface;
+        border: none;
+        color: $text-muted;
+    }
+    SessionTable .filter-tool-btn.active {
+        background: $primary;
+        color: $text;
+        text-style: bold;
+    }
+    SessionTable .hidden {
+        display: none;
+    }
+    """
+
+    # ── Messages ──────────────────────────────────────────────────────────────
+
+    class RowSelected(Message):
+        """Emitted when the user highlights a session row."""
+        def __init__(self, session_id: str, tool: str) -> None:
+            super().__init__()
+            self.session_id = session_id
+            self.tool = tool
+
+    class FilterChanged(Message):
+        """Emitted when filter state changes."""
+        def __init__(self, text: str, tool: str | None, match_count: int) -> None:
+            super().__init__()
+            self.text = text
+            self.tool = tool
+            self.match_count = match_count
+
+    # ── State ─────────────────────────────────────────────────────────────────
+
+    def __init__(self, **kwargs) -> None:
+        super().__init__(**kwargs)
+        self._rows: list[dict] = []
+        self._display_rows: list[dict] = []
+        self._filter_text: str = ""
+        self._filter_tool: str | None = None
+        self._filter_visible: bool = False
+
+    # ── Compose ───────────────────────────────────────────────────────────────
+
+    def compose(self) -> ComposeResult:
+        with Vertical():
+            with Horizontal(id="filter-bar", classes="hidden"):
+                yield Label("Filter:", id="filter-label")
+                yield Input(placeholder="title or project…", id="filter-input")
+                yield Button("All",   id="ftool-all",    classes="filter-tool-btn active")
+                yield Button("CC",    id="ftool-cc",     classes="filter-tool-btn")
+                yield Button("Codex", id="ftool-codex",  classes="filter-tool-btn")
+                yield Button("Alt",   id="ftool-alt",    classes="filter-tool-btn")
+            yield DataTable(cursor_type="row", zebra_stripes=True)
+
+    def on_mount(self) -> None:
+        self.border_title = "Sessions"
+        table = self.query_one(DataTable)
+        table.add_columns("Tool", "Title", "Updated", "Tokens", "ID")
+        self.reload()
+
+    # ── Public API ────────────────────────────────────────────────────────────
+
+    def toggle_filter(self) -> None:
+        """Show or hide the filter bar."""
+        bar = self.query_one("#filter-bar")
+        if self._filter_visible:
+            self._filter_visible = False
+            bar.add_class("hidden")
+            self._filter_text = ""
+            self._filter_tool = None
+            self.query_one("#filter-input", Input).value = ""
+            # Reset active button
+            for btn_id in _FTOOL_MAP:
+                btn = self.query_one(f"#{btn_id}", Button)
+                if btn_id == "ftool-all":
+                    btn.add_class("active")
+                else:
+                    btn.remove_class("active")
+            self.reload()
+            self.post_message(self.FilterChanged("", None, len(self._rows)))
+        else:
+            self._filter_visible = True
+            bar.remove_class("hidden")
+            self.query_one("#filter-input", Input).focus()
+
+    def reload(self, tool_filter: str | None = None) -> None:
+        """Re-read sessions from DB and refresh the table."""
+        db_path: Path | None = getattr(self.app, "db_path", None)
+        if db_path is None:
+            return
+
+        from contextforge.core.db import get_db, get_sessions
+        db = get_db(db_path)
+        rows = get_sessions(db, tool=tool_filter, limit=500)
+        self._rows = rows
+        self._display_rows = rows
+        self._render_rows(rows)
+
+    # ── Internal rendering ────────────────────────────────────────────────────
+
+    def _render_rows(self, rows: list[dict]) -> None:
+        """Populate the DataTable from a list of session dicts."""
+        self._display_rows = rows
+        table = self.query_one(DataTable)
+        table.clear()
+
+        for row in rows:
+            tool = row.get("tool", "")
+            tool_label = TOOL_MARKUP.get(tool, f"⚪ {tool}")
+
+            from contextforge.utils.display import _clean_title
+            title = _clean_title(row.get("title") or "", max_len=38) or "(no title)"
+
+            updated_ms = row.get("updated_at") or 0
+            try:
+                dt = datetime.fromtimestamp(updated_ms / 1000, tz=timezone.utc)
+                updated = dt.strftime("%m-%d %H:%M")
+            except Exception:
+                updated = "?"
+
+            tokens = _token_markup(row.get("token_count"))
+            session_id = str(row.get("id", ""))[:12]
+
+            table.add_row(tool_label, title, updated, tokens, session_id)
+
+    def _apply_filter(self) -> None:
+        """Filter self._rows in memory and re-render."""
+        text = self._filter_text.lower()
+        tool = self._filter_tool
+
+        filtered = [
+            row for row in self._rows
+            if (not tool or row.get("tool") == tool)
+            and (
+                not text
+                or text in (row.get("title") or "").lower()
+                or text in (row.get("cwd") or "").lower()
+            )
+        ]
+        self._render_rows(filtered)
+        self.post_message(self.FilterChanged(self._filter_text, tool, len(filtered)))
+
+    # ── Event handlers ────────────────────────────────────────────────────────
+
+    def on_input_changed(self, event: Input.Changed) -> None:
+        if event.input.id == "filter-input":
+            self._filter_text = event.value
+            self._apply_filter()
+
+    def on_key(self, event: Key) -> None:
+        if event.key == "escape" and self._filter_visible:
+            self.toggle_filter()
+            event.stop()
+
+    def on_button_pressed(self, event: Button.Pressed) -> None:
+        btn_id = event.button.id
+        if btn_id not in _FTOOL_MAP:
+            return
+        self._filter_tool = _FTOOL_MAP[btn_id]
+        for bid in _FTOOL_MAP:
+            btn = self.query_one(f"#{bid}", Button)
+            if bid == btn_id:
+                btn.add_class("active")
+            else:
+                btn.remove_class("active")
+        self._apply_filter()
+
+    def on_data_table_row_highlighted(self, event: DataTable.RowHighlighted) -> None:
+        idx = event.cursor_row
+        display = self._display_rows
+        if 0 <= idx < len(display):
+            row = display[idx]
+            self.post_message(self.RowSelected(
+                session_id=str(row["id"]),
+                tool=str(row.get("tool", "")),
+            ))