Support "Custom Class" for Transformable attributes:

generated classes will have this attribute typed.
rentzsch · Aug 5, 2018 · 3fd2a5a · 3fd2a5a
1 parent fdc15ed
commit 3fd2a5a
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/mogenerator.xcodeproj/xcshareddata/xcschemes/mogenerator.xcscheme b/mogenerator.xcodeproj/xcshareddata/xcschemes/mogenerator.xcscheme
@@ -61,6 +61,20 @@
             ReferencedContainer = "container:mogenerator.xcodeproj">
          </BuildableReference>
       </BuildableProductRunnable>
+      <CommandLineArguments>
+         <CommandLineArgument
+            argument = "--model /Users/tomek/Code/mobile-bo-ios/HiringStoreKit/HiringStoreKit/DataModel/hiring.xcdatamodeld"
+            isEnabled = "YES">
+         </CommandLineArgument>
+         <CommandLineArgument
+            argument = "--output-dir /Users/tomek/Desktop"
+            isEnabled = "YES">
+         </CommandLineArgument>
+         <CommandLineArgument
+            argument = "--swift"
+            isEnabled = "YES">
+         </CommandLineArgument>
+      </CommandLineArguments>
       <AdditionalOptions>
       </AdditionalOptions>
    </LaunchAction>

diff --git a/momcom/NSAttributeDescription+momcom.m b/momcom/NSAttributeDescription+momcom.m
@@ -11,6 +11,7 @@
 
 static NSDictionary *attributeTypeForString;
 const NSString *const kUsesScalarAttributeType = @"mogenerator.usesScalarAttributeType";
+const NSString *const kAttributeValueClassName = @"attributeValueClassName";
 
 @implementation NSAttributeDescription (momcom)
 
@@ -50,6 +51,13 @@ + (NSAttributeDescription *)baseEntityForXML:(NSXMLElement *)xmlNode
             [attributeDescription setAttributeType:[attributeType integerValue]];
         }
     }
+
+    NSXMLNode *customClassNameElement = [xmlNode attributeForName:@"customClassName"];
+    if (customClassNameElement != nil) {
+        NSMutableDictionary *userInfo = [[attributeDescription userInfo] mutableCopy];
+        userInfo[kAttributeValueClassName] = [customClassNameElement stringValue];
+        [attributeDescription setUserInfo:userInfo.copy];
+    }
 
     NSXMLNode *userScalarElement = [xmlNode attributeForName:@"usesScalarValueType"];
     if (userScalarElement != nil) {