tohaya 0.1.2

Convert citation file formats to hayagriva YAML
Documentation 
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

mod path_arg;

use crate::path_arg::PathArg;
use clap::{Args, Parser};
use itertools::Itertools;
use std::io::Write;
use tohaya::{CitationFormat, tohaya};

#[derive(Parser)]
#[command(version, about)]
struct Cli {
    #[command(flatten)]
    format: FormatArg,

    /// Append to output file
    #[clap(short, long)]
    append: bool,
    /// Output file where to write hayagriva YAML
    #[clap(short, long, default_value = "-")]
    output_file: PathArg,
    /// Input citation file
    #[clap(required = true)]
    input_files: Vec<PathArg>,
}

#[derive(Args, Copy, Clone)]
#[group(required = false, multiple = false)]
struct FormatArg {
    /// Specify input file is BibLaTeX format (.bib file)
    #[clap(short, long)]
    bibtex: bool,
    /// Specify input file is PubMed format, (.nbib file)
    #[clap(short, long)]
    pubmed: bool,
}

impl FormatArg {
    fn resolve(self) -> Option<CitationFormat> {
        if self.bibtex {
            Some(CitationFormat::Bibtex)
        } else if self.pubmed {
            Some(CitationFormat::Pubmed)
        } else {
            None
        }
    }
}

fn main() -> color_eyre::Result<()> {
    color_eyre::install()?;
    let args = Cli::parse();
    let mut out = args
        .output_file
        .open(args.append)
        .map(std::io::BufWriter::new)?;
    let inputs: Vec<_> = args
        .input_files
        .into_iter()
        .map(|s| s.read())
        .try_collect()?;
    let output = tohaya(inputs, args.format.resolve())?;
    out.write_all(output.as_ref())?;
    Ok(())
}