[formMapping] Ability to inherit from other field classes

kokomo · Jun 16, 2012 · caf6d77 · caf6d77
1 parent a3d1499
commit caf6d77
Show file tree

Hide file tree

Showing 4 changed files with 44 additions and 17 deletions.
diff --git a/Code/FormMapping/BKFormMapper.m b/Code/FormMapping/BKFormMapper.m
@@ -241,52 +241,52 @@ - (BKSimpleField *)cellWithAttributeMapping:(BKFormAttributeMapping *)attributeM
     BKFormAttributeMappingType type = attributeMapping.type;
 
     if (type == BKFormAttributeMappingTypeText) {
-        field = [BKTextField cellForTableView:self.tableView];
+        field = [_formMapping.textFieldClass cellForTableView:self.tableView];
         [[(BKTextField *)field textField] setDelegate:self];
         [[(BKTextField *)field textField] setFormAttributeMapping:attributeMapping];
 
     } else if (type == BKFormAttributeMappingTypeFloat) {
-        field = [BKFloatField cellForTableView:self.tableView];
+        field = [_formMapping.floatFieldClass cellForTableView:self.tableView];
 
     } else if (type == BKFormAttributeMappingTypeInteger) {
-        field = [BKIntegerField cellForTableView:self.tableView];
+        field = [_formMapping.integerFieldClass cellForTableView:self.tableView];
         [[(BKIntegerField *)field textField] setDelegate:self];
         [[(BKIntegerField *)field textField] setFormAttributeMapping:attributeMapping];
 
     } else if (type == BKFormAttributeMappingTypeLabel) {
-        field = [BKLabelField cellForTableView:self.tableView];
+        field = [_formMapping.labelFieldClass cellForTableView:self.tableView];
 
     } else if (type == BKFormAttributeMappingTypePassword) {
-        field = [BKPasswordTextField cellForTableView:self.tableView];
+        field = [_formMapping.passwordFieldClass cellForTableView:self.tableView];
         [[(BKPasswordTextField *)field textField] setDelegate:self];
         [[(BKPasswordTextField *)field textField] setFormAttributeMapping:attributeMapping];
 
     } else if (type == BKFormAttributeMappingTypeBoolean) {
-        field = [BKSwitchField cellForTableView:self.tableView];
+        field = [_formMapping.switchFieldClass cellForTableView:self.tableView];
 
     } else if (type == BKFormAttributeMappingTypeSaveButton) {
-        field = [BKSaveButtonField cellForTableView:self.tableView];
+        field = [_formMapping.saveButtonFieldClass cellForTableView:self.tableView];
 
     } else if (type == BKFormAttributeMappingTypeButton) {
-        field = [BKButtonField cellForTableView:self.tableView];
+        field = [_formMapping.buttonFieldClass cellForTableView:self.tableView];
 
     } else if (type == BKFormAttributeMappingTypeSelect ||
                type == BKFormAttributeMappingTypeTimePicker ||
                type == BKFormAttributeMappingTypeDatePicker ||
                type == BKFormAttributeMappingTypeDateTimePicker) {
-        field = [BKLabelField cellForTableView:self.tableView];
+        field = [_formMapping.labelFieldClass cellForTableView:self.tableView];
 
     } else if (type == BKFormAttributeMappingTypeBigText) {
-        field = [BKBigTextField cellForTableView:self.tableView];
+        field = [_formMapping.bigTextFieldClass cellForTableView:self.tableView];
 
     } else if (type == BKFormAttributeMappingTypeCustomCell) {
         field = [attributeMapping.customCell cellForTableView:self.tableView];
 
     } else if (type == BKFormAttributeMappingTypeSlider) {
-        field = [BKSliderField cellForTableView:self.tableView];
+        field = [_formMapping.sliderFieldClass cellForTableView:self.tableView];
 
     } else {
-        field = [BKLabelField cellForTableView:self.tableView];
+        field = [_formMapping.labelFieldClass cellForTableView:self.tableView];
     }
 
     return field;

diff --git a/Code/FormMapping/BKFormMapping.h b/Code/FormMapping/BKFormMapping.h
@@ -31,6 +31,16 @@
 @property (nonatomic, readonly) NSDictionary *sectionTitles;
 @property (nonatomic, retain) NSArray *fieldsOrder;
 @property (nonatomic, retain) BKFormAttributeMapping *saveAttribute;
+@property (nonatomic, assign) Class textFieldClass;
+@property (nonatomic, assign) Class floatFieldClass;
+@property (nonatomic, assign) Class integerFieldClass;
+@property (nonatomic, assign) Class labelFieldClass;
+@property (nonatomic, assign) Class passwordFieldClass;
+@property (nonatomic, assign) Class switchFieldClass;
+@property (nonatomic, assign) Class saveButtonFieldClass;
+@property (nonatomic, assign) Class bigTextFieldClass;
+@property (nonatomic, assign) Class sliderFieldClass;
+@property (nonatomic, assign) Class buttonFieldClass;
 
 - (id)initWithObjectClass:(Class)objectClass;
 

diff --git a/Code/FormMapping/BKFormMapping.m b/Code/FormMapping/BKFormMapping.m
@@ -19,6 +19,7 @@
 
 #import "BKFormAttributeMapping.h"
 #import "BKMacrosDefinitions.h"
+#import "BaseKitFormField.h"
 
 
 ////////////////////////////////////////////////////////////////////////////////////////////////////
@@ -40,30 +41,46 @@ @implementation BKFormMapping
 @synthesize objectClass = _objectClass;
 @synthesize fieldsOrder = _fieldsOrder;
 @synthesize saveAttribute = _saveAttribute;
+@synthesize textFieldClass = _textFieldClass;
+@synthesize floatFieldClass = _floatFieldClass;
+@synthesize integerFieldClass = _integerFieldClass;
+@synthesize labelFieldClass = _labelFieldClass;
+@synthesize passwordFieldClass = _passwordFieldClass;
+@synthesize switchFieldClass = _switchFieldClass;
+@synthesize saveButtonFieldClass = _saveButtonFieldClass;
+@synthesize bigTextFieldClass = _bigTextFieldClass;
+@synthesize sliderFieldClass = _sliderFieldClass;
+@synthesize buttonFieldClass = _buttonFieldClass;
 
 
 ////////////////////////////////////////////////////////////////////////////////////////////////////
 - (id)init {
     self = [super init];
-
     if (self) {
         _attributeMappings = [[NSMutableDictionary alloc] init];
         _sectionTitles = [[NSMutableDictionary alloc] init];
         _fieldsOrder = [[NSMutableArray alloc] init];
+        _textFieldClass = [BKTextField class];
+        _floatFieldClass = [BKFloatField class];
+        _integerFieldClass = [BKIntegerField class];
+        _labelFieldClass = [BKLabelField class];
+        _passwordFieldClass = [BKPasswordTextField class];
+        _switchFieldClass = [BKSwitchField class];
+        _saveButtonFieldClass = [BKSaveButtonField class];
+        _bigTextFieldClass = [BKBigTextField class];
+        _sliderFieldClass = [BKSliderField class];
+        _buttonFieldClass = [BKButtonField class];
     }
-
     return self;
 }
 
 
 ////////////////////////////////////////////////////////////////////////////////////////////////////
 - (id)initWithObjectClass:(Class)objectClass {
     self = [self init];
-
     if (self) {
         self.objectClass = objectClass;
     }
-
     return self;
 }
 

diff --git a/Examples/FormMappingExample/FormMappingExample/FormTableViewExampleController.m b/Examples/FormMappingExample/FormMappingExample/FormTableViewExampleController.m
@@ -50,7 +50,7 @@ - (void)viewDidLoad {
 
     self.movie = movie;
 
-    [BKFormMapping mappingForClass:[Movie class] block:^(BKFormMapping *formMapping) {
+    [BKFormMapping mappingForClass:[Movie class] block:^(BKFormMapping *formMapping) {        
         [formMapping sectiontTitle:@"Information section" identifier:@"info"];
         [formMapping mapAttribute:@"title" title:@"Title" type:BKFormAttributeMappingTypeText];
         [formMapping mapAttribute:@"releaseDate" title:@"ReleaseDate" type:BKFormAttributeMappingTypeDatePicker dateFormat:@"yyyy-MM-dd HH:mm:ss"];