#import "PhoneCodeButton.h"

@interface PhoneCodeButton ()

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

@property (assign, nonatomic) NSTimeInterval durationToValidity;

@implementation PhoneCodeButton

- (instancetype)initWithFrame:(CGRect)frame

{

self = [super initWithFrame:frame];

if (self) {

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

self.enabled = YES;

}

return self;

}

- (void)setEnabled:(BOOL)enabled{

[super setEnabled:enabled];

UIColor *foreColor = enabled? kColorDark2: kColorDarkA;

[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];

}

}