Expose allAttributes and allRelationships, related to #328

rentzsch · Jan 11, 2017 · f5fbfe9 · f5fbfe9
1 parent 9e403ad
commit f5fbfe9
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/mogenerator.m b/mogenerator.m
@@ -201,6 +201,11 @@ - (NSString*)forcedCustomBaseClass {
     return userInfoCustomBaseClass ? userInfoCustomBaseClass : gCustomBaseClassForced;
 }
 /** @TypeInfo NSAttributeDescription */
+- (NSArray*)allAttributes {
+    NSArray *sortDescriptors = [NSArray arrayWithObject:[NSSortDescriptor sortDescriptorWithKey:@"name" ascending:YES]];
+    return [[[self attributesByName] allValues] sortedArrayUsingDescriptors:sortDescriptors];
+}
+/** @TypeInfo NSAttributeDescription */
 - (NSArray*)noninheritedAttributes {
     NSArray *sortDescriptors = [NSArray arrayWithObject:[NSSortDescriptor sortDescriptorWithKey:@"name" ascending:YES]];
     NSEntityDescription *superentity = [self superentity];
@@ -233,7 +238,12 @@ - (NSArray*)noninheritedAttributesSansType {
     }
     return filteredAttributeDescriptions;
 }
-/** @TypeInfo NSAttributeDescription */
+/** @TypeInfo NSRelationshipDescription */
+- (NSArray*)allRelationships {
+    NSArray *sortDescriptors = [NSArray arrayWithObject:[NSSortDescriptor sortDescriptorWithKey:@"name" ascending:YES]];
+    return [[[self relationshipsByName] allValues] sortedArrayUsingDescriptors:sortDescriptors];
+}
+/** @TypeInfo NSRelationshipDescription */
 - (NSArray*)noninheritedRelationships {
     NSArray *sortDescriptors = [NSArray arrayWithObject:[NSSortDescriptor sortDescriptorWithKey:@"name" ascending:YES]];
     NSEntityDescription *superentity = [self superentity];