code_moniker_cli/
extract.rs

1use std::path::Path;
2
3use code_moniker_core::core::code_graph::CodeGraph;
4use code_moniker_core::core::moniker::{Moniker, MonikerBuilder};
5use code_moniker_core::lang::{Lang, ts};
6
7use crate::tsconfig::TsResolution;
8
9#[derive(Debug, Clone, Default)]
10pub struct Context {
11	pub ts: TsResolution,
12	pub project: Option<String>,
13}
14
15pub fn extract(lang: Lang, source: &str, path: &Path) -> CodeGraph {
16	extract_with(lang, source, path, &Context::default())
17}
18
19pub fn extract_with(lang: Lang, source: &str, path: &Path, ctx: &Context) -> CodeGraph {
20	let uri = path.to_str().unwrap_or("single-file");
21	let project = ctx.project.as_deref().map(str::as_bytes).unwrap_or(b".");
22	let anchor = anchor_moniker(project);
23	let deep = true;
24	match lang {
25		Lang::Ts => {
26			let presets = ts::Presets {
27				path_aliases: ctx.ts.aliases.clone(),
28				..ts::Presets::default()
29			};
30			ts::extract(uri, source, &anchor, deep, &presets)
31		}
32		Lang::Rs => code_moniker_core::lang::rs::extract(
33			uri,
34			source,
35			&anchor,
36			deep,
37			&code_moniker_core::lang::rs::Presets::default(),
38		),
39		Lang::Java => code_moniker_core::lang::java::extract(
40			uri,
41			source,
42			&anchor,
43			deep,
44			&code_moniker_core::lang::java::Presets::default(),
45		),
46		Lang::Python => code_moniker_core::lang::python::extract(
47			uri,
48			source,
49			&anchor,
50			deep,
51			&code_moniker_core::lang::python::Presets::default(),
52		),
53		Lang::Go => code_moniker_core::lang::go::extract(
54			uri,
55			source,
56			&anchor,
57			deep,
58			&code_moniker_core::lang::go::Presets::default(),
59		),
60		Lang::Cs => code_moniker_core::lang::cs::extract(
61			uri,
62			source,
63			&anchor,
64			deep,
65			&code_moniker_core::lang::cs::Presets::default(),
66		),
67		Lang::Sql => code_moniker_core::lang::sql::extract(
68			uri,
69			source,
70			&anchor,
71			deep,
72			&code_moniker_core::lang::sql::Presets::default(),
73		),
74	}
75}
76
77fn anchor_moniker(project: &[u8]) -> Moniker {
78	let mut b = MonikerBuilder::new();
79	b.project(project);
80	b.build()
81}
82
83pub fn file_uri(path: &Path) -> String {
84	let abs = path.canonicalize().unwrap_or_else(|_| path.to_path_buf());
85	format!("file://{}", abs.display())
86}
code_moniker_cli/extract.rs

code_moniker_cli/
extract.rs
