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

mod error;
mod guess;
mod pubmed;

use crate::error::TohayaError;
use crate::guess::guess_format;
use crate::pubmed::parse_pubmed;
pub use error::ParseError;
use itertools::Itertools;

#[cfg(feature = "wasm")]
use wasm_bindgen::prelude::*;

/// Citation file formats supported by _tohaya_.
#[cfg_attr(feature = "wasm", wasm_bindgen)]
#[derive(Copy, Clone, Eq, PartialEq, Debug)]
pub enum CitationFormat {
    /// BibTeX format
    Bibtex,
    /// PubMed `.nbib` file format
    Pubmed,
}

/// Convert citation file contents to hayagriva YAML.
#[cfg_attr(feature = "wasm", wasm_bindgen)]
pub fn tohaya(inputs: Vec<String>, format: Option<CitationFormat>) -> Result<String, TohayaError> {
    let library = inputs
        .into_iter()
        .map(|s| to_library(s, format).map(|l| l.into_iter()))
        .flatten_ok()
        .try_collect()?;
    let yaml = hayagriva::io::to_yaml_str(&library)?;
    Ok(yaml)
}

fn to_library<S: AsRef<str>>(
    input: S,
    format: Option<CitationFormat>,
) -> Result<hayagriva::Library, ParseError> {
    let text = input.as_ref();
    let format = if let Some(f) = format {
        f
    } else {
        guess_format(text).ok_or_else(|| ParseError("Unknown format".to_string()))?
    };
    match format {
        CitationFormat::Pubmed => parse_pubmed(input),
        CitationFormat::Bibtex => from_bibtex(input),
    }
}

fn from_bibtex<S: AsRef<str>>(input: S) -> Result<hayagriva::Library, ParseError> {
    hayagriva::io::from_biblatex_str(input.as_ref()).map_err(|e| {
        let msg = e.into_iter().map(|e| e.to_string()).join(", ");
        ParseError(msg)
    })
}