BoilVisitor.MessageSend for Bracket, proper class Extension in ST

1 files changed, 8 insertions, 2 deletions

diff --git a/st/SymbolTable.java b/st/SymbolTable.java
index 7f3c788..90640c6 100644
--- a/st/SymbolTable.java
+++ b/st/SymbolTable.java
@@ -55,11 +55,19 @@ public class SymbolTable {
                                              t.getName(), t.getType(),
                                              cls.getName(), cls.getType()));
             cls.addLocal(t);
+            this.symt.put(new TokenKey(t.getName(),
+                                       cls,
+                                       null),
+                          t);
         }
         for (MethodInstance m : ext.getMethods()) {
             MinimalLogger.info(String.format("Added %s (%s) as a method of %s (%s)",
                                              m.getName(), m.getType(),
                                              cls.getName(), cls.getType()));
+            this.symt.put(new TokenKey(m.getName(),
+                                       cls,
+                                       null),
+                          m);
             cls.addMethod(m);
         }
     }
@@ -202,8 +210,6 @@ public class SymbolTable {
         if (ret == null)
             MinimalLogger.severe(String.format("getClass returning null for missing alias %s!",
                                                id));
-        MinimalLogger.severe(String.format("It was: %s",
-                                           this.symt.get(id)));
         return ret;
     }