Add key value coding for count on NSArray

rentzsch · Jan 25, 2012 · 9fa14f9 · 9fa14f9
1 parent 6aff9fd
commit 9fa14f9
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/mogenerator.m b/mogenerator.m
@@ -57,6 +57,17 @@ - (NSArray*)entitiesWithACustomSubclassVerbose:(BOOL)verbose_ {
 @end
 
 
+@implementation NSArray (countKeyValueCoding)
+
+- (id)valueForKey:(NSString *)key {
+
+	if ([key isEqualToString:@"count"])
+		return [NSNumber numberWithUnsignedInteger:[self count]];
+
+	return [super valueForKey:key];
+}
+@end
+
 @implementation NSEntityDescription (customBaseClass)
 - (BOOL)hasCustomSuperentity {
 	NSEntityDescription *superentity = [self superentity];