code-moniker-core 0.2.0

Core symbol-graph types and per-language extractors for code-moniker (pure Rust, no pgrx). Consumed by the CLI and the PostgreSQL extension.
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

use tree_sitter::Node;

use crate::core::moniker::{Moniker, MonikerBuilder};

use super::kinds;

pub(super) fn compute_module_moniker(
	anchor: &Moniker,
	uri: &str,
	package_pieces: &[&str],
) -> Moniker {
	let class_name = file_stem(uri);
	let mut b = MonikerBuilder::from_view(anchor.as_view());
	b.segment(crate::lang::kinds::LANG, b"java");
	for piece in package_pieces.iter().filter(|s| !s.is_empty()) {
		b.segment(kinds::PACKAGE, piece.as_bytes());
	}
	b.segment(kinds::MODULE, class_name.as_bytes());
	b.build()
}

pub(super) fn file_stem(uri: &str) -> &str {
	let after_slash = uri.rsplit('/').next().unwrap_or(uri);
	after_slash.strip_suffix(".java").unwrap_or(after_slash)
}

pub(super) fn read_package_name<'src>(root: Node<'_>, source: &'src [u8]) -> &'src str {
	let mut cursor = root.walk();
	for child in root.children(&mut cursor) {
		if child.kind() != "package_declaration" {
			continue;
		}
		let mut nc = child.walk();
		for nm in child.named_children(&mut nc) {
			if let Ok(s) = nm.utf8_text(source) {
				return s;
			}
		}
	}
	""
}

#[cfg(test)]
mod tests {
	use super::*;

	#[test]
	fn file_stem_strips_directory_and_extension() {
		assert_eq!(file_stem("src/main/java/com/acme/Foo.java"), "Foo");
		assert_eq!(file_stem("Foo.java"), "Foo");
		assert_eq!(file_stem("Foo"), "Foo");
	}
}