// .m文件

#import "DZDeleteButton.h"
#import "UIView+Extension.h" // 這個(gè)只是為了方便取寬高的一個(gè)分類，代碼就不貼了

@interface DZDeleteButton ()

// 是否抖動(dòng)
@property (nonatomic, assign, getter=isShaking) BOOL shaking;
// 右上角的按鈕，
@property (nonatomic, weak) UIImageView *iconBtn;
// 遮蓋，在抖動(dòng)時(shí)出現(xiàn)
@property (nonatomic, weak) UIView *coverView;
@end

@implementation DZDeleteButton

- (UIImageView *)iconBtn {
 if (!_iconBtn) {
  UIImageView *iconBtn = [[UIImageView alloc] initWithImage:[UIImage imageNamed:@"delete"]];
  iconBtn.userInteractionEnabled = YES;
  iconBtn.hidden = YES;
  _iconBtn = iconBtn;
  UITapGestureRecognizer *tap = [[UITapGestureRecognizer alloc] initWithTarget:self action:@selector(iconClick)];
  [iconBtn addGestureRecognizer:tap];
  [self addSubview:iconBtn];
 }
 return _iconBtn;
}

- (UIView *)coverView {
 if (!_coverView) {
  UIView *view = [[UIView alloc] init];
  view.backgroundColor = [UIColor clearColor];
  view.hidden = YES;
  UITapGestureRecognizer *tap = [[UITapGestureRecognizer alloc] initWithTarget:self action:@selector(coverClick)];
  [view addGestureRecognizer:tap];
  [self addSubview:view];
  _coverView = view;
 }
 return _coverView;
}

- (instancetype)initWithFrame:(CGRect)frame
{
 self = [super initWithFrame:frame];
 if (self) {
  [self addLongPressGestureRecognizer];
 }
 return self;
}

- (instancetype)init
{
 self = [super init];
 if (self) {
  [self addLongPressGestureRecognizer];
 }
 return self;
}

- (instancetype)initWithCoder:(NSCoder *)coder
{
 self = [super initWithCoder:coder];
 if (self) {
  [self addLongPressGestureRecognizer];
 }
 return self;
}

// 添加長按手勢
- (void)addLongPressGestureRecognizer {
 UILongPressGestureRecognizer *longPress = [[UILongPressGestureRecognizer alloc] initWithTarget:self action:@selector(longClick)];
 [self addGestureRecognizer:longPress];
}

- (void)delete {
 [self.iconBtn.superview removeFromSuperview];
}

// 是否執(zhí)行動(dòng)畫
- (void)setShaking:(BOOL)shaking {
 if (shaking) {
  [self shakingAnimation];
  self.coverView.hidden = NO;
  self.iconBtn.hidden = NO;
 } else {
  [self.layer removeAllAnimations];
  self.coverView.hidden = YES;
  self.iconBtn.hidden = YES;
 }
}

#pragma mark - 抖動(dòng)動(dòng)畫
#define Angle2Radian(angle) ((angle) / 180.0 * M_PI)
- (void)shakingAnimation {
 CAKeyframeAnimation *anim = [CAKeyframeAnimation animation];
 anim.keyPath = @"transform.rotation";

 anim.values = @[@(Angle2Radian(-5)), @(Angle2Radian(5)), @(Angle2Radian(-5))];
 anim.duration = 0.25;

 // 動(dòng)畫次數(shù)設(shè)置為最大
 anim.repeatCount = MAXFLOAT;
 // 保持動(dòng)畫執(zhí)行完畢后的狀態(tài)
 anim.removedOnCompletion = NO;
 anim.fillMode = kCAFillModeForwards;

 [self.layer addAnimation:anim forKey:@"shake"];
}

- (void)longClick {
 if (self.shaking) return;
 self.shaking = YES;
}

// 點(diǎn)擊右上角按鈕
- (void)iconClick {
 [self removeFromSuperview];
// 設(shè)計(jì)一個(gè)代理，為了在自己被刪除后做一些事情(例如，對頁面進(jìn)行布局)
 if ([self.delegate respondsToSelector:@selector(deleteButtonRemoveSelf:)]) {
  [self.delegate deleteButtonRemoveSelf:self];
 }
}

- (void)coverClick {
 self.shaking = NO;
}

- (void)layoutSubviews {
 [super layoutSubviews];

 // 調(diào)整位置
 self.imageView.x = 0;
 self.imageView.y = 0;
 self.imageView.width = self.width;
 self.imageView.height = self.width;

 self.titleLabel.x = 0;
 self.titleLabel.width = self.width;
 if (self.width >= self.height) {
  self.titleLabel.height = 20;
  self.titleLabel.y = self.height - self.titleLabel.height;
 } else {
  self.titleLabel.y = self.imageView.height;
  self.titleLabel.height = self.height - self.titleLabel.y;
 }

 self.titleLabel.textAlignment = NSTextAlignmentCenter;
 self.iconBtn.size = CGSizeMake(self.width * 0.3, self.width * 0.3);
 self.iconBtn.x = self.width - self.iconBtn.width;
 self.iconBtn.y = 0;

 self.coverView.frame = self.bounds;
 [self bringSubviewToFront:self.iconBtn];
}

@end

// .h文件 只有一個(gè)代理
#import <UIKit/UIKit.h>

@class DZDeleteButton;
@protocol DZDeleteButtonDelegate <NSObject>
@optional
- (void)deleteButtonRemoveSelf:(DZDeleteButton *)button;
@end

@interface DZDeleteButton : UIButton
@property (nonatomic, weak) id<DZDeleteButtonDelegate> delegate;

@end

- (void)viewDidLoad {
 [super viewDidLoad];

 DZDeleteButton *button = [[DZDeleteButton alloc] init];
 [button setImage:[UIImage imageNamed:@"bj"] forState:UIControlStateNormal];
 [button setTitle:@"百思" forState:UIControlStateNormal];
 button.delegate = self;
 button.frame = CGRectMake(20, 20, 60, 80);
 [button setTitleColor:[UIColor redColor] forState:UIControlStateNormal];
 [button addTarget:self action:@selector(btnClick) forControlEvents:UIControlEventTouchUpInside];
 [self.view addSubview:button];
}

- (void)btnClick {
 NSLog(@"點(diǎn)擊button");
}

- (void)deleteButtonRemoveSelf:(DZDeleteButton *)button {
 NSLog(@"已經(jīng)刪除，要做什么事");
}