provenant/parsers/
citation.rs

1use std::fs;
2use std::path::Path;
3
4use crate::models::{DatasourceId, PackageData, PackageType, Party};
5use crate::parser_warn as warn;
6
7use super::PackageParser;
8use super::license_normalization::normalize_spdx_declared_license;
9
10pub struct CitationCffParser;
11
12impl PackageParser for CitationCffParser {
13    const PACKAGE_TYPE: PackageType = PackageType::Generic;
14
15    fn is_match(path: &Path) -> bool {
16        path.file_name().and_then(|name| name.to_str()) == Some("CITATION.cff")
17    }
18
19    fn extract_packages(path: &Path) -> Vec<PackageData> {
20        let content = match fs::read_to_string(path) {
21            Ok(content) => content,
22            Err(error) => {
23                warn!("Failed to read CITATION.cff at {:?}: {}", path, error);
24                return vec![default_package_data()];
25            }
26        };
27
28        let yaml: yaml_serde::Value = match yaml_serde::from_str(&content) {
29            Ok(yaml) => yaml,
30            Err(error) => {
31                warn!("Failed to parse CITATION.cff at {:?}: {}", path, error);
32                return vec![default_package_data()];
33            }
34        };
35
36        vec![parse_citation_cff(&yaml)]
37    }
38}
39
40fn default_package_data() -> PackageData {
41    PackageData {
42        package_type: Some(CitationCffParser::PACKAGE_TYPE),
43        datasource_id: Some(DatasourceId::CitationCff),
44        ..Default::default()
45    }
46}
47
48fn parse_citation_cff(yaml: &yaml_serde::Value) -> PackageData {
49    if yaml
50        .get("cff-version")
51        .and_then(yaml_serde::Value::as_str)
52        .is_none()
53    {
54        return default_package_data();
55    }
56
57    let mut package = default_package_data();
58    package.name = yaml
59        .get("title")
60        .and_then(yaml_serde::Value::as_str)
61        .map(str::to_string);
62    package.version = yaml
63        .get("version")
64        .and_then(yaml_serde::Value::as_str)
65        .map(str::to_string);
66    package.description = yaml
67        .get("abstract")
68        .and_then(yaml_serde::Value::as_str)
69        .or_else(|| yaml.get("message").and_then(yaml_serde::Value::as_str))
70        .map(str::to_string);
71    package.homepage_url = yaml
72        .get("url")
73        .and_then(yaml_serde::Value::as_str)
74        .map(str::to_string);
75    package.vcs_url = yaml
76        .get("repository-code")
77        .and_then(yaml_serde::Value::as_str)
78        .map(str::to_string);
79    package.parties = extract_author_parties(yaml.get("authors"));
80
81    if let Some(license) = yaml.get("license").and_then(yaml_serde::Value::as_str) {
82        let license = license.to_string();
83        package.extracted_license_statement = Some(license.clone());
84        let (declared, declared_spdx, detections) = normalize_spdx_declared_license(Some(&license));
85        package.declared_license_expression = declared;
86        package.declared_license_expression_spdx = declared_spdx;
87        package.license_detections = detections;
88    }
89
90    package
91}
92
93fn extract_author_parties(value: Option<&yaml_serde::Value>) -> Vec<Party> {
94    value
95        .and_then(yaml_serde::Value::as_sequence)
96        .into_iter()
97        .flatten()
98        .filter_map(|entry| {
99            let name = entry
100                .get("name")
101                .and_then(yaml_serde::Value::as_str)
102                .map(str::to_string)
103                .or_else(|| {
104                    let given = entry.get("given-names").and_then(yaml_serde::Value::as_str);
105                    let family = entry
106                        .get("family-names")
107                        .and_then(yaml_serde::Value::as_str);
108                    match (given, family) {
109                        (Some(given), Some(family)) => Some(format!("{given} {family}")),
110                        (Some(given), None) => Some(given.to_string()),
111                        (None, Some(family)) => Some(family.to_string()),
112                        (None, None) => None,
113                    }
114                });
115            let email = entry
116                .get("email")
117                .and_then(yaml_serde::Value::as_str)
118                .map(str::to_string);
119            let url = entry
120                .get("orcid")
121                .and_then(yaml_serde::Value::as_str)
122                .map(str::to_string);
123
124            if name.is_none() && email.is_none() && url.is_none() {
125                return None;
126            }
127
128            Some(Party {
129                r#type: Some("person".to_string()),
130                role: Some("author".to_string()),
131                name,
132                email,
133                url,
134                organization: None,
135                organization_url: None,
136                timezone: None,
137            })
138        })
139        .collect()
140}
141
142crate::register_parser!(
143    "citation cff metadata",
144    &["**/CITATION.cff"],
145    "generic",
146    "Text",
147    Some("https://citation-file-format.github.io/"),
148);
provenant/parsers/citation.rs

provenant/parsers/
citation.rs
