agentgear 0.1.1

//! The JetBrains-Copilot backend: GitHub Copilot's JetBrains IDE plugin. MCP-only
//! in practice — the plugin has no CLI and its only user-writable, file-backed
//! surface is `<config>/github-copilot/intellij/mcp.json`, except when
//! `XDG_CONFIG_HOME` is set: the plugin's resolver checks that first, on every
//! platform, and that branch has **no** `intellij` segment (`mcp_path()`). Root key
//! `servers`, stdio entries shaped `{type:"stdio",command,args,env}`. Written through the
//! shared json renderer (`ServerShape::typed()`), so `remove` is exact (only our
//! server keys) and a second reconcile is a true `NoOp`. Hooks/commands/agents are
//! repo-level `.github` surfaces (Copilot-CLI / vscode-copilot territory) with no
//! JetBrains-plugin file consumer, so they are skipped — see
//! `docs/harness/jetbrains-copilot.md`.
//!
//! User-scope only: capabilities advertise just `user` and the orchestration skips
//! any other scope, so the lifecycle methods ignore their `scope` argument.

use std::path::PathBuf;

use serde_json::Value;

use super::mcpjson::{self, RemoteShape, ServerShape};
use super::report;
use super::{AgentBackend, BackendState};
use crate::components::McpServer;
use crate::doctor::{CheckStatus, DoctorCheck, DoctorReport};
use crate::error::{Error, Result};
use crate::host::{Capabilities, Desired, Outcome, Plugin, Scope, Source};

pub(crate) struct JetbrainsCopilotBackend;

/// mcp.json nests servers under a bare `servers` object (not VS Code's / CC's
/// `mcpServers`); stdio entries carry an explicit `"type":"stdio"`.
const MCP_KEY: &[&str] = &["servers"];
// Remote entries are `{type, url}`: the bundled MCP SDK reads fetch headers only
// from `requestInit.headers`, so a flat `headers` key is never written.
const SHAPE: ServerShape = ServerShape::typed().with_remote(RemoteShape::TypeUrl);

impl AgentBackend for JetbrainsCopilotBackend {
    fn id(&self) -> &'static str {
        "jetbrains-copilot"
    }

    fn detect(&self) -> bool {
        // No CLI on PATH (it's an IDE plugin bundling copilot-language-server), so the
        // JetBrains-specific `github-copilot/intellij` config dir is the only signal.
        // Keyed on that subdir, not the shared `github-copilot` parent, so we don't fire
        // when only the VS Code Copilot backend's config is present. On Unix `config_base()`
        // is `$HOME`-based, so a HOME-redirecting test also redirects detection; on Windows
        // it resolves LocalAppData via SHGetKnownFolderPath (env-independent), so the
        // hermetic test stays Unix-only.
        //
        // Stays on this `intellij`-suffixed dir even though `mcp_path()` also honors
        // `XDG_CONFIG_HOME`: the plugin's own resolver never appends `intellij` on that
        // branch, so there is no marker under an XDG-only install to key detection on.
        config_base().is_some_and(|b| b.join("github-copilot").join("intellij").is_dir())
    }

    fn capabilities(&self) -> Capabilities {
        // mcp-only: the JetBrains plugin's sole writable surface is its mcp.json;
        // hooks/commands/agents/skills are repo-level `.github` surfaces with no
        // JetBrains-plugin file consumer (module doc).
        Capabilities {
            plugins: false,
            mcp: true,
            hooks: false,
            commands: false,
            agents: false,
            skills: false,
            instructions: false,
            statusline: false,
            scopes: &["user"],
        }
    }

    fn probe(&self, plugin: &Plugin, _scope: &Scope, source: &Source) -> Result<BackendState> {
        // Ownership is our mcp server keys; the shared probe returns Healthy — never
        // Absent — for a plugin with no portable servers, so a present marker is never
        // dropped. `source` is the one self_heal resolved for this agent (rehydrated
        // `--path`, else the compile-time default), so probe and reconcile render
        // identical bytes.
        let comp = plugin.components(source)?.with_client(self.id());
        mcpjson::probe(&mcp_path()?, MCP_KEY, &comp.mcp_servers, SHAPE)
    }

    fn reconcile(&self, plugin: &Plugin, desired: &Desired, _scope: &Scope) -> Result<Outcome> {
        let comp = plugin.components(&desired.source)?.with_client(self.id());
        mcpjson::reconcile(&mcp_path()?, MCP_KEY, &comp.mcp_servers, SHAPE)
    }

    fn remove(&self, plugin: &Plugin, _scope: &Scope, source: &Source) -> Result<Outcome> {
        let comp = plugin.components(source)?.with_client(self.id());
        // Key removal off the same portable-server set reconcile writes: an unfiltered name
        // could delete an unrelated user server sharing a name with a non-portable entry we
        // never wrote (e.g. a `${CLAUDE_PLUGIN_ROOT}`-bearing one).
        mcpjson::remove(&mcp_path()?, MCP_KEY, &comp.mcp_servers, SHAPE)
    }

    fn report(&self, plugin: &Plugin, source: &Source) -> DoctorReport {
        DoctorReport::from_checks(report_checks(self, plugin, source))
    }
}

// --- paths -------------------------------------------------------------------

/// GitHub Copilot's config root **fallback**: `~/.config` on every Unix (macOS
/// included — NOT `~/Library`) and `%LOCALAPPDATA%` on Windows — what every
/// non-`XDG_CONFIG_HOME` branch of the plugin's own resolver joins with
/// `github-copilot/intellij` (`config_dir_from` below). Built from `dirs::home_dir()`
/// on Unix rather than `dirs::config_dir()` (which resolves to `~/Library/Application
/// Support` on macOS and would miss the real path), so a test redirecting `$HOME`
/// also redirects us.
#[cfg(not(windows))]
fn config_base() -> Option<PathBuf> {
    dirs::home_dir().map(|h| h.join(".config"))
}

#[cfg(windows)]
fn config_base() -> Option<PathBuf> {
    dirs::data_local_dir()
}

/// Pure resolver mirroring the plugin's own `McpConfigurationService.getConfigPath()`
/// (decompiled, `docs/research/verify-jetbrains-copilot.md`): an absolute
/// `XDG_CONFIG_HOME` wins outright on every platform, Windows included, and lands
/// directly under `github-copilot` with **no** `intellij` segment. Unset, empty, or
/// relative falls back to `fallback_base` (`config_base()`), which every real
/// fallback branch (Windows `LOCALAPPDATA`, Unix `$HOME/.config`) joins with
/// `github-copilot/intellij`. Takes the env value as a parameter (rather than
/// reading it itself) so the branch asymmetry is unit-testable without mutating
/// process env.
fn config_dir_from(xdg_config_home: Option<PathBuf>, fallback_base: Option<PathBuf>) -> Option<PathBuf> {
    xdg_config_home
        .filter(|p| p.is_absolute())
        .map(|xdg| xdg.join("github-copilot"))
        .or_else(|| fallback_base.map(|b| b.join("github-copilot").join("intellij")))
}

fn config_dir() -> Option<PathBuf> {
    config_dir_from(env_nonempty("XDG_CONFIG_HOME").map(PathBuf::from), config_base())
}

fn env_nonempty(var: &str) -> Option<std::ffi::OsString> {
    std::env::var_os(var).filter(|v| !v.is_empty())
}

/// `<xdg>/github-copilot/mcp.json` when `XDG_CONFIG_HOME` is set (absolute), else
/// `<config_base>/github-copilot/intellij/mcp.json` — the JetBrains plugin's only
/// user-scope MCP file (no project-level file exists).
fn mcp_path() -> Result<PathBuf> {
    let dir = config_dir()
        .ok_or_else(|| Error::Tree("no config directory (XDG_CONFIG_HOME/HOME/LOCALAPPDATA unset); cannot locate github-copilot".into()))?;
    Ok(dir.join("mcp.json"))
}

/// Server names `reconcile` actually writes (the shared renderer skips non-portable
/// ones); `remove` must key off the same set.
fn portable_names(servers: &[McpServer]) -> Vec<&str> {
    servers.iter().filter(|s| s.is_portable()).map(|s| s.name.as_str()).collect()
}

// --- report ------------------------------------------------------------------

fn check_mcp_registered(servers: &[McpServer], root: Option<&Value>) -> DoctorCheck {
    let name = "mcp server registered";
    let portable: Vec<&str> = portable_names(servers);
    let skipped = report::skipped_mcp(servers, &portable);
    if portable.is_empty() {
        return report::note_skipped(DoctorCheck { name, status: CheckStatus::Ok(report::NO_MCP.into()) }, &skipped);
    }
    let obj = root.and_then(|r| r.get("servers")).and_then(Value::as_object);
    let missing: Vec<&str> = portable.iter().copied().filter(|n| obj.is_none_or(|o| !o.contains_key(*n))).collect();
    if !missing.is_empty() {
        return DoctorCheck {
            name,
            status: CheckStatus::Fail {
                problem: format!("mcp server(s) not in mcp.json: {}", missing.join(", ")),
                fix: "run the host's `setup`".into(),
            },
        };
    }
    report::note_skipped(DoctorCheck { name, status: CheckStatus::Ok(format!("{} registered", portable.join(", "))) }, &skipped)
}

fn report_checks(backend: &JetbrainsCopilotBackend, plugin: &Plugin, source: &Source) -> Vec<DoctorCheck> {
    let mut checks = Vec::new();

    checks.push(if backend.detect() {
        DoctorCheck { name: "jetbrains-copilot detected", status: CheckStatus::Ok("`github-copilot/intellij` config dir present".into()) }
    } else {
        DoctorCheck {
            name: "jetbrains-copilot detected",
            status: CheckStatus::Warn("no `github-copilot/intellij` config dir; the JetBrains Copilot plugin isn't set up here".into()),
        }
    });

    let path = match mcp_path() {
        Ok(path) => path,
        Err(e) => {
            checks.push(DoctorCheck { name: "mcp.json", status: CheckStatus::Warn(e.to_string()) });
            return checks;
        }
    };

    let root = report::read_json_config(&mut checks, "mcp.json", &path);

    let Some(comp) = report::components(&mut checks, plugin, source).map(|c| c.with_client(backend.id())) else {
        return checks;
    };

    checks.push(check_mcp_registered(&comp.mcp_servers, root.as_ref()));
    checks.push(report::check_mcp_command(&comp.mcp_servers));

    checks
}

#[cfg(test)]
#[path = "../../tests/unit/jetbrains_copilot.rs"]
mod jetbrains_copilot_tests;