#import "PhoneCodeButton.h"

@interface PhoneCodeButton ()

@property (nonatomic, strong, readwrite) NSTimer *timer;

@property (assign, nonatomic) NSTimeInterval durationToValidity;

@property (strong, nonatomic) UIView *lineView;

@implementation PhoneCodeButton

- (instancetype)initWithFrame:(CGRect)frame

{

self = [super initWithFrame:frame];

if (self) {

self.titleLabel.font = [UIFont systemFontOfSize:14];

self.enabled = YES;

_lineView = [[UIView alloc] initWithFrame:CGRectMake(-10, 5, 0.5, CGRectGetHeight(frame) - 2*5)];

_lineView.backgroundColor = [UIColor colorWithHexString:@"0xD8D8D8"];

[self addSubview:_lineView];

}

return self;

}

- (void)setEnabled:(BOOL)enabled{

[super setEnabled:enabled];

UIColor *foreColor = [UIColor colorWithHexString:enabled? @"0x3BBD79": @"0xCCCCCC"];

[self setTitleColor:foreColor forState:UIControlStateNormal];

if (enabled) {

[self setTitle:@"发送验证码" forState:UIControlStateNormal];

}else if ([self.titleLabel.text isEqualToString:@"发送验证码"]){

[self setTitle:@"正在发送..." forState:UIControlStateNormal];

}

}

- (void)startUpTimer{

_durationToValidity = 60;

if (self.isEnabled) {

self.enabled = NO;

}

[self setTitle:[NSString stringWithFormat:@"%.0f 秒", _durationToValidity] forState:UIControlStateNormal];

self.timer = [NSTimer scheduledTimerWithTimeInterval:1

target:self

selector:@selector(redrawTimer:)

userInfo:nil

repeats:YES];

}

- (void)invalidateTimer{

if (!self.isEnabled) {

self.enabled = YES;

}

[self.timer invalidate];

self.timer = nil;

}

- (void)redrawTimer:(NSTimer *)timer {

_durationToValidity--;

if (_durationToValidity > 0) {

self.titleLabel.text = [NSString stringWithFormat:@"%.0f 秒", _durationToValidity];//防止 button_title 闪烁

[self setTitle:[NSString stringWithFormat:@"%.0f 秒", _durationToValidity] forState:UIControlStateNormal];

}else{

[self invalidateTimer];

}

}