Merge pull request klaus01#51 from doncheng/master

klaus01 · web-flow · commit 032ec4f74e9e · 2018-12-05T08:55:50.000+08:00
修改前缀时，添加是category时的处理
diff --git a/KLGenerateSpamCode/main.m b/KLGenerateSpamCode/main.m
@@ -900,7 +900,15 @@ void modifyClassNamePrefix(NSMutableString *projectContent, NSString *sourceCode
         if ([fileName hasPrefix:oldName]) {
             newClassName = [newName stringByAppendingString:[fileName substringFromIndex:oldName.length]];
         } else {
-            newClassName = [newName stringByAppendingString:fileName];
+            //处理是category的情况。当是category时，修改+号后面的类名前缀
+            NSString *oldNamePlus = [NSString stringWithFormat:@"+%@",oldName];
+            if ([fileName containsString:oldNamePlus]) {
+                NSMutableString *fileNameStr = [[NSMutableString alloc] initWithString:fileName];
+                [fileNameStr replaceCharactersInRange:[fileName rangeOfString:oldNamePlus] withString:[NSString stringWithFormat:@"+%@",newName]];
+                newClassName = fileNameStr;
+            }else{
+                newClassName = [newName stringByAppendingString:fileName];
+            }
         }
         
         // 文件名 Const.ext > DDConst.ext
