Add support for KVC.

lembacon · lembacon · commit 15414a39cd44 · 2013-03-08T02:57:44.000+08:00
diff --git a/src/VSDataModel.h b/src/VSDataModel.h
@@ -44,6 +44,9 @@
 - (void)dataObject:(VSDataObject *)dataObject willChangeValueForKey:(NSString *)key;
 - (void)dataObject:(VSDataObject *)dataObject didChangeValueForKey:(NSString *)key;
 
+- (BOOL)dataObject:(VSDataObject *)dataObject getValue:(__strong id *)value forKey:(NSString *)key;
+- (BOOL)dataObject:(VSDataObject *)dataObject setValue:(id)value forKey:(NSString *)key;
+
 - (BOOL)dataManager:(VSDataManager *)dataManager eraseAllValuesForDataObject:(VSDataObject *)dataObject;
 - (BOOL)dataManager:(VSDataManager *)dataManager setAllValuesForDataObject:(VSDataObject *)dataObject;
 
diff --git a/src/VSDataModel.m b/src/VSDataModel.m
@@ -665,6 +665,51 @@ - (void)dataObject:(VSDataObject *)dataObject didChangeValueForKey:(NSString *)k
   }
 }
 
+- (BOOL)dataObject:(VSDataObject *)dataObject getValue:(__strong id *)value forKey:(NSString *)key
+{
+  if (key == nil) {
+    return NO;
+  }
+
+  NSDictionary *properties = [self _propertiesForDataObjectClass:[dataObject class]];
+  VSDataObjectPropertyInfo *propInfo = nil;
+  if (properties == nil || (propInfo = [properties objectForKey:key]) == nil) {
+    return NO;
+  }
+
+  NSInvocation *invocation = [NSInvocation invocationWithMethodSignature:[propInfo getterSignature]];
+  [invocation setTarget:dataObject];
+  [invocation setSelector:[propInfo getter]];
+  [invocation invoke];
+
+  __unsafe_unretained id valueObject = nil;
+  [invocation getReturnValue:&valueObject];
+  *value = valueObject;
+
+  return YES;
+}
+
+- (BOOL)dataObject:(VSDataObject *)dataObject setValue:(id)value forKey:(NSString *)key
+{
+  if (key == nil) {
+    return NO;
+  }
+
+  NSDictionary *properties = [self _propertiesForDataObjectClass:[dataObject class]];
+  VSDataObjectPropertyInfo *propInfo = nil;
+  if (properties == nil || (propInfo = [properties objectForKey:key]) == nil) {
+    return NO;
+  }
+
+  NSInvocation *invocation = [NSInvocation invocationWithMethodSignature:[propInfo setterSignature]];
+  [invocation setTarget:dataObject];
+  [invocation setSelector:[propInfo setter]];
+  [invocation setArgument:&value atIndex:2];
+  [invocation invoke];
+
+  return YES;
+}
+
 - (BOOL)dataManager:(VSDataManager *)dataManager eraseAllValuesForDataObject:(VSDataObject *)dataObject
 {
   if ([dataObject dataManager] != dataManager) {
diff --git a/src/VSDataObject.m b/src/VSDataObject.m
@@ -139,4 +139,21 @@ - (void)didChangeValueForKey:(NSString *)key withSetMutation:(NSKeyValueSetMutat
   [super didChangeValueForKey:key withSetMutation:mutationKind usingObjects:objects];
 }
 
+- (id)valueForKey:(NSString *)key
+{
+  id value = nil;
+  if ([[VSDataModel sharedModel] dataObject:self getValue:&value forKey:key]) {
+    return value;
+  }
+
+  return [super valueForKey:key];
+}
+
+- (void)setValue:(id)value forKey:(NSString *)key
+{
+  if (![[VSDataModel sharedModel] dataObject:self setValue:value forKey:key]) {
+    [super setValue:value forKey:key];
+  }
+}
+
 @end
