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::config::{
12    MAX_PROVIDER_API_KEY_BYTES as MAX_API_KEY_BYTES, MAX_PROVIDER_LABEL_BYTES,
13};
14use mobius_gateway::wire::{ReadyPayload, SessionReadyPayload};
15use ratatui::Terminal;
16use ratatui::backend::CrosstermBackend;
17
18use self::runtime::{apply, apply_gateway, edit};
19use self::state::SetupState;
20use super::terminal::TerminalGuard;
21
22const MAX_ENDPOINT_BYTES: usize = 4 * 1024;
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.owner_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    preferred_provider: Option<&str>,
99    sender: &GatewaySender,
100    events: &mut GatewayEvents,
101    gateway: &mut ReadyPayload,
102) -> Result<()> {
103    let original = gateway
104        .bot_defaults
105        .as_ref()
106        .map(|default| default.config.clone())
107        .unwrap_or_default();
108    if mode == SetupMode::Bot && gateway.bot_defaults.is_none() {
109        return Err(Error::Config(
110            "configure a provider before changing the Bot template".into(),
111        ));
112    }
113    let mut state = SetupState::new(mode, preferred_provider, gateway, original, true)?;
114    terminal.clear()?;
115    if !edit(terminal, &mut state, sender, events, gateway).await? {
116        return Ok(());
117    }
118    apply_gateway(terminal, &mut state, sender, events, gateway).await
119}
120
121/// Runs gateway-scoped provider setup before any chat exists.
122pub async fn run_gateway_login(
123    sender: &GatewaySender,
124    events: &mut GatewayEvents,
125    gateway: &mut ReadyPayload,
126) -> Result<()> {
127    let mut guard = TerminalGuard::alternate()?;
128    guard.set_mouse_capture(false)?;
129    let mut terminal = Terminal::new(CrosstermBackend::new(io::stdout()))?;
130    run_gateway(
131        &mut terminal,
132        SetupMode::Login,
133        None,
134        sender,
135        events,
136        gateway,
137    )
138    .await
139}
140
141#[cfg(test)]
142mod tests;
mobius_cli/frontend/setup.rs

mobius_cli/frontend/
setup.rs
