BROKEN WIP first crack at obj-c lightweight generics

rentzsch · Mar 4, 2016 · c141670 · c141670
1 parent f48402f
commit c141670
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/mogenerator.m b/mogenerator.m
@@ -584,12 +584,23 @@ - (BOOL)isReadonly {
 
 @implementation NSRelationshipDescription (collectionClassName)
 
+- (NSString*)jr_CollectionClassStringWithOrderedClassName:(NSString*)orderedClassName
+                                       unorderedClassName:(NSString*)unorderedClassName
+{
+    NSString *generic = [NSString stringWithFormat:@"<%@*>", self.destinationEntity.managedObjectClassName];
+    return [self jr_isOrdered]
+        ? [orderedClassName stringByAppendingString:generic]
+        : [unorderedClassName stringByAppendingString:generic];
+}
+
 - (NSString*)mutableCollectionClassName {
-    return [self jr_isOrdered] ? @"NSMutableOrderedSet" : @"NSMutableSet";
+    return [self jr_CollectionClassStringWithOrderedClassName:@"NSMutableOrderedSet"
+                                           unorderedClassName:@"NSMutableSet"];
 }
 
 - (NSString*)immutableCollectionClassName {
-    return [self jr_isOrdered] ? @"NSOrderedSet" : @"NSSet";
+    return [self jr_CollectionClassStringWithOrderedClassName:@"NSOrderedSet"
+                                           unorderedClassName:@"NSSet"];
 }
 
 - (BOOL)jr_isOrdered {