Merge pull request #394 from julia-vscode/sp/fix-public-iterate

fix: correctly iterate over :publics
julia-vscode · Jul 14, 2024 · 44f622f · 44f622f
2 parents 5e9255b + 12980ca
commit 44f622f
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/src/iterate.jl b/src/iterate.jl
@@ -43,6 +43,8 @@ function _getindex(x::EXPR, i)
         x.args[i]
     elseif headof(x) === :export
         oddt_evena(x, i)
+    elseif headof(x) === :public
+        oddt_evena(x, i)
     elseif headof(x) === :filter
         _filter(x, i)
     elseif headof(x) === :flatten

diff --git a/test/iterate/test_iterators.jl b/test/iterate/test_iterators.jl
@@ -744,6 +744,21 @@ end
     @test x[6] === x.args[3]
 end
 
+if VERSION > v"1.11-"
+    @testitem ":public" begin
+        using CSTParser: @cst_str, headof, valof
+
+        x = cst"public a, b, c"
+        @test length(x) == 6
+        @test x[1] === x.trivia[1]
+        @test x[2] === x.args[1]
+        @test x[3] === x.trivia[2]
+        @test x[4] === x.args[2]
+        @test x[5] === x.trivia[3]
+        @test x[6] === x.args[3]
+    end
+end
+
 @testitem ":parameters" begin
     using CSTParser: @cst_str, headof, valof