mirror of
https://github.com/mistralai/mistral-vibe
synced 2026-04-25 17:14:55 +02:00
6a50d1d521
Co-authored-by: Quentin Torroba <quentin.torroba@mistral.ai> Co-authored-by: Clément Drouin <clement.drouin@mistral.ai> Co-authored-by: Clément Sirieix <clement.sirieix@mistral.ai> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
113 lines
3.4 KiB
Python
113 lines
3.4 KiB
Python
from __future__ import annotations
|
|
|
|
from datetime import UTC, datetime
|
|
from typing import TYPE_CHECKING, Any, ClassVar
|
|
|
|
from rich.text import Text
|
|
from textual.app import ComposeResult
|
|
from textual.binding import Binding, BindingType
|
|
from textual.containers import Container, Vertical
|
|
from textual.message import Message
|
|
from textual.widgets import OptionList
|
|
from textual.widgets.option_list import Option
|
|
|
|
from vibe.cli.textual_ui.widgets.no_markup_static import NoMarkupStatic
|
|
|
|
if TYPE_CHECKING:
|
|
from vibe.core.session.session_loader import SessionInfo
|
|
|
|
_SECONDS_PER_MINUTE = 60
|
|
_SECONDS_PER_HOUR = 3600
|
|
_SECONDS_PER_DAY = 86400
|
|
_SECONDS_PER_WEEK = 604800
|
|
|
|
|
|
def _format_relative_time(iso_time: str | None) -> str:
|
|
if not iso_time:
|
|
return "unknown"
|
|
try:
|
|
dt = datetime.fromisoformat(iso_time.replace("Z", "+00:00"))
|
|
now = datetime.now(UTC)
|
|
delta = now - dt
|
|
seconds = int(delta.total_seconds())
|
|
|
|
if seconds < _SECONDS_PER_MINUTE:
|
|
return "just now"
|
|
for threshold, divisor, unit in [
|
|
(_SECONDS_PER_HOUR, _SECONDS_PER_MINUTE, "m"),
|
|
(_SECONDS_PER_DAY, _SECONDS_PER_HOUR, "h"),
|
|
(_SECONDS_PER_WEEK, _SECONDS_PER_DAY, "d"),
|
|
(float("inf"), _SECONDS_PER_WEEK, "w"),
|
|
]:
|
|
if seconds < threshold:
|
|
return f"{seconds // divisor}{unit} ago"
|
|
except (ValueError, OSError):
|
|
pass
|
|
return "unknown"
|
|
|
|
|
|
def _build_option_text(session: SessionInfo, message: str) -> Text:
|
|
text = Text(no_wrap=True)
|
|
time_str = _format_relative_time(session.get("end_time"))
|
|
session_id = session["session_id"][:8]
|
|
text.append(f"{time_str:10}", style="dim")
|
|
text.append(" ")
|
|
text.append(f"{session_id} ", style="dim")
|
|
text.append(message)
|
|
return text
|
|
|
|
|
|
class SessionPickerApp(Container):
|
|
"""Session picker for /resume command."""
|
|
|
|
can_focus_children = True
|
|
|
|
BINDINGS: ClassVar[list[BindingType]] = [
|
|
Binding("escape", "cancel", "Cancel", show=False)
|
|
]
|
|
|
|
class SessionSelected(Message):
|
|
def __init__(self, session_id: str) -> None:
|
|
self.session_id = session_id
|
|
super().__init__()
|
|
|
|
class Cancelled(Message):
|
|
pass
|
|
|
|
def __init__(
|
|
self,
|
|
sessions: list[SessionInfo],
|
|
latest_messages: dict[str, str],
|
|
**kwargs: Any,
|
|
) -> None:
|
|
super().__init__(id="sessionpicker-app", **kwargs)
|
|
self._sessions = sessions
|
|
self._latest_messages = latest_messages
|
|
|
|
def compose(self) -> ComposeResult:
|
|
options = [
|
|
Option(
|
|
_build_option_text(
|
|
session,
|
|
self._latest_messages.get(session["session_id"], "(empty session)"),
|
|
),
|
|
id=session["session_id"],
|
|
)
|
|
for session in self._sessions
|
|
]
|
|
with Vertical(id="sessionpicker-content"):
|
|
yield OptionList(*options, id="sessionpicker-options")
|
|
yield NoMarkupStatic(
|
|
"↑↓ Navigate Enter Select Esc Cancel", classes="sessionpicker-help"
|
|
)
|
|
|
|
def on_mount(self) -> None:
|
|
self.query_one(OptionList).focus()
|
|
|
|
def on_option_list_option_selected(self, event: OptionList.OptionSelected) -> None:
|
|
if event.option.id:
|
|
self.post_message(self.SessionSelected(event.option.id))
|
|
|
|
def action_cancel(self) -> None:
|
|
self.post_message(self.Cancelled())
|