summaryrefslogtreecommitdiff
path: root/support/texlab/crates/symbols/src/workspace/sort.rs
diff options
context:
space:
mode:
Diffstat (limited to 'support/texlab/crates/symbols/src/workspace/sort.rs')
-rw-r--r--support/texlab/crates/symbols/src/workspace/sort.rs203
1 files changed, 203 insertions, 0 deletions
diff --git a/support/texlab/crates/symbols/src/workspace/sort.rs b/support/texlab/crates/symbols/src/workspace/sort.rs
new file mode 100644
index 0000000000..bc30728da9
--- /dev/null
+++ b/support/texlab/crates/symbols/src/workspace/sort.rs
@@ -0,0 +1,203 @@
+use base_db::{graph, Document, Workspace};
+use itertools::Itertools;
+use url::Url;
+
+#[derive(Debug, PartialEq, Eq, Clone)]
+pub struct ProjectOrdering<'a> {
+ inner: Vec<&'a Document>,
+}
+
+impl<'a> ProjectOrdering<'a> {
+ pub fn get(&self, uri: &Url) -> usize {
+ self.inner
+ .iter()
+ .position(|doc| doc.uri == *uri)
+ .unwrap_or(std::usize::MAX)
+ }
+}
+
+impl<'a> From<&'a Workspace> for ProjectOrdering<'a> {
+ fn from(workspace: &'a Workspace) -> Self {
+ let inner = workspace
+ .iter()
+ .filter(|document| {
+ let data = document.data.as_tex();
+ data.map_or(false, |data| data.semantics.can_be_root)
+ })
+ .chain(workspace.iter())
+ .flat_map(|document| {
+ let graph = graph::Graph::new(workspace, document);
+ graph.preorder().rev().collect_vec()
+ })
+ .unique()
+ .collect_vec();
+
+ Self { inner }
+ }
+}
+
+#[cfg(test)]
+mod tests {
+ use base_db::Owner;
+ use distro::Language;
+ use rowan::TextSize;
+
+ use super::{ProjectOrdering, Url, Workspace};
+
+ #[test]
+ fn test_no_cycles() {
+ let mut workspace = Workspace::default();
+
+ let a = Url::parse("http://example.com/a.tex").unwrap();
+ let b = Url::parse("http://example.com/b.tex").unwrap();
+ let c = Url::parse("http://example.com/c.tex").unwrap();
+
+ workspace.open(
+ a.clone(),
+ String::new(),
+ Language::Tex,
+ Owner::Client,
+ TextSize::default(),
+ );
+
+ workspace.open(
+ b.clone(),
+ String::new(),
+ Language::Tex,
+ Owner::Client,
+ TextSize::default(),
+ );
+
+ workspace.open(
+ c.clone(),
+ r#"\documentclass{article}\include{b}\include{a}"#.to_string(),
+ Language::Tex,
+ Owner::Client,
+ TextSize::default(),
+ );
+
+ let ordering = ProjectOrdering::from(&workspace);
+ assert_eq!(ordering.get(&a), 0);
+ assert_eq!(ordering.get(&b), 1);
+ assert_eq!(ordering.get(&c), 2);
+ }
+
+ #[test]
+ fn test_two_layers() {
+ let mut workspace = Workspace::default();
+
+ let a = Url::parse("http://example.com/a.tex").unwrap();
+ let b = Url::parse("http://example.com/b.tex").unwrap();
+ let c = Url::parse("http://example.com/c.tex").unwrap();
+
+ workspace.open(
+ a.clone(),
+ String::new(),
+ Language::Tex,
+ Owner::Client,
+ TextSize::default(),
+ );
+
+ workspace.open(
+ b.clone(),
+ r#"\include{a}"#.to_string(),
+ Language::Tex,
+ Owner::Client,
+ TextSize::default(),
+ );
+
+ workspace.open(
+ c.clone(),
+ r#"\begin{documnent}\include{b}\end{document}"#.to_string(),
+ Language::Tex,
+ Owner::Client,
+ TextSize::default(),
+ );
+
+ let ordering = ProjectOrdering::from(&workspace);
+ assert_eq!(ordering.get(&a), 0);
+ assert_eq!(ordering.get(&b), 1);
+ assert_eq!(ordering.get(&c), 2);
+ }
+
+ #[test]
+ fn test_cycles() {
+ let mut workspace = Workspace::default();
+
+ let a = Url::parse("http://example.com/a.tex").unwrap();
+ let b = Url::parse("http://example.com/b.tex").unwrap();
+ let c = Url::parse("http://example.com/c.tex").unwrap();
+ workspace.open(
+ a.clone(),
+ r#"\begin{document}\include{b}\end{document}"#.to_string(),
+ Language::Tex,
+ Owner::Client,
+ TextSize::default(),
+ );
+
+ workspace.open(
+ b.clone(),
+ r#"\include{a}"#.to_string(),
+ Language::Tex,
+ Owner::Client,
+ TextSize::default(),
+ );
+
+ workspace.open(
+ c.clone(),
+ r#"\include{a}"#.to_string(),
+ Language::Tex,
+ Owner::Client,
+ TextSize::default(),
+ );
+
+ let ordering = ProjectOrdering::from(&workspace);
+ assert_ne!(ordering.get(&a), 0);
+ }
+
+ #[test]
+ fn test_multiple_roots() {
+ let mut workspace = Workspace::default();
+
+ let a = Url::parse("http://example.com/a.tex").unwrap();
+ let b = Url::parse("http://example.com/b.tex").unwrap();
+ let c = Url::parse("http://example.com/c.tex").unwrap();
+ let d = Url::parse("http://example.com/d.tex").unwrap();
+
+ workspace.open(
+ a.clone(),
+ r#"\begin{document}\include{b}\end{document}"#.to_string(),
+ Language::Tex,
+ Owner::Client,
+ TextSize::default(),
+ );
+
+ workspace.open(
+ b.clone(),
+ String::new(),
+ Language::Tex,
+ Owner::Client,
+ TextSize::default(),
+ );
+
+ workspace.open(
+ c.clone(),
+ String::new(),
+ Language::Tex,
+ Owner::Client,
+ TextSize::default(),
+ );
+
+ workspace.open(
+ d.clone(),
+ r#"\begin{document}\include{c}\end{document}"#.to_string(),
+ Language::Tex,
+ Owner::Client,
+ TextSize::default(),
+ );
+
+ let ordering = ProjectOrdering::from(&workspace);
+ assert!(ordering.get(&b) < ordering.get(&a));
+ assert!(ordering.get(&c) < ordering.get(&d));
+ }
+}