diff options
Diffstat (limited to 'support/texlab/src/component_db.rs')
-rw-r--r-- | support/texlab/src/component_db.rs | 137 |
1 files changed, 137 insertions, 0 deletions
diff --git a/support/texlab/src/component_db.rs b/support/texlab/src/component_db.rs new file mode 100644 index 0000000000..a0c728a7ae --- /dev/null +++ b/support/texlab/src/component_db.rs @@ -0,0 +1,137 @@ +use itertools::Itertools; +use lsp_types::{MarkupContent, MarkupKind}; +use once_cell::sync::Lazy; +use serde::{Deserialize, Serialize}; +use smol_str::SmolStr; + +use crate::WorkspaceSubset; + +#[derive(Debug, PartialEq, Eq, Clone, Serialize, Deserialize)] +#[serde(rename_all = "camelCase")] +pub struct ComponentDatabase { + pub components: Vec<Component>, + pub metadata: Vec<ComponentMetadata>, +} + +impl ComponentDatabase { + pub fn find(&self, name: &str) -> Option<&Component> { + self.components.iter().find(|component| { + component + .file_names + .iter() + .any(|file_name| file_name == name) + }) + } + + pub fn find_no_ext(&self, name: &str) -> Option<&Component> { + self.components.iter().find(|component| { + component + .file_names + .iter() + .any(|file_name| &file_name[0..file_name.len() - 4] == name) + }) + } + + pub fn linked_components<'a>(&self, subset: &WorkspaceSubset) -> Vec<&Component> { + let mut start_components = vec![self.kernel()]; + for document in &subset.documents { + if let Some(data) = document.data.as_latex() { + data.extras + .explicit_links + .iter() + .filter_map(|link| link.as_component_name()) + .filter_map(|name| self.find(&name)) + .for_each(|component| start_components.push(component)); + } + } + + let mut all_components = Vec::new(); + for component in start_components { + all_components.push(component); + component + .references + .iter() + .flat_map(|file| self.find(&file)) + .for_each(|component| all_components.push(component)) + } + + all_components + .into_iter() + .unique_by(|component| &component.file_names) + .collect() + } + + pub fn contains(&self, short_name: &str) -> bool { + let sty = format!("{}.sty", short_name); + let cls = format!("{}.cls", short_name); + self.find(&sty).is_some() || self.find(&cls).is_some() + } + + pub fn kernel(&self) -> &Component { + self.components + .iter() + .find(|component| component.file_names.is_empty()) + .unwrap() + } + + pub fn exists(&self, file_name: &str) -> bool { + self.components + .iter() + .any(|component| component.file_names.iter().any(|f| f == file_name)) + } + + pub fn documentation(&self, name: &str) -> Option<MarkupContent> { + let metadata = self + .metadata + .iter() + .find(|metadata| metadata.name == name)?; + + let desc = metadata.description.to_owned()?; + Some(MarkupContent { + kind: MarkupKind::PlainText, + value: desc, + }) + } +} + +#[derive(Debug, PartialEq, Eq, Clone, Serialize, Deserialize)] +#[serde(rename_all = "camelCase")] +pub struct Component { + pub file_names: Vec<SmolStr>, + pub references: Vec<SmolStr>, + pub commands: Vec<ComponentCommand>, + pub environments: Vec<SmolStr>, +} + +#[derive(Debug, PartialEq, Eq, Clone, Serialize, Deserialize)] +#[serde(rename_all = "camelCase")] +pub struct ComponentCommand { + pub name: SmolStr, + pub image: Option<String>, + pub glyph: Option<SmolStr>, + pub parameters: Vec<ComponentParameter>, +} + +#[derive(Debug, PartialEq, Eq, Clone, Serialize, Deserialize)] +#[serde(rename_all = "camelCase")] +pub struct ComponentParameter(pub Vec<ComponentArgument>); + +#[derive(Debug, PartialEq, Eq, Clone, Serialize, Deserialize)] +#[serde(rename_all = "camelCase")] +pub struct ComponentArgument { + pub name: SmolStr, + pub image: Option<String>, +} + +#[derive(Debug, PartialEq, Eq, Clone, Serialize, Deserialize)] +#[serde(rename_all = "camelCase")] +pub struct ComponentMetadata { + pub name: String, + pub caption: Option<String>, + pub description: Option<String>, +} + +const JSON: &str = include_str!("../data/components.json"); + +pub static COMPONENT_DATABASE: Lazy<ComponentDatabase> = + Lazy::new(|| serde_json::from_str(JSON).unwrap()); |