diff options
author | Aleksey Kladov <[email protected]> | 2020-06-25 08:13:46 +0100 |
---|---|---|
committer | Aleksey Kladov <[email protected]> | 2020-06-25 08:13:46 +0100 |
commit | 0ec5d4f55c6a5ac3fadcd48ae17b70379aba17fa (patch) | |
tree | 89fb78d856e36225f7f09560020729b1200a8ee6 /crates/ra_flycheck/src | |
parent | 5a184fe85517507fd3b07c6fb36b017e558665f7 (diff) |
Rename ra_flycheck -> flycheck
Diffstat (limited to 'crates/ra_flycheck/src')
-rw-r--r-- | crates/ra_flycheck/src/lib.rs | 342 |
1 files changed, 0 insertions, 342 deletions
diff --git a/crates/ra_flycheck/src/lib.rs b/crates/ra_flycheck/src/lib.rs deleted file mode 100644 index af75adbe2..000000000 --- a/crates/ra_flycheck/src/lib.rs +++ /dev/null | |||
@@ -1,342 +0,0 @@ | |||
1 | //! cargo_check provides the functionality needed to run `cargo check` or | ||
2 | //! another compatible command (f.x. clippy) in a background thread and provide | ||
3 | //! LSP diagnostics based on the output of the command. | ||
4 | |||
5 | use std::{ | ||
6 | fmt, | ||
7 | io::{self, BufReader}, | ||
8 | path::PathBuf, | ||
9 | process::{Command, Stdio}, | ||
10 | time::Instant, | ||
11 | }; | ||
12 | |||
13 | use cargo_metadata::Message; | ||
14 | use crossbeam_channel::{never, select, unbounded, Receiver, RecvError, Sender}; | ||
15 | |||
16 | pub use cargo_metadata::diagnostic::{ | ||
17 | Applicability, Diagnostic, DiagnosticLevel, DiagnosticSpan, DiagnosticSpanMacroExpansion, | ||
18 | }; | ||
19 | |||
20 | #[derive(Clone, Debug, PartialEq, Eq)] | ||
21 | pub enum FlycheckConfig { | ||
22 | CargoCommand { | ||
23 | command: String, | ||
24 | all_targets: bool, | ||
25 | all_features: bool, | ||
26 | features: Vec<String>, | ||
27 | extra_args: Vec<String>, | ||
28 | }, | ||
29 | CustomCommand { | ||
30 | command: String, | ||
31 | args: Vec<String>, | ||
32 | }, | ||
33 | } | ||
34 | |||
35 | impl fmt::Display for FlycheckConfig { | ||
36 | fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { | ||
37 | match self { | ||
38 | FlycheckConfig::CargoCommand { command, .. } => write!(f, "cargo {}", command), | ||
39 | FlycheckConfig::CustomCommand { command, args } => { | ||
40 | write!(f, "{} {}", command, args.join(" ")) | ||
41 | } | ||
42 | } | ||
43 | } | ||
44 | } | ||
45 | |||
46 | /// Flycheck wraps the shared state and communication machinery used for | ||
47 | /// running `cargo check` (or other compatible command) and providing | ||
48 | /// diagnostics based on the output. | ||
49 | /// The spawned thread is shut down when this struct is dropped. | ||
50 | #[derive(Debug)] | ||
51 | pub struct FlycheckHandle { | ||
52 | // XXX: drop order is significant | ||
53 | cmd_send: Sender<CheckCommand>, | ||
54 | handle: jod_thread::JoinHandle<()>, | ||
55 | } | ||
56 | |||
57 | impl FlycheckHandle { | ||
58 | pub fn spawn( | ||
59 | sender: Box<dyn Fn(CheckTask) + Send>, | ||
60 | config: FlycheckConfig, | ||
61 | workspace_root: PathBuf, | ||
62 | ) -> FlycheckHandle { | ||
63 | let (cmd_send, cmd_recv) = unbounded::<CheckCommand>(); | ||
64 | let handle = jod_thread::spawn(move || { | ||
65 | FlycheckActor::new(sender, config, workspace_root).run(&cmd_recv); | ||
66 | }); | ||
67 | FlycheckHandle { cmd_send, handle } | ||
68 | } | ||
69 | |||
70 | /// Schedule a re-start of the cargo check worker. | ||
71 | pub fn update(&self) { | ||
72 | self.cmd_send.send(CheckCommand::Update).unwrap(); | ||
73 | } | ||
74 | } | ||
75 | |||
76 | #[derive(Debug)] | ||
77 | pub enum CheckTask { | ||
78 | /// Request a clearing of all cached diagnostics from the check watcher | ||
79 | ClearDiagnostics, | ||
80 | |||
81 | /// Request adding a diagnostic with fixes included to a file | ||
82 | AddDiagnostic { workspace_root: PathBuf, diagnostic: Diagnostic }, | ||
83 | |||
84 | /// Request check progress notification to client | ||
85 | Status(Status), | ||
86 | } | ||
87 | |||
88 | #[derive(Debug)] | ||
89 | pub enum Status { | ||
90 | Being, | ||
91 | Progress(String), | ||
92 | End, | ||
93 | } | ||
94 | |||
95 | pub enum CheckCommand { | ||
96 | /// Request re-start of check thread | ||
97 | Update, | ||
98 | } | ||
99 | |||
100 | struct FlycheckActor { | ||
101 | sender: Box<dyn Fn(CheckTask) + Send>, | ||
102 | config: FlycheckConfig, | ||
103 | workspace_root: PathBuf, | ||
104 | last_update_req: Option<Instant>, | ||
105 | // XXX: drop order is significant | ||
106 | message_recv: Receiver<CheckEvent>, | ||
107 | /// WatchThread exists to wrap around the communication needed to be able to | ||
108 | /// run `cargo check` without blocking. Currently the Rust standard library | ||
109 | /// doesn't provide a way to read sub-process output without blocking, so we | ||
110 | /// have to wrap sub-processes output handling in a thread and pass messages | ||
111 | /// back over a channel. | ||
112 | check_process: Option<jod_thread::JoinHandle<()>>, | ||
113 | } | ||
114 | |||
115 | impl FlycheckActor { | ||
116 | fn new( | ||
117 | sender: Box<dyn Fn(CheckTask) + Send>, | ||
118 | config: FlycheckConfig, | ||
119 | workspace_root: PathBuf, | ||
120 | ) -> FlycheckActor { | ||
121 | FlycheckActor { | ||
122 | sender, | ||
123 | config, | ||
124 | workspace_root, | ||
125 | last_update_req: None, | ||
126 | message_recv: never(), | ||
127 | check_process: None, | ||
128 | } | ||
129 | } | ||
130 | |||
131 | fn run(&mut self, cmd_recv: &Receiver<CheckCommand>) { | ||
132 | // If we rerun the thread, we need to discard the previous check results first | ||
133 | self.clean_previous_results(); | ||
134 | |||
135 | loop { | ||
136 | select! { | ||
137 | recv(&cmd_recv) -> cmd => match cmd { | ||
138 | Ok(cmd) => self.handle_command(cmd), | ||
139 | Err(RecvError) => { | ||
140 | // Command channel has closed, so shut down | ||
141 | break; | ||
142 | }, | ||
143 | }, | ||
144 | recv(self.message_recv) -> msg => match msg { | ||
145 | Ok(msg) => self.handle_message(msg), | ||
146 | Err(RecvError) => { | ||
147 | // Watcher finished, replace it with a never channel to | ||
148 | // avoid busy-waiting. | ||
149 | self.message_recv = never(); | ||
150 | self.check_process = None; | ||
151 | }, | ||
152 | } | ||
153 | }; | ||
154 | |||
155 | if self.should_recheck() { | ||
156 | self.last_update_req = None; | ||
157 | self.send(CheckTask::ClearDiagnostics); | ||
158 | self.restart_check_process(); | ||
159 | } | ||
160 | } | ||
161 | } | ||
162 | |||
163 | fn clean_previous_results(&self) { | ||
164 | self.send(CheckTask::ClearDiagnostics); | ||
165 | self.send(CheckTask::Status(Status::End)); | ||
166 | } | ||
167 | |||
168 | fn should_recheck(&mut self) -> bool { | ||
169 | if let Some(_last_update_req) = &self.last_update_req { | ||
170 | // We currently only request an update on save, as we need up to | ||
171 | // date source on disk for cargo check to do it's magic, so we | ||
172 | // don't really need to debounce the requests at this point. | ||
173 | return true; | ||
174 | } | ||
175 | false | ||
176 | } | ||
177 | |||
178 | fn handle_command(&mut self, cmd: CheckCommand) { | ||
179 | match cmd { | ||
180 | CheckCommand::Update => self.last_update_req = Some(Instant::now()), | ||
181 | } | ||
182 | } | ||
183 | |||
184 | fn handle_message(&self, msg: CheckEvent) { | ||
185 | match msg { | ||
186 | CheckEvent::Begin => { | ||
187 | self.send(CheckTask::Status(Status::Being)); | ||
188 | } | ||
189 | |||
190 | CheckEvent::End => { | ||
191 | self.send(CheckTask::Status(Status::End)); | ||
192 | } | ||
193 | |||
194 | CheckEvent::Msg(Message::CompilerArtifact(msg)) => { | ||
195 | self.send(CheckTask::Status(Status::Progress(msg.target.name))); | ||
196 | } | ||
197 | |||
198 | CheckEvent::Msg(Message::CompilerMessage(msg)) => { | ||
199 | self.send(CheckTask::AddDiagnostic { | ||
200 | workspace_root: self.workspace_root.clone(), | ||
201 | diagnostic: msg.message, | ||
202 | }); | ||
203 | } | ||
204 | |||
205 | CheckEvent::Msg(Message::BuildScriptExecuted(_msg)) => {} | ||
206 | CheckEvent::Msg(Message::BuildFinished(_)) => {} | ||
207 | CheckEvent::Msg(Message::TextLine(_)) => {} | ||
208 | CheckEvent::Msg(Message::Unknown) => {} | ||
209 | } | ||
210 | } | ||
211 | |||
212 | fn restart_check_process(&mut self) { | ||
213 | // First, clear and cancel the old thread | ||
214 | self.message_recv = never(); | ||
215 | self.check_process = None; | ||
216 | |||
217 | let mut cmd = match &self.config { | ||
218 | FlycheckConfig::CargoCommand { | ||
219 | command, | ||
220 | all_targets, | ||
221 | all_features, | ||
222 | extra_args, | ||
223 | features, | ||
224 | } => { | ||
225 | let mut cmd = Command::new(ra_toolchain::cargo()); | ||
226 | cmd.arg(command); | ||
227 | cmd.args(&["--workspace", "--message-format=json", "--manifest-path"]) | ||
228 | .arg(self.workspace_root.join("Cargo.toml")); | ||
229 | if *all_targets { | ||
230 | cmd.arg("--all-targets"); | ||
231 | } | ||
232 | if *all_features { | ||
233 | cmd.arg("--all-features"); | ||
234 | } else if !features.is_empty() { | ||
235 | cmd.arg("--features"); | ||
236 | cmd.arg(features.join(" ")); | ||
237 | } | ||
238 | cmd.args(extra_args); | ||
239 | cmd | ||
240 | } | ||
241 | FlycheckConfig::CustomCommand { command, args } => { | ||
242 | let mut cmd = Command::new(command); | ||
243 | cmd.args(args); | ||
244 | cmd | ||
245 | } | ||
246 | }; | ||
247 | cmd.current_dir(&self.workspace_root); | ||
248 | |||
249 | let (message_send, message_recv) = unbounded(); | ||
250 | self.message_recv = message_recv; | ||
251 | self.check_process = Some(jod_thread::spawn(move || { | ||
252 | // If we trigger an error here, we will do so in the loop instead, | ||
253 | // which will break out of the loop, and continue the shutdown | ||
254 | let _ = message_send.send(CheckEvent::Begin); | ||
255 | |||
256 | let res = run_cargo(cmd, &mut |message| { | ||
257 | // Skip certain kinds of messages to only spend time on what's useful | ||
258 | match &message { | ||
259 | Message::CompilerArtifact(artifact) if artifact.fresh => return true, | ||
260 | Message::BuildScriptExecuted(_) => return true, | ||
261 | Message::Unknown => return true, | ||
262 | _ => {} | ||
263 | } | ||
264 | |||
265 | // if the send channel was closed, we want to shutdown | ||
266 | message_send.send(CheckEvent::Msg(message)).is_ok() | ||
267 | }); | ||
268 | |||
269 | if let Err(err) = res { | ||
270 | // FIXME: make the `message_send` to be `Sender<Result<CheckEvent, CargoError>>` | ||
271 | // to display user-caused misconfiguration errors instead of just logging them here | ||
272 | log::error!("Cargo watcher failed {:?}", err); | ||
273 | } | ||
274 | |||
275 | // We can ignore any error here, as we are already in the progress | ||
276 | // of shutting down. | ||
277 | let _ = message_send.send(CheckEvent::End); | ||
278 | })) | ||
279 | } | ||
280 | |||
281 | fn send(&self, check_task: CheckTask) { | ||
282 | (self.sender)(check_task) | ||
283 | } | ||
284 | } | ||
285 | |||
286 | enum CheckEvent { | ||
287 | Begin, | ||
288 | Msg(cargo_metadata::Message), | ||
289 | End, | ||
290 | } | ||
291 | |||
292 | fn run_cargo( | ||
293 | mut command: Command, | ||
294 | on_message: &mut dyn FnMut(cargo_metadata::Message) -> bool, | ||
295 | ) -> io::Result<()> { | ||
296 | let mut child = | ||
297 | command.stdout(Stdio::piped()).stderr(Stdio::null()).stdin(Stdio::null()).spawn()?; | ||
298 | |||
299 | // We manually read a line at a time, instead of using serde's | ||
300 | // stream deserializers, because the deserializer cannot recover | ||
301 | // from an error, resulting in it getting stuck, because we try to | ||
302 | // be resillient against failures. | ||
303 | // | ||
304 | // Because cargo only outputs one JSON object per line, we can | ||
305 | // simply skip a line if it doesn't parse, which just ignores any | ||
306 | // erroneus output. | ||
307 | let stdout = BufReader::new(child.stdout.take().unwrap()); | ||
308 | let mut read_at_least_one_message = false; | ||
309 | for message in cargo_metadata::Message::parse_stream(stdout) { | ||
310 | let message = match message { | ||
311 | Ok(message) => message, | ||
312 | Err(err) => { | ||
313 | log::error!("Invalid json from cargo check, ignoring ({})", err); | ||
314 | continue; | ||
315 | } | ||
316 | }; | ||
317 | |||
318 | read_at_least_one_message = true; | ||
319 | |||
320 | if !on_message(message) { | ||
321 | break; | ||
322 | } | ||
323 | } | ||
324 | |||
325 | // It is okay to ignore the result, as it only errors if the process is already dead | ||
326 | let _ = child.kill(); | ||
327 | |||
328 | let exit_status = child.wait()?; | ||
329 | if !exit_status.success() && !read_at_least_one_message { | ||
330 | // FIXME: Read the stderr to display the reason, see `read2()` reference in PR comment: | ||
331 | // https://github.com/rust-analyzer/rust-analyzer/pull/3632#discussion_r395605298 | ||
332 | return Err(io::Error::new( | ||
333 | io::ErrorKind::Other, | ||
334 | format!( | ||
335 | "the command produced no valid metadata (exit code: {:?}): {:?}", | ||
336 | exit_status, command | ||
337 | ), | ||
338 | )); | ||
339 | } | ||
340 | |||
341 | Ok(()) | ||
342 | } | ||