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

use std::{io, process};

use clap::{Parser, Subcommand};
use mdbook::{
  errors::Error,
  preprocess::{CmdPreprocessor, Preprocessor},
};
use semver::{Version, VersionReq};

mod processor;
#[cfg(feature = "testing")]
pub mod testing;

pub use mdbook;
pub use processor::{Asset, SimplePreprocessor};
pub use rayon;

#[derive(Parser)]
#[clap(author, version, about)]
struct Args {
  #[clap(subcommand)]
  command: Option<Command>,
}

#[derive(Subcommand)]
enum Command {
  Supports { renderer: String },
}

pub fn main<P: SimplePreprocessor>() {
  let args = Args::parse();
  let preprocessor = processor::SimplePreprocessorDriver::<P>::new();

  if let Some(Command::Supports { renderer }) = args.command {
    handle_supports(&preprocessor, &renderer);
  } else if let Err(e) = handle_preprocessing(&preprocessor) {
    eprintln!("{}", e);
    process::exit(1);
  }
}

fn handle_preprocessing(pre: &dyn Preprocessor) -> Result<(), Error> {
  let (ctx, book) = CmdPreprocessor::parse_input(io::stdin())?;

  let book_version = Version::parse(&ctx.mdbook_version)?;
  let version_req = VersionReq::parse(mdbook::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::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, renderer: &str) -> ! {
  let supported = pre.supports_renderer(renderer);

  // Signal whether the renderer is supported by exiting with 1 or 0.
  if supported {
    process::exit(0);
  } else {
    process::exit(1);
  }
}