/*
 Copyright (c) 2015, Apple Inc. All rights reserved.
 Copyright (c) 2015, Ricardo Sánchez-Sáez.
 
 Redistribution and use in source and binary forms, with or without modification,
 are permitted provided that the following conditions are met:
 
 1.  Redistributions of source code must retain the above copyright notice, this
 list of conditions and the following disclaimer.
 
 2.  Redistributions in binary form must reproduce the above copyright notice,
 this list of conditions and the following disclaimer in the documentation and/or
 other materials provided with the distribution.
 
 3.  Neither the name of the copyright holder(s) nor the names of any contributors
 may be used to endorse or promote products derived from this software without
 specific prior written permission. No license is granted to the trademarks of
 the copyright holders even if such marks are included in this software.
 
 THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE
 FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
 SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
 CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
 OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */


#import "ORKTextFieldView.h"

#import "ORKAccessibility.h"
#import "ORKSkin.h"


static NSString *const EmptyBulletString = @"\u25CB";
static NSString *const FilledBulletString = @"\u25CF";

@implementation ORKCaretOptionalTextField

- (UIView *)hitTest:(CGPoint)point withEvent:(UIEvent *)event {
    _hitClearButton = NO;
    if ([self allowsSelection]) {
        return [super hitTest:point withEvent:event];
    } else {
        // Make exception for clear button, which is hittable
        if ( CGRectContainsPoint([self clearButtonRectForBounds:self.bounds], point)) {
            UIView *hitView = [super hitTest:point withEvent:event];
            // Where we are using a picker for date, time interval, and choice
            // Turn on flag to avoid bring up the keyboard when the field is not active.
            _hitClearButton = [hitView isKindOfClass:[UIButton class]];
            return hitView;
        }
        return nil;
    }
}

- (CGRect)caretRectForPosition:(UITextPosition *)position {
    if (_allowsSelection) {
        return [super caretRectForPosition:position];
    } else {
        return CGRectZero;
    }
}

@end


@implementation ORKPasscodeTextField

- (instancetype)init {
    self = [super init];
    if (self) {
        self.font = [UIFont fontWithName:@"Courier" size:35.0];
        self.textAlignment = NSTextAlignmentCenter;
    }
    return self;
}

- (UIKeyboardType)keyboardType {
    return UIKeyboardTypeNumberPad;
}

- (BOOL)allowsSelection {
    return NO;
}

- (void)updateTextWithNumberOfFilledBullets:(NSInteger)filledBullets {
    // Error checking.
    if (filledBullets > self.numberOfDigits) {
        @throw [NSException exceptionWithName:NSInvalidArgumentException
                                       reason:@"The number of filled bullets cannot exceed the number of pin digits."
                                     userInfo:nil];
    }
    
    // Append the string with the correct number of filled and empty bullets.
    NSString *text = [NSString new];
    text = [text stringByPaddingToLength:filledBullets withString:FilledBulletString startingAtIndex:0];
    text = [text stringByPaddingToLength:self.numberOfDigits withString:EmptyBulletString startingAtIndex:0];
    
    // Apply spacing attribute to string.
    NSMutableAttributedString *attributedText = [[NSMutableAttributedString alloc] initWithString:text];
    [attributedText addAttribute:NSKernAttributeName
                             value:@(20.0f)
                             range:NSMakeRange(0, [text length]-1)];

    // Set the textfield's text property.
    self.attributedText = attributedText;
}

- (void)setNumberOfDigits:(NSInteger)numberOfDigits {
    _numberOfDigits = numberOfDigits;
    [self updateTextWithNumberOfFilledBullets:0];
}

#pragma mark - Accessibility

- (NSString *)accessibilityLabel {
    return ORKLocalizedString(@"PASSCODE_TEXTFIELD_ACCESSIBILITY_LABEL", nil);
}

- (NSString *)accessibilityValue {
    NSRegularExpression *regularExpression = [NSRegularExpression regularExpressionWithPattern:FilledBulletString options:NSRegularExpressionCaseInsensitive error:nil];
    NSUInteger numberOfFilledBullets = [regularExpression numberOfMatchesInString:self.text options:0 range:NSMakeRange(0, [self.text length])];
    return [NSString stringWithFormat:ORKLocalizedString(@"PASSCODE_TEXTFIELD_ACCESSIBILTIY_VALUE", nil), ORKLocalizedStringFromNumber(@(numberOfFilledBullets)), ORKLocalizedStringFromNumber(@([self.text length]))];
}

- (UIAccessibilityTraits)accessibilityTraits {
    return UIAccessibilityTraitNone;
}

@end


@implementation ORKUnitTextField {
    NSString *_managedPlaceholder;
    
    NSString *_unitWithNumber;
 
    UIColor *_unitRegularColor;
    UIColor *_unitActiveColor;
    
    UIColor *_savedSuffixColor;
    NSString *_savedSuffixText;
    
    UILabel *_suffixLabel;
}

- (instancetype)init {
    self = [super init];
    if (self) {
        [[NSNotificationCenter defaultCenter]  addObserver:self selector:@selector(textFieldTextDidBeginEditing:) name:UITextFieldTextDidBeginEditingNotification object:self];
        [[NSNotificationCenter defaultCenter]  addObserver:self selector:@selector(textFieldTextDidEndEditing:) name:UITextFieldTextDidEndEditingNotification object:self];
        [[NSNotificationCenter defaultCenter]  addObserver:self selector:@selector(textFieldTextDidChange:) name:UITextFieldTextDidChangeNotification object:self];
        
    }
    return self;
}

- (id)ork_createTextLabelWithTextColor:(UIColor *)textColor {
    UILabel *textLabel = [[UILabel alloc] initWithFrame:CGRectZero];
    textLabel.font = [UIFont systemFontOfSize:17];
    [textLabel setOpaque:NO];
    [textLabel setBackgroundColor:nil];
    textLabel.contentMode = UIViewContentModeRedraw;
    if (textColor != nil) {
        textLabel.textColor = textColor;
    }
    return textLabel;
}

- (void)ork_setSuffix:(NSString *)suffix withColor:(UIColor *)color {
    CGRect previousSuffixFrame = CGRectZero;
    if (_suffixLabel) {
        previousSuffixFrame = _suffixLabel.frame;
        [_suffixLabel removeFromSuperview];
        _suffixLabel = nil;
        [self setNeedsLayout];
    } else {
        previousSuffixFrame.size.height = CGRectGetHeight(self.bounds);
    }
    if (suffix.length == 0) {
        return;
    }

    _suffixLabel = [self ork_createTextLabelWithTextColor:(color ? : [UIColor placeholderTextColor])];
    _suffixLabel.text = suffix;
    _suffixLabel.font = self.font;
    _suffixLabel.textAlignment = NSTextAlignmentLeft;
    _suffixLabel.userInteractionEnabled = NO;
    _suffixLabel.lineBreakMode = NSLineBreakByTruncatingMiddle;
    _suffixLabel.frame = previousSuffixFrame;

    // re-layout to position the suffix
    [self setNeedsLayout];
}

- (void)ork_updateSuffix:(NSString *)suffix {
    if (!_suffixLabel) {
        [self ork_setSuffix:suffix withColor:nil];
    } else {
        _suffixLabel.text = suffix;
        [self setNeedsLayout];
    }
}

- (void)ork_updateSuffix:(NSString *)suffix withColor:(UIColor *)color {
    if (NO == [color isEqual:_savedSuffixColor]) {
        if (suffix != nil) {
            _savedSuffixColor = color;
        }
        _savedSuffixText = suffix;
        [self ork_setSuffix:suffix withColor:color];
        return;
    }
    
    if (NO == [suffix isEqualToString:_savedSuffixText]) {
        _savedSuffixText = suffix;
        [self ork_updateSuffix:suffix];
    }
}

- (void)setManageUnitAndPlaceholder:(BOOL)manageUnitAndPlaceholder {
    _manageUnitAndPlaceholder = manageUnitAndPlaceholder;
    [self updateManagedUnitAndPlaceholder];
}

- (void)setPlaceholder:(NSString *)placeholder {
    _managedPlaceholder = placeholder;
    [self updateManagedUnitAndPlaceholder];
}

- (void)ork_setPlaceholder:(NSString *)placeholder {
    [super setPlaceholder:placeholder];
}

- (void)setUnit:(NSString *)unit {
    _unit = unit;
    
    if (_unit.length > 0) {
        _unitWithNumber = [NSString stringWithFormat:@" %@", unit];
        _unitRegularColor = [UIColor placeholderTextColor];
        _unitActiveColor = [UIColor placeholderTextColor];
    } else {
        _unitWithNumber = nil;
    }
    
    [self updateManagedUnitAndPlaceholder];
}

- (void)updateManagedUnitAndPlaceholder {
    if (_manageUnitAndPlaceholder) {
        BOOL isEditing = self.isEditing;
        
        UIColor *suffixColor = isEditing ? _unitActiveColor : _unitRegularColor;
            if (_managedPlaceholder.length > 0) {
                [self ork_setPlaceholder:((isEditing && _unit.length > 0) ? nil : _managedPlaceholder)];
                if (!(_hideUnitWhenAnswerEmpty && !isEditing && self.text.length == 0)) {
                    [self ork_updateSuffix:_unitWithNumber withColor:suffixColor];
                } else {
                    [self ork_updateSuffix:nil withColor:suffixColor];
                }
            } else {
                if (self.text.length > 0 || isEditing) {
                    [self ork_setPlaceholder:nil];
                    [self ork_updateSuffix:_unitWithNumber withColor:suffixColor];
                } else {
                    if (!(_hideUnitWhenAnswerEmpty && !isEditing && self.text.length == 0)) {
                        [self ork_setPlaceholder:_unit];
                        [self ork_updateSuffix:nil withColor:suffixColor];
                    } else {
                        [self ork_setPlaceholder:nil];
                        [self ork_updateSuffix:nil withColor:suffixColor];
                    }
                }
        }
    } else {
        // remove unit string
        if (_savedSuffixText.length > 0) {
            [self ork_updateSuffix:nil withColor:nil];
        }
        // put back unit string
        if ([self.placeholder isEqualToString:_managedPlaceholder] == NO) {
            [self ork_setPlaceholder:_managedPlaceholder];
        }
    }
    [self invalidateIntrinsicContentSize];
    [self layoutIfNeeded]; // layout immediatly to avoid animation glitch in which the unit label frame grows from left to right
}

- (void)textFieldTextDidBeginEditing:(NSNotification *)notification {
    [self updateManagedUnitAndPlaceholder];
}

- (void)textFieldTextDidEndEditing:(NSNotification *)notification {
    [self updateManagedUnitAndPlaceholder];
    
}

- (void)textFieldTextDidChange:(NSNotification *)notification {
    [self updateManagedUnitAndPlaceholder];
}

- (void)setText:(NSString *)text {
    [super setText:text];
    [self updateManagedUnitAndPlaceholder];
}

- (BOOL)isPlaceholderVisible {
    BOOL isEditing = self.isEditing;
    return (self.placeholder.length > 0) &&
            ((!isEditing && self.text.length == 0) || (isEditing && self.text.length == 0 && _unit.length == 0));
}

- (CGFloat)suffixWidthForBounds:(CGRect)bounds {
    CGFloat suffixWidth = [_suffixLabel.text sizeWithAttributes:@{NSFontAttributeName: _suffixLabel.font}].width;
    suffixWidth = MIN(suffixWidth, bounds.size.width / 2);
    return suffixWidth;
}

static const UIEdgeInsets paddingGuess = (UIEdgeInsets){.left = 2, .right = 6};

- (CGRect)textRectForBounds:(CGRect)bounds {
    CGRect textRect = [super textRectForBounds:bounds];
    
    // Leave room for the suffix label
    if (_suffixLabel.text.length) {
        CGFloat suffixWidth = [self suffixWidthForBounds:bounds];
        if (suffixWidth > 0) {
            suffixWidth += paddingGuess.right;
        }
        textRect.size.width = MAX(0, textRect.size.width - suffixWidth);
    }
    return textRect;
}


- (CGRect)editingRectForBounds:(CGRect)bounds {
    CGRect rect = [super editingRectForBounds:bounds];
    
    // Leave room for the suffix label
    if (_suffixLabel.text.length) {
        CGFloat suffixWidth = [self suffixWidthForBounds:bounds];
        if (suffixWidth > 0) {
            suffixWidth += paddingGuess.right;
        }
        rect.size.width = MAX(0, rect.size.width - suffixWidth);
    }
    
    return rect;
}

- (CGRect)ork_suffixFrame {
    // Get the text currently 'in' the edit field
    NSString *textToMeasure = [self isPlaceholderVisible] ? self.placeholder : self.text;
    CGSize sizeOfText = [textToMeasure sizeWithAttributes:[self defaultTextAttributes]];
    
    // Get the maximum size of the actual editable area (taking into account prefix/suffix/views/clear button
    CGRect textFrame = [self textRectForBounds:self.bounds];
    
    // Work out the size of our suffix frame
    CGRect suffixFrame = [super placeholderRectForBounds:self.bounds];
    suffixFrame.size.width = [self suffixWidthForBounds:self.bounds];
    
    // Take padding into account
    CGFloat xMaximum = CGRectGetMaxX(textFrame);
    if (sizeOfText.width < (textFrame.size.width - (paddingGuess.left + paddingGuess.right))) {
        // Adjust the rectangle to include the padding
        textFrame.origin.x += paddingGuess.left;
        textFrame.size.width -= paddingGuess.left + paddingGuess.right;
    } else {
        // Cover the fringe case where the padding is not applied, but the field editor has not scrolled, so the prefix/suffix could
        // overlap the text slightly.
        sizeOfText.width += paddingGuess.left + paddingGuess.right;
    }
    
    // Calculate position for alignment
    CGFloat xOffset = CGRectGetMinX(textFrame) + sizeOfText.width;
    
    // Make sure it can't escape out the right of the view
    suffixFrame.origin.x = MIN(xOffset, xMaximum);
    
    suffixFrame.size.height = CGRectGetHeight(self.bounds);
    suffixFrame.origin.y = 0;
    
    return suffixFrame;
}

- (void)layoutSubviews {
    [super layoutSubviews];
    
    if (_suffixLabel) {
        [self addSubview:_suffixLabel];
        _suffixLabel.frame = [self ork_suffixFrame];
        [_suffixLabel setNeedsDisplay];
    }
}

- (void)dealloc {
    [[NSNotificationCenter defaultCenter]  removeObserver:self];
}

#pragma mark Accessibility

- (NSString *)accessibilityValue {
    if (self.text.length > 0) {
        return ORKAccessibilityStringForVariables([super accessibilityValue], _unitWithNumber);
    
    }
    else if ( _managedPlaceholder ) {
        return ORKAccessibilityStringForVariables(_managedPlaceholder, _unitWithNumber);
    }

    return [super accessibilityValue];
}

- (BOOL)accessibilityActivate
{
    return [self becomeFirstResponder];
}

@end


@implementation ORKTextFieldView {

    NSMutableArray<NSLayoutConstraint *> *_errorLabelConstraints;

}

- (instancetype)initWithFrame:(CGRect)frame {
    self = [super initWithFrame:frame];
    if (self) {
        _textField = [[ORKUnitTextField alloc] init];
        _textField.clearButtonMode = UITextFieldViewModeWhileEditing;
        _textField.translatesAutoresizingMaskIntoConstraints = NO;
        
        [self addSubview:_textField];
        [self setUpConstraints];
    }
    return self;
}

- (void)setUpConstraints {
    NSMutableArray *constraints = [NSMutableArray new];
    
    NSDictionary *views = NSDictionaryOfVariableBindings(_textField);
    
    [constraints addObjectsFromArray:[NSLayoutConstraint constraintsWithVisualFormat:@"H:|[_textField]|"
                                                                             options:NSLayoutFormatDirectionLeadingToTrailing
                                                                             metrics:nil
                                                                               views:views]];
    
    [constraints addObjectsFromArray:[NSLayoutConstraint constraintsWithVisualFormat:@"V:|[_textField]|"
                                                                             options:NSLayoutFormatDirectionLeadingToTrailing
                                                                             metrics:nil
                                                                               views:views]];
    
    NSLayoutConstraint *bottomConstraint = [NSLayoutConstraint constraintWithItem:self
                                                             attribute:NSLayoutAttributeBottom
                                                             relatedBy:NSLayoutRelationEqual
                                                                toItem:_textField
                                                             attribute:NSLayoutAttributeBottom
                                                            multiplier:1.0
                                                              constant:0.0];
    
    // Ask to fill the available horizontal space
    NSLayoutConstraint *widthConstraint = [NSLayoutConstraint constraintWithItem:_textField
                                                                       attribute:NSLayoutAttributeWidth
                                                                       relatedBy:NSLayoutRelationEqual
                                                                          toItem:nil
                                                                       attribute:NSLayoutAttributeNotAnAttribute
                                                                      multiplier:1.0
                                                                        constant:ORKScreenMetricMaxDimension];

    widthConstraint.priority = UILayoutPriorityDefaultLow;
    [constraints addObject:bottomConstraint];
    [constraints addObject:widthConstraint];
    
    [NSLayoutConstraint activateConstraints:constraints];
}

- (CGFloat)estimatedWidth {
    
    NSString *placeholderAndUnit = self.textField.placeholder;
    NSString *textAndUnit = self.textField.text;
    
    if (self.textField.unit.length > 0) {
        NSString *unitString = [NSString stringWithFormat:@"  %@", self.textField.unit];
        placeholderAndUnit = [placeholderAndUnit stringByAppendingString:unitString];
        textAndUnit = [textAndUnit stringByAppendingString:unitString];
    }
    
    NSDictionary *attributes = @{ NSFontAttributeName : self.textField.font };
    CGFloat fieldWidth = MAX([placeholderAndUnit sizeWithAttributes:attributes].width,
                             [textAndUnit sizeWithAttributes:attributes].width);
    
    return fieldWidth;
}

- (void)setHideUnitWhenAnswerEmpty:(BOOL)hideUnitWhenAnswerEmpty {
    _textField.hideUnitWhenAnswerEmpty = hideUnitWhenAnswerEmpty;
}

- (BOOL)hideUnitWhenAnswerEmpty {
    return _textField.hideUnitWhenAnswerEmpty;
}

@end