devobin 1.0.0__py3-none-any.whl

devobin/ui/model_modal.py

@@ -0,0 +1,143 @@
+"""Model selection modal for DevObin AI."""
+
+from __future__ import annotations
+
+from textual.app import ComposeResult
+from textual.screen import ModalScreen
+from textual.widgets import Static, OptionList, Button, Input
+from textual.containers import Vertical
+from textual.binding import Binding
+from textual import on
+
+from devobin.config.settings import get_config
+
+
+class ModelModal(ModalScreen[None]):
+    """Modal for selecting AI model with search."""
+
+    CSS = """
+    ModelModal {
+        align: center middle;
+    }
+    #model-container {
+        width: 60;
+        max-height: 35;
+        background: $surface;
+        border: tall $primary;
+        padding: 2;
+    }
+    #model-title {
+        text-align: center;
+        text-style: bold;
+        color: $primary;
+        margin-bottom: 1;
+    }
+    #current-model {
+        text-align: center;
+        color: $text-muted;
+        margin-bottom: 1;
+    }
+    #search-input {
+        width: 100%;
+        margin-bottom: 1;
+    }
+    #model-list {
+        height: auto;
+        max-height: 18;
+    }
+    #close-btn {
+        width: 100%;
+        margin-top: 1;
+    }
+    """
+
+    BINDINGS = [
+        Binding("escape", "close", "Close", show=True),
+    ]
+
+    def __init__(self) -> None:
+        super().__init__()
+        self.config = get_config()
+        self.all_models: list[str] = []
+
+    def compose(self) -> ComposeResult:
+        with Vertical(id="model-container"):
+            yield Static("Select Model", id="model-title")
+
+            current = self.config.active_model or "None"
+            yield Static(f"Current: [bold green]{current}[/]", id="current-model")
+
+            yield Input(placeholder="Search models...", id="search-input")
+
+            self.all_models = self.get_models()
+
+            if self.all_models:
+                yield OptionList(*self.all_models, id="model-list")
+            else:
+                yield Static("[dim]No models available. Connect a provider first.[/]")
+
+            yield Button("Close", id="close-btn", variant="default")
+
+    def get_models(self) -> list[str]:
+        """Get available models from current provider."""
+        provider_name = self.config.active_provider
+        if not provider_name:
+            return []
+
+        provider_cfg = self.config.get_provider(provider_name)
+        if not provider_cfg:
+            return []
+
+        try:
+            provider_type = provider_cfg.get("type", provider_name)
+
+            if provider_type in ("openai", "openai-compatible"):
+                from devobin.providers.openai_provider import OpenAIProvider
+                provider = OpenAIProvider(provider_cfg)
+                return provider.available_models()
+            elif provider_type == "anthropic":
+                from devobin.providers.anthropic_provider import AnthropicProvider
+                provider = AnthropicProvider(provider_cfg)
+                return provider.available_models()
+            elif provider_type == "google":
+                from devobin.providers.google_provider import GoogleProvider
+                provider = GoogleProvider(provider_cfg)
+                return provider.available_models()
+            elif provider_type == "ollama":
+                from devobin.providers.ollama_provider import OllamaProvider
+                provider = OllamaProvider(provider_cfg)
+                return provider.available_models()
+        except Exception:
+            pass
+
+        return []
+
+    @on(Input.Changed, "#search-input")
+    def on_search_changed(self, event: Input.Changed) -> None:
+        """Filter models based on search."""
+        try:
+            table = self.query_one("#model-list", OptionList)
+        except Exception:
+            return
+        query = event.value.lower()
+        table.clear_options()
+        for model in (m for m in self.all_models if query in m.lower()):
+            table.add_option(model)
+
+    @on(OptionList.OptionSelected, "#model-list")
+    def on_option_selected(self, event: OptionList.OptionSelected) -> None:
+        """Handle model selection (works with filtered lists too)."""
+        model = str(event.option.prompt)
+        self.config.active_model = model
+
+        current = self.query_one("#current-model", Static)
+        current.update(f"Current: [bold green]{model}[/]")
+
+    @on(Button.Pressed, "#close-btn")
+    def on_close_pressed(self) -> None:
+        """Close the modal."""
+        self.dismiss()
+
+    def action_close(self) -> None:
+        """Close the modal."""
+        self.dismiss()

devobin/ui/permission_modal.py

@@ -0,0 +1,92 @@
+"""Permission modal — asks the user before DevObin reads project files."""
+
+from __future__ import annotations
+
+from textual.app import ComposeResult
+from textual.screen import ModalScreen
+from textual.widgets import Static, Button
+from textual.containers import Vertical
+from textual.binding import Binding
+from textual import on
+
+
+class PermissionModal(ModalScreen[str | None]):
+    """Modal that asks the user whether DevObin may read their project files.
+
+    Returns:
+        "all"    — allow reading all files for this session
+        "one"    — ask before each file read
+        None     — deny all file reads
+    """
+
+    CSS = """
+    PermissionModal {
+        align: center middle;
+    }
+    #perm-container {
+        width: 55;
+        max-height: 25;
+        background: $surface;
+        border: tall $primary;
+        padding: 2;
+    }
+    #perm-title {
+        text-align: center;
+        text-style: bold;
+        color: $primary;
+        margin-bottom: 1;
+    }
+    #perm-body {
+        height: auto;
+        max-height: 12;
+        overflow-y: auto;
+        margin-bottom: 1;
+        color: $text;
+    }
+    #perm-buttons {
+        height: auto;
+        margin-top: 1;
+    }
+    #perm-buttons Button {
+        margin: 0 1;
+        width: 1fr;
+    }
+    """
+
+    BINDINGS = [
+        Binding("escape", "deny", "Deny", show=True),
+    ]
+
+    def __init__(self, file_count: int = 0) -> None:
+        super().__init__()
+        self._file_count = file_count
+
+    def compose(self) -> ComposeResult:
+        with Vertical(id="perm-container"):
+            yield Static("File Access Permission", id="perm-title")
+            body = (
+                f"DevObin found **{self._file_count} files** in your project.\n\n"
+                "To build an accurate engineering prompt, it needs to read\n"
+                "the key files and understand the architecture.\n\n"
+                "Allow file access?"
+            )
+            yield Static(body, id="perm-body")
+            with Vertical(id="perm-buttons"):
+                yield Button("Yes — Read All Files", id="perm-all", variant="primary")
+                yield Button("Ask Before Each File", id="perm-one", variant="default")
+                yield Button("No — Skip File Reading", id="perm-deny", variant="default")
+
+    @on(Button.Pressed, "#perm-all")
+    def on_all_pressed(self) -> None:
+        self.dismiss("all")
+
+    @on(Button.Pressed, "#perm-one")
+    def on_one_pressed(self) -> None:
+        self.dismiss("one")
+
+    @on(Button.Pressed, "#perm-deny")
+    def on_deny_pressed(self) -> None:
+        self.dismiss(None)
+
+    def action_deny(self) -> None:
+        self.dismiss(None)

devobin/ui/project_modal.py

@@ -0,0 +1,133 @@
+"""Project management modal for DevObin AI."""
+
+from __future__ import annotations
+
+from textual.app import ComposeResult
+from textual.screen import ModalScreen
+from textual.widgets import Static, Input, OptionList, Button
+from textual.containers import Vertical
+from textual.binding import Binding
+from textual import on
+
+from devobin.config.settings import get_config
+from devobin.storage.database import ProjectStorage
+
+
+class ProjectModal(ModalScreen[None]):
+    """Modal for managing projects."""
+
+    CSS = """
+    ProjectModal {
+        align: center middle;
+    }
+    #project-container {
+        width: 50;
+        max-height: 30;
+        background: $surface;
+        border: tall $primary;
+        padding: 2;
+    }
+    #project-title {
+        text-align: center;
+        text-style: bold;
+        color: $primary;
+        margin-bottom: 1;
+    }
+    #project-list {
+        height: auto;
+        max-height: 12;
+    }
+    #new-project-input {
+        width: 100%;
+        margin-top: 1;
+    }
+    #create-btn {
+        width: 100%;
+        margin-top: 1;
+    }
+    #close-btn {
+        width: 100%;
+        margin-top: 1;
+    }
+    #status-message {
+        margin-top: 1;
+        text-align: center;
+    }
+    """
+
+    BINDINGS = [
+        Binding("escape", "close", "Close", show=True),
+    ]
+
+    def __init__(self) -> None:
+        super().__init__()
+        self.config = get_config()
+        self._project_names: list[str] = []
+
+    def compose(self) -> ComposeResult:
+        with Vertical(id="project-container"):
+            yield Static("Manage Projects", id="project-title")
+
+            projects = self.get_projects()
+            active = self.config.get("active_project")
+
+            if projects:
+                self._project_names = sorted(projects)
+                labels = [
+                    f"{'→ ' if p == active else ''}{p}"
+                    for p in self._project_names
+                ]
+                yield OptionList(*labels, id="project-list")
+            else:
+                yield Static("[dim]No projects yet.[/]")
+
+            yield Static("New Project:", classes="field-label")
+            yield Input(placeholder="project-name", id="new-project-input")
+            yield Button("Create & Switch", id="create-btn", variant="primary")
+            yield Button("Close", id="close-btn", variant="default")
+            yield Static("", id="status-message")
+
+    def get_projects(self) -> list[str]:
+        """Get list of projects."""
+        projects_dir = self.config.projects_dir()
+        return [p.name for p in projects_dir.iterdir() if p.is_dir()]
+
+    @on(OptionList.OptionSelected, "#project-list")
+    def on_option_selected(self, event: OptionList.OptionSelected) -> None:
+        """Switch to selected project."""
+        idx = event.option_index
+        if 0 <= idx < len(self._project_names):
+            project = self._project_names[idx]
+            self.config.set("active_project", project)
+            self.update_status(f"Switched to: {project}", "success")
+
+    @on(Button.Pressed, "#create-btn")
+    def on_create_pressed(self) -> None:
+        """Create new project."""
+        name = self.query_one("#new-project-input", Input).value.strip()
+        if not name:
+            self.update_status("Enter a project name", "error")
+            return
+
+        ProjectStorage(name)
+        self.config.set("active_project", name)
+        self.update_status(f"Created: {name}", "success")
+
+    @on(Button.Pressed, "#close-btn")
+    def on_close_pressed(self) -> None:
+        """Close the modal."""
+        self.dismiss()
+
+    def update_status(self, message: str, style: str) -> None:
+        """Update status message."""
+        from rich.markup import escape as _escape
+        status = self.query_one("#status-message", Static)
+        safe = _escape(str(message))
+        if style == "success":
+            status.update(f"[green]✓[/] {safe}")
+        else:
+            status.update(f"[red]✗[/] {safe}")
+
+    def action_close(self) -> None:
+        """Close the modal."""
+        self.dismiss()

devobin/ui/settings_modal.py

@@ -0,0 +1,89 @@
+"""Settings modal for DevObin AI."""
+
+from __future__ import annotations
+
+from textual.app import ComposeResult
+from textual.screen import ModalScreen
+from textual.widgets import Static, Button
+from textual.containers import Vertical
+from textual.binding import Binding
+from textual import on
+
+from devobin.config.settings import get_config, CONFIG_DIR
+
+
+class SettingsModal(ModalScreen[None]):
+    """Modal for viewing settings."""
+
+    CSS = """
+    SettingsModal {
+        align: center middle;
+    }
+    #settings-container {
+        width: 50;
+        max-height: 25;
+        background: $surface;
+        border: tall $primary;
+        padding: 2;
+    }
+    #settings-title {
+        text-align: center;
+        text-style: bold;
+        color: $primary;
+        margin-bottom: 1;
+    }
+    .setting-row {
+        margin: 0 0 1 0;
+    }
+    .setting-label {
+        color: $text-muted;
+    }
+    .setting-value {
+        color: $text;
+        text-style: bold;
+    }
+    #close-btn {
+        width: 100%;
+        margin-top: 2;
+    }
+    """
+
+    BINDINGS = [
+        Binding("escape", "close", "Close", show=True),
+    ]
+
+    def __init__(self) -> None:
+        super().__init__()
+        self.config = get_config()
+
+    def compose(self) -> ComposeResult:
+        with Vertical(id="settings-container"):
+            yield Static("Settings", id="settings-title")
+
+            yield Static(
+                f"[dim]Provider:[/] [bold]{self.config.active_provider or 'None'}[/]",
+                classes="setting-row",
+            )
+            yield Static(
+                f"[dim]Model:[/] [bold]{self.config.active_model or 'None'}[/]",
+                classes="setting-row",
+            )
+            yield Static(
+                f"[dim]Project:[/] [bold]{self.config.get('active_project') or 'None'}[/]",
+                classes="setting-row",
+            )
+            yield Static(
+                f"[dim]Config:[/] [bold]{CONFIG_DIR}[/]",
+                classes="setting-row",
+            )
+
+            yield Button("Close", id="close-btn", variant="default")
+
+    @on(Button.Pressed, "#close-btn")
+    def on_close_pressed(self) -> None:
+        """Close the modal."""
+        self.dismiss()
+
+    def action_close(self) -> None:
+        """Close the modal."""
+        self.dismiss()