rust-meth 0.1.6

//! `rust-meth`: A CLI utility to discover and filter methods available on Rust types.
//! It leverages `rust-analyzer` via the Language Server Protocol (LSP) to provide
//! accurate, context-aware method resolution.

#![deny(missing_docs)]

mod analyzer;
mod lsp;
mod probe;
mod ui;

use std::process;
use std::time::Instant;

fn main() {
    if let Err(err) = run() {
        eprintln!("error: {err}");
        process::exit(1);
    }
}

/// Orchestrates the tool's execution flow:
/// 1. Finds rust-analyzer.
/// 2. Queries for methods.
/// 3. Routes to interactive or batch output modes.
fn run() -> Result<(), String> {
    let opts = ui::parse_args()?;

    let ra_path = analyzer::find_rust_analyzer().map_err(|e| e.to_string())?;

    // Handle --gd before querying methods. Avoids spinning up RA twice.
    if let Some(method_name) = &opts.goto_def {
        let spinner = ui::definition(&opts.type_name, method_name);

        // match analyzer::query_definition(&opts.type_name, method_name, &ra_path) {
        match analyzer::query_definition(
            &opts.type_name,
            method_name,
            &ra_path,
            opts.deps.as_deref(),
        ) {
            Ok(Some(def)) => {
                spinner.finish_with_message("✓ Found definition");

                println!(
                    "{}::{}  {}:{}",
                    opts.type_name,
                    method_name,
                    def.path,
                    def.line + 1
                );

                if opts.open_def {
                    ui::open_in_editor(&def)?;
                }

                if opts.open_doc {
                    let url = ui::build_doc_url(&opts.type_name, method_name, &def);
                    ui::open_in_browser(&url)?;
                }
            }
            Ok(None) => {
                spinner.finish_with_message("✗ Not found");

                return Err(format!(
                    "No definition found for `{}::{}` — is rust-src installed?\n\
                     Run: rustup component add rust-src",
                    opts.type_name, method_name
                ));
            }
            Err(e) => {
                spinner.finish_with_message("✗ Query failed");
                return Err(e.to_string());
            }
        }
        return Ok(());
    }

    let start = Instant::now();
    let spinner = ui::indexing(&opts.type_name);

    let methods = match analyzer::query_methods(&opts.type_name, &ra_path, opts.deps.as_deref()) {
        Ok(methods) => {
            let elapsed = start.elapsed();
            spinner.finish_with_message(format!(
                "✓ Found {} methods ({:.1}s)",
                methods.len(),
                elapsed.as_secs_f64()
            ));
            methods
        }
        Err(e) => {
            spinner.finish_with_message("✗ Query failed");
            return Err(e.to_string());
        }
    };

    if opts.interactive {
        return ui::run_interactive(&opts, &methods);
    }

    let matched = ui::filter_methods(&methods, opts.filter.as_deref());

    if matched.is_empty() {
        return Err(match opts.filter.as_deref() {
            Some(pat) => format!("No methods on `{}` matching {pat:?}", opts.type_name),
            None => format!("No methods found for type `{}`", opts.type_name),
        });
    }

    match opts.filter.as_deref() {
        Some(pat) => println!(
            "\n{}: methods on `{}` matching {pat:?}\n",
            opts.bin, opts.type_name
        ),
        None => println!("\n{}: methods on `{}`\n", opts.bin, opts.type_name),
    }

    let name_width = matched.iter().map(|m| m.name.len()).max().unwrap_or(0);

    for m in &matched {
        ui::print_method(m, name_width, opts.show_doc);
    }

    println!("\n{} method(s)", matched.len());
    Ok(())
}