use clap::{Arg, ArgMatches, Command};
use env_logger::Builder;
use log::{LevelFilter, info};
use mdbook_preprocessor::Preprocessor;
use semver::{Version, VersionReq};
use std::env;
use std::{io, process};
use mdbook_djot::Djot;
fn main() {
init_logger();
let matches = make_app().get_matches();
let preprocessor = Djot;
if let Some(sub_args) = matches.subcommand_matches("supports") {
handle_supports(&preprocessor, sub_args);
} else if let Err(e) = handle_preprocessing(&preprocessor) {
eprintln!("{e}");
process::exit(1);
}
}
#[must_use]
pub fn make_app() -> Command {
Command::new("nop-preprocessor")
.about("A mdbook preprocessor which does precisely nothing")
.subcommand(
Command::new("supports")
.arg(Arg::new("renderer").required(true))
.about("Check whether a renderer is supported by this preprocessor"),
)
}
fn handle_preprocessing(pre: &dyn Preprocessor) -> anyhow::Result<()> {
info!("Running the preprocessor");
let (ctx, book) = mdbook_preprocessor::parse_input(io::stdin())?;
let book_version = Version::parse(&ctx.mdbook_version)?;
let version_req = VersionReq::parse(mdbook_preprocessor::MDBOOK_VERSION)?;
if !version_req.matches(&book_version) {
eprintln!(
"Warning: The {} plugin was built against version {} of mdbook, \
but we're being called from version {}",
pre.name(),
mdbook_preprocessor::MDBOOK_VERSION,
ctx.mdbook_version
);
}
let processed_book = pre.run(&ctx, book)?;
serde_json::to_writer(io::stdout(), &processed_book)?;
Ok(())
}
fn handle_supports(pre: &dyn Preprocessor, sub_args: &ArgMatches) -> ! {
let renderer = sub_args
.get_one::<String>("renderer")
.expect("Required argument");
let supported = pre.supports_renderer(renderer);
if let Ok(true) = supported {
process::exit(0)
} else {
process::exit(1)
}
}
pub fn init_logger() {
let mut builder = Builder::new();
if let Ok(var) = env::var("RUST_LOG") {
builder.parse_filters(&var);
} else {
builder.filter(None, LevelFilter::Info);
}
builder.init();
}