fixed swift class resolution when app name has spaces

ClassyKit · Aug 31, 2015 · 78b1f6f · 78b1f6f
1 parent 67a17a7
commit 78b1f6f
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/Classy/Parser/CASParser.m b/Classy/Parser/CASParser.m
@@ -643,7 +643,7 @@ - (NSArray *)nextStyleNodes {
                 styleSelector.styleClass = [tokenValue substringFromIndex:1];
             } else {
                 styleSelector.objectClass = NSClassFromString(tokenValue);
-				if (!styleSelector.objectClass) {
+                if (!styleSelector.objectClass) {
                      // Maybe it's a custom Swift class
                     styleSelector.objectClass = [self swiftClassFromString:tokenValue];
                 }
@@ -779,9 +779,10 @@ - (CASStyleProperty *)nextStylePropertyIsParent:(BOOL *)isParent {
 }
 
 - (Class)swiftClassFromString:(NSString *)className {
-	NSString *appName = [[NSBundle mainBundle] objectForInfoDictionaryKey:@"CFBundleName"];
-	NSString *classStringName = [NSString stringWithFormat:@"_TtC%lu%@%lu%@", (unsigned long)appName.length, appName, (unsigned long)className.length, className];
-	return NSClassFromString(classStringName);
+    NSString *appName = [[NSBundle mainBundle] objectForInfoDictionaryKey:@"CFBundleName"];
+    NSString *sanitizedAppName = [appName stringByReplacingOccurrencesOfString:@" " withString:@"_"];
+    NSString *classStringName = [NSString stringWithFormat:@"_TtC%lu%@%lu%@", (unsigned long)sanitizedAppName.length, sanitizedAppName, (unsigned long)className.length, className];
+    return NSClassFromString(classStringName);
 }