Introducing 'compact' mode - in this mode do not print superclasses' properties for each class

Author: Bartosz Waresiak

src/MOMPrinter.h

@@ -8,8 +8,15 @@
 
 #import <Foundation/Foundation.h>
 
+typedef NS_ENUM(NSUInteger, MOMPrinterPeropertiesOutputMode) {
+    MOMPrinterIncludeSuperclassProperties = 1,
+    MOMPrinterOmitSuperclassProperties = 2
+};
+
 @interface MOMPrinter : NSObject
 
+- (id)initWithMode:(MOMPrinterPeropertiesOutputMode)mode;
+
 - (BOOL)printPath:(NSString *)path;
 
 @end

src/MOMPrinter.m

@@ -11,8 +11,31 @@
 #import "NSData+Base64.h"
 #import "utils.h"
 
+@interface MOMPrinter ()
+
+@property (nonatomic, assign) MOMPrinterPeropertiesOutputMode _mode;
+
+@end
+
 @implementation MOMPrinter
 
+#pragma mark - init methods
+
+- (id)init {
+  return [self initWithMode:MOMPrinterIncludeSuperclassProperties];
+}
+
+- (id)initWithMode:(MOMPrinterPeropertiesOutputMode)mode {
+    
+  self = [super init];
+  if (self) {
+    self._mode = mode;
+  }
+  return self;
+}
+
+#pragma mark -
+
 NSString *commonFlagsStringForProperty(NSPropertyDescription *property) {
   char ochar = [property isOptional] ? 'O' : ' ';
   char tchar = [property isTransient] ? 'T' : ' ';
@@ -119,7 +142,20 @@ - (BOOL)printPath:(NSString *)path {
     }
     [entityStr appendFormat:@" (%@)", [entity managedObjectClassName]];
     NSPrintf(@"%@\n", entityStr);
-    NSMutableArray *properties = [NSMutableArray arrayWithArray:[entity properties]];
+      
+    NSMutableArray *properties = nil;
+      
+    if (self._mode == MOMPrinterIncludeSuperclassProperties) {
+      properties = [NSMutableArray arrayWithArray:[entity properties]];
+    } else {
+      // Filter out properties that belong to superclasses.
+      NSSet *superentityProperties = [NSSet setWithArray:[superentity properties]];
+      NSMutableSet *entityProperties = [NSMutableSet setWithArray:[entity properties]];
+      [entityProperties minusSet:superentityProperties];
+          
+      properties = [[entityProperties allObjects] mutableCopy];
+    }
+    
     [properties sortUsingComparator:^(id obj1, id obj2) {
       NSNumber *n1 = orderNumberForClassOfProperty(obj1);
       NSNumber *n2 = orderNumberForClassOfProperty(obj2);
@@ -130,6 +166,7 @@ - (BOOL)printPath:(NSString *)path {
         return result;
       }
     }];
+      
     for (id property in properties) {
       const char *name = [[property name] UTF8String];
       const char *commonFlags = [commonFlagsStringForProperty(property) UTF8String];

src/main.m

@@ -11,26 +11,50 @@
 #import "MOMCompiler.h"
 #import "MOMPrinter.h"
 
+int run(NSString *path, BOOL includeSuperclassProperites);
+
 int main(int argc, const char * argv[])
 {
-
   @autoreleasepool {
     if (argc == 2) {
       NSString *path = [NSString stringWithCString:argv[1] encoding:NSUTF8StringEncoding];
-      MOMCompiler *compiler = [[[MOMCompiler alloc] init] autorelease];
-      NSString *compiledPath = [compiler compilePath:path];
-      if (compiledPath == nil) {
-        return 2;
+    
+      return run(path, YES);
+        
+    } else if (argc == 3) {
+      NSString *parameter = [NSString stringWithCString:argv[1] encoding:NSASCIIStringEncoding];
+      if ([parameter isEqualToString:@"--compact"] ||
+          [parameter isEqualToString:@"-c"] ) {
+        NSString *path = [NSString stringWithCString:argv[2] encoding:NSUTF8StringEncoding];
+        
+        return run(path, NO);
       }
-      MOMPrinter *printer = [[[MOMPrinter alloc] init] autorelease];
-      if (![printer printPath:compiledPath]) {
+    }
+  }
+    
+  NSPrintf(@"Usage: [--compact] %@ path_to_xcdatamodel_file\n", [[NSProcessInfo processInfo] processName]);
+  return 1;
+}
+
+int run(NSString *path, BOOL includeSuperclassProperites) {
+
+    MOMCompiler *compiler = [[[MOMCompiler alloc] init] autorelease];
+    NSString *compiledPath = [compiler compilePath:path];
+    if (compiledPath == nil) {
         return 2;
-      }
+    }
+    MOMPrinter *printer = nil;
+    
+    if (includeSuperclassProperites) {
+      printer = [[[MOMPrinter alloc] init] autorelease];
     } else {
-      NSPrintf(@"Usage: %@ path_to_xcdatamodel_file\n", [[NSProcessInfo processInfo] processName]);
-      return 1;
+      printer = [[[MOMPrinter alloc] initWithMode:MOMPrinterOmitSuperclassProperties] autorelease];
     }
-  }
-  return 0;
+
+    if (![printer printPath:compiledPath]) {
+        return 2;
+    }
+    
+    return 0;
 }