chore: minor SDK work

crates/core/src/lib.rs

@@ -16,7 +16,6 @@ mod foreign_function_definition;
 pub mod fs;
 mod inline_snippets;
 
-#[cfg(any(feature = "napi", feature = "wasm_core"))]
 pub mod sdk;
 
 mod limits;

crates/core/src/pattern_compiler/auto_wrap.rs

@@ -470,7 +470,7 @@ fn wrap_pattern_in_file(
     Ok(pattern)
 }
 
-pub(crate) fn wrap_pattern_in_before_and_after_each_file(
+pub fn wrap_pattern_in_before_and_after_each_file(
     pattern: Pattern<MarzanoQueryContext>,
     context: &mut dyn SnippetCompilationContext,
 ) -> Result<Pattern<MarzanoQueryContext>> {

crates/core/src/pattern_compiler/builder.rs

@@ -1,8 +1,9 @@
 use super::{
     auto_wrap::auto_wrap_pattern,
     compiler::{
-        filter_libs, get_definition_info, get_definitions, CompilationContext, DefinitionInfo,
-        DefinitionInfoKinds, NodeCompilationContext, VariableLocations,
+        build_standard_global_vars, filter_libs, get_definition_info, get_definitions,
+        CompilationContext, DefinitionInfo, DefinitionInfoKinds, NodeCompilationContext,
+        VariableLocations,
     },
     pattern_compiler::PatternCompiler,
     CompilationResult, NodeCompiler,
@@ -305,12 +306,3 @@ impl CompiledPatternBuilder {
         Ok(result)
     }
 }
-
-pub fn build_standard_global_vars() -> BTreeMap<String, usize> {
-    BTreeMap::from([
-        ("$new_files".to_owned(), NEW_FILES_INDEX),
-        ("$filename".to_owned(), FILENAME_INDEX),
-        ("$program".to_owned(), PROGRAM_INDEX),
-        ("$absolute_filename".to_owned(), ABSOLUTE_PATH_INDEX),
-    ])
-}

crates/core/src/pattern_compiler/compiler.rs

@@ -16,7 +16,10 @@ use crate::{
 };
 use anyhow::{anyhow, bail, Result};
 use grit_pattern_matcher::{
-    constants::{DEFAULT_FILE_NAME, GLOBAL_VARS_SCOPE_INDEX, MATCH_VAR},
+    constants::{
+        ABSOLUTE_PATH_INDEX, DEFAULT_FILE_NAME, FILENAME_INDEX, GLOBAL_VARS_SCOPE_INDEX, MATCH_VAR,
+        NEW_FILES_INDEX, PROGRAM_INDEX,
+    },
     pattern::{
         DynamicSnippetPart, GritFunctionDefinition, Pattern, PatternDefinition,
         PredicateDefinition, Variable, VariableContent, VariableSource,
@@ -395,7 +398,7 @@ pub(crate) fn get_definitions(
         global_vars
             .iter()
             .sorted_by(|x, y| Ord::cmp(x.1, y.1))
-            .map(|x| VariableSource::new(x.0.clone(), context.file.to_owned()))
+            .map(|x| VariableSource::new(x.0.clone(), DEFAULT_FILE_NAME.to_owned()))
             .collect(),
     );
 
@@ -702,7 +705,7 @@ pub fn src_to_problem(src: String, default_lang: TargetLanguage) -> Result<Probl
     Ok(problem)
 }
 
-#[derive(Debug, Default)]
+#[derive(Debug)]
 pub struct VariableLocations {
     /// List of scopes, each scope with an array of variables
     pub(crate) locations: Vec<Vec<VariableSource>>,
@@ -746,6 +749,15 @@ impl VariableLocations {
         }
         variables
     }
+
+    pub(crate) fn globals() -> Self {
+        let global_vars = build_standard_global_vars();
+        let locations = global_vars
+            .iter()
+            .map(|(name, index)| VariableSource::new_global(name.to_owned()))
+            .collect();
+        Self::new(vec![locations])
+    }
 }
 
 #[cfg(test)]
@@ -818,3 +830,12 @@ mod tests {
         );
     }
 }
+
+pub fn build_standard_global_vars() -> BTreeMap<String, usize> {
+    BTreeMap::from([
+        ("$new_files".to_owned(), NEW_FILES_INDEX),
+        ("$filename".to_owned(), FILENAME_INDEX),
+        ("$program".to_owned(), PROGRAM_INDEX),
+        ("$absolute_filename".to_owned(), ABSOLUTE_PATH_INDEX),
+    ])
+}

crates/core/src/pattern_compiler/mod.rs

@@ -77,7 +77,8 @@ pub(crate) mod variable_compiler;
 pub(crate) mod where_compiler;
 pub(crate) mod within_compiler;
 
-pub use builder::build_standard_global_vars;
+pub use auto_wrap::wrap_pattern_in_before_and_after_each_file;
 pub use builder::CompiledPatternBuilder;
+pub use compiler::build_standard_global_vars;
 pub use compiler::{src_to_problem_libs, CompilationResult};
 pub(crate) use node_compiler::NodeCompiler;

crates/core/src/problem.rs

@@ -7,7 +7,7 @@ use crate::{
     marzano_code_snippet::MarzanoCodeSnippet,
     marzano_context::MarzanoContext,
     marzano_resolved_pattern::{MarzanoFile, MarzanoResolvedPattern},
-    pattern_compiler::compiler::VariableLocations,
+    pattern_compiler::{build_standard_global_vars, compiler::VariableLocations},
 };
 use anyhow::{bail, Result};
 use grit_pattern_matcher::{
@@ -83,6 +83,26 @@ impl Problem {
         }
         defs
     }
+
+    pub fn new(
+        pattern: Pattern<MarzanoQueryContext>,
+        language: TargetLanguage,
+        built_ins: BuiltIns,
+    ) -> Self {
+        Self::new_from_pattern(
+            pattern,
+            language,
+            built_ins,
+            false, // is_multifile
+            false, // has_limit
+            None,  // name
+            VariableLocations::globals(),
+            vec![], // pattern_definitions
+            vec![], // predicate_definitions
+            vec![], // function_definitions
+            vec![], // foreign_function_definitions
+        )
+    }
 }
 
 enum FilePattern {

crates/core/src/sdk/language_sdk.rs

@@ -5,9 +5,7 @@ use marzano_language::target_language::TargetLanguage;
 
 use crate::{
     built_in_functions::BuiltIns,
-    pattern_compiler::{
-        auto_wrap::auto_wrap_pattern, build_standard_global_vars, compiler::VariableLocations,
-    },
+    pattern_compiler::{auto_wrap::auto_wrap_pattern, compiler::VariableLocations},
     problem::{MarzanoQueryContext, Problem},
 };
 
@@ -48,7 +46,6 @@ impl LanguageSdk {
         pattern: Pattern<MarzanoQueryContext>,
     ) -> Result<Problem> {
         let _logs: AnalysisLogs = vec![].into();
-        let global_vars = build_standard_global_vars();
         let mut pattern_definitions = vec![];
 
         let is_multifile = false;
@@ -69,10 +66,7 @@ impl LanguageSdk {
             is_multifile,
             false,
             None,
-            VariableLocations::new(vec![global_vars
-                .into_keys()
-                .map(VariableSource::new_global)
-                .collect()]),
+            VariableLocations::globals(),
             pattern_definitions,
             vec![],
             vec![],

crates/core/src/sdk/mod.rs

@@ -6,6 +6,6 @@ mod test_js;
 #[cfg(feature = "napi")]
 mod binding;
 
-pub(crate) use compiler::StatelessCompilerContext;
+pub use compiler::StatelessCompilerContext;
 pub use language_sdk::LanguageSdk;
 pub use pattern_sdk::UncompiledPatternBuilder;

crates/language/src/markdown_inline.rs

@@ -28,7 +28,7 @@ pub struct MarkdownInline {
 }
 
 impl MarkdownInline {
-    pub(crate) fn new(lang: Option<TSLanguage>) -> Self {
+    pub fn new(lang: Option<TSLanguage>) -> Self {
         let language = LANGUAGE.get_or_init(|| lang.unwrap_or_else(language));
         let node_types = NODE_TYPES.get_or_init(|| fields_for_nodes(language, NODE_TYPES_STRING));
         let metavariable_sort = language.id_for_node_kind("grit_metavariable", true);