* Change: processing IncludeButton data -> runtime method

tezpark · Jun 26, 2017 · 6a01bbb · 6a01bbb
1 parent c412139
commit 6a01bbb
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 2 deletions.
diff --git a/TezButton/TezButton.h b/TezButton/TezButton.h
@@ -331,5 +331,6 @@
 #pragma mark - Include data Button
 /** Incluging data button */
 @interface IncludeDataButton : UIButton
-@property (nonatomic, strong) NSDictionary* info;
+- (void)setData:(id)data;
+- (id)data;
 @end
diff --git a/TezButton/TezButton.m b/TezButton/TezButton.m
@@ -6,11 +6,13 @@
 //
 
 #import "TezButton.h"
+#import <objc/runtime.h>
 
 static const double DefaultMagnification = 1;
 static const CGFloat DefaultFontSize = 13;
 static const CGFloat DefaultCornerRadius = 8;
 static const UIRectCorner DefaultCornerOption = UIRectCornerAllCorners;
+static void *DataKey = "DataKey";
 
 #pragma mark - TezButton
 @interface TezButton()
@@ -622,4 +624,13 @@ - (void) drawRect:(CGRect)rect {
 
 #pragma mark - Include data Button
 @implementation IncludeDataButton : UIButton
+- (void)setData:(id)data {
+    objc_setAssociatedObject(self, DataKey, data, OBJC_ASSOCIATION_RETAIN);
+}
+
+- (id)data {
+    id data = objc_getAssociatedObject(self, DataKey);
+
+    return data;
+}
 @end
diff --git a/TezButtonExample/ViewController.m b/TezButtonExample/ViewController.m
@@ -142,7 +142,7 @@ - (void)makeButtons {
 
     // Include Data option
     IncludeDataButton* includeDataBtn = [[IncludeDataButton alloc] initWithFrame:CGRectMake(originX, originY, self.view.frame.size.width/2 - 20*1.5, 60)];
-    [includeDataBtn setInfo:[NSDictionary new]];
+    [includeDataBtn setData:[NSDictionary new]];
     [includeDataBtn setBackgroundColor:colorPink];
     [includeDataBtn setTitleColor:colorWhite forState:UIControlStateNormal];
     [includeDataBtn setTitle:@"Include data btn" forState:UIControlStateNormal];