lspz 0.10.1

//! CLI entry point for lspz — LSP compression proxy and MCP server.
//!
//! ```bash
//! lspz --backend rust-analyzer       # Proxy mode (default)
//! lspz mcp                           # MCP server mode
//! ```

use std::process::ExitCode;
use std::str::FromStr;
use std::sync::Arc;

use clap::Parser;
use lspz::Transport;
use lspz::config_watcher::ConfigWatcher;
use lspz::interceptors::Interceptor;
use lspz::interceptors::InterceptorChain;
use lspz::interceptors::capping::CappingInterceptor;
use lspz::interceptors::completions::CompletionCompressor;
use lspz::interceptors::diagnostics::DiagnosticsCompressor;
use lspz::interceptors::hover::HoverCompressor;
use lspz::interceptors::locations::LocationCompressor;
use lspz::interceptors::symbols::DocumentSymbolCompressor;
use lspz::interceptors::workspace_diagnostics::WorkspaceDiagnosticCompressor;
use lspz::interceptors::workspace_symbols::WorkspaceSymbolCompressor;
use lspz::metrics::MetredInterceptor;
use lspz::{
    CappingConfig, Config, MetricsConfig, OutputFormat, Proxy, StdioTransport, TcpTransport,
};
use tokio::sync::RwLock;
use tracing_subscriber::EnvFilter;

#[cfg(feature = "mcp")]
use rmcp::ServiceExt;
#[cfg(feature = "mcp")]
use rmcp::transport::stdio;

#[cfg(feature = "transport-websocket")]
use lspz::WsTransport;

/// Grouped CLI arguments for proxy mode to reduce function parameter count.
struct ProxyArgs {
    backend: String,
    backend_args: Vec<String>,
    transport_scheme: String,
    output: String,
    log_level: String,
    max_diags: usize,
    max_completions: usize,
    max_symbols: usize,
    metrics_enabled: bool,
    metrics_interval: u64,
    config_file: Option<String>,
    compress_diag: bool,
    compress_completion: bool,
    compress_hover: bool,
    compress_document_symbol: bool,
    compress_location: bool,
    compress_workspace_symbol: bool,
    compress_workspace_diag: bool,
}

#[derive(Parser, Debug)]
#[command(version, about)]
enum Cli {
    /// Run in proxy mode — transparent LSP proxy with diagnostic compression
    #[command(name = "proxy", alias = "p")]
    Proxy {
        /// Backend LSP server command (e.g. "rust-analyzer", "gopls")
        #[arg(short, long, env = "LSPZ_BACKEND_CMD")]
        backend: String,

        /// Arguments to pass through to the backend LSP server
        #[arg(trailing_var_arg = true, allow_hyphen_values = true)]
        backend_args: Vec<String>,

        /// Transport type: stdio, tcp://host:port, ws://url, wss://url
        #[arg(
            short = 't',
            long = "transport",
            env = "LSPZ_TRANSPORT",
            default_value = "stdio"
        )]
        transport: String,

        /// Enable diagnostic compression (default: true)
        #[arg(
            short = 'd',
            long = "compress-diag",
            env = "LSPZ_ENABLE_DIAG_COMPRESS",
            default_value_t = true
        )]
        compress_diag: bool,

        /// Enable completion compression (default: true)
        #[arg(
            short = 'c',
            long = "compress-completion",
            env = "LSPZ_ENABLE_COMPLETION_COMPRESS",
            default_value_t = true
        )]
        compress_completion: bool,

        /// Enable hover compression (default: true)
        #[arg(
            short = 'H',
            long = "compress-hover",
            env = "LSPZ_ENABLE_HOVER_COMPRESS",
            default_value_t = true
        )]
        compress_hover: bool,

        /// Enable document symbol compression (default: true)
        #[arg(
            short = 'S',
            long = "compress-document-symbol",
            env = "LSPZ_ENABLE_DOCUMENT_SYMBOL_COMPRESS",
            default_value_t = true
        )]
        compress_document_symbol: bool,

        /// Enable location compression (default: true)
        #[arg(
            short = 'L',
            long = "compress-location",
            env = "LSPZ_ENABLE_LOCATION_COMPRESS",
            default_value_t = true
        )]
        compress_location: bool,

        /// Enable workspace symbol compression (default: true)
        #[arg(
            short = 'W',
            long = "compress-workspace-symbol",
            env = "LSPZ_ENABLE_WORKSPACE_SYMBOL_COMPRESS",
            default_value_t = true
        )]
        compress_workspace_symbol: bool,

        /// Enable workspace diagnostic compression (default: true)
        #[arg(
            long = "compress-workspace-diag",
            env = "LSPZ_ENABLE_WORKSPACE_DIAG_COMPRESS",
            default_value_t = true
        )]
        compress_workspace_diag: bool,

        /// Output format: toon, json (compact), or passthrough
        #[arg(
            short = 'o',
            long = "output",
            env = "LSPZ_OUTPUT_FORMAT",
            default_value = "toon"
        )]
        output: String,

        /// Log level (trace, debug, info, warn, error)
        #[arg(short, long, env = "LSPZ_LOG_LEVEL", default_value = "info")]
        log_level: String,

        /// Maximum number of diagnostics to keep (0 = unlimited)
        #[arg(long, env = "LSPZ_MAX_DIAGS", default_value_t = 0)]
        max_diags: usize,

        /// Maximum number of completion items to keep (0 = unlimited)
        #[arg(long, env = "LSPZ_MAX_COMPLETIONS", default_value_t = 0)]
        max_completions: usize,

        /// Maximum number of document symbols to keep (0 = unlimited)
        #[arg(long, env = "LSPZ_MAX_SYMBOLS", default_value_t = 0)]
        max_symbols: usize,

        /// Enable runtime metrics collection
        #[arg(long, env = "LSPZ_METRICS_ENABLED", default_value_t = false)]
        metrics: bool,

        /// Metrics report interval in seconds (0 = only on shutdown)
        #[arg(long, env = "LSPZ_METRICS_INTERVAL", default_value_t = 0)]
        metrics_interval: u64,

        /// Path to TOML config file for hot-reload support
        #[arg(long, env = "LSPZ_CONFIG_FILE")]
        config: Option<String>,
    },

    /// Run as MCP server — exposes LSP tools via Model Context Protocol
    #[cfg(feature = "mcp")]
    #[command(name = "mcp")]
    Mcp {
        /// Log level (trace, debug, info, warn, error)
        #[arg(short, long, env = "LSPZ_LOG_LEVEL", default_value = "info")]
        log_level: String,
    },

    /// Initialize Claude Code integration (MCP server registration, context injection)
    #[command(name = "init")]
    Init {
        /// Add to global config (~/.claude/settings.json) instead of project-local
        #[arg(short, long)]
        global: bool,

        /// Auto-patch settings.json without prompting
        #[arg(long = "auto-patch")]
        auto_patch: bool,

        /// Skip settings.json patching (print manual instructions)
        #[arg(long = "no-patch")]
        no_patch: bool,

        /// Show current lspz configuration
        #[arg(long)]
        show: bool,

        /// Remove lspz artifacts from Claude Code settings
        #[arg(long)]
        uninstall: bool,

        /// Preview changes without writing any files
        #[arg(long = "dry-run")]
        dry_run: bool,
    },
}

#[tokio::main]
async fn main() -> ExitCode {
    match Cli::parse() {
        Cli::Proxy {
            backend,
            backend_args,
            transport,
            output,
            log_level,
            max_diags,
            max_completions,
            max_symbols,
            metrics,
            metrics_interval,
            config,
            compress_diag,
            compress_completion,
            compress_hover,
            compress_document_symbol,
            compress_location,
            compress_workspace_symbol,
            compress_workspace_diag,
        } => {
            let args = ProxyArgs {
                backend,
                backend_args,
                transport_scheme: transport,
                output,
                log_level,
                max_diags,
                max_completions,
                max_symbols,
                metrics_enabled: metrics,
                metrics_interval,
                config_file: config,
                compress_diag,
                compress_completion,
                compress_hover,
                compress_document_symbol,
                compress_location,
                compress_workspace_symbol,
                compress_workspace_diag,
            };
            run_proxy(args).await
        }
        #[cfg(feature = "mcp")]
        Cli::Mcp { log_level } => run_mcp(log_level).await,
        Cli::Init {
            global,
            auto_patch,
            no_patch,
            show,
            uninstall,
            dry_run,
        } => lspz::init::run(global, auto_patch, no_patch, show, uninstall, dry_run),
    }
}

async fn run_proxy(args: ProxyArgs) -> ExitCode {
    if let Some(code) = handle_backend_help(&args.backend, &args.backend_args).await {
        return code;
    }

    tracing_subscriber::fmt()
        .with_env_filter(EnvFilter::builder().parse_lossy(&args.log_level))
        .with_target(false)
        .init();

    let output_format = OutputFormat::from_str(&args.output)
        .ok()
        .unwrap_or(OutputFormat::Json);

    let base_config = match build_config(&args, output_format) {
        Ok(c) => c,
        Err(code) => return code,
    };

    let config = match &args.config_file {
        Some(path) => match Config::from_file(path) {
            Ok(file_config) => {
                tracing::info!(path = %path, "Loaded config file");
                Config {
                    backend_cmd: base_config.backend_cmd.clone(),
                    ..file_config
                }
            }
            Err(e) => {
                eprintln!("Failed to load config file '{path}': {e}");
                return ExitCode::FAILURE;
            }
        },
        None => base_config,
    };

    let shared_config = Arc::new(RwLock::new(config));

    let backend_cmd = shared_config.read().await.backend_cmd.clone();
    let transport: Box<dyn Transport> =
        match create_transport(&args.transport_scheme, &backend_cmd, &args.backend_args).await {
            Ok(t) => t,
            Err(code) => return code,
        };

    let interceptor_chain = build_interceptor_chain(&shared_config);

    if let Some(ref path) = args.config_file {
        match ConfigWatcher::spawn(path, shared_config.clone()) {
            Ok(_) => tracing::info!(path = %path, "Config hot-reload enabled"),
            Err(e) => {
                tracing::warn!(error = %e, "Config watcher failed, running without hot-reload");
            }
        }
    }

    let mut proxy = Proxy::new(shared_config, transport, interceptor_chain);

    if let Err(e) = proxy.start().await {
        tracing::error!(error = %e, "Proxy exited with error");
        return ExitCode::FAILURE;
    }

    ExitCode::SUCCESS
}

async fn handle_backend_help(backend: &str, backend_args: &[String]) -> Option<ExitCode> {
    if !backend_args.iter().any(|a| a == "--help" || a == "-h") {
        return None;
    }

    let parts = shell_words::split(backend).ok()?;
    let mut iter = parts.into_iter();
    let program = iter.next()?;
    let mut args: Vec<String> = iter.collect();
    args.extend(backend_args.iter().cloned());

    let status = tokio::process::Command::new(&program)
        .args(&args)
        .stdin(std::process::Stdio::null())
        .stdout(std::process::Stdio::inherit())
        .stderr(std::process::Stdio::inherit())
        .status()
        .await;

    match status {
        Ok(s) if s.success() => Some(ExitCode::SUCCESS),
        _ => Some(ExitCode::FAILURE),
    }
}

async fn create_transport(
    scheme: &str,
    backend_cmd: &str,
    backend_args: &[String],
) -> Result<Box<dyn Transport>, ExitCode> {
    if scheme == "stdio" {
        StdioTransport::spawn(backend_cmd, backend_args)
            .map(|t| Box::new(t) as Box<dyn Transport>)
            .map_err(|e| {
                eprintln!("Failed to start backend server: {e}");
                ExitCode::FAILURE
            })
    } else if let Some(addr) = scheme.strip_prefix("tcp://") {
        TcpTransport::connect(addr)
            .await
            .map(|t| {
                tracing::info!(addr = %addr, "Connected to TCP LSP server");
                Box::new(t) as Box<dyn Transport>
            })
            .map_err(|e| {
                eprintln!("Failed to connect to TCP server '{addr}': {e}");
                ExitCode::FAILURE
            })
    } else if scheme.starts_with("ws://") || scheme.starts_with("wss://") {
        connect_websocket(scheme).await
    } else {
        eprintln!(
            "Unknown transport scheme '{scheme}'. Use 'stdio', 'tcp://host:port', or 'ws://url'."
        );
        Err(ExitCode::FAILURE)
    }
}

async fn connect_websocket(scheme: &str) -> Result<Box<dyn Transport>, ExitCode> {
    #[cfg(feature = "transport-websocket")]
    {
        WsTransport::connect(scheme)
            .await
            .map(|t| {
                tracing::info!(url = %scheme, "Connected to WebSocket LSP server");
                Box::new(t) as Box<dyn Transport>
            })
            .map_err(|e| {
                eprintln!("Failed to connect to WebSocket server '{scheme}': {e}");
                ExitCode::FAILURE
            })
    }
    #[cfg(not(feature = "transport-websocket"))]
    {
        let _ = scheme;
        eprintln!("WebSocket transport is not enabled. Build with `transport-websocket` feature.");
        Err(ExitCode::FAILURE)
    }
}

fn build_config(args: &ProxyArgs, output_format: OutputFormat) -> Result<Config, ExitCode> {
    Config::builder()
        .backend_cmd(&args.backend)
        .capping(CappingConfig {
            max_diags: args.max_diags,
            max_completions: args.max_completions,
            max_symbols: args.max_symbols,
        })
        .enable_diag_compress(args.compress_diag)
        .enable_completion_compress(args.compress_completion)
        .enable_hover_compress(args.compress_hover)
        .enable_document_symbol_compress(args.compress_document_symbol)
        .enable_location_compress(args.compress_location)
        .enable_workspace_symbol_compress(args.compress_workspace_symbol)
        .enable_workspace_diag_compress(args.compress_workspace_diag)
        .output_format(output_format)
        .log_level(&args.log_level)
        .metrics(MetricsConfig {
            enabled: args.metrics_enabled,
            report_interval_secs: args.metrics_interval,
        })
        .build()
        .map_err(|e| {
            eprintln!("Configuration error: {e}");
            ExitCode::FAILURE
        })
}

fn build_interceptor_chain(shared_config: &Arc<RwLock<Config>>) -> InterceptorChain {
    let config = shared_config.blocking_read();

    let interceptors: Vec<Box<dyn Interceptor>> = vec![
        Box::new(CappingInterceptor::new(
            config.capping.max_diags,
            config.capping.max_completions,
            config.capping.max_symbols,
        )),
        Box::new(DiagnosticsCompressor::default()),
        Box::new(CompletionCompressor::default()),
        Box::new(HoverCompressor::default()),
        Box::new(DocumentSymbolCompressor),
        Box::new(LocationCompressor),
        Box::new(WorkspaceSymbolCompressor),
        Box::new(WorkspaceDiagnosticCompressor),
    ];

    tracing::info!(
        capping = %config.capping.any_enabled(),
        diagnostics = %config.enable_diag_compress,
        completions = %config.enable_completion_compress,
        hover = %config.enable_hover_compress,
        document_symbols = %config.enable_document_symbol_compress,
        locations = %config.enable_location_compress,
        workspace_symbols = %config.enable_workspace_symbol_compress,
        workspace_diags = %config.enable_workspace_diag_compress,
        "Interceptor chain built (all interceptors, gated by runtime config)"
    );

    drop(config);

    if shared_config.blocking_read().metrics.enabled {
        tracing::info!("Metrics collection enabled");
        let wrapped: Vec<Box<dyn Interceptor>> = interceptors
            .into_iter()
            .map(|i| Box::new(MetredInterceptor::new(i).enable()) as Box<dyn Interceptor>)
            .collect();
        InterceptorChain::new(wrapped, shared_config.clone())
    } else {
        InterceptorChain::new(interceptors, shared_config.clone())
    }
}

#[cfg(feature = "mcp")]
async fn run_mcp(log_level: String) -> ExitCode {
    tracing_subscriber::fmt()
        .with_env_filter(EnvFilter::builder().parse_lossy(&log_level))
        .with_target(false)
        .init();

    tracing::info!("Starting lspz MCP server");

    let server = lspz::mcp::McpServer::new();

    match server.serve(stdio()).await {
        Ok(service) => {
            tracing::info!("MCP server ready (stdio)");
            if let Err(e) = service.waiting().await {
                tracing::error!(error = %e, "MCP server exited with error");
                return ExitCode::FAILURE;
            }
        }
        Err(e) => {
            tracing::error!(error = %e, "Failed to start MCP server");
            return ExitCode::FAILURE;
        }
    }

    ExitCode::SUCCESS
}