115 lines
4.3 KiB
Objective-C
115 lines
4.3 KiB
Objective-C
#import "PhoneNumberDirectoryFilterManager.h"
|
|
|
|
#import "Environment.h"
|
|
#import "NotificationManifest.h"
|
|
#import "PreferencesUtil.h"
|
|
#import "SignalUtil.h"
|
|
#import "ThreadManager.h"
|
|
#import "Util.h"
|
|
|
|
#define MINUTE (60.0)
|
|
#define HOUR (MINUTE*60.0)
|
|
|
|
#define DIRECTORY_UPDATE_TIMEOUT_PERIOD (1.0*MINUTE)
|
|
#define DIRECTORY_UPDATE_RETRY_PERIOD (1.0*HOUR)
|
|
|
|
@implementation PhoneNumberDirectoryFilterManager {
|
|
@private TOCCancelTokenSource* currentUpdateLifetime;
|
|
}
|
|
|
|
-(id) init {
|
|
if (self = [super init]) {
|
|
phoneNumberDirectoryFilter = PhoneNumberDirectoryFilter.phoneNumberDirectoryFilterDefault;
|
|
}
|
|
return self;
|
|
}
|
|
-(void) startUntilCancelled:(TOCCancelToken*)cancelToken {
|
|
lifetimeToken = cancelToken;
|
|
|
|
phoneNumberDirectoryFilter = [Environment.preferences tryGetSavedPhoneNumberDirectory];
|
|
if (phoneNumberDirectoryFilter == nil) {
|
|
phoneNumberDirectoryFilter = PhoneNumberDirectoryFilter.phoneNumberDirectoryFilterDefault;
|
|
}
|
|
|
|
[self scheduleUpdate];
|
|
}
|
|
|
|
-(PhoneNumberDirectoryFilter*) getCurrentFilter {
|
|
@synchronized(self) {
|
|
return phoneNumberDirectoryFilter;
|
|
}
|
|
}
|
|
-(void)forceUpdate {
|
|
[self scheduleUpdateAt:NSDate.date];
|
|
}
|
|
-(void) scheduleUpdate {
|
|
return [self scheduleUpdateAt:self.getCurrentFilter.getExpirationDate];
|
|
}
|
|
-(void) scheduleUpdateAt:(NSDate*)date {
|
|
void(^doUpdate)(void) = ^{
|
|
[self update];
|
|
};
|
|
|
|
[currentUpdateLifetime cancel];
|
|
currentUpdateLifetime = [TOCCancelTokenSource new];
|
|
[lifetimeToken whenCancelledDo:^{ [currentUpdateLifetime cancel]; }];
|
|
[TimeUtil scheduleRun:doUpdate
|
|
at:date
|
|
onRunLoop:[ThreadManager normalLatencyThreadRunLoop]
|
|
unlessCancelled:currentUpdateLifetime.token];
|
|
}
|
|
|
|
-(TOCFuture*) asyncQueryCurrentDirectory {
|
|
TOCUntilOperation startAwaitDirectoryOperation = ^(TOCCancelToken* untilCancelledToken) {
|
|
HttpRequest* directoryRequest = [HttpRequest httpRequestForPhoneNumberDirectoryFilter];
|
|
|
|
TOCFuture* futureDirectoryResponse = [HttpManager asyncOkResponseFromMasterServer:directoryRequest
|
|
unlessCancelled:untilCancelledToken
|
|
andErrorHandler:Environment.errorNoter];
|
|
|
|
return [futureDirectoryResponse thenTry:^(HttpResponse* response) {
|
|
return [PhoneNumberDirectoryFilter phoneNumberDirectoryFilterFromHttpResponse:response];
|
|
}];
|
|
};
|
|
|
|
return [TOCFuture futureFromUntilOperation:[TOCFuture operationTry:startAwaitDirectoryOperation]
|
|
withOperationTimeout:DIRECTORY_UPDATE_TIMEOUT_PERIOD
|
|
until:lifetimeToken];
|
|
}
|
|
|
|
-(PhoneNumberDirectoryFilter*) sameDirectoryWithRetryTimeout {
|
|
BloomFilter* filter = [phoneNumberDirectoryFilter bloomFilter];
|
|
NSDate* retryDate = [NSDate dateWithTimeInterval:DIRECTORY_UPDATE_RETRY_PERIOD
|
|
sinceDate:[NSDate date]];
|
|
return [PhoneNumberDirectoryFilter phoneNumberDirectoryFilterWithBloomFilter:filter
|
|
andExpirationDate:retryDate];
|
|
}
|
|
-(void) signalDirectoryQueryFailed:(id)failure {
|
|
NSString* desc = [NSString stringWithFormat:@"Failed to retrieve directory. Retrying in %f hours.",
|
|
DIRECTORY_UPDATE_RETRY_PERIOD/HOUR];
|
|
Environment.errorNoter(desc, failure, false);
|
|
}
|
|
-(TOCFuture*) asyncQueryCurrentDirectoryWithDefaultOnFail {
|
|
TOCFuture* futureDirectory = [self asyncQueryCurrentDirectory];
|
|
|
|
return [futureDirectory catchTry:^PhoneNumberDirectoryFilter*(id error) {
|
|
[self signalDirectoryQueryFailed:error];
|
|
return [self sameDirectoryWithRetryTimeout];
|
|
}];
|
|
}
|
|
|
|
-(void) update {
|
|
TOCFuture* eventualDirectory = [self asyncQueryCurrentDirectoryWithDefaultOnFail];
|
|
|
|
[eventualDirectory thenDo:^(PhoneNumberDirectoryFilter* directory) {
|
|
@synchronized(self) {
|
|
phoneNumberDirectoryFilter = directory;
|
|
}
|
|
[Environment.preferences setSavedPhoneNumberDirectory:directory];
|
|
[[NSNotificationCenter defaultCenter] postNotificationName:NOTIFICATION_DIRECTORY_WAS_UPDATED object:nil];
|
|
[self scheduleUpdate];
|
|
}];
|
|
}
|
|
|
|
@end
|