// Inherit and extend TAVWidgetContentView
@interface MYWidgetContentView : TAVWidgetContentView
@property (nonatomic, weak) TAVEditor *editor;

@end


@implementation MYWidgetContentView

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

- (void)setupUI {
    // Custom UI
}
@end


// Inherit and extend TAVStickerView
@interface MyStickerView : TAVStickerView
// Add an edit button
@property (nonatomic, strong, readonly) UIButton *editButton;


@end

@implementation MyStickerView

- (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 imagedName:@"icon_pip_edit"] forState:UIControlStateNormal];
    [_editButton addTarget:self action:@selector(didClickEditItem) forControlEvents:UIControlEventTouchUpInside];
    [self addSubview:_editButton];
}

// Override interface to customize hiding buttons when dragging view
- (void)handlePanRecognizer:(UIPanGestureRecognizer *)recognizer {
    [super handlePanRecognizer:recognizer];
    [self handleRecognizer:recognizer];
}

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