#import "SWTableViewCell.h"

#import <UIKit/UIGestureRecognizerSubclass.h>

#import "SWUtilityButtonView.h"

static NSString * const kTableViewCellContentView = @"UITableViewCellContentView";

#pragma mark - SWUtilityButtonView

@interface SWTableViewCell () <UIScrollViewDelegate>

{

SWCellState _cellState; // The state of the cell within the scroll view, can be left, right or middle

CGFloat additionalRightPadding;

dispatch_once_t onceToken;

}

@property (nonatomic, strong) SWUtilityButtonView *scrollViewButtonViewLeft;

@property (nonatomic, strong) SWUtilityButtonView *scrollViewButtonViewRight;

@property (nonatomic, weak) UIView *scrollViewContentView;

@property (nonatomic) CGFloat height;

@property (nonatomic, strong) SWLongPressGestureRecognizer *longPressGestureRecognizer;

@property (nonatomic, strong) UITapGestureRecognizer *tapGestureRecognizer;

@implementation SWTableViewCell

#pragma mark Initializers

- (instancetype)initWithStyle:(UITableViewCellStyle)style reuseIdentifier:(NSString *)reuseIdentifier containingTableView:(UITableView *)containingTableView leftUtilityButtons:(NSArray *)leftUtilityButtons rightUtilityButtons:(NSArray *)rightUtilityButtons

{

self = [super initWithStyle:style reuseIdentifier:reuseIdentifier];

if (self)

{

self.height = containingTableView.rowHeight;

self.containingTableView = containingTableView;

self.highlighted = NO;

[self initializer];

self.rightUtilityButtons = rightUtilityButtons;

self.leftUtilityButtons = leftUtilityButtons;

}

return self;

}

- (instancetype)initWithCoder:(NSCoder *)aDecoder

{

self = [super initWithCoder:aDecoder];

if (self)

{

[self initializer];

}

return self;

}

- (instancetype)init

{

self = [super init];

if (self)

{

[self initializer];

}

return self;

}

- (instancetype)initWithStyle:(UITableViewCellStyle)style reuseIdentifier:(NSString *)reuseIdentifier

{

self = [super initWithStyle:style reuseIdentifier:reuseIdentifier];

if (self)

{

[self initializer];

}

return self;

}

- (void)initializer

{

// Check if the UITableView will display Indices on the right. If that's the case, add a padding

if([self.containingTableView.dataSource respondsToSelector:@selector(sectionIndexTitlesForTableView:)])

{

NSArray *indices = [self.containingTableView.dataSource sectionIndexTitlesForTableView:self.containingTableView];

additionalRightPadding = indices == nil ? 0 : kSectionIndexWidth;

}

// Set up scroll view that will host our cell content

SWCellScrollView *cellScrollView = [[SWCellScrollView alloc] initWithFrame:CGRectMake(0, 0, CGRectGetWidth(self.bounds), self.height)];

cellScrollView.delegate = self;

cellScrollView.showsHorizontalScrollIndicator = NO;

cellScrollView.scrollsToTop = NO;

cellScrollView.scrollEnabled = YES;

UITapGestureRecognizer *tapGestureRecognizer = [[UITapGestureRecognizer alloc] initWithTarget:self

action:@selector(scrollViewUp:)];

tapGestureRecognizer.cancelsTouchesInView = NO;

[cellScrollView addGestureRecognizer:tapGestureRecognizer];

self.tapGestureRecognizer = tapGestureRecognizer;

SWLongPressGestureRecognizer *longPressGestureRecognizer = [[SWLongPressGestureRecognizer alloc] initWithTarget:self

action:@selector(scrollViewPressed:)];

longPressGestureRecognizer.cancelsTouchesInView = NO;

longPressGestureRecognizer.minimumPressDuration = 0.1;

[cellScrollView addGestureRecognizer:longPressGestureRecognizer];

self.longPressGestureRecognizer = longPressGestureRecognizer;

self.cellScrollView = cellScrollView;

// Create the content view that will live in our scroll view

UIView *scrollViewContentView = [[UIView alloc] initWithFrame:CGRectMake([self leftUtilityButtonsWidth], 0, CGRectGetWidth(self.bounds), self.height)];

scrollViewContentView.backgroundColor = [UIColor whiteColor];

[self.cellScrollView addSubview:scrollViewContentView];

self.scrollViewContentView = scrollViewContentView;

// Add the cell scroll view to the cell

UIView *contentViewParent = self;

if (![NSStringFromClass([[self.subviews objectAtIndex:0] class]) isEqualToString:kTableViewCellContentView])

{

// iOS 7

contentViewParent = [self.subviews objectAtIndex:0];

}

NSArray *cellSubviews = [contentViewParent subviews];

[self insertSubview:cellScrollView atIndex:0];

for (UIView *subview in cellSubviews)

{

[self.scrollViewContentView addSubview:subview];

}

self.containingTableView.directionalLockEnabled = YES;

}

- (void)layoutSubviews

{

[super layoutSubviews];

self.cellScrollView.frame = CGRectMake(0, 0, CGRectGetWidth(self.bounds), self.height);

self.cellScrollView.contentSize = CGSizeMake(CGRectGetWidth(self.bounds) + [self utilityButtonsPadding], self.height);

self.cellScrollView.contentOffset = CGPointMake([self leftUtilityButtonsWidth], 0);

self.scrollViewButtonViewLeft.frame = CGRectMake([self leftUtilityButtonsWidth], 0, 0, self.height);

self.scrollViewButtonViewLeft.layer.masksToBounds = YES;

self.scrollViewButtonViewRight.frame = CGRectMake(CGRectGetWidth(self.bounds), 0, 0, self.height);

self.scrollViewButtonViewRight.layer.masksToBounds = YES;

self.scrollViewContentView.frame = CGRectMake([self leftUtilityButtonsWidth], 0, CGRectGetWidth(self.bounds), self.height);

self.cellScrollView.scrollEnabled = YES;

self.tapGestureRecognizer.enabled = YES;

}

#pragma mark - Properties

- (void)setLeftUtilityButtons:(NSArray *)leftUtilityButtons

{

_leftUtilityButtons = leftUtilityButtons;

SWUtilityButtonView *departingLeftButtons = self.scrollViewButtonViewLeft;

SWUtilityButtonView *scrollViewButtonViewLeft = [[SWUtilityButtonView alloc] initWithUtilityButtons:leftUtilityButtons

parentCell:self

utilityButtonSelector:@selector(leftUtilityButtonHandler:)];

self.scrollViewButtonViewLeft = scrollViewButtonViewLeft;

[scrollViewButtonViewLeft setFrame:CGRectMake([self leftUtilityButtonsWidth], 0, [self leftUtilityButtonsWidth], self.height)];

[self.cellScrollView insertSubview:scrollViewButtonViewLeft belowSubview:self.scrollViewContentView];

[departingLeftButtons removeFromSuperview];

[scrollViewButtonViewLeft populateUtilityButtons];

[self setNeedsLayout];

}

- (void)setRightUtilityButtons:(NSArray *)rightUtilityButtons

{

_rightUtilityButtons = rightUtilityButtons;

SWUtilityButtonView *departingLeftButtons = self.scrollViewButtonViewRight;

SWUtilityButtonView *scrollViewButtonViewRight = [[SWUtilityButtonView alloc] initWithUtilityButtons:rightUtilityButtons

parentCell:self

utilityButtonSelector:@selector(rightUtilityButtonHandler:)];

self.scrollViewButtonViewRight = scrollViewButtonViewRight;

[scrollViewButtonViewRight setFrame:CGRectMake(CGRectGetWidth(self.bounds), 0, [self rightUtilityButtonsWidth], self.height)];

[self.cellScrollView insertSubview:scrollViewButtonViewRight belowSubview:self.scrollViewContentView];

[departingLeftButtons removeFromSuperview];

[scrollViewButtonViewRight populateUtilityButtons];

[self setNeedsLayout];

}

#pragma mark Selection

- (void)scrollViewPressed:(id)sender

{

SWLongPressGestureRecognizer *longPressGestureRecognizer = (SWLongPressGestureRecognizer *)sender;

if (longPressGestureRecognizer.state == UIGestureRecognizerStateEnded)

{

// Gesture recognizer ended without failing so we select the cell

[self selectCell];

// Set back to deselected

[self setSelected:NO];

}

else

{

// Handle the highlighting of the cell

if (self.isHighlighted)

{

[self setHighlighted:NO];

}

else

{

[self highlightCell];

}

}

}

- (void)scrollViewUp:(id)sender

{

[self selectCellWithTimedHighlight];

}

- (void)selectCell

{

if (_cellState == kCellStateCenter)

{

if ([self.containingTableView.delegate respondsToSelector:@selector(tableView:didSelectRowAtIndexPath:)])

{

NSIndexPath *cellIndexPath = [self.containingTableView indexPathForCell:self];

[self.containingTableView selectRowAtIndexPath:cellIndexPath animated:NO scrollPosition:UITableViewScrollPositionNone];

[self.containingTableView.delegate tableView:self.containingTableView didSelectRowAtIndexPath:cellIndexPath];

[self.containingTableView deselectRowAtIndexPath:cellIndexPath animated:NO];

}

}

}

- (void)selectCellWithTimedHighlight

{

if(_cellState == kCellStateCenter)

{

// Selection

if ([self.containingTableView.delegate respondsToSelector:@selector(tableView:didSelectRowAtIndexPath:)])

{

NSIndexPath *cellIndexPath = [self.containingTableView indexPathForCell:self];

[self setSelected:YES];

[self.containingTableView selectRowAtIndexPath:cellIndexPath animated:NO scrollPosition:UITableViewScrollPositionNone];

[self.containingTableView.delegate tableView:self.containingTableView didSelectRowAtIndexPath:cellIndexPath];

[self.containingTableView deselectRowAtIndexPath:cellIndexPath animated:NO];

// Make the selection visible

NSTimer *endHighlightTimer = [NSTimer scheduledTimerWithTimeInterval:0.20

target:self

selector:@selector(timerEndCellHighlight:)

userInfo:nil

repeats:NO];

[[NSRunLoop currentRunLoop] addTimer:endHighlightTimer forMode:NSRunLoopCommonModes];

}

}

else

{

// Scroll back to center

[self hideUtilityButtonsAnimated:YES];

}

}

- (void)highlightCell

{

if (_cellState == kCellStateCenter)

{

[self setHighlighted:YES];

}

}

- (void)timerEndCellHighlight:(id)sender

{

[self setSelected:NO];

}

#pragma mark UITableViewCell overrides

- (void)setBackgroundColor:(UIColor *)backgroundColor

{

[super setBackgroundColor:backgroundColor];

_scrollViewContentView.backgroundColor = backgroundColor;

}

- (void)setHighlighted:(BOOL)highlighted

{

[super setHighlighted:highlighted animated:NO];

self.scrollViewButtonViewLeft.hidden = highlighted;

self.scrollViewButtonViewRight.hidden = highlighted;

}

- (void)setHighlighted:(BOOL)highlighted animated:(BOOL)animated

{

[super setHighlighted:highlighted animated:animated];

self.scrollViewButtonViewLeft.hidden = highlighted;

self.scrollViewButtonViewRight.hidden = highlighted;

}

- (void)setSelected:(BOOL)selected

{

[self setHighlighted:selected];

}

- (void)setSelected:(BOOL)selected animated:(BOOL)animated

{

[self setHighlighted:selected];

}

#pragma mark Height methods

- (void)setCellHeight:(CGFloat)height

{

_height = height;

// update the utility button height

[self.scrollViewButtonViewLeft setHeight:height];

[self.scrollViewButtonViewRight setHeight:height];

[self layoutSubviews];

}

#pragma mark - Utility buttons handling

- (void)rightUtilityButtonHandler:(id)sender

{

SWUtilityButtonTapGestureRecognizer *utilityButtonTapGestureRecognizer = (SWUtilityButtonTapGestureRecognizer *)sender;

NSUInteger utilityButtonIndex = utilityButtonTapGestureRecognizer.buttonIndex;

if ([self.delegate respondsToSelector:@selector(swipeableTableViewCell:didTriggerRightUtilityButtonWithIndex:)])

{

[self.delegate swipeableTableViewCell:self didTriggerRightUtilityButtonWithIndex:utilityButtonIndex];

}

}

- (void)leftUtilityButtonHandler:(id)sender

{

SWUtilityButtonTapGestureRecognizer *utilityButtonTapGestureRecognizer = (SWUtilityButtonTapGestureRecognizer *)sender;

NSUInteger utilityButtonIndex = utilityButtonTapGestureRecognizer.buttonIndex;

if ([self.delegate respondsToSelector:@selector(swipeableTableViewCell:didTriggerLeftUtilityButtonWithIndex:)])

{

[self.delegate swipeableTableViewCell:self didTriggerLeftUtilityButtonWithIndex:utilityButtonIndex];

}

}

- (void)hideUtilityButtonsAnimated:(BOOL)animated

{

// Scroll back to center

// Force the scroll back to run on the main thread because of weird scroll view bugs

dispatch_async(dispatch_get_main_queue(), ^{

[self.cellScrollView setContentOffset:CGPointMake([self leftUtilityButtonsWidth], 0) animated:YES];

});

_cellState = kCellStateCenter;

if ([self.delegate respondsToSelector:@selector(swipeableTableViewCell:scrollingToState:)])

{

[self.delegate swipeableTableViewCell:self scrollingToState:kCellStateCenter];

}

}

#pragma mark - Setup helpers

- (CGFloat)leftUtilityButtonsWidth

{

return [self.scrollViewButtonViewLeft utilityButtonsWidth];

}

- (CGFloat)rightUtilityButtonsWidth

{

return [self.scrollViewButtonViewRight utilityButtonsWidth] + additionalRightPadding;

}

- (CGFloat)utilityButtonsPadding

{

return [self leftUtilityButtonsWidth] + [self rightUtilityButtonsWidth];

}

- (CGPoint)scrollViewContentOffset

{

return CGPointMake([self.scrollViewButtonViewLeft utilityButtonsWidth], 0);

}

- (void) setAppearanceWithBlock:(void (^)())appearanceBlock force:(BOOL)force

{

if (force)

{

appearanceBlock();

}

else

{

dispatch_once(&onceToken, ^{

appearanceBlock();

});

}

}

#pragma mark UIScrollView helpers

- (void)scrollToRight:(inout CGPoint *)targetContentOffset

{

targetContentOffset->x = [self utilityButtonsPadding];

_cellState = kCellStateRight;

self.longPressGestureRecognizer.enabled = NO;

self.tapGestureRecognizer.enabled = NO;

if ([self.delegate respondsToSelector:@selector(swipeableTableViewCell:scrollingToState:)])

{

[self.delegate swipeableTableViewCell:self scrollingToState:kCellStateRight];

}

if ([self.delegate respondsToSelector:@selector(swipeableTableViewCellShouldHideUtilityButtonsOnSwipe:)])

{

for (SWTableViewCell *cell in [self.containingTableView visibleCells]) {

if (cell != self && [cell isKindOfClass:[SWTableViewCell class]] && [self.delegate swipeableTableViewCellShouldHideUtilityButtonsOnSwipe:cell]) {

[cell hideUtilityButtonsAnimated:YES];

}

}

}

}

- (void)scrollToCenter:(inout CGPoint *)targetContentOffset

{

targetContentOffset->x = [self leftUtilityButtonsWidth];

_cellState = kCellStateCenter;

self.longPressGestureRecognizer.enabled = YES;

self.tapGestureRecognizer.enabled = NO;

if ([self.delegate respondsToSelector:@selector(swipeableTableViewCell:scrollingToState:)])

{

[self.delegate swipeableTableViewCell:self scrollingToState:kCellStateCenter];

}

}

- (void)scrollToLeft:(inout CGPoint *)targetContentOffset

{

targetContentOffset->x = 0;

_cellState = kCellStateLeft;

self.longPressGestureRecognizer.enabled = NO;

self.tapGestureRecognizer.enabled = NO;

if ([self.delegate respondsToSelector:@selector(swipeableTableViewCell:scrollingToState:)])

{

[self.delegate swipeableTableViewCell:self scrollingToState:kCellStateLeft];

}

if ([self.delegate respondsToSelector:@selector(swipeableTableViewCellShouldHideUtilityButtonsOnSwipe:)])

{

for (SWTableViewCell *cell in [self.containingTableView visibleCells]) {

if (cell != self && [cell isKindOfClass:[SWTableViewCell class]] && [self.delegate swipeableTableViewCellShouldHideUtilityButtonsOnSwipe:cell]) {

[cell hideUtilityButtonsAnimated:YES];

}

}

}

}

#pragma mark UIScrollViewDelegate

- (void)scrollViewWillEndDragging:(UIScrollView *)scrollView withVelocity:(CGPoint)velocity targetContentOffset:(inout CGPoint *)targetContentOffset

{

switch (_cellState)

{

case kCellStateCenter:

if (velocity.x >= 0.5f)

{

[self scrollToRight:targetContentOffset];

}

else if (velocity.x <= -0.5f)

{

[self scrollToLeft:targetContentOffset];

}

else

{

CGFloat rightThreshold = [self utilityButtonsPadding] - ([self rightUtilityButtonsWidth] / 2);

CGFloat leftThreshold = [self leftUtilityButtonsWidth] / 2;

if (targetContentOffset->x > rightThreshold)

[self scrollToRight:targetContentOffset];

else if (targetContentOffset->x < leftThreshold)

[self scrollToLeft:targetContentOffset];

else

[self scrollToCenter:targetContentOffset];

}

break;

case kCellStateLeft:

if (velocity.x >= 0.5f)

{

[self scrollToCenter:targetContentOffset];

}

else if (velocity.x <= -0.5f)

{

// No-op

}

else

{

if (targetContentOffset->x >= ([self utilityButtonsPadding] - [self rightUtilityButtonsWidth] / 2))

[self scrollToRight:targetContentOffset];

else if (targetContentOffset->x > [self leftUtilityButtonsWidth] / 2)

[self scrollToCenter:targetContentOffset];

else

[self scrollToLeft:targetContentOffset];

}

break;

case kCellStateRight:

if (velocity.x >= 0.5f)

{

// No-op

}

else if (velocity.x <= -0.5f)

{

[self scrollToCenter:targetContentOffset];

}

else

{

if (targetContentOffset->x <= [self leftUtilityButtonsWidth] / 2)

[self scrollToLeft:targetContentOffset];

else if (targetContentOffset->x < ([self utilityButtonsPadding] - [self rightUtilityButtonsWidth] / 2))

[self scrollToCenter:targetContentOffset];

else

[self scrollToRight:targetContentOffset];

}

break;

default:

break;

}

}

- (void)scrollViewDidScroll:(UIScrollView *)scrollView

{

self.tapGestureRecognizer.enabled = NO;

if (scrollView.contentOffset.x > [self leftUtilityButtonsWidth])

{

if ([self rightUtilityButtonsWidth] > 0)

{

if (self.delegate && [self.delegate respondsToSelector:@selector(swipeableTableViewCell:canSwipeToState:)])

{

scrollView.scrollEnabled = [self.delegate swipeableTableViewCell:self canSwipeToState:kCellStateRight];

if (!scrollView.scrollEnabled)

{

return;

}

}

CGFloat scrollViewWidth = MIN(scrollView.contentOffset.x - [self leftUtilityButtonsWidth], [self rightUtilityButtonsWidth]);

// Expose the right button view

self.scrollViewButtonViewRight.frame = CGRectMake(scrollView.contentOffset.x + (CGRectGetWidth(self.bounds) - scrollViewWidth), 0.0f, scrollViewWidth,self.height);

CGRect scrollViewBounds = self.scrollViewButtonViewRight.bounds;

scrollViewBounds.origin.x = MAX([self rightUtilityButtonsWidth] - scrollViewWidth, [self rightUtilityButtonsWidth] - scrollView.contentOffset.x);

self.scrollViewButtonViewRight.bounds = scrollViewBounds;

}

}

else

{

// Expose the left button view

if ([self leftUtilityButtonsWidth] > 0)

{

if (self.delegate && [self.delegate respondsToSelector:@selector(swipeableTableViewCell:canSwipeToState:)])

{

scrollView.scrollEnabled = [self.delegate swipeableTableViewCell:self canSwipeToState:kCellStateLeft];

if (!scrollView.scrollEnabled)

{

return;

}

}

CGFloat scrollViewWidth = MIN(scrollView.contentOffset.x - [self leftUtilityButtonsWidth], [self leftUtilityButtonsWidth]);

self.scrollViewButtonViewLeft.frame = CGRectMake([self leftUtilityButtonsWidth], 0.0f, scrollViewWidth, self.height);

}

}

}

- (void)scrollViewDidEndDecelerating:(UIScrollView *)scrollView

{

self.tapGestureRecognizer.enabled = YES;

}

- (void)scrollViewDidEndScrollingAnimation:(UIScrollView *)scrollView

{

// Called when setContentOffset in hideUtilityButtonsAnimated: is done

self.tapGestureRecognizer.enabled = YES;

if (_cellState == kCellStateCenter)

{

self.longPressGestureRecognizer.enabled = YES;

}

}

#pragma mark - Cleaning up

- (void)dealloc

{

self.cellScrollView.delegate = nil;

self.cellScrollView = nil;

}