Merge pull request #158 from github/analyze-tsconfig

Analyze package.json

Author: Hendrik van Antwerpen

languages/tree-sitter-stack-graphs-typescript/Cargo.toml

@@ -25,6 +25,8 @@ harness = false
 anyhow = "1.0"
 clap = "3"
 glob = "0.3"
+serde = { version = "1.0", features = ["derive"] }
+serde_json = "1.0"
 stack-graphs = { version = "~0.10", path = "../../stack-graphs" }
 tree-sitter-stack-graphs = { version = "~0.5", path = "../../tree-sitter-stack-graphs", features=["cli"] }
 tree-sitter-typescript = { git = "https://github.com/tree-sitter/tree-sitter-typescript", rev="082da44a5263599186dadafd2c974c19f3a73d28" }

languages/tree-sitter-stack-graphs-typescript/rust/lib.rs

@@ -5,12 +5,16 @@
 // Please see the LICENSE-APACHE or LICENSE-MIT files in this distribution for license details.
 // ------------------------------------------------------------------------------------------------
 
-use crate::tsconfig::TsConfigAnalyzer;
 use tree_sitter_stack_graphs::loader::FileAnalyzers;
 use tree_sitter_stack_graphs::loader::LanguageConfiguration;
 use tree_sitter_stack_graphs::CancellationFlag;
 
+use crate::npm_package::NpmPackageAnalyzer;
+use crate::tsconfig::TsConfigAnalyzer;
+
+pub mod npm_package;
 pub mod tsconfig;
+pub mod util;
 
 /// The stack graphs tsg source for this language
 pub const STACK_GRAPHS_TSG_SOURCE: &str = include_str!("../src/stack-graphs.tsg");
@@ -20,6 +24,11 @@ pub const STACK_GRAPHS_BUILTINS_CONFIG: &str = include_str!("../src/builtins.cfg
 /// The stack graphs builtins source for this language
 pub const STACK_GRAPHS_BUILTINS_SOURCE: &str = include_str!("../src/builtins.ts");
 
+/// The name of the file path global variable
+pub const FILE_PATH_VAR: &str = "FILE_PATH";
+/// The name of the project name global variable
+pub const PROJECT_NAME_VAR: &str = "PROJECT_NAME";
+
 pub fn language_configuration(cancellation_flag: &dyn CancellationFlag) -> LanguageConfiguration {
     LanguageConfiguration::from_tsg_str(
         tree_sitter_typescript::language_typescript(),
@@ -29,7 +38,9 @@ pub fn language_configuration(cancellation_flag: &dyn CancellationFlag) -> Langu
         STACK_GRAPHS_TSG_SOURCE,
         Some(STACK_GRAPHS_BUILTINS_SOURCE),
         Some(STACK_GRAPHS_BUILTINS_CONFIG),
-        FileAnalyzers::new().add("tsconfig.json".to_string(), TsConfigAnalyzer {}),
+        FileAnalyzers::new()
+            .add("tsconfig.json".to_string(), TsConfigAnalyzer {})
+            .add("package.json".to_string(), NpmPackageAnalyzer {}),
         cancellation_flag,
     )
     .unwrap()

languages/tree-sitter-stack-graphs-typescript/rust/npm_package.rs

@@ -0,0 +1,76 @@
+// -*- coding: utf-8 -*-
+// ------------------------------------------------------------------------------------------------
+// Copyright © 2022, stack-graphs authors.
+// Licensed under either of Apache License, Version 2.0, or MIT license, at your option.
+// Please see the LICENSE-APACHE or LICENSE-MIT files in this distribution for license details.
+// ------------------------------------------------------------------------------------------------
+
+use serde::Deserialize;
+use std::collections::HashMap;
+use std::path::Path;
+
+use stack_graphs::arena::Handle;
+use stack_graphs::graph::File;
+use stack_graphs::graph::StackGraph;
+use tree_sitter_stack_graphs::FileAnalyzer;
+use tree_sitter_stack_graphs::LoadError;
+
+use crate::util::*;
+
+pub struct NpmPackageAnalyzer {}
+
+impl FileAnalyzer for NpmPackageAnalyzer {
+    fn build_stack_graph_into<'a>(
+        &self,
+        graph: &mut StackGraph,
+        file: Handle<File>,
+        _path: &Path,
+        source: &str,
+        _all_paths: &mut dyn Iterator<Item = &'a Path>,
+        globals: &HashMap<String, String>,
+        _cancellation_flag: &dyn tree_sitter_stack_graphs::CancellationFlag,
+    ) -> Result<(), tree_sitter_stack_graphs::LoadError> {
+        // read globals
+        let proj_name = globals
+            .get(crate::PROJECT_NAME_VAR)
+            .map(String::as_str)
+            .unwrap_or("");
+
+        // parse source
+        let npm_pkg: NpmPackage =
+            serde_json::from_str(source).map_err(|_| LoadError::ParseError)?;
+
+        // root node
+        let root = StackGraph::root_node();
+
+        // project scope
+        let proj_scope_id = graph.new_node_id(file);
+        let proj_scope = graph.add_scope_node(proj_scope_id, false).unwrap();
+        add_debug_name(graph, proj_scope, "npm_package.proj_scope");
+
+        // project definition
+        let proj_def = add_ns_pop(graph, file, root, PROJ_NS, proj_name);
+        add_debug_name(graph, proj_def, "npm_package.proj_def");
+        add_edge(graph, proj_def, proj_scope, 0);
+
+        // project reference
+        let proj_ref = add_ns_push(graph, file, root, PROJ_NS, proj_name);
+        add_debug_name(graph, proj_ref, "npm_package.proj_ref");
+        add_edge(graph, proj_scope, proj_ref, 0);
+
+        // package definition
+        let pkg_def = add_module_pops(graph, file, NON_REL_M_NS, Path::new(&npm_pkg.name), root);
+        add_debug_name(graph, pkg_def, "npm_package.pkg_def");
+        let pkg_ref = add_push(graph, file, proj_scope, PKG_M_NS);
+        add_debug_name(graph, pkg_ref, "npm_package.pkg_ref");
+        add_edge(graph, pkg_def, pkg_ref, 0);
+
+        Ok(())
+    }
+}
+
+#[derive(Default, Debug, Clone, PartialEq, Deserialize)]
+#[serde(rename_all = "camelCase")]
+pub struct NpmPackage {
+    pub name: String,
+}