1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
//! 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_stdout, clippy::print_stderr)]
#[cfg(feature = "in-rust-tree")]
extern crate rustc_driver as _;
mod version;
use clap::{Command, ValueEnum};
use proc_macro_api::ProtocolFormat;
#[cfg(feature = "sysroot-abi")]
use proc_macro_srv_cli::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-legacy")
.value_parser(clap::builder::EnumValueParser::<ProtocolFormatArg>::new()),
clap::Arg::new("version")
.long("version")
.action(clap::ArgAction::SetTrue)
.help("Prints the version of the proc-macro-srv"),
])
.get_matches();
if matches.get_flag("version") {
println!("rust-analyzer-proc-macro-srv {}", version::version());
return Ok(());
}
let &format = matches
.get_one::<ProtocolFormatArg>("format")
.expect("format value should always be present");
let mut stdin = std::io::BufReader::new(std::io::stdin());
let mut stdout = std::io::stdout();
run(&mut stdin, &mut stdout, format.into())
}
/// Wrapper for CLI argument parsing that implements `ValueEnum`.
#[derive(Copy, Clone)]
struct ProtocolFormatArg(ProtocolFormat);
impl From<ProtocolFormatArg> for ProtocolFormat {
fn from(arg: ProtocolFormatArg) -> Self {
arg.0
}
}
impl ValueEnum for ProtocolFormatArg {
fn value_variants<'a>() -> &'a [Self] {
&[
ProtocolFormatArg(ProtocolFormat::JsonLegacy),
ProtocolFormatArg(ProtocolFormat::BidirectionalPostcardPrototype),
]
}
fn to_possible_value(&self) -> Option<clap::builder::PossibleValue> {
match self.0 {
ProtocolFormat::JsonLegacy => Some(clap::builder::PossibleValue::new("json-legacy")),
ProtocolFormat::BidirectionalPostcardPrototype => {
Some(clap::builder::PossibleValue::new("bidirectional-postcard-prototype"))
}
}
}
fn from_str(input: &str, _ignore_case: bool) -> Result<Self, String> {
match input {
"json-legacy" => Ok(ProtocolFormatArg(ProtocolFormat::JsonLegacy)),
"bidirectional-postcard-prototype" => {
Ok(ProtocolFormatArg(ProtocolFormat::BidirectionalPostcardPrototype))
}
_ => Err(format!("unknown protocol format: {input}")),
}
}
}
#[cfg(not(feature = "sysroot-abi"))]
fn run(
_: &mut std::io::BufReader<std::io::Stdin>,
_: &mut std::io::Stdout,
_: 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(),
))
}
|