NSTimer+Blocks

NSTimer+Blocks

typedef void (^SGBlock)(); 

@interface NSTimer (Blocks)

+ (id)scheduledTimerWithTimeInterval:(NSTimeInterval)interval repeats:(BOOL)shouldRepeat block:(SGBlock)block;
+ (id)timerWithTimeInterval:(NSTimeInterval)interval repeats:(BOOL)shouldRepeat block:(SGBlock)block;

@end



@implementation NSTimer (Blocks)

+ (id)scheduledTimerWithTimeInterval:(NSTimeInterval)theInterval repeats:(BOOL)shouldRepeat block:(SGBlock)theBlock
{
  SGBlock block = [theBlock copy];
  id ret = [self scheduledTimerWithTimeInterval:theInterval target:self selector:@selector(sg_fireBlock:) userInfo:block repeats:shouldRepeat];
  [block release];
  return ret;
}

+ (id)timerWithTimeInterval:(NSTimeInterval)theInterval repeats:(BOOL)shouldRepeat block:(SGBlock)theBlock
{
  SGBlock block = [theBlock copy];
  id ret = [self timerWithTimeInterval:theInterval target:self selector:@selector(sg_fireBlock:) userInfo:block repeats:shouldRepeat];
  [block release];
  return ret;
}

+ (void)sg_fireBlock:(NSTimer *)theTimer;
{
  if([theTimer userInfo])
  {
    SGBlock block = (SGBlock)[theTimer userInfo];
    block();
  }
}

@end