wisp/
runtime_state.rs

1use crate::cli::Cli;
2use crate::error::AppError;
3use crate::settings::{StatusLineSettings, WispSettings};
4use crate::workspace_status::WorkspaceStatus;
5use acp_utils::client::{AcpEvent, AcpPromptHandle, spawn_acp_session};
6use agent_client_protocol::schema::{
7    AuthMethod, Implementation, InitializeRequest, NewSessionRequest, PromptCapabilities, ProtocolVersion,
8    SessionConfigOption, SessionId,
9};
10use std::env::current_dir;
11use tokio::sync::mpsc;
12use tui::Theme;
13
14#[doc = include_str!("docs/runtime_state.md")]
15pub struct RuntimeState {
16    pub session_id: SessionId,
17    pub agent_name: String,
18    pub prompt_capabilities: PromptCapabilities,
19    pub config_options: Vec<SessionConfigOption>,
20    pub auth_methods: Vec<AuthMethod>,
21    pub theme: Theme,
22    pub settings: WispSettings,
23    pub event_rx: mpsc::UnboundedReceiver<AcpEvent>,
24    pub prompt_handle: AcpPromptHandle,
25    pub working_dir: std::path::PathBuf,
26    pub workspace_status: WorkspaceStatus,
27}
28
29impl RuntimeState {
30    pub async fn new(agent_command: &str, settings: WispSettings) -> Result<Self, AppError> {
31        let cwd = current_dir().unwrap_or_else(|_| std::path::PathBuf::from("."));
32        let workspace_status = WorkspaceStatus::resolve(&cwd).await;
33        let new_session_request = NewSessionRequest::new(cwd.clone());
34        let init_request = InitializeRequest::new(ProtocolVersion::LATEST)
35            .client_info(Implementation::new("wisp", env!("CARGO_PKG_VERSION")));
36
37        let session =
38            spawn_acp_session(agent_command, init_request, new_session_request).await.map_err(AppError::Acp)?;
39
40        let theme = crate::settings::load_theme(&settings);
41
42        Ok(Self {
43            session_id: session.session_id,
44            agent_name: session.agent_name,
45            prompt_capabilities: session.prompt_capabilities,
46            config_options: session.config_options,
47            auth_methods: session.auth_methods,
48            theme,
49            settings,
50            event_rx: session.event_rx,
51            prompt_handle: session.prompt_handle,
52            working_dir: cwd,
53            workspace_status,
54        })
55    }
56
57    pub async fn from_cli(cli: &Cli) -> Result<Self, AppError> {
58        Self::new(&cli.agent, crate::settings::load_or_create_settings(StatusLineSettings::defaults())).await
59    }
60}
wisp/runtime_state.rs

wisp/
runtime_state.rs
