diff --git a/src/main/java/de/maishai/typedast/typedclass/TypedLocalVariable.java b/src/main/java/de/maishai/typedast/typedclass/TypedLocalVariable.java index 639dcff..ad73cbf 100644 --- a/src/main/java/de/maishai/typedast/typedclass/TypedLocalVariable.java +++ b/src/main/java/de/maishai/typedast/typedclass/TypedLocalVariable.java @@ -32,12 +32,14 @@ public final class TypedLocalVariable implements TypedNode { if (typedProgram.getCurrentClass().getCurrentMethod().isLocalVariableInMethod(name)) { throw new RuntimeException("Variable " + name + " already declared"); } + typedProgram.getCurrentClass().getCurrentMethod().getLocalVariables().add(this); return type; } if (!typedProgram.getCurrentClass().isCurrentMethodPresent() && typedProgram.getCurrentClass().isCurrentConstructorPresent()) { if (typedProgram.getCurrentClass().getCurrentConstructor().isLocalVariableInConstructor(name)) { throw new RuntimeException("Variable " + name + " already declared"); } + typedProgram.getCurrentClass().getCurrentConstructor().getLocalVariables().add(this); return type; } throw new RuntimeException("not found method or constructor in class");