PS: Add 'raw' AST classes coming directly from the extractor.

Author: MathiasVP

powershell/ql/lib/semmle/code/powershell/ast/internal/Raw/ArrayExpression.qll

@@ -0,0 +1,11 @@
+private import Raw
+
+class ArrayExpr extends @array_expression, Expr {
+  override SourceLocation getLocation() { array_expression_location(this, result) }
+
+  StmtBlock getStmtBlock() { array_expression(this, result) }
+
+  final override Ast getChild(ChildIndex i) {
+    i = ArrayExprStmtBlock() and result = this.getStmtBlock()
+  }
+}

powershell/ql/lib/semmle/code/powershell/ast/internal/Raw/ArrayLiteral.qll

@@ -0,0 +1,16 @@
+private import Raw
+
+class ArrayLiteral extends @array_literal, Expr {
+  override SourceLocation getLocation() { array_literal_location(this, result) }
+
+  Expr getElement(int index) { array_literal_element(this, index, result) }
+
+  Expr getAnElement() { array_literal_element(this, _, result) }
+
+  final override Ast getChild(ChildIndex i) {
+    exists(int index |
+      i = ArrayLiteralExpr(index) and
+      result = this.getElement(index)
+    )
+  }
+}

powershell/ql/lib/semmle/code/powershell/ast/internal/Raw/AssignmentStatement.qll

@@ -0,0 +1,19 @@
+private import Raw
+
+class AssignStmt extends @assignment_statement, PipelineBase {
+  override SourceLocation getLocation() { assignment_statement_location(this, result) }
+
+  int getKind() { assignment_statement(this, result, _, _) }
+
+  Expr getLeftHandSide() { assignment_statement(this, _, result, _) }
+
+  Stmt getRightHandSide() { assignment_statement(this, _, _, result) }
+
+  final override Ast getChild(ChildIndex i) {
+    i = AssignStmtLeftHandSide() and
+    result = this.getLeftHandSide()
+    or
+    i = AssignStmtRightHandSide() and
+    result = this.getRightHandSide()
+  }
+}

powershell/ql/lib/semmle/code/powershell/ast/internal/Raw/Ast.qll

@@ -0,0 +1,17 @@
+private import Raw
+import Location
+private import Scope
+
+class Ast extends @ast {
+  final string toString() { none() }
+
+  final Ast getParent() { result.getAChild() = this }
+
+  Ast getChild(ChildIndex i) { none() }
+
+  final Ast getAChild() { result = this.getChild(_) }
+
+  Location getLocation() { none() }
+
+  Scope getScope() { result = scopeOf(this) }
+}

powershell/ql/lib/semmle/code/powershell/ast/internal/Raw/Attribute.qll

@@ -0,0 +1,33 @@
+private import Raw
+
+class Attribute extends @attribute, AttributeBase {
+  override SourceLocation getLocation() { attribute_location(this, result) }
+
+  string getName() { attribute(this, result, _, _) }
+
+  int getNumNamedArguments() { attribute(this, _, result, _) }
+
+  int getNumPositionalArguments() { attribute(this, _, _, result) }
+
+  NamedAttributeArgument getNamedArgument(int i) { attribute_named_argument(this, i, result) }
+
+  final override Ast getChild(ChildIndex i) {
+    exists(int index |
+      i = AttributeNamedArg(index) and
+      result = this.getNamedArgument(index)
+      or
+      i = AttributePosArg(index) and
+      result = this.getPositionalArgument(index)
+    )
+  }
+
+  NamedAttributeArgument getANamedArgument() { result = this.getNamedArgument(_) }
+
+  int getNumberOfArguments() { result = count(this.getAPositionalArgument()) }
+
+  Expr getPositionalArgument(int i) { attribute_positional_argument(this, i, result) }
+
+  Expr getAPositionalArgument() { result = this.getPositionalArgument(_) }
+
+  int getNumberOfPositionalArguments() { result = count(this.getAPositionalArgument()) }
+}

powershell/ql/lib/semmle/code/powershell/ast/internal/Raw/AttributeBase.qll

@@ -0,0 +1,3 @@
+private import Raw
+
+class AttributeBase extends @attribute_base, Ast { }

powershell/ql/lib/semmle/code/powershell/ast/internal/Raw/AttributedExpr.qll

@@ -0,0 +1,17 @@
+private import Raw
+
+class AttributedExpr extends AttributedExprBase, @attributed_expression {
+  final override Expr getExpr() { attributed_expression(this, _, result) }
+
+  final override Attribute getAttribute() { attributed_expression(this, result, _) }
+
+  override Location getLocation() { attributed_expression_location(this, result) }
+
+  override Ast getChild(ChildIndex i) {
+    i = AttributedExprExpr() and
+    result = this.getExpr()
+    or
+    i = AttributedExprAttr() and
+    result = this.getAttribute()
+  }
+}

powershell/ql/lib/semmle/code/powershell/ast/internal/Raw/AttributedExprBase.qll

@@ -0,0 +1,7 @@
+private import Raw
+
+class AttributedExprBase extends @attributed_expression_ast, Expr {
+  Expr getExpr() { none() }
+
+  AttributeBase getAttribute() { none() }
+}

powershell/ql/lib/semmle/code/powershell/ast/internal/Raw/BaseConstantExpression.qll

@@ -0,0 +1,10 @@
+private import Raw
+
+/** The base class for constant expressions. */
+class BaseConstExpr extends @base_constant_expression, Expr {
+  /** Gets the type of this constant expression. */
+  string getType() { none() }
+
+  /** Gets a string literal of this constant expression. */
+  StringLiteral getValue() { none() }
+}

powershell/ql/lib/semmle/code/powershell/ast/internal/Raw/BinaryExpression.qll

@@ -0,0 +1,35 @@
+private import Raw
+
+class BinaryExpr extends @binary_expression, Expr {
+  override SourceLocation getLocation() { binary_expression_location(this, result) }
+
+  int getKind() { binary_expression(this, result, _, _) }
+
+  /** Gets an operand of this binary expression. */
+  Expr getAnOperand() {
+    result = this.getLeft()
+    or
+    result = this.getRight()
+  }
+
+  final override Ast getChild(ChildIndex i) {
+    i = BinaryExprLeft() and
+    result = this.getLeft()
+    or
+    i = BinaryExprRight() and
+    result = this.getRight()
+  }
+
+  /** Holds if this binary expression has the operands `e1` and `e2`. */
+  predicate hasOperands(Expr e1, Expr e2) {
+    e1 = this.getLeft() and
+    e2 = this.getRight()
+    or
+    e1 = this.getRight() and
+    e2 = this.getLeft()
+  }
+
+  Expr getLeft() { binary_expression(this, _, result, _) }
+
+  Expr getRight() { binary_expression(this, _, _, result) }
+}