threema-ios/Vendor/JKLLockScreenViewController/JKLLockScreenPincodeView.m

// This file is based on third party code, see below for the original author
// and original license.
// Modifications are (c) by Threema GmbH and licensed under the AGPLv3.

// See Resources/License.html for original license

#import "JKLLockScreenPincodeView.h"
#import "KKKeychain.h"

static const NSUInteger LSPMaxPincodeLength = 6;

@interface JKLLockScreenPincodeView() {
    NSUInteger _wasCompleted;
}

@property (nonatomic, strong) NSString * pincode;

@end

@implementation JKLLockScreenPincodeView

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

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

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

- (void)lsp_initialize {
    
    _maxPincodeLength = LSPMaxPincodeLength;
    self.enabled = YES;
    [self initPincode];
}

- (void)initPincode {
    self.pincode = @"";
}

/**
 핀코드를 설정하는 프로퍼티 메소드
 @param NSString PIN code
 */
- (void)setPincode:(NSString *)pincode {
    
    if (_pincode != pincode) {
        _pincode = pincode;
        
        [self setNeedsDisplay];
        
        
        BOOL length = ([pincode length] == _maxPincodeLength);
        if (length && [_delegate respondsToSelector:@selector(lockScreenPincodeView:pincode:)]) {
            //------------------ Threema edit begin ---------------------------
            [NSTimer scheduledTimerWithTimeInterval:0.1 target:self selector:@selector(checkPinCode:) userInfo:pincode repeats:NO];
            //------------------ Threema edit end ---------------------------
        }
    }
}
//------------------ Threema edit begin ---------------------------
- (void)checkPinCode:(NSTimer*)timer {
    [_delegate lockScreenPincodeView:self pincode:timer.userInfo];
}
//------------------ Threema edit end ---------------------------

/**
 핀코드를 추가하는 메소드
 @param NSString PIN code
 */
- (void)appendingPincode:(NSString *)pincode {

    if (!_enabled) return;

    NSString * appended = [_pincode stringByAppendingString:pincode];

    // 최대 자릿수를 넘는다면 최대 자릿수만큼만 설정
    NSUInteger length = MIN([appended length], _maxPincodeLength);
    self.pincode = [appended substringToIndex:length];
}

/**
 마지막 핀코드를 제거하는 메소드
 */
- (void)removeLastPincode {
    
    if (!_enabled) return;
    
    NSUInteger index = ([_pincode length] - 1);
    if ([_pincode length] > index) {
        self.pincode = [_pincode substringToIndex:index];
    }
}

- (void)wasCompleted {
    for (NSUInteger idx = 0; idx < _maxPincodeLength; idx++) {
        dispatch_time_t delayInSeconds = dispatch_time(DISPATCH_TIME_NOW, (int64_t)(idx * 0.01f * NSEC_PER_SEC));
        dispatch_after(delayInSeconds, dispatch_get_main_queue(), ^(void){
            
            _wasCompleted++;
            [self setNeedsDisplay];
        });
    }
}

// Only override drawRect: if you perform custom drawing.
// An empty implementation adversely affects performance during animation.
- (void)drawRect:(CGRect)rect {
    // Drawing code
    [super drawRect:rect];
    
    [_pincodeColor setFill];
    
    // 1 character box size
    CGSize boxSize  = CGSizeMake(CGRectGetWidth(rect) / _maxPincodeLength, CGRectGetHeight(rect));
    CGSize charSize = CGSizeMake(self.frame.size.height/1.8, self.frame.size.height/1.8);
    
    CGFloat y = rect.origin.y;
    
    NSUInteger completed = MAX([_pincode length], _wasCompleted);
    
    // draw a circle : '.'
    NSInteger str = MIN(completed, _maxPincodeLength);
    for (NSUInteger idx = 0; idx < str; idx++) {
        CGFloat x = boxSize.width * idx;
        CGRect rounded = CGRectMake(x + floorf((boxSize.width  - charSize.width) / 2),
                                    y + floorf((boxSize.height - charSize.width) / 2),
                                    charSize.width,
                                    charSize.height);
        
        CGContextRef context = UIGraphicsGetCurrentContext();
        CGContextSetFillColorWithColor(context, _pincodeColor.CGColor);
        CGContextSetLineWidth(context, 1.0);
        CGContextFillEllipseInRect(context, rounded);
        CGContextFillPath(context);
    }
    
    // draw a dash : '-'
    for (NSUInteger idx = str; idx < _maxPincodeLength; idx++) {
        CGFloat x = boxSize.width * idx;
        CGRect rounded = CGRectMake(x + floorf((boxSize.width  - charSize.width)  / 2),
                                    y + floorf((boxSize.height - charSize.height) / 2),
                                    charSize.width,
                                    charSize.height);
        
        CGContextRef context = UIGraphicsGetCurrentContext();
        CGContextSetStrokeColorWithColor(context, _pincodeColor.CGColor);
        CGContextSetLineWidth(context, 1.0);
        CGContextAddEllipseInRect(context, rounded);
        CGContextStrokePath(context);
    }
}

@end