Extract driving a node to parse or begin phase

rakudo · Jun 5, 2023 · bb4dd9c · bb4dd9c
1 parent 1600bd7
commit bb4dd9c
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 6 deletions.
diff --git a/src/Raku/Actions.nqp b/src/Raku/Actions.nqp
@@ -22,12 +22,7 @@ role Raku::CommonActions {
     # as we go. This factors out that process and attaches the AST to the match
     # object.
     method attach($/, $node, :$as-key-origin) {
-        if nqp::istype($node, self.r('ParseTime')) {
-            $node.ensure-parse-performed($*R, $*CU.context);
-        }
-        if nqp::istype($node, self.r('BeginTime')) {
-            $node.ensure-begin-performed($*R, $*CU.context);
-        }
+        $node.to-begin-time($*R, $*CU.context);
         if nqp::istype($node, self.r('ImplicitLookups')) {
             $node.resolve-implicit-lookups-with($*R);
         }

diff --git a/src/Raku/ast/base.rakumod b/src/Raku/ast/base.rakumod
@@ -81,6 +81,25 @@ class RakuAST::Node {
         self
     }
 
+    # Bring the node up to parse time. Returns the node itself.
+    method to-parse-time(RakuAST::Resolver $resolver, RakuAST::IMPL::QASTContext $context) {
+        if nqp::istype(self, RakuAST::ParseTime) {
+            self.ensure-parse-performed($resolver, $context);
+        }
+        self
+    }
+
+    # Bring the node up to begin time. Returns the node itself.
+    method to-begin-time(RakuAST::Resolver $resolver, RakuAST::IMPL::QASTContext $context) {
+        if nqp::istype(self, RakuAST::ParseTime) {
+            self.ensure-parse-performed($resolver, $context);
+        }
+        if nqp::istype(self, RakuAST::BeginTime) {
+            self.ensure-begin-performed($resolver, $context);
+        }
+        self
+    }
+
     # Perform CHECK-time activities on this node.
     method IMPL-CHECK(RakuAST::Resolver $resolver, RakuAST::IMPL::QASTContext $context, Bool $resolve-only) {
         # Ensure parse time was performed already before visiting children, when it is a