Coding-iOS/Coding_iOS/Util/OC_Category/UIView+PressMenu.m at master · coding/Coding-iOS · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
//
//  UIView+PressMenu.m
//  Coding_iOS
//
//  Created by Ease on 15/6/9.
//  Copyright (c) 2015年 Coding. All rights reserved.
//


#import "UIView+PressMenu.h"
#import <objc/runtime.h>

@implementation UIView (PressMenu)

static const NSString *kPressMenuSelectorPrefix = @"easePressMenuClicked_";
static char PressMenuTitlesKey, PressMenuBlockKey, PressMenuGestureKey, MenuVCKey;

#pragma mark M
- (void)addPressMenuTitles:(NSArray *)menuTitles menuClickedBlock:(void(^)(NSInteger index, NSString *title))block{
    self.userInteractionEnabled = YES;
    self.menuClickedBlock = block;
    self.menuTitles = menuTitles;
    if (self.pressGR == nil) {
        self.pressGR = [[UILongPressGestureRecognizer alloc] initWithTarget:self action:@selector(handlePress:)];
    }
    [self addGestureRecognizer:self.pressGR];
}

- (void)showMenuTitles:(NSArray *)menuTitles menuClickedBlock:(void(^)(NSInteger index, NSString *title))block{
    self.menuClickedBlock = block;
    self.menuTitles = menuTitles;
    [self p_showMenu];
}

- (BOOL)isMenuVCVisible{
    if (self.menuVC) {
        return [self.menuVC isMenuVisible];
    }
    return NO;
}

- (void)removePressMenu{
    if (self.menuVC) {
        [self.menuVC setMenuVisible:NO animated:YES];
        self.menuVC = nil;
    }
    if ([self.pressGR isKindOfClass:[UILongPressGestureRecognizer class]]) {
        [self removeGestureRecognizer:self.pressGR];
        self.pressGR = nil;
    }
    if (self.menuClickedBlock) {
        self.menuClickedBlock = nil;
    }
    if (self.menuTitles) {
        self.menuTitles = nil;
    }
}

#pragma mark SET_GET
- (void)setMenuTitles:(NSArray *)menuTitles{
    objc_setAssociatedObject(self, &PressMenuTitlesKey, menuTitles, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}
- (NSArray *)menuTitles{
    return objc_getAssociatedObject(self, &PressMenuTitlesKey);
}

- (void)setMenuClickedBlock:(void (^)(NSInteger, NSString *))menuClickedBlock{
    objc_setAssociatedObject(self, &PressMenuBlockKey, menuClickedBlock, OBJC_ASSOCIATION_COPY_NONATOMIC);
}
- (void (^)(NSInteger, NSString *))menuClickedBlock{
    return objc_getAssociatedObject(self, &PressMenuBlockKey);
}

- (void)setPressGR:(UILongPressGestureRecognizer *)pressGR{
    objc_setAssociatedObject(self, &PressMenuGestureKey, pressGR, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}
- (UILongPressGestureRecognizer *)pressGR{
    return objc_getAssociatedObject(self, &PressMenuGestureKey);
}

- (void)setMenuVC:(UIMenuController *)menuVC{
    objc_setAssociatedObject(self, &MenuVCKey, menuVC, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}
- (UIMenuController *)menuVC{
    return objc_getAssociatedObject(self, &MenuVCKey);
}

#pragma mark canPerformAction
- (BOOL)canBecomeFirstResponder{
    if (self.menuClickedBlock) {
        return YES;
    }else{
        return [super canBecomeFirstResponder];
    }
}

-(BOOL)canPerformAction:(SEL)action withSender:(id)sender{
    if (self.menuClickedBlock) {
        for (int i=0; i<self.menuTitles.count; i++) {
            if (action == NSSelectorFromString([NSString stringWithFormat:@"%@%d:", kPressMenuSelectorPrefix, i])) {
                return YES;
            }
        }
        return NO;
    }else{
        return [super canPerformAction:action withSender:sender];
    }
}

-(void)handlePress:(UIGestureRecognizer*)recognizer{
    if (recognizer.state == UIGestureRecognizerStateBegan) {
        [self p_showMenu];
    }
}

- (void)p_showMenu{
    [self becomeFirstResponder];
    NSMutableArray *menuItems = [[NSMutableArray alloc] initWithCapacity:self.menuTitles.count];
    Class cls = [self class];
    SEL imp = @selector(pressMenuClicked:);
    for (int i=0; i<self.menuTitles.count; i++) {
        NSString *title = [self.menuTitles objectAtIndex:i];
        //注册名添加方法sel，sel的具体实现在imp(pressMenuClicked:)
        SEL sel = sel_registerName([[NSString stringWithFormat:@"%@%d:", kPressMenuSelectorPrefix, i] UTF8String]);
        class_addMethod(cls, sel, [cls instanceMethodForSelector:imp], "v@");
        UIMenuItem *menuItem = [[UIMenuItem alloc] initWithTitle:title action:sel];
        [menuItems addObject:menuItem];
    }
    UIMenuController *menu = [UIMenuController sharedMenuController];
    [menu setMenuItems:menuItems];
    [menu setTargetRect:self.frame inView:self.superview];
    [menu setMenuVisible:YES animated:YES];
    self.menuVC = menu;
}

- (void)pressMenuClicked:(id)sender {
    NSString *selStr = NSStringFromSelector(_cmd);
    NSString *indexStr = [selStr substringFromIndex:kPressMenuSelectorPrefix.length];
    NSInteger index = indexStr.integerValue;
    if (index >=0 && index<self.menuTitles.count) {
        NSString *title = [self.menuTitles objectAtIndex:index];
        if (self.menuClickedBlock) {
            self.menuClickedBlock(index, title);
        }
    }
}

@end