feat(ast): Enhance AST classes with additional methods for expression handling

Author: GeekMasher

ql/lib/codeql/bicep/ast/Expr.qll

@@ -30,12 +30,37 @@ final class Expr extends AstNode instanceof ExprImpl { }
 /**
  * A AssignmentExpression expression in the AST.
  */
-final class AssignmentExpression extends Expr instanceof AssignmentExpressionImpl { }
+class AssignmentExpression extends Expr instanceof AssignmentExpressionImpl {
+  /**
+   * Gets the left operand of the assignment expression.
+   */
+  Expr getLeft() { result = AssignmentExpressionImpl.super.getLeft() }
+
+  /**
+   * Gets the right operand of the assignment expression.
+   */
+  Expr getRight() { result = AssignmentExpressionImpl.super.getRight() }
+}
 
 /**
  * A BinaryExpression expression in the AST.
  */
-final class BinaryExpression extends Expr instanceof BinaryExpressionImpl { }
+class BinaryExpression extends Expr instanceof BinaryExpressionImpl {
+  /**
+   * Gets the left operand of the binary expression.
+   */
+  Expr getLeft() { result = BinaryExpressionImpl.super.getLeft() }
+
+  /**
+   * Gets the right operand of the binary expression.
+   */
+  Expr getRight() { result = BinaryExpressionImpl.super.getRight() }
+
+  /**
+   * Gets the operator of the binary expression.
+   */
+  string getOperator() { result = BinaryExpressionImpl.super.getOperator() }
+}
 
 /**
  * A Expression expression in the AST.
@@ -45,7 +70,18 @@ final class Expression extends Expr instanceof ExpressionImpl { }
 /**
  * A Interpolation literal in the AST.
  */
-final class Interpolation extends Expr instanceof InterpolationImpl { }
+final class Interpolation extends Expr instanceof InterpolationImpl {
+  /**
+   *  Gets the expression contained within the interpolation.
+   */
+  Expr getExpression() {
+    result = InterpolationImpl.super.getExpression()
+  }
+
+  string getValue() {
+    result = "${" + this.getExpression().toString() + "}"
+  }
+}
 
 /**
  * A LambdaExpression expression in the AST.
@@ -69,9 +105,7 @@ class MemberExpression extends Expr instanceof MemberExpressionImpl {
   /**
    * Gets the full name of the member expression, which includes the namespace and the member name.
    */
-  string getFullName() {
-    result = this.getNamespace().getName() + "." + this.getName().getName()
-  }
+  string getFullName() { result = this.getNamespace().getName() + "." + this.getName().getName() }
 }
 
 /**

ql/lib/codeql/bicep/ast/Stmts.qll

@@ -50,7 +50,13 @@ final class ForStatementStmt extends Stmts instanceof ForStatementImpl { }
 /**
  * A IfStatement statement
  */
-final class IfStatementStmt extends Stmts instanceof IfStatementImpl { }
+class IfStatement extends Stmts instanceof IfStatementImpl { 
+  /** Gets the condition of the if statement. */
+  Expr getCondition() { result = IfStatementImpl.super.getCondition() }
+
+  /** Gets the body of the if statement. */
+  Expr getBody() { result = IfStatementImpl.super.getBody() }
+}
 
 /**
  * A ImportStatement statement

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

@@ -3,12 +3,12 @@
  *
  *  WARNING: this file is generated, do not edit manually
  */
+
 private import AstNodes
 private import TreeSitter
 private import codeql.bicep.ast.AstNodes
 private import Expr
 
-
 /**
  *  A AssignmentExpression AST Node.
  */
@@ -21,6 +21,7 @@ class AssignmentExpressionImpl extends TAssignmentExpression, ExprImpl {
 
   override string toString() { result = ast.toString() }
 
+  ExprImpl getLeft() { toTreeSitter(result) = ast.getLeft() }
 
-
-}
+  ExprImpl getRight() { toTreeSitter(result) = ast.getRight() }
+}

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

@@ -21,6 +21,12 @@ class BinaryExpressionImpl extends TBinaryExpression, ExprImpl {
 
   override string toString() { result = ast.toString() }
 
+  ExprImpl getLeft() { toTreeSitter(result) = ast.getLeft() }
 
+  ExprImpl getRight() { toTreeSitter(result) = ast.getRight() }
+
+  string getOperator() {
+    result = ast.getOperator()
+  }
 
 }

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

@@ -3,11 +3,13 @@
  *
  *  WARNING: this file is generated, do not edit manually
  */
+
 private import AstNodes
 private import TreeSitter
 private import codeql.bicep.ast.AstNodes
 private import Stmts
-
+private import ParenthesizedExpression
+private import Object
 
 /**
  *  A IfStatement AST Node.
@@ -21,6 +23,7 @@ class IfStatementImpl extends TIfStatement, StmtsImpl {
 
   override string toString() { result = ast.toString() }
 
+  ParenthesizedExpressionImpl getCondition() { toTreeSitter(result) = ast.getChild(0) }
 
-
-}
+  ObjectImpl getBody() { toTreeSitter(result) = ast.getChild(1) }
+}