[Fix] capitalizedString issue

MrPans · Nov 21, 2017 · 56efa1b · 56efa1b
1 parent 41576a0
commit 56efa1b
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/Source/AccessCodeGenerator.m b/Source/AccessCodeGenerator.m
@@ -148,16 +148,22 @@ + (NSString *)getterWithPSProperty:(PSProperty *)model
 
 + (NSString *)setterWithPSProperty:(PSProperty *)model {
     NSString *setter = @"";
+    if (model.name.length < 1) {
+        return setter;
+    }
+
+    NSString *firstLetter = [model.name substringToIndex:1];
+    NSString *name = [model.name stringByReplacingCharactersInRange:NSMakeRange(0, 1) withString:firstLetter.uppercaseString];
     if (model.isObjectType) {
         setter = [NSString stringWithFormat:@"\n- (void)set%@:(%@ *)%@ {\n    _%@ = %@;\n}",
-                  [model.name capitalizedString],
+                  name,
                   model.dataType,
                   model.name,
                   model.name,
                   model.name];
     } else {
         setter = [NSString stringWithFormat:@"\n- (void)set%@:(%@)%@ {\n    _%@ = %@;\n}",
-                  [model.name capitalizedString],
+                  name,
                   model.dataType,
                   model.name,
                   model.name,