openidea/crates/ide-agent/src/config.rs

//! Session settings (model, mode, thinking effort…), slash commands and context use, as
//! the agent reports them over ACP.

use crate::model::{AgentCommand, AgentConfigOption, AgentConfigValue, AgentConfigVia, AgentUsage};
use serde_json::{Value, json};

/// The settings in a `session/new` or `session/load` result: its select `configOptions`,
/// plus the older `modes` / `models` fields for what those don't cover.
pub(crate) fn options_of(result: &Value) -> Vec<AgentConfigOption> {
    let mut out = result.get("configOptions").map(parse_options).unwrap_or_default();
    let has = |out: &[AgentConfigOption], category: &str| out.iter().any(|o| o.category.as_deref() == Some(category));
    if !has(&out, "mode")
        && let Some(modes) = result.get("modes")
    {
        out.push(legacy(modes, "mode", "Mode", "availableModes", "id", "currentModeId", AgentConfigVia::Mode));
    }
    if !has(&out, "model")
        && let Some(models) = result.get("models")
    {
        out.push(legacy(
            models,
            "model",
            "Model",
            "availableModels",
            "modelId",
            "currentModelId",
            AgentConfigVia::Model,
        ));
    }
    out.retain(|o| !o.values.is_empty());
    out
}

/// A `configOptions` array; only select options are kept (the UI shows them as dropdowns).
pub(crate) fn parse_options(options: &Value) -> Vec<AgentConfigOption> {
    let Some(list) = options.as_array() else { return Vec::new() };
    list.iter()
        .filter(|o| o.get("type").and_then(Value::as_str).is_none_or(|t| t == "select"))
        .filter_map(|o| {
            let id = o.get("id").or_else(|| o.get("configId")).and_then(Value::as_str)?.to_owned();
            let mut values = Vec::new();
            for v in o.get("options").and_then(Value::as_array).into_iter().flatten() {
                // Grouped options: `{ group, name, options: [...] }`.
                match v.get("options").and_then(Value::as_array) {
                    Some(group) => values.extend(group.iter().filter_map(value_of)),
                    None => values.extend(value_of(v)),
                }
            }
            Some(AgentConfigOption {
                name: str_of(o, "name").unwrap_or_else(|| id.clone()),
                description: str_of(o, "description"),
                category: str_of(o, "category").or_else(|| matches!(id.as_str(), "model" | "mode").then(|| id.clone())),
                current: str_of(o, "currentValue"),
                values,
                via: AgentConfigVia::Option,
                id,
            })
        })
        .collect()
}

/// A `config_option_update`: its options replace the config-option ones; `modes` / `models`
/// stand-ins stay.
pub(crate) fn merge_update(config: &mut Vec<AgentConfigOption>, options: &Value) {
    let fresh = parse_options(options);
    let covered = |o: &AgentConfigOption| fresh.iter().any(|f| f.category.is_some() && f.category == o.category);
    config.retain(|o| o.via != AgentConfigVia::Option && !covered(o));
    let legacy = std::mem::take(config);
    config.extend(fresh);
    config.extend(legacy);
    config.retain(|o| !o.values.is_empty());
}

/// `current_mode_update`.
pub(crate) fn set_mode(config: &mut [AgentConfigOption], mode: &str) {
    if let Some(o) = config.iter_mut().find(|o| o.category.as_deref() == Some("mode")) {
        o.current = Some(mode.to_owned());
    }
}

/// The request (method, params without the session id) that sets `option` to `value`.
pub(crate) fn set_request(option: &AgentConfigOption, value: &str) -> (&'static str, Value) {
    match option.via {
        AgentConfigVia::Option => ("session/set_config_option", json!({ "configId": option.id, "value": value })),
        AgentConfigVia::Mode => ("session/set_mode", json!({ "modeId": value })),
        AgentConfigVia::Model => ("session/set_model", json!({ "modelId": value })),
    }
}

/// `available_commands_update`.
pub(crate) fn commands_of(update: &Value) -> Vec<AgentCommand> {
    update
        .get("availableCommands")
        .and_then(Value::as_array)
        .into_iter()
        .flatten()
        .filter_map(|c| {
            Some(AgentCommand {
                name: c.get("name")?.as_str()?.trim_start_matches('/').to_owned(),
                description: str_of(c, "description").unwrap_or_default(),
                hint: c.pointer("/input/hint").and_then(Value::as_str).filter(|h| !h.is_empty()).map(str::to_owned),
            })
        })
        .collect()
}

/// `usage_update`: `{ used, size, cost?: { amount, currency } }`.
pub(crate) fn usage_of(update: &Value) -> Option<AgentUsage> {
    let tokens = |key: &str| update.get(key)?.as_f64().map(|n| n.clamp(0.0, f64::from(u32::MAX)) as u32);
    let used = tokens("used")?;
    let size = tokens("size").filter(|s| *s > 0)?;
    Some(AgentUsage {
        used,
        size,
        cost: update.pointer("/cost/amount").and_then(Value::as_f64),
        currency: update.pointer("/cost/currency").and_then(Value::as_str).map(str::to_owned),
    })
}

fn legacy(
    v: &Value,
    id: &str,
    name: &str,
    list: &str,
    key: &str,
    current: &str,
    via: AgentConfigVia,
) -> AgentConfigOption {
    let values = v
        .get(list)
        .and_then(Value::as_array)
        .into_iter()
        .flatten()
        .filter_map(|m| {
            let value = m.get(key)?.as_str()?.to_owned();
            Some(AgentConfigValue {
                name: str_of(m, "name").unwrap_or_else(|| value.clone()),
                description: str_of(m, "description"),
                value,
            })
        })
        .collect();
    AgentConfigOption {
        id: id.to_owned(),
        name: name.to_owned(),
        description: None,
        category: Some(id.to_owned()),
        current: str_of(v, current),
        values,
        via,
    }
}

fn value_of(v: &Value) -> Option<AgentConfigValue> {
    let value = v.get("value")?.as_str()?.to_owned();
    Some(AgentConfigValue {
        name: str_of(v, "name").unwrap_or_else(|| value.clone()),
        description: str_of(v, "description"),
        value,
    })
}

fn str_of(v: &Value, key: &str) -> Option<String> {
    v.get(key).and_then(Value::as_str).map(str::to_owned)
}

#[cfg(test)]
mod tests {
    use super::*;

    fn ids(c: &[AgentConfigOption]) -> Vec<(&str, Option<&str>)> {
        c.iter().map(|o| (o.id.as_str(), o.current.as_deref())).collect()
    }

    #[test]
    fn config_options_win_over_the_older_fields() {
        let r = json!({
            "modes": {"currentModeId": "default", "availableModes": [{"id": "default", "name": "Manual"}]},
            "configOptions": [
                {"id": "mode", "category": "mode", "type": "select", "currentValue": "plan",
                 "options": [{"value": "default", "name": "Manual"}, {"value": "plan", "name": "Plan"}]},
                {"id": "effort", "category": "thought_level", "type": "select", "currentValue": "medium",
                 "options": [{"group": "g", "name": "G", "options": [{"value": "low"}, {"value": "medium", "name": "Medium"}]}]},
                {"id": "fast", "type": "boolean", "currentValue": false}
            ],
            "models": {"currentModelId": "m1", "availableModels": [{"modelId": "m1", "name": "One"}]}
        });
        let c = options_of(&r);
        assert_eq!(ids(&c), [("mode", Some("plan")), ("effort", Some("medium")), ("model", Some("m1"))]);
        assert_eq!(c[1].values.iter().map(|v| v.name.as_str()).collect::<Vec<_>>(), ["low", "Medium"]);
        assert_eq!(set_request(&c[0], "default").0, "session/set_config_option");
        assert_eq!(set_request(&c[2], "m1"), ("session/set_model", json!({"modelId": "m1"})));
    }

    #[test]
    fn updates_replace_options_and_track_the_mode() {
        let mut c = options_of(&json!({
            "modes": {"currentModeId": "ask", "availableModes": [{"id": "ask"}, {"id": "code"}]},
            "configOptions": [{"id": "model", "type": "select", "currentValue": "a", "options": [{"value": "a"}]}]
        }));
        assert_eq!(c[0].category.as_deref(), Some("model"), "the category defaults from the id");
        set_mode(&mut c, "code");
        assert_eq!(c[1].current.as_deref(), Some("code"));
        assert_eq!(set_request(&c[1], "ask").0, "session/set_mode");
        merge_update(
            &mut c,
            &json!([{"id": "model", "type": "select", "currentValue": "b", "options": [{"value": "b"}]}]),
        );
        assert_eq!(ids(&c), [("model", Some("b")), ("mode", Some("code"))]);
    }

    #[test]
    fn commands_and_usage() {
        let cmds = commands_of(&json!({"availableCommands": [
            {"name": "review", "description": "Review changes", "input": {"hint": "what to review"}},
            {"name": "/init", "description": "Write CLAUDE.md", "input": null}
        ]}));
        assert_eq!(cmds[0].hint.as_deref(), Some("what to review"));
        assert_eq!((cmds[1].name.as_str(), cmds[1].hint.as_ref()), ("init", None));
        let u = usage_of(&json!({"used": 1200, "size": 200000, "cost": {"amount": 0.42, "currency": "USD"}})).unwrap();
        assert_eq!((u.used, u.cost, u.currency.as_deref()), (1200, Some(0.42), Some("USD")));
        assert!(usage_of(&json!({"used": 1, "size": 0})).is_none());
    }
}