Adds logic to infer messages to literal classes

Issue: #17
0x7CFE · May 25, 2016 · b0d8b87 · b0d8b87
1 parent feddc12
commit b0d8b87
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/TypeAnalyzer.cpp b/src/TypeAnalyzer.cpp
@@ -636,7 +636,12 @@ InferContext* TypeSystem::inferMessage(TSelector selector, const Type& arguments
     TClass* receiver = 0;
 
     if (self.getKind() == Type::tkLiteral) {
-        receiver = isSmallInteger(self.getValue()) ? globals.smallIntClass : self.getValue()->getClass();
+        if (isSmallInteger(self.getValue()))
+            receiver = globals.smallIntClass;
+        else if (self.getValue()->getClass()->getClass() == globals.stringClass->getClass()->getClass())
+            receiver = self.getValue()->cast<TClass>();
+        else
+            receiver = self.getValue()->getClass();
     } else {
         receiver = self.getValue()->cast<TClass>();
     }