Adds two booleans you can use in the templates to figure out does the…

… current attribute uses custom type (setup through userInfo)
rentzsch · Jul 13, 2017 · 1930cd9 · 1930cd9
1 parent db41d58
commit 1930cd9
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/mogenerator.h b/mogenerator.h
@@ -32,12 +32,14 @@
 @interface NSAttributeDescription (typing)
 - (BOOL)hasScalarAttributeType;
 - (BOOL)usesScalarAttributeType;
+- (BOOL)usesCustomScalarAttributeType;
 - (NSString*)scalarAttributeType;
 - (NSString*)scalarAccessorMethodName;
 - (NSString*)scalarFactoryMethodName;
 - (BOOL)hasDefinedAttributeType;
 - (NSArray*)objectAttributeTransformableProtocols;
 - (BOOL)hasAttributeTransformableProtocols;
+- (BOOL)usesCustomObjectAttributeType;
 - (NSString*)objectAttributeClassName;
 - (NSString*)objectAttributeType;
 - (BOOL)hasTransformableAttributeType;

diff --git a/mogenerator.m b/mogenerator.m
@@ -443,6 +443,11 @@ - (BOOL)usesScalarAttributeType {
     }
 }
 
+- (BOOL)usesCustomScalarAttributeType {
+    NSString *attributeValueScalarType = [[self userInfo] objectForKey:kAttributeValueScalarTypeKey];
+    return (attributeValueScalarType != nil);
+}
+
 - (NSString*)scalarAttributeType {
     BOOL isUnsigned = [self isUnsigned];
 
@@ -584,6 +589,12 @@ - (NSArray*)objectAttributeTransformableProtocols {
 - (BOOL)hasAttributeTransformableProtocols {
     return [self hasTransformableAttributeType] && [[self userInfo] objectForKey:@"attributeTransformableProtocols"];
 }
+
+- (BOOL)usesCustomObjectAttributeType {
+    NSString *attributeValueClassName = [[self userInfo] objectForKey:@"attributeValueClassName"];
+    return (attributeValueClassName != nil);
+}
+
 - (NSString*)objectAttributeType {
     NSString *result = [self objectAttributeClassName];
     if ([result isEqualToString:@"Class"]) {