mobius_cli/frontend/
setup.rs

1//! Gateway-native provider login and Bot setup wizard.
2
3mod runtime;
4mod state;
5pub(in crate::frontend) mod view;
6
7use std::io;
8
9use mobius::{Error, Result};
10use mobius_gateway::client::{GatewayEvents, GatewaySender};
11use mobius_gateway::wire::{ReadyPayload, SessionReadyPayload};
12use ratatui::Terminal;
13use ratatui::backend::CrosstermBackend;
14
15use self::runtime::{apply, apply_gateway, edit};
16use self::state::SetupState;
17use super::terminal::TerminalGuard;
18
19const MAX_API_KEY_BYTES: usize = 16 * 1024;
20const MAX_ENDPOINT_BYTES: usize = 4 * 1024;
21// Matches the gateway's provider label bound.
22const MAX_PROVIDER_LABEL_BYTES: usize = 128;
23const MAX_MODEL_IDS_BYTES: usize = 16 * 1024;
24const MIN_INLINE_DESCRIPTION_WIDTH: usize = 20;
25
26const UPDATE_BOT_LABEL: &str = "Update this Bot";
27const UPDATE_BOT_DESCRIPTION: &str = "Apply these settings to every chat owned by this Bot";
28const SAVE_DEFAULT_LABEL: &str = "Save Bot template";
29const SAVE_DEFAULT_DESCRIPTION: &str = "Use these settings when creating Bots";
30
31pub(in crate::frontend) type SetupTerminal = Terminal<CrosstermBackend<io::Stdout>>;
32
33/// The focused setup flow requested by the CLI shell.
34#[derive(Debug, Clone, Copy, PartialEq, Eq)]
35pub(crate) enum SetupMode {
36    Login,
37    Bot,
38    BotModel,
39}
40
41/// Runs one gateway-backed setup flow and updates its machine and chat snapshots.
42pub(crate) async fn run(
43    terminal: &mut SetupTerminal,
44    mode: SetupMode,
45    preferred_provider: Option<&str>,
46    sender: &GatewaySender,
47    events: &mut GatewayEvents,
48    gateway: &mut ReadyPayload,
49    session: &mut SessionReadyPayload,
50) -> Result<()> {
51    run_bot(
52        terminal,
53        mode,
54        preferred_provider,
55        sender,
56        events,
57        gateway,
58        &session.session.context.bot_id,
59    )
60    .await
61}
62
63/// Edits one durable Bot without requiring an open chat.
64pub(in crate::frontend) async fn run_bot(
65    terminal: &mut SetupTerminal,
66    mode: SetupMode,
67    preferred_provider: Option<&str>,
68    sender: &GatewaySender,
69    events: &mut GatewayEvents,
70    gateway: &mut ReadyPayload,
71    bot_id: &str,
72) -> Result<()> {
73    let bot = gateway
74        .bots
75        .iter()
76        .find(|bot| bot.id == bot_id)
77        .ok_or_else(|| Error::Config("the selected Bot is not in the gateway catalog".into()))?;
78    let mut state = SetupState::new(
79        mode,
80        preferred_provider,
81        gateway,
82        bot.config.config.clone(),
83        false,
84    )?;
85    terminal.clear()?;
86
87    if !edit(terminal, &mut state, sender, events, gateway).await? {
88        return Ok(());
89    }
90    apply(terminal, &mut state, sender, events, gateway, bot_id).await?;
91    Ok(())
92}
93
94/// Runs provider or Bot-template setup without creating or changing a chat.
95pub(crate) async fn run_gateway(
96    terminal: &mut SetupTerminal,
97    mode: SetupMode,
98    sender: &GatewaySender,
99    events: &mut GatewayEvents,
100    gateway: &mut ReadyPayload,
101) -> Result<()> {
102    let original = gateway
103        .bot_defaults
104        .as_ref()
105        .map(|default| default.config.clone())
106        .unwrap_or_default();
107    if mode == SetupMode::Bot && gateway.bot_defaults.is_none() {
108        return Err(Error::Config(
109            "configure a provider before changing the Bot template".into(),
110        ));
111    }
112    let mut state = SetupState::new(mode, None, gateway, original, true)?;
113    terminal.clear()?;
114    if !edit(terminal, &mut state, sender, events, gateway).await? {
115        return Ok(());
116    }
117    apply_gateway(terminal, &mut state, sender, events, gateway).await
118}
119
120/// Runs gateway-scoped provider setup before any chat exists.
121pub async fn run_gateway_login(
122    sender: &GatewaySender,
123    events: &mut GatewayEvents,
124    gateway: &mut ReadyPayload,
125) -> Result<()> {
126    let mut guard = TerminalGuard::alternate()?;
127    guard.set_mouse_capture(false)?;
128    let mut terminal = Terminal::new(CrosstermBackend::new(io::stdout()))?;
129    run_gateway(&mut terminal, SetupMode::Login, sender, events, gateway).await
130}
131
132#[cfg(test)]
133mod tests;
mobius_cli/frontend/setup.rs

mobius_cli/frontend/
setup.rs
