Expose some private AST node members

Author: heimskr

Source/JavaScriptCore/parser/Nodes.cpp

@@ -66,6 +66,11 @@ StatementNode* SourceElements::singleStatement() const
     return m_head == m_tail ? m_head : nullptr;
 }
 
+StatementNode* SourceElements::firstStatement() const
+{
+    return m_head;
+}
+
 StatementNode* SourceElements::lastStatement() const
 {
     return m_tail;

Source/JavaScriptCore/parser/Nodes.h

@@ -1649,6 +1649,7 @@ namespace JSC {
         void append(StatementNode*);
 
         StatementNode* singleStatement() const;
+        StatementNode* firstStatement() const;
         StatementNode* lastStatement() const;
 
         bool hasCompletionValue() const;
@@ -1989,6 +1990,7 @@ namespace JSC {
             return m_numConstants + 2;
         }
 
+        SourceElements* statements() const { return m_statements; }
         StatementNode* singleStatement() const;
 
         bool hasCompletionValue() const override;
@@ -2121,6 +2123,8 @@ namespace JSC {
     class ModuleDeclarationNode : public StatementNode {
     public:
         virtual bool analyzeModule(ModuleAnalyzer&) = 0;
+        virtual bool isImportDeclarationNode() const { return false; }
+        virtual bool hasAttributesList() const { return false; }
         bool hasCompletionValue() const override { return false; }
         bool isModuleDeclarationNode() const override { return true; }
 
@@ -2136,6 +2140,9 @@ namespace JSC {
         };
         ImportDeclarationNode(const JSTokenLocation&, ImportType, ImportSpecifierListNode*, ModuleNameNode*, ImportAttributesListNode*);
 
+        bool isImportDeclarationNode() const override { return true; }
+        bool hasAttributesList() const override { return true; }
+
         ImportSpecifierListNode* specifierList() const { return m_specifierList; }
         ModuleNameNode* moduleName() const { return m_moduleName; }
         ImportAttributesListNode* attributesList() const { return m_attributesList; }
@@ -2154,6 +2161,8 @@ namespace JSC {
     public:
         ExportAllDeclarationNode(const JSTokenLocation&, ModuleNameNode*, ImportAttributesListNode*);
 
+        bool hasAttributesList() const override { return true; }
+
         ModuleNameNode* moduleName() const { return m_moduleName; }
         ImportAttributesListNode* attributesList() const { return m_attributesList; }
 
@@ -2222,6 +2231,8 @@ namespace JSC {
     public:
         ExportNamedDeclarationNode(const JSTokenLocation&, ExportSpecifierListNode*, ModuleNameNode*, ImportAttributesListNode*);
 
+        bool hasAttributesList() const override { return true; }
+
         ExportSpecifierListNode* specifierList() const { return m_specifierList; }
         ModuleNameNode* moduleName() const { return m_moduleName; }
         ImportAttributesListNode* attributesList() const { return m_attributesList; }