GenericASTPass.h 992 B

123456789101112131415161718192021222324252627282930313233343536373839404142434445
  1. /*
  2. * Copyright (c) 2023, Dan Klishch <danilklishch@gmail.com>
  3. *
  4. * SPDX-License-Identifier: BSD-2-Clause
  5. */
  6. #pragma once
  7. #include <AK/NonnullRefPtr.h>
  8. #include <AK/RecursionDecision.h>
  9. #include "Compiler/CompilerPass.h"
  10. namespace JSSpecCompiler {
  11. class RecursiveASTVisitor {
  12. public:
  13. virtual ~RecursiveASTVisitor() = default;
  14. void run_in_const_subtree(NullableTree tree);
  15. void run_in_subtree(Tree& tree);
  16. protected:
  17. virtual RecursionDecision on_entry(Tree) { return RecursionDecision::Recurse; }
  18. virtual void on_leave(Tree) { }
  19. void replace_current_node_with(NullableTree tree);
  20. private:
  21. RecursionDecision recurse(Tree root, NodeSubtreePointer& pointer);
  22. NodeSubtreePointer* m_current_subtree_pointer = nullptr;
  23. };
  24. class GenericASTPass
  25. : public IntraproceduralCompilerPass
  26. , protected RecursiveASTVisitor {
  27. public:
  28. using IntraproceduralCompilerPass::IntraproceduralCompilerPass;
  29. protected:
  30. void process_function() override;
  31. };
  32. }