2019-01-14 17:19:46 -08:00
|
|
|
#[cfg(unix)]
|
|
|
|
|
use std::path::PathBuf;
|
|
|
|
|
#[cfg(unix)]
|
2019-01-08 21:03:51 -08:00
|
|
|
use std::process::{Child, ChildStdin, Command, Stdio};
|
|
|
|
|
use tree_sitter::Parser;
|
|
|
|
|
|
2019-01-15 19:18:33 -08:00
|
|
|
#[cfg(unix)]
|
2019-01-14 17:19:46 -08:00
|
|
|
const HTML_HEADER: &[u8] = b"<!DOCTYPE html>\n<style>svg { width: 100%; }</style>\n\n";
|
|
|
|
|
|
2019-01-10 15:50:54 -08:00
|
|
|
#[cfg(windows)]
|
2019-02-01 14:39:37 -08:00
|
|
|
pub struct LogSession();
|
2019-01-10 15:50:54 -08:00
|
|
|
|
2019-01-14 17:19:46 -08:00
|
|
|
#[cfg(unix)]
|
2019-02-01 14:39:37 -08:00
|
|
|
pub struct LogSession(PathBuf, Option<Child>, Option<ChildStdin>);
|
2019-01-10 15:50:54 -08:00
|
|
|
|
|
|
|
|
#[cfg(windows)]
|
2019-02-01 14:39:37 -08:00
|
|
|
pub fn log_graphs(_parser: &mut Parser, _path: &str) -> std::io::Result<LogSession> {
|
2019-01-14 17:19:46 -08:00
|
|
|
Ok(LogSession())
|
2019-01-10 15:50:54 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[cfg(unix)]
|
2019-02-01 14:39:37 -08:00
|
|
|
pub fn log_graphs(parser: &mut Parser, path: &str) -> std::io::Result<LogSession> {
|
2019-01-14 17:19:46 -08:00
|
|
|
use std::io::Write;
|
2019-01-08 21:03:51 -08:00
|
|
|
|
2019-01-14 17:19:46 -08:00
|
|
|
let mut dot_file = std::fs::File::create(path)?;
|
|
|
|
|
dot_file.write(HTML_HEADER)?;
|
2019-01-08 21:03:51 -08:00
|
|
|
let mut dot_process = Command::new("dot")
|
|
|
|
|
.arg("-Tsvg")
|
|
|
|
|
.stdin(Stdio::piped())
|
|
|
|
|
.stdout(dot_file)
|
|
|
|
|
.spawn()
|
|
|
|
|
.expect("Failed to run Dot");
|
|
|
|
|
let dot_stdin = dot_process
|
|
|
|
|
.stdin
|
|
|
|
|
.take()
|
|
|
|
|
.expect("Failed to open stdin for Dot");
|
|
|
|
|
parser.print_dot_graphs(&dot_stdin);
|
2019-01-17 17:16:04 -08:00
|
|
|
Ok(LogSession(
|
|
|
|
|
PathBuf::from(path),
|
|
|
|
|
Some(dot_process),
|
|
|
|
|
Some(dot_stdin),
|
|
|
|
|
))
|
2019-01-08 21:03:51 -08:00
|
|
|
}
|
|
|
|
|
|
2019-01-10 15:50:54 -08:00
|
|
|
#[cfg(unix)]
|
2019-01-14 17:19:46 -08:00
|
|
|
impl Drop for LogSession {
|
|
|
|
|
fn drop(&mut self) {
|
|
|
|
|
use std::fs;
|
|
|
|
|
|
|
|
|
|
drop(self.2.take().unwrap());
|
|
|
|
|
let output = self.1.take().unwrap().wait_with_output().unwrap();
|
|
|
|
|
if output.status.success() {
|
2019-01-17 17:16:04 -08:00
|
|
|
if cfg!(target_os = "macos")
|
|
|
|
|
&& fs::metadata(&self.0).unwrap().len() > HTML_HEADER.len() as u64
|
|
|
|
|
{
|
2019-01-20 16:58:49 -08:00
|
|
|
Command::new("open").arg(&self.0).output().unwrap();
|
2019-01-14 17:19:46 -08:00
|
|
|
}
|
|
|
|
|
} else {
|
2019-01-17 17:16:04 -08:00
|
|
|
eprintln!(
|
|
|
|
|
"Dot failed: {} {}",
|
|
|
|
|
String::from_utf8_lossy(&output.stdout),
|
|
|
|
|
String::from_utf8_lossy(&output.stderr)
|
|
|
|
|
);
|
2019-01-14 17:19:46 -08:00
|
|
|
}
|
2019-01-10 15:50:54 -08:00
|
|
|
}
|
2019-01-08 21:03:51 -08:00
|
|
|
}
|
