mdbook_translator/
command_handler.rs

1use mdbook::preprocess::{Preprocessor, CmdPreprocessor};
2use clap::{Arg, ArgMatches, Command};
3use semver::{Version, VersionReq};
4use std::io;
5use mdbook::errors::Error;
6use anyhow::Result;
7use std::process;
8use toml::value::Value;
9use crate::translate_preprocessor::DeepSeekTranslator;
10
11pub fn handle_preprocessing(pre: &mut DeepSeekTranslator) -> Result<(), Error> {
12    let (ctx, book) = CmdPreprocessor::parse_input(io::stdin())?;
13
14    let book_version = Version::parse(&ctx.mdbook_version)?;
15    let version_req = VersionReq::parse(mdbook::MDBOOK_VERSION)?;
16
17    if !version_req.matches(&book_version) {
18        eprintln!(
19            "Warning: The {} plugin was built against version {} of mdbook, \
20             but we're being called from version {}",
21            pre.name(),
22            mdbook::MDBOOK_VERSION,
23            ctx.mdbook_version
24        );
25    }
26
27    let language = 
28        ctx.config.get("preprocessor")
29            .and_then(|p| p.get("translator"))
30            .and_then(|t| t.get("language"));
31    let ext_prompt = 
32        ctx.config.get("preprocessor")
33            .and_then(|p| p.get("translator"))
34            .and_then(|t| t.get("prompt"));
35    let proxy = 
36        ctx.config.get("preprocessor")
37            .and_then(|p| p.get("translator"))
38            .and_then(|t| t.get("proxy"));
39
40    if let Some(Value::String(language_config)) = language {
41        if !language_config.is_empty() {
42            pre.set_language(language_config);
43        }
44    }
45
46    if let Some(Value::String(prompt_config)) = ext_prompt {
47        if !prompt_config.is_empty() {
48            pre.set_prompt(prompt_config);
49        }
50    }
51
52    if let Some(Value::String(proxy_config)) = proxy {
53        if !proxy_config.is_empty() {
54            pre.set_proxy(proxy_config);
55        }
56    }
57
58    eprintln!("target_lang: {:?}", pre.target_lang);
59    eprintln!("prompt: {:?}", pre.prompt);
60
61    let processed_book = pre.run(&ctx, book)?;
62    serde_json::to_writer(io::stdout(), &processed_book)?;
63
64    Ok(())
65}
66
67pub fn handle_supports(pre: &dyn Preprocessor, sub_args: &ArgMatches) -> ! {
68    let renderer = sub_args
69        .get_one::<String>("renderer")
70        .expect("Required argument");
71    let supported = pre.supports_renderer(renderer);
72
73    // Signal whether the renderer is supported by exiting with 1 or 0.
74    if supported {
75        process::exit(0);
76    } else {
77        process::exit(1);
78    }
79}
80
81pub fn make_app() -> Command {
82    Command::new("mdbook-translator")
83        .about("A translation preprocessor plugin for mdBook that automatically translates Markdown documents using the DeepSeek API.")
84        .subcommand(
85            Command::new("supports")
86                .arg(Arg::new("renderer").required(true))
87                .about("Check whether a renderer is supported by this preprocessor"),
88        )
89}
mdbook_translator/command_handler.rs

mdbook_translator/
command_handler.rs
