Swift: add getSequence back to ForEachStmt

Robert Marsh · Robert Marsh · commit 5dd7c14d3607 · 2023-10-04T20:05:00.000Z
diff --git a/swift/ql/lib/codeql/swift/elements/stmt/ForEachStmt.qll b/swift/ql/lib/codeql/swift/elements/stmt/ForEachStmt.qll
@@ -6,4 +6,12 @@ class ForEachStmt extends Generated::ForEachStmt {
     then result = "for ... in ... where ... { ... }"
     else result = "for ... in ... { ... }"
   }
+
+  /*
+   * Gets the sequence of this for each statement.
+   */
+
+  final Expr getSequence() {
+    result = this.getIteratorVar().getInit(0)
+  }
 }

Original file line number	Diff line number	Diff line change
`@@ -6,4 +6,12 @@ class ForEachStmt extends Generated::ForEachStmt {`
`6`	`6`	`then result = "for ... in ... where ... { ... }"`
`7`	`7`	`else result = "for ... in ... { ... }"`
`8`	`8`	`}`
	`9`	`+`
	`10`	`+ /*`
	`11`	`+ * Gets the sequence of this for each statement.`
	`12`	`+ */`
	`13`	`+`
	`14`	`+ final Expr getSequence() {`
	`15`	`+ result = this.getIteratorVar().getInit(0)`
	`16`	`+ }`
`9`	`17`	`}`