mdbook-yapp 2.0.1

//! # Yapp - Yet another preprocessor for `mdBook`
//!
//! This preprocessor simply replaces text in all chapters.

use crate::config::load_config_from_file;
use crate::replacement::Replacements;
use clap::{Arg, ArgMatches, Command};
use mdbook_preprocessor::book::{Book, BookItem};
use mdbook_preprocessor::errors::{Error, Result};
use mdbook_preprocessor::{Preprocessor, PreprocessorContext};
use semver::{Version, VersionReq};
use std::{io, process};

mod config;
mod replacement;
mod tests;

/// A preprocessor for simple text replacements.
struct Yapp {
  replacements: Replacements,
}

impl Yapp {
  /// Creates a preprocessor.
  pub fn new() -> Self {
    if let Some(replacements) = load_config_from_file() {
      Self { replacements }
    } else {
      Self {
        replacements: Replacements::default(),
      }
    }
  }
}

impl Preprocessor for Yapp {
  /// Returns the name of the preprocessor.
  fn name(&self) -> &str {
    "yapp-preprocessor"
  }

  /// Runs preprocessing, replaces text in all chapters.
  fn run(&self, _ctx: &PreprocessorContext, mut book: Book) -> Result<Book, Error> {
    book.for_each_mut(|book_item| {
      if let BookItem::Chapter(chapter) = book_item {
        chapter.content = self.replacements.replace(&chapter.content)
      }
    });
    Ok(book)
  }

  /// Checks for supported renderers.
  fn supports_renderer(&self, renderer: &str) -> Result<bool> {
    Ok(renderer != "not-supported")
  }
}

/// Returns application commands.
fn make_app() -> Command {
  Command::new("yapp-preprocessor").about("mdBook preprocessor for simple text replacements").subcommand(
    Command::new("supports")
      .arg(Arg::new("renderer").required(true))
      .about("Check whether a renderer is supported by this preprocessor"),
  )
}

/// Handles preprocessing command.
fn handle_preprocessing(pre: &dyn Preprocessor) -> Result<(), Error> {
  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 is 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(())
}

/// Handles `supports` command.
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).unwrap();
  if supported {
    process::exit(0);
  } else {
    process::exit(1);
  }
}

/// Main entrypoint.
fn main() {
  let matches = make_app().get_matches();
  let preprocessor = Yapp::new();
  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);
  }
}