feat(ast): Update internal AST

Author: GeekMasher

ql/lib/codeql/bicep/ast/internal/Arguments.qll

@@ -6,6 +6,7 @@
 private import AstNodes
 private import TreeSitter
 private import codeql.bicep.ast.AstNodes
+private import Expr
 
 /**
  *  A Arguments AST Node.
@@ -19,6 +20,11 @@ class ArgumentsImpl extends TArguments, AstNode {
 
   override string toString() { result = ast.toString() }
 
+  ExprImpl getArgument(int index ) {
+    toTreeSitter(result) = ast.getChild(index)
+  }
 
-
+  ExprImpl getArguments() {
+    toTreeSitter(result) = ast.getChild(_)
+  }
 }

ql/lib/codeql/bicep/ast/internal/AstNodes.qll

@@ -87,8 +87,8 @@ class TIdents = TIdentifier or TPropertyIdentifier;
  * A statement in a Bicep program
  */
 class TStmts =
-  TAssertStatement or TForStatement or TIfStatement or TImportStatement or TImportWithStatement or
-      TStatement or TUsingStatement;
+  TInfrastructure or TAssertStatement or TForStatement or TIfStatement or TImportStatement or
+      TImportWithStatement or TStatement or TUsingStatement;
 
 /**
  * A expersion value in a Bicep program

ql/lib/codeql/bicep/ast/internal/CallExpression.qll

@@ -7,7 +7,10 @@ private import AstNodes
 private import TreeSitter
 private import codeql.bicep.ast.AstNodes
 private import Expr
-
+private import Expression
+private import Idents
+private import Arguments
+private import NullableReturnType
 
 /**
  *  A CallExpression AST Node.
@@ -21,6 +24,15 @@ class CallExpressionImpl extends TCallExpression, ExprImpl {
 
   override string toString() { result = ast.toString() }
 
+  IdentsImpl getIdentifier() {
+    toTreeSitter(result) = ast.getFunction()
+  }
 
+  ArgumentsImpl getArguments() {
+    toTreeSitter(result) = ast.getArguments()
+  }
 
+  NullableReturnTypeImpl getReturnType() {
+    toTreeSitter(result) = ast.getChild()
+  }
 }

ql/lib/codeql/bicep/ast/internal/Infrastructure.qll

@@ -6,6 +6,8 @@
 private import AstNodes
 private import TreeSitter
 private import codeql.bicep.ast.AstNodes
+private import Stmts
+
 
 /**
  *  A Infrastructure AST Node.
@@ -19,6 +21,7 @@ class InfrastructureImpl extends TInfrastructure, AstNode {
 
   override string toString() { result = ast.toString() }
 
-
-
+  StmtsImpl getStatement(int index) {
+    toTreeSitter(result) = ast.getChild(index)
+  }
 }

ql/lib/codeql/bicep/ast/internal/UserDefinedFunction.qll

@@ -6,6 +6,9 @@
 private import AstNodes
 private import TreeSitter
 private import codeql.bicep.ast.AstNodes
+private import Identifier
+private import Stmts
+private import Type
 
 /**
  *  A UserDefinedFunction AST Node.
@@ -19,6 +22,16 @@ class UserDefinedFunctionImpl extends TUserDefinedFunction, AstNode {
 
   override string toString() { result = ast.toString() }
 
+  IdentifierImpl getName() {
+    toTreeSitter(result) = ast.getName()
+  }
 
+  TypeImpl getReturnType() {
+    toTreeSitter(result) = ast.getReturns()
+  }
+
+  StmtsImpl getBody() {
+    toTreeSitter(result) = ast.getChild(_)
+  } 
 
 }