Store 'color' in CBCanvas model

drpancake · Apr 21, 2012 · 403fdda · 403fdda
1 parent dd9d82f
commit 403fdda
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/layout/CBCanvas.h b/layout/CBCanvas.h
@@ -11,5 +11,6 @@
 @interface CBCanvas : CBLayoutElement
 
 @property (nonatomic, strong, readwrite) NSArray *points;
+@property (nonatomic, strong, readwrite) NSString *color;
 
 @end
diff --git a/layout/CBCanvas.m b/layout/CBCanvas.m
@@ -11,12 +11,14 @@
 @implementation CBCanvas
 
 @synthesize points;
+@synthesize color;
 
 - (id)initWithJSON:(NSDictionary *)json
 {
     self = [super initWithJSON:json];
     if (self) {
         points = [json objectForKey:@"points"];
+        color = [json objectForKey:@"color"];
     }
     return self;
 }
@@ -26,9 +28,11 @@ - (NSDictionary *)toJSON
     NSDictionary *base = [super toJSON];
     NSMutableDictionary *json = [[NSMutableDictionary alloc] initWithDictionary:base];
 
-    if (self.points) {
+    if (self.points)
         [json setObject:self.points forKey:@"points"];
-    }
+
+    if (self.color)
+        [json setObject:self.color forKey:@"color"];
 
     // Copy to immutable
     return [NSDictionary dictionaryWithDictionary:json];