vala: Add missing replace_expression() implementations

1 files changed, 16 insertions, 0 deletions

diff --git a/vala/valaforstatement.vala b/vala/valaforstatement.vala
index 80ff8dbeb..79bd063f7 100644
--- a/vala/valaforstatement.vala
+++ b/vala/valaforstatement.vala
@@ -136,6 +136,22 @@ public class Vala.ForStatement : CodeNode, Statement {
 		body.accept (visitor);
 	}
 
+	public override void replace_expression (Expression old_node, Expression new_node) {
+		if (condition == old_node) {
+			condition = new_node;
+		}
+		for (int i=0; i < initializer.size; i++) {
+			if (initializer[i] == old_node) {
+				initializer[i] = new_node;
+			}
+		}
+		for (int i=0; i < iterator.size; i++) {
+			if (iterator[i] == old_node) {
+				iterator[i] = new_node;
+			}
+		}
+	}
+
 	bool always_true (Expression condition) {
 		var literal = condition as BooleanLiteral;
 		return (literal != null && literal.value);