feat(go): Scope goto statements

alexpovel · Aug 16, 2024 · 4133575 · 4133575
1 parent 308e28f
commit 4133575
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -1448,6 +1448,7 @@ Language scopes:
           - go:          `go` blocks
           - switch:      `switch` blocks
           - labeled:     Labeled statements
+          - goto:        `goto` statements
           - struct-tags: Struct tags
 
       --go-query <TREE-SITTER-QUERY>

diff --git a/src/scoping/langs/go.rs b/src/scoping/langs/go.rs
@@ -53,6 +53,8 @@ pub enum PreparedGoQuery {
     Switch,
     /// Labeled statements.
     Labeled,
+    /// `goto` statements.
+    Goto,
     /// Struct tags.
     StructTags,
 }
@@ -105,6 +107,7 @@ impl From<PreparedGoQuery> for TSQuery {
                 PreparedGoQuery::Go => "(go_statement) @go",
                 PreparedGoQuery::Switch => "(expression_switch_statement) @switch",
                 PreparedGoQuery::Labeled => "(labeled_statement) @labeled",
+                PreparedGoQuery::Goto => "(goto_statement) @goto",
                 PreparedGoQuery::StructTags => "(field_declaration tag: (raw_string_literal) @tag)",
             },
         )

diff --git a/tests/langs/mod.rs b/tests/langs/mod.rs
@@ -542,6 +542,11 @@ impl InScopeLinePart {
     include_str!("go/base.go"),
     Go::new(CodeQuery::Prepared(PreparedGoQuery::Labeled)),
 )]
+#[case(
+    "base.go_goto",
+    include_str!("go/base.go"),
+    Go::new(CodeQuery::Prepared(PreparedGoQuery::Goto)),
+)]
 #[case(
     "base.go_struct-tags",
     include_str!("go/base.go"),

diff --git a/tests/langs/snapshots/r#mod__langs__base.go_goto.snap b/tests/langs/snapshots/r#mod__langs__base.go_goto.snap
@@ -0,0 +1,7 @@
+---
+source: tests/langs/mod.rs
+expression: inscope_parts
+---
+- n: 292
+  l: "\t\tgoto Loop\n"
+  m: "    ^^^^^^^^^  "