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
|
use futures::{channel::mpsc, prelude::*};
use jsonrpc::MessageHandler;
use log::LevelFilter;
use std::path::PathBuf;
use std::{env, error, fs::OpenOptions, sync::Arc};
use structopt::StructOpt;
use texlab::{
protocol::{LatexLspClient, LspCodec},
server::LatexLspServer,
tex::Distribution,
};
use tokio_util::codec::{FramedRead, FramedWrite};
/// An implementation of the Language Server Protocol for LaTeX
#[derive(Debug, StructOpt)]
struct Opts {
/// Increase message verbosity (-vvvv for max verbosity)
#[structopt(short, long, parse(from_occurrences))]
verbosity: u8,
/// No output printed to stderr
#[structopt(short, long)]
quiet: bool,
/// Write the logging output to FILE
#[structopt(long, name = "FILE", parse(from_os_str))]
log_file: Option<PathBuf>,
}
#[tokio::main]
async fn main() -> Result<(), Box<dyn error::Error>> {
let opts = Opts::from_args();
setup_logger(opts);
let mut stdin = FramedRead::new(tokio::io::stdin(), LspCodec);
let (stdout_tx, mut stdout_rx) = mpsc::channel(0);
let client = Arc::new(LatexLspClient::new(stdout_tx.clone()));
let server = Arc::new(LatexLspServer::new(
Distribution::detect().await,
Arc::clone(&client),
Arc::new(env::current_dir().expect("failed to get working directory")),
));
let mut handler = MessageHandler {
server,
client,
output: stdout_tx,
};
tokio::spawn(async move {
let mut stdout = FramedWrite::new(tokio::io::stdout(), LspCodec);
loop {
let message = stdout_rx.next().await.unwrap();
stdout.send(message).await.unwrap();
}
});
while let Some(json) = stdin.next().await {
handler.handle(&json.unwrap()).await;
}
Ok(())
}
fn setup_logger(opts: Opts) {
let verbosity_level = if !opts.quiet {
match opts.verbosity {
0 => LevelFilter::Error,
1 => LevelFilter::Warn,
2 => LevelFilter::Info,
3 => LevelFilter::Debug,
_ => LevelFilter::Trace,
}
} else {
LevelFilter::Off
};
let logger = fern::Dispatch::new()
.format(|out, message, record| out.finish(format_args!("{} - {}", record.level(), message)))
.level(verbosity_level)
.filter(|metadata| metadata.target() == "jsonrpc" || metadata.target().contains("texlab"))
.chain(std::io::stderr());
let logger = match opts.log_file {
Some(log_file) => logger.chain(
OpenOptions::new()
.write(true)
.create(true)
.open(log_file)
.expect("failed to open log file"),
),
None => logger,
};
logger.apply().expect("failed to initialize logger");
}
|
