summaryrefslogtreecommitdiff
path: root/support/texlab/src/citeproc/mod.rs
diff options
context:
space:
mode:
authorNorbert Preining <norbert@preining.info>2019-12-03 03:01:24 +0000
committerNorbert Preining <norbert@preining.info>2019-12-03 03:01:24 +0000
commitb8d4bb76703bcb15578e2b23c5d256532180b894 (patch)
treebedd1df7a00521a2bd986b3c0289d6556a59e39b /support/texlab/src/citeproc/mod.rs
parent02e4625a78a5029e8b5dc2a4ec70193b232f497e (diff)
CTAN sync 201912030301
Diffstat (limited to 'support/texlab/src/citeproc/mod.rs')
-rw-r--r--support/texlab/src/citeproc/mod.rs69
1 files changed, 69 insertions, 0 deletions
diff --git a/support/texlab/src/citeproc/mod.rs b/support/texlab/src/citeproc/mod.rs
new file mode 100644
index 0000000000..9827c190f3
--- /dev/null
+++ b/support/texlab/src/citeproc/mod.rs
@@ -0,0 +1,69 @@
+mod name;
+mod ris;
+
+use self::ris::*;
+use crate::formatting::bibtex::{format_entry, format_string, BibtexFormattingParams};
+use crate::syntax::*;
+use bibutils::{InputFormat, OutputFormat};
+use citeproc::prelude::*;
+use citeproc_db::PredefinedLocales;
+use lsp_types::{MarkupContent, MarkupKind};
+use std::sync::Arc;
+
+static APA_STYLE: &str = include_str!("apa.csl");
+
+pub fn render_citation(tree: &BibtexSyntaxTree, key: &str) -> Option<MarkupContent> {
+ let reference: Reference = convert_to_ris(tree, key)?.into();
+
+ let html = generate_bibliography(reference)?;
+ let markdown = html2md::parse_html(&html).trim().to_owned();
+ if markdown == "" {
+ return None;
+ }
+
+ let content = MarkupContent {
+ kind: MarkupKind::Markdown,
+ value: markdown.replace("..", "."),
+ };
+ Some(content)
+}
+
+fn convert_to_ris(tree: &BibtexSyntaxTree, key: &str) -> Option<RisReference> {
+ let bib_params = BibtexFormattingParams::default();
+ let mut bib_code = String::new();
+
+ for string in tree.strings() {
+ bib_code.push_str(&format_string(string, &bib_params));
+ bib_code.push('\n');
+ }
+
+ let entry = tree.find_entry(key)?;
+ if let Some(crossref) = tree.resolve_crossref(entry) {
+ bib_code.push_str(&format_entry(crossref, &bib_params));
+ bib_code.push('\n');
+ }
+
+ bib_code.push_str(&format_entry(entry, &bib_params));
+ bib_code.push('\n');
+
+ let ris_code = bibutils::convert(bib_code, InputFormat::Biblatex, OutputFormat::Ris)?;
+ let ris_lib = RisLibrary::parse(ris_code.lines());
+ ris_lib
+ .references
+ .into_iter()
+ .find(|reference| reference.id.as_ref().map(AsRef::as_ref) == Some(key))
+}
+
+fn generate_bibliography(reference: Reference) -> Option<String> {
+ let locales = Arc::new(PredefinedLocales::bundled_en_us());
+ let mut processor = Processor::new(APA_STYLE, locales, false, SupportedFormat::Html).unwrap();
+ let cite = Cite::basic(&reference.id);
+ let cluster = Cluster2::Note {
+ id: 1,
+ note: IntraNote::Single(1),
+ cites: vec![cite],
+ };
+ processor.insert_reference(reference);
+ processor.init_clusters(vec![cluster]);
+ processor.get_bibliography().pop()
+}