code_moniker_core/lang/java/
mod.rs

1use tree_sitter::{Language, Parser, Tree};
2
3use crate::core::code_graph::CodeGraph;
4use crate::core::moniker::Moniker;
5use crate::core::shape::Shape;
6
7use crate::lang::{ExtractionContext, KindSpec, LangExtractor, ParsedDocument};
8
9mod kinds;
10pub mod pom_manifest;
11mod sdk_pipeline;
12
13#[derive(Clone, Debug, Default)]
14pub struct Presets {
15	pub external_packages: Vec<String>,
16}
17
18pub fn parse(source: &str) -> Tree {
19	let mut parser = Parser::new();
20	let language: Language = tree_sitter_java::LANGUAGE.into();
21	parser.set_language(&language).unwrap_or_else(|err| {
22		panic!("failed to load tree-sitter Java grammar: {err}");
23	});
24	parser.parse(source, None).unwrap_or_else(|| {
25		panic!("tree-sitter parse returned None on a non-cancelled call");
26	})
27}
28
29pub fn extract(
30	uri: &str,
31	source: &str,
32	anchor: &Moniker,
33	deep: bool,
34	presets: &Presets,
35) -> CodeGraph {
36	<Lang as LangExtractor>::extract(uri, source, anchor, deep, presets)
37}
38
39pub fn extract_sdk(
40	uri: &str,
41	source: &str,
42	anchor: &Moniker,
43	deep: bool,
44	presets: &Presets,
45) -> CodeGraph {
46	<Lang as LangExtractor>::extract(uri, source, anchor, deep, presets)
47}
48
49pub struct Lang;
50
51const DEF_KINDS: &[&str] = &[
52	"class",
53	"interface",
54	"enum",
55	"record",
56	"annotation_type",
57	"method",
58	"constructor",
59	"field",
60	"enum_constant",
61];
62
63const DEF_KIND_SPECS: &[KindSpec] = &[
64	KindSpec::new("class", Shape::Type, 20, "class"),
65	KindSpec::new("interface", Shape::Type, 21, "interface"),
66	KindSpec::new("enum", Shape::Type, 22, "enum"),
67	KindSpec::new("record", Shape::Type, 23, "record"),
68	KindSpec::new("annotation_type", Shape::Type, 24, "annotation_type"),
69	KindSpec::new("enum_constant", Shape::Value, 30, "enum_constant"),
70	KindSpec::new("field", Shape::Value, 31, "field"),
71	KindSpec::new("constructor", Shape::Callable, 40, "constructor"),
72	KindSpec::new("method", Shape::Callable, 41, "method"),
73];
74
75impl crate::lang::LangExtractor for Lang {
76	type Presets = Presets;
77	const LANG_TAG: &'static str = "java";
78	const ALLOWED_KINDS: &'static [&'static str] = DEF_KINDS;
79	const KIND_SPECS: &'static [KindSpec] = DEF_KIND_SPECS;
80	const ALLOWED_VISIBILITIES: &'static [&'static str] =
81		&["public", "protected", "package", "private"];
82
83	fn parse(_uri: &str, source: &str) -> ParsedDocument {
84		ParsedDocument::new(parse(source))
85	}
86
87	fn file_root(uri: &str, anchor: &Moniker) -> Option<Moniker> {
88		sdk_pipeline::standard_path_module_moniker(anchor, uri)
89	}
90
91	fn extract_parsed(
92		context: ExtractionContext<'_, Self::Presets>,
93		document: &ParsedDocument,
94	) -> CodeGraph {
95		sdk_pipeline::extract(
96			context.uri,
97			context.source,
98			document,
99			context.anchor,
100			context.deep,
101			context.presets,
102		)
103	}
104}
code_moniker_core/lang/java/mod.rs

code_moniker_core/lang/java/
mod.rs
