traxys/tree-sitter
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Tweak cancellation logic for CLI commands

Browse source 
In 'parse' and 'highlight' cancel on stdin if stdin is a tty.
This commit is contained in:
Max Brunsfeld 2020-10-22 15:55:51 -07:00
parent 5caa83e020
commit b972a7158d
6 changed files with 358 additions and 343 deletions
Download patch file Download diff file Expand all files Collapse all files

18
cli/src/util.rs
View file

@ -15,14 +15,16 @@ const HTML_HEADER: &[u8] = b"<!DOCTYPE html>\n<style>svg { width: 100%; }</style
pub fn cancel_on_stdin() -> Arc<AtomicUsize> {
let result = Arc::new(AtomicUsize::new(0));
thread::spawn({
let flag = result.clone();
move || {
let mut line = String::new();
io::stdin().read_line(&mut line).unwrap();
flag.store(1, Ordering::Relaxed);
}
});
if atty::is(atty::Stream::Stdin) {
thread::spawn({
let flag = result.clone();
move || {
let mut line = String::new();
io::stdin().read_line(&mut line).unwrap();
flag.store(1, Ordering::Relaxed);
}
});
}
result
}
#[cfg(windows)]