feat: go package identifier (#9)

getgrit · Mar 12, 2024 · 3865d22 · 3865d22
1 parent 45cc15a
commit 3865d22
Show file tree

Hide file tree

Showing 4 changed files with 55 additions and 3 deletions.
diff --git a/crates/core/src/test.rs b/crates/core/src/test.rs
@@ -12583,3 +12583,47 @@ fn go_field_identifier() {
     })
     .unwrap();
 }
+
+#[test]
+fn go_package_identifier() {
+    run_test_match(TestArg {
+        pattern: r#"
+            |language go
+            |
+            |`mypackage`
+            |"#
+        .trim_margin()
+        .unwrap(),
+        source: r#"
+            |func ZoneLockdown(api *mypackage.API) (resp interface{}, err error) {
+            |   resp, err = api.Call()
+            |   return
+            |}
+            |"#
+        .trim_margin()
+        .unwrap(),
+    })
+    .unwrap();
+}
+
+#[test]
+fn go_package_type() {
+    run_test_match(TestArg {
+        pattern: r#"
+            |language go
+            |
+            |`API`
+            |"#
+        .trim_margin()
+        .unwrap(),
+        source: r#"
+            |func ZoneLockdown(api *mypackage.API) (resp interface{}, err error) {
+            |   resp, err = api.Call()
+            |   return
+            |}
+            |"#
+        .trim_margin()
+        .unwrap(),
+    })
+    .unwrap();
+}
diff --git a/crates/language/src/csharp.rs b/crates/language/src/csharp.rs
@@ -2,7 +2,8 @@ use std::sync::OnceLock;
 
 use crate::language::{fields_for_nodes, Field, Language, SortId, TSLanguage};
 
-static NODE_TYPES_STRING: &str = include_str!("../../../resources/node-types/csharp-node-types.json");
+static NODE_TYPES_STRING: &str =
+    include_str!("../../../resources/node-types/csharp-node-types.json");
 static NODE_TYPES: OnceLock<Vec<Vec<Field>>> = OnceLock::new();
 static LANGUAGE: OnceLock<TSLanguage> = OnceLock::new();
 

diff --git a/crates/language/src/go.rs b/crates/language/src/go.rs
@@ -49,7 +49,13 @@ impl Language for Go {
         "Go"
     }
     fn snippet_context_strings(&self) -> &[(&'static str, &'static str)] {
-        &[("", ""), ("", ";"), ("GO_OBJECT.", ";")]
+        &[
+            ("", ""),
+            ("", ";"),
+            ("GO_OBJECT.", ";"),
+            ("func GRIT_FUNC(GRIT_ARG *", ".GRIT_TYPE) {}"),
+            ("func GRIT_FUNC(GRIT_ARG *GRIT_PACKAGE.", ") {}"),
+        ]
     }
 
     fn node_types(&self) -> &[Vec<Field>] {

diff --git a/crates/language/src/python.rs b/crates/language/src/python.rs
@@ -2,7 +2,8 @@ use std::sync::OnceLock;
 
 use crate::language::{fields_for_nodes, Field, Language, SortId, TSLanguage};
 
-static NODE_TYPES_STRING: &str = include_str!("../../../resources/node-types/python-node-types.json");
+static NODE_TYPES_STRING: &str =
+    include_str!("../../../resources/node-types/python-node-types.json");
 static NODE_TYPES: OnceLock<Vec<Vec<Field>>> = OnceLock::new();
 static LANGUAGE: OnceLock<TSLanguage> = OnceLock::new();