X-Git-Url: http://git.ithinksw.org/MenuTunes.git/blobdiff_plain/97eb97cd6af1e3019980ed380a6cbaf1feae0abe..276579746a9cbe94b358155d6eb1aff0b07760a0:/AudioscrobblerController.m?ds=sidebyside diff --git a/AudioscrobblerController.m b/AudioscrobblerController.m index 695c684..29c8ecf 100644 --- a/AudioscrobblerController.m +++ b/AudioscrobblerController.m @@ -12,15 +12,25 @@ */ #import "AudioscrobblerController.h" +#import "PreferencesController.h" +#import +#import + +#define AUDIOSCROBBLER_ID @"tst" +#define AUDIOSCROBBLER_VERSION [[[NSBundle mainBundle] infoDictionary] objectForKey:@"CFBundleVersion"] static AudioscrobblerController *_sharedController = nil; @implementation AudioscrobblerController -+ (void)load +/*+ (void)load { - //[[AudioscrobblerController sharedController] attemptHandshake]; -} + NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init]; + [[AudioscrobblerController sharedController] submitTrack:@"Immigrant Song" artist:@"Led Zeppelin" album:@"How The West Was Won" length:221]; + [[AudioscrobblerController sharedController] submitTrack:@"Comfortably Numb" artist:@"Pink Floyd" album:@"The Wall" length:384]; + [[AudioscrobblerController sharedController] submitTracks]; + [pool release]; +}*/ + (AudioscrobblerController *)sharedController { @@ -34,27 +44,60 @@ static AudioscrobblerController *_sharedController = nil; { if ( (self = [super init]) ) { _handshakeCompleted = NO; + _md5Challenge = nil; + _postURL = nil; + + /*_handshakeCompleted = YES; + _md5Challenge = @"rawr"; + _postURL = [NSURL URLWithString:@"http://audioscrobbler.com/"];*/ + + _delayDate = [[NSDate date] retain]; _responseData = nil; + _tracks = [[NSMutableArray alloc] init]; + _submitTracks = [[NSMutableArray alloc] init]; + [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(handleAudioscrobblerNotification:) name:nil object:self]; } return self; } - (void)dealloc { + [_md5Challenge release]; + [_postURL release]; [_responseData release]; + [_submitTracks release]; + [_tracks release]; + [_delayDate release]; [super dealloc]; } - (void)attemptHandshake { - NSString *version = [[[NSBundle bundleWithPath:[[NSWorkspace sharedWorkspace] fullPathForApplication:@"iTunes.app"]] infoDictionary] objectForKey:@"CFBundleVersion"], *user = @"Tristrex"; - NSURL *url = [NSURL URLWithString:[NSString stringWithFormat:@"http://post.audioscrobbler.com/?hs=true&p=1.1&c=est&v=\"%@\"&u=\"%@\"", version, user]]; - NSURLConnection *connection; + [self attemptHandshake:NO]; +} + +- (void)attemptHandshake:(BOOL)force +{ + if (_handshakeCompleted && !force) { + return; + } - _responseData = [[NSMutableData alloc] init]; - connection = [NSURLConnection connectionWithRequest:[NSURLRequest requestWithURL:url] delegate:self]; + //Delay if we haven't met the interval time limit + NSTimeInterval interval = [_delayDate timeIntervalSinceNow]; + if (interval > 0) { + ITDebugLog(@"Audioscrobbler: Delaying handshake attempt for %i seconds", interval); + [self performSelector:@selector(attemptHandshake) withObject:nil afterDelay:interval + 1]; + return; + } - _handshakeCompleted = YES; + NSString *user = [[NSUserDefaults standardUserDefaults] stringForKey:@"audioscrobblerUser"]; + if (!_handshakeCompleted && user) { + NSURL *url = [NSURL URLWithString:[NSString stringWithFormat:@"http://post.audioscrobbler.com/?hs=true&p=1.1&c=%@&v=%@&u=%@", AUDIOSCROBBLER_ID, AUDIOSCROBBLER_VERSION, user]]; + + _currentStatus = AudioscrobblerRequestingHandshakeStatus; + _responseData = [[NSMutableData alloc] init]; + [NSURLConnection connectionWithRequest:[NSURLRequest requestWithURL:url cachePolicy:NSURLRequestReloadIgnoringCacheData timeoutInterval:15] delegate:self]; + } } - (BOOL)handshakeCompleted @@ -62,11 +105,126 @@ static AudioscrobblerController *_sharedController = nil; return _handshakeCompleted; } +- (void)submitTrack:(NSString *)title artist:(NSString *)artist album:(NSString *)album length:(int)length +{ + ITDebugLog(@"Audioscrobbler: Adding a new track to the submission queue."); + NSDictionary *newTrack = [NSDictionary dictionaryWithObjectsAndKeys:title, + @"title", + artist, + @"artist", + (album == nil) ? @"" : album, + @"album", + [NSString stringWithFormat:@"%i", length], + @"length", + [[NSDate date] descriptionWithCalendarFormat:@"%Y-%m-%d %H:%M:%S" timeZone:nil locale:nil], + @"time", + nil, nil]; + [_tracks addObject:newTrack]; + [self submitTracks]; +} + +- (void)submitTracks +{ + if (!_handshakeCompleted) { + [self attemptHandshake:NO]; + return; + } + + NSString *user = [[NSUserDefaults standardUserDefaults] stringForKey:@"audioscrobblerUser"], *passString = [PreferencesController getKeychainItemPasswordForUser:user]; + char *pass = (char *)[passString UTF8String]; + + if (passString == nil) { + ITDebugLog(@"Audioscrobbler: Access denied to user password"); + return; + } + + NSTimeInterval interval = [_delayDate timeIntervalSinceNow]; + if (interval > 0) { + ITDebugLog(@"Audioscrobbler: Delaying track submission for %f seconds", interval); + [self performSelector:@selector(submitTracks) withObject:nil afterDelay:interval + 1]; + return; + } + + int i; + NSMutableString *requestString; + NSString *authString, *responseHash = @""; + unsigned char *buffer; + EVP_MD_CTX ctx; + + ITDebugLog(@"Audioscrobbler: Submitting queued tracks"); + + if ([_tracks count] == 0) { + ITDebugLog(@"Audioscrobbler: No queued tracks to submit."); + return; + } + + //Build the MD5 response string we send along with the request + buffer = malloc(EVP_MD_size(EVP_md5())); + EVP_DigestInit(&ctx, EVP_md5()); + EVP_DigestUpdate(&ctx, pass, strlen(pass)); + EVP_DigestFinal(&ctx, buffer, NULL); + + for (i = 0; i < 16; i++) { + responseHash = [responseHash stringByAppendingFormat:@"%0.2x", buffer[i]]; + } + + free(buffer); + buffer = malloc(EVP_MD_size(EVP_md5())); + char *cat = (char *)[[responseHash stringByAppendingString:_md5Challenge] UTF8String]; + EVP_DigestInit(&ctx, EVP_md5()); + EVP_DigestUpdate(&ctx, cat, strlen(cat)); + EVP_DigestFinal(&ctx, buffer, NULL); + + responseHash = @""; + for (i = 0; i < 16; i++) { + responseHash = [responseHash stringByAppendingFormat:@"%0.2x", buffer[i]]; + } + free(buffer); + + authString = (NSString *)CFURLCreateStringByAddingPercentEscapes(NULL, (CFStringRef)[NSString stringWithFormat:@"u=%@&s=%@", user, responseHash], NULL, NULL, kCFStringEncodingUTF8); + requestString = [[NSMutableString alloc] initWithString:authString]; + [authString release]; + + //We can only submit ten tracks at a time + for (i = 0; (i < [_tracks count]) && (i < 10); i++) { + NSDictionary *nextTrack = [_tracks objectAtIndex:i]; + NSString *trackString; + + trackString = (NSString *)CFURLCreateStringByAddingPercentEscapes(NULL, (CFStringRef)[NSString stringWithFormat:@"&a[%i]=%@&t[%i]=%@&b[%i]=%@&m[%i]=&l[%i]=%@&i[%i]=%@", i, [nextTrack objectForKey:@"artist"], i, [nextTrack objectForKey:@"title"], i, [nextTrack objectForKey:@"album"], i, i, [nextTrack objectForKey:@"length"], i, [nextTrack objectForKey:@"time"]], NULL, NULL, kCFStringEncodingUTF8); + [requestString appendString:trackString]; + [trackString release]; + [_submitTracks addObject:nextTrack]; + } + + ITDebugLog(@"Audioscrobbler: Sending track submission request"); + + //Create and send the request + NSMutableURLRequest *request = [[NSURLRequest requestWithURL:_postURL cachePolicy:NSURLRequestReloadIgnoringCacheData timeoutInterval:15] mutableCopy]; + [request setHTTPMethod:@"POST"]; + [request setHTTPBody:[requestString dataUsingEncoding:NSUTF8StringEncoding]]; + _currentStatus = AudioscrobblerSubmittingTracksStatus; + _responseData = [[NSMutableData alloc] init]; + [NSURLConnection connectionWithRequest:request delegate:self]; + [requestString release]; + [request release]; + + //If we have tracks left, submit again after the interval seconds +} + +- (void)handleAudioscrobblerNotification:(NSNotification *)note +{ + if ([[note name] isEqualToString:@"AudioscrobblerHandshakeComplete"]) { + if ([_tracks count] > 0) { + [self performSelector:@selector(submitTracks) withObject:nil afterDelay:2]; + } + } +} + #pragma mark - - (void)connection:(NSURLConnection *)connection didFailWithError:(NSError *)error { - NSLog(@"Failed with an error: %@", error); + ITDebugLog(@"Audioscrobbler: Connection error \"%@\"", error); } - (void)connection:(NSURLConnection *)connection didReceiveData:(NSData *)data @@ -77,8 +235,67 @@ static AudioscrobblerController *_sharedController = nil; - (void)connectionDidFinishLoading:(NSURLConnection *)connection { NSString *string = [[NSString alloc] initWithData:_responseData encoding:NSASCIIStringEncoding]; - NSLog(@"Rawr: %@", string); + NSArray *lines = [string componentsSeparatedByString:@"\n"]; + NSString *responseAction = nil; + + if ([lines count] > 0) { + responseAction = [lines objectAtIndex:0]; + } + ITDebugLog(@"Audioscrobbler: Response %@", string); + if (_currentStatus == AudioscrobblerRequestingHandshakeStatus) { + if ([lines count] < 2) { + //We have a protocol error + } + if ([responseAction isEqualToString:@"UPTODATE"] || (([responseAction length] > 5) && [[responseAction substringToIndex:5] isEqualToString:@"UPDATE"])) { + if ([lines count] >= 4) { + _md5Challenge = [[lines objectAtIndex:1] retain]; + _postURL = [[NSURL alloc] initWithString:[lines objectAtIndex:2]]; + _handshakeCompleted = YES; + [[NSNotificationCenter defaultCenter] postNotificationName:@"AudioscrobblerHandshakeComplete" object:self]; + } else { + //We have a protocol error + } + } else if (([responseAction length] > 5) && [[responseAction substringToIndex:5] isEqualToString:@"FAILED"]) { + ITDebugLog(@"Audioscrobbler: Handshake failed (%@)", [responseAction substringFromIndex:6]); + //We have a error + } else if ([responseAction isEqualToString:@"BADUSER"]) { + ITDebugLog(@"Audioscrobbler: Bad user name"); + //We have a bad user + } else { + ITDebugLog(@"Audioscrobbler: Handshake failed, protocol error"); + //We have a protocol error + } + } else if (_currentStatus == AudioscrobblerSubmittingTracksStatus) { + if ([responseAction isEqualToString:@"OK"]) { + ITDebugLog(@"Audioscrobbler: Submission successful, clearing queue."); + [_tracks removeObjectsInArray:_submitTracks]; + [_submitTracks removeAllObjects]; + if ([_tracks count] > 0) { + ITDebugLog(@"Audioscrobbler: Tracks remaining in queue, submitting remaining tracks"); + [self performSelector:@selector(submitTracks) withObject:nil afterDelay:2]; + } + } else if ([responseAction isEqualToString:@"BADAUTH"]) { + ITDebugLog(@"Audioscrobbler: Bad password"); + //Bad auth + } else if (([responseAction length] > 5) && [[responseAction substringToIndex:5] isEqualToString:@"FAILED"]) { + ITDebugLog(@"Audioscrobbler: Submission failed (%@)", [responseAction substringFromIndex:6]); + //Failed + } + } + + //Handle the final INTERVAL response + if (([[lines objectAtIndex:[lines count] - 2] length] > 9) && [[[lines objectAtIndex:[lines count] - 2] substringToIndex:8] isEqualToString:@"INTERVAL"]) { + int seconds = [[[lines objectAtIndex:[lines count] - 2] substringFromIndex:9] intValue]; + ITDebugLog(@"Audioscrobbler: INTERVAL %i", seconds); + [_delayDate release]; + _delayDate = [[NSDate dateWithTimeIntervalSinceNow:seconds] retain]; + } else { + ITDebugLog(@"No interval response."); + //We have a protocol error + } + [string release]; + [_responseData release]; } @end