crates/proc-macro-srv-cli/src/main.rs

//! A standalone binary for `proc-macro-srv`.
//! Driver for proc macro server
#![cfg_attr(feature = "in-rust-tree", feature(rustc_private))]
#![cfg_attr(not(feature = "sysroot-abi"), allow(unused_crate_dependencies))]
#![allow(clippy::print_stderr)]

#[cfg(feature = "in-rust-tree")]
extern crate rustc_driver as _;

#[cfg(any(feature = "sysroot-abi", rust_analyzer))]
mod main_loop;
use clap::{Command, ValueEnum};
#[cfg(any(feature = "sysroot-abi", rust_analyzer))]
use main_loop::run;

fn main() -> std::io::Result<()> {
    let v = std::env::var("RUST_ANALYZER_INTERNALS_DO_NOT_USE");
    if v.is_err() {
        eprintln!(
            "This is an IDE implementation detail, you can use this tool by exporting RUST_ANALYZER_INTERNALS_DO_NOT_USE."
        );
        eprintln!(
            "Note that this tool's API is highly unstable and may break without prior notice"
        );
        std::process::exit(122);
    }
    let matches = Command::new("proc-macro-srv")
        .args(&[clap::Arg::new("format")
            .long("format")
            .action(clap::ArgAction::Set)
            .default_value("json")
            .value_parser(clap::builder::EnumValueParser::<ProtocolFormat>::new())])
        .get_matches();
    let &format =
        matches.get_one::<ProtocolFormat>("format").expect("format value should always be present");
    run(format)
}

#[derive(Copy, Clone)]
enum ProtocolFormat {
    Json,
    #[cfg(feature = "postcard")]
    Postcard,
}

impl ValueEnum for ProtocolFormat {
    fn value_variants<'a>() -> &'a [Self] {
        &[ProtocolFormat::Json]
    }

    fn to_possible_value(&self) -> Option<clap::builder::PossibleValue> {
        match self {
            ProtocolFormat::Json => Some(clap::builder::PossibleValue::new("json")),
            #[cfg(feature = "postcard")]
            ProtocolFormat::Postcard => Some(clap::builder::PossibleValue::new("postcard")),
        }
    }
    fn from_str(input: &str, _ignore_case: bool) -> Result<Self, String> {
        match input {
            "json" => Ok(ProtocolFormat::Json),
            #[cfg(feature = "postcard")]
            "postcard" => Ok(ProtocolFormat::Postcard),
            _ => Err(format!("unknown protocol format: {input}")),
        }
    }
}

#[cfg(not(any(feature = "sysroot-abi", rust_analyzer)))]
fn run(_: ProtocolFormat) -> std::io::Result<()> {
    Err(std::io::Error::new(
        std::io::ErrorKind::Unsupported,
        "proc-macro-srv-cli needs to be compiled with the `sysroot-abi` feature to function"
            .to_owned(),
    ))
}