// 继承扩展TAVWidgetContentView
@interface MYWidgetContentView : TAVWidgetContentView
@property (nonatomic, weak) TAVEditor *editor;

@end


@implementation TAVMagicWidgetContentView

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

- (void)setupUI {
    // 自定义UI
}
@end


// 继承扩展TAVStickerView
@interface MyStickerView : TAVStickerView
// 添加一个编辑按钮
@property (nonatomic, strong, readonly) UIButton *editButton;


@end

@implementation TAVMagicStickerView

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

- (void)setupUI 
{
    _editButton = [UIButton buttonWithType:UIButtonTypeCustom];
    _editButton.imageView.contentMode = UIViewContentModeScaleAspectFit;
    [_editButton setImage:[UIImage imageWithName:@"icon_pip_edit"] forState:UIControlStateNormal];
    [_editButton addTarget:self action:@selector(didClickEditItem) forControlEvents:UIControlEventTouchUpInside];
    [self addSubview:_editButton];
}

// 重写接口 自定义实现拖动view时隐藏按钮
- (void)handlePanRecognizer:(UIPanGestureRecognizer *)recognizer {
    [super handlePanRecognizer:recognizer];
    [self handleRecognizer:recognizer];
}

- (void)handleRecognizer:(UIGestureRecognizer *)recognizer {
    switch (recognizer.state) {
        case UIGestureRecognizerStateBegan:
            self.editButtom.hidden = YES;
            break;
        case UIGestureRecognizerStateChanged:
            self.editButtom.hidden = YES;
            break;
        case UIGestureRecognizerStateCancelled:
        case UIGestureRecognizerStateEnded:
        case UIGestureRecognizerStateFailed:
        {
            self.editButtom.hidden = NO;
            break;
        }
        default:
            break;
    }
}
@end