Translate END node

Author: andrykonchin

spec/tags/truffle/parsing/parsing_tags.txt

@@ -1,5 +1,4 @@
 fails:Parsing a BEGIN block (BEGIN { ... }) case is parsed correctly
-fails:Parsing a END block (END { ... }) case is parsed correctly
 fails:Parsing a For operator (for ... in ... operator) case is parsed correctly
 
 # Not supported yet by Prism

spec/truffle/parsing/fixtures/END.yaml

@@ -1,7 +1,9 @@
 subject: "END block"
 description: "END { ... }"
 notes: >
-  Is trasnalted into a `Truffle::KernelOperations.at_exit(false) { ... }` call.
+  Is translated into a `Truffle::KernelOperations.at_exit(false) { ... }` call.
+yarp_specific: true # don't optimize reading KernelOperations constant
+                    # with TruffleKernelOperationsModuleLiteralNode
 focused_on_node: "org.truffleruby.language.control.OnceNode"
 ruby: |
   END {
@@ -11,7 +13,7 @@ ruby: |
 ast: |
   OnceNode
       attributes:
-          flags = 0
+          flags = 1
           holder = org.truffleruby.language.control.OnceNode$Holder@...
       children:
           child =
@@ -56,7 +58,7 @@ ast: |
                                                           callTargets = ProcCallTargets(callTargetForProc = block in <top (required)>, callTargetForLambda = null, altCallTargetCompiler = ...$$Lambda$.../0x...@...)
                                                           flags = 0
                                                           frameOnStackMarkerSlot = 2
-                                                          sharedMethodInfo = SharedMethodInfo(staticLexicalScope =  :: Object, arity = Arity{preRequired = 0, optional = 0, hasRest = false, postRequired = 0, keywordArguments = [], requiredKeywordArgumentsCount = 0, hasKeywordsRest = false}, originName = block in <top (required)>, blockDepth = 1, parseName = block in <top (required)>, notes = <top (required)>, argumentDescriptors = [])
+                                                          sharedMethodInfo = SharedMethodInfo(staticLexicalScope =  :: Object, arity = Arity{preRequired = 0, optional = 0, hasRest = false, isImplicitRest = false, postRequired = 0, keywordArguments = [], requiredKeywordArgumentsCount = 0, hasKeywordsRest = false}, originName = block in <top (required)>, blockDepth = 1, parseName = block in <top (required)>, notes = <top (required)>, argumentDescriptors = [])
                                                           type = PROC
                                                       call targets:
                                                           RubyProcRootNode
@@ -70,7 +72,7 @@ ast: |
                                                                   redoProfile = false
                                                                   retryProfile = false
                                                                   returnID = org.truffleruby.language.control.ReturnID@...
-                                                                  sharedMethodInfo = SharedMethodInfo(staticLexicalScope =  :: Object, arity = Arity{preRequired = 0, optional = 0, hasRest = false, postRequired = 0, keywordArguments = [], requiredKeywordArgumentsCount = 0, hasKeywordsRest = false}, originName = block in <top (required)>, blockDepth = 1, parseName = block in <top (required)>, notes = <top (required)>, argumentDescriptors = [])
+                                                                  sharedMethodInfo = SharedMethodInfo(staticLexicalScope =  :: Object, arity = Arity{preRequired = 0, optional = 0, hasRest = false, isImplicitRest = false, postRequired = 0, keywordArguments = [], requiredKeywordArgumentsCount = 0, hasKeywordsRest = false}, originName = block in <top (required)>, blockDepth = 1, parseName = block in <top (required)>, notes = <top (required)>, argumentDescriptors = [])
                                                                   split = HEURISTIC
                                                               children:
                                                                   body =
@@ -103,6 +105,16 @@ ast: |
                                                                                           value = 2
                                                                               ]
                                               receiver =
-                                                  TruffleKernelOperationsModuleLiteralNode
+                                                  ReadConstantWithLexicalScopeNode
                                                       attributes:
-                                                          flags = 0
+                                                          flags = 0
+                                                          lexicalScope =  :: Object
+                                                          name = "KernelOperations"
+                                                      children:
+                                                          getConstantNode =
+                                                              GetConstantNodeGen
+                                                          lookupConstantNode =
+                                                              LookupConstantWithLexicalScopeNodeGen
+                                                                  attributes:
+                                                                      lexicalScope =  :: Object
+                                                                      name = "KernelOperations"

src/main/java/org/truffleruby/parser/YARPTranslator.java

@@ -2425,7 +2425,17 @@ public RubyNode visitPinnedVariableNode(Nodes.PinnedVariableNode node) {
 
     @Override
     public RubyNode visitPostExecutionNode(Nodes.PostExecutionNode node) {
-        return defaultVisit(node);
+        // END blocks run after any other code - not just code in the same file
+        // Turn into a call to Truffle::KernelOperations.at_exit
+
+        // Create Prism CallNode to avoid duplication block literal related logic
+        final var receiver = new Nodes.ConstantReadNode("KernelOperations", 0, 0);
+        final var arguments = new Nodes.ArgumentsNode(NO_FLAGS, new Nodes.Node[]{ new Nodes.FalseNode(0, 0) }, 0, 0);
+        final var block = new Nodes.BlockNode(StringUtils.EMPTY_STRING_ARRAY, 0, null, node.statements, 0, 0);
+
+        final var callNode = new Nodes.CallNode(NO_FLAGS, receiver, "at_exit", arguments, block, 0, 0).accept(this);
+        final RubyNode rubyNode = new OnceNode(callNode);
+        return assignPositionAndFlags(node, rubyNode);
     }
 
     @Override