mirror of
https://github.com/typst/typst
synced 2025-05-15 01:25:28 +08:00
102 lines
2.6 KiB
Rust
102 lines
2.6 KiB
Rust
mod args;
|
|
mod compile;
|
|
mod download;
|
|
mod fonts;
|
|
mod package;
|
|
mod query;
|
|
mod timings;
|
|
#[cfg(feature = "self-update")]
|
|
mod update;
|
|
mod watch;
|
|
mod world;
|
|
|
|
use std::cell::Cell;
|
|
use std::io::{self, IsTerminal, Write};
|
|
use std::process::ExitCode;
|
|
|
|
use clap::Parser;
|
|
use codespan_reporting::term::{self, termcolor};
|
|
use once_cell::sync::Lazy;
|
|
use termcolor::{ColorChoice, WriteColor};
|
|
|
|
use crate::args::{CliArguments, Command};
|
|
use crate::timings::Timer;
|
|
|
|
thread_local! {
|
|
/// The CLI's exit code.
|
|
static EXIT: Cell<ExitCode> = Cell::new(ExitCode::SUCCESS);
|
|
}
|
|
|
|
/// The parsed commandline arguments.
|
|
static ARGS: Lazy<CliArguments> = Lazy::new(CliArguments::parse);
|
|
|
|
/// Entry point.
|
|
fn main() -> ExitCode {
|
|
let timer = Timer::new(&ARGS);
|
|
let res = match &ARGS.command {
|
|
Command::Compile(command) => crate::compile::compile(timer, command.clone()),
|
|
Command::Watch(command) => crate::watch::watch(timer, command.clone()),
|
|
Command::Query(command) => crate::query::query(command),
|
|
Command::Fonts(command) => crate::fonts::fonts(command),
|
|
Command::Update(command) => crate::update::update(command),
|
|
};
|
|
|
|
if let Err(msg) = res {
|
|
set_failed();
|
|
print_error(&msg).expect("failed to print error");
|
|
}
|
|
|
|
EXIT.with(|cell| cell.get())
|
|
}
|
|
|
|
/// Ensure a failure exit code.
|
|
fn set_failed() {
|
|
EXIT.with(|cell| cell.set(ExitCode::FAILURE));
|
|
}
|
|
|
|
/// Print an application-level error (independent from a source file).
|
|
fn print_error(msg: &str) -> io::Result<()> {
|
|
let mut w = color_stream();
|
|
let styles = term::Styles::default();
|
|
|
|
w.set_color(&styles.header_error)?;
|
|
write!(w, "error")?;
|
|
|
|
w.reset()?;
|
|
writeln!(w, ": {msg}.")
|
|
}
|
|
|
|
/// Get stderr with color support if desirable.
|
|
fn color_stream() -> termcolor::StandardStream {
|
|
termcolor::StandardStream::stderr(match ARGS.color {
|
|
clap::ColorChoice::Auto => {
|
|
if std::io::stderr().is_terminal() {
|
|
ColorChoice::Auto
|
|
} else {
|
|
ColorChoice::Never
|
|
}
|
|
}
|
|
clap::ColorChoice::Always => ColorChoice::Always,
|
|
clap::ColorChoice::Never => ColorChoice::Never,
|
|
})
|
|
}
|
|
|
|
/// Used by `args.rs`.
|
|
fn typst_version() -> &'static str {
|
|
env!("TYPST_VERSION")
|
|
}
|
|
|
|
#[cfg(not(feature = "self-update"))]
|
|
mod update {
|
|
use crate::args::UpdateCommand;
|
|
use typst::diag::{bail, StrResult};
|
|
|
|
pub fn update(_: &UpdateCommand) -> StrResult<()> {
|
|
bail!(
|
|
"self-updating is not enabled for this executable, \
|
|
please update with the package manager or mechanism \
|
|
used for initial installation"
|
|
)
|
|
}
|
|
}
|