X-Git-Url: http://git.ithinksw.org/MenuTunes.git/blobdiff_plain/97eb97cd6af1e3019980ed380a6cbaf1feae0abe..b9b2d350b10207b3e749085e97c25e7c8a6b1d24:/AudioscrobblerController.m diff --git a/AudioscrobblerController.m b/AudioscrobblerController.m index 695c684..62653ba 100644 --- a/AudioscrobblerController.m +++ b/AudioscrobblerController.m @@ -12,6 +12,8 @@ */ #import "AudioscrobblerController.h" +#import +#import static AudioscrobblerController *_sharedController = nil; @@ -19,7 +21,11 @@ static AudioscrobblerController *_sharedController = nil; + (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 @@ -33,28 +39,49 @@ static AudioscrobblerController *_sharedController = nil; - (id)init { if ( (self = [super init]) ) { - _handshakeCompleted = NO; + /*_handshakeCompleted = NO; + _md5Challenge = nil; + _postURL = nil;*/ + + _handshakeCompleted = YES; + _md5Challenge = @"rawr"; + _postURL = [NSURL URLWithString:@"http://audioscrobbler.com/"]; + + _delayDate = nil; _responseData = nil; + _tracks = [[NSMutableArray alloc] init]; + [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(handleAudioscrobblerNotification:) name:nil object:self]; } return self; } - (void)dealloc { + [_md5Challenge release]; + [_postURL release]; [_responseData release]; + [_tracks 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; - - _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; + if (!_handshakeCompleted) { + 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=tst&v=%@&u=%@", version, user]]; + NSURLConnection *connection; + + _currentStatus = AudioscrobblerRequestingHandshakeStatus; + connection = [NSURLConnection connectionWithRequest:[NSURLRequest requestWithURL:url cachePolicy:NSURLRequestReloadIgnoringCacheData timeoutInterval:30] delegate:self]; + } } - (BOOL)handshakeCompleted @@ -62,11 +89,121 @@ static AudioscrobblerController *_sharedController = nil; return _handshakeCompleted; } +- (void)submitTrack:(NSString *)title artist:(NSString *)artist album:(NSString *)album length:(int)length +{ + if (!_handshakeCompleted) { + [self attemptHandshake]; + return; + } + ITDebugLog(@"Audioscrobbler: Adding a new track to the submission queue."); + NSDictionary *newTrack = [NSDictionary dictionaryWithObjectsAndKeys:title, + @"title", + artist, + @"artist", + 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]; +} + +- (void)submitTracks +{ + NSTimeInterval interval = [_delayDate timeIntervalSinceNow]; + if (interval > 0) { + ITDebugLog(@"Audioscrobbler: Delaying track submission for %i seconds", interval); + [self performSelector:@selector(attemptHandshake) withObject:nil afterDelay:interval + 1]; + return; + } + + int i; + NSMutableString *requestString; + NSString *authString, *responseHash; + char *pass = "waffles"; + unsigned char *buffer; + EVP_MD_CTX ctx; + + if (!_handshakeCompleted) { + [self attemptHandshake]; + return; + } + + 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=%@", @"Tristrex", responseHash], NULL, NULL, kCFStringEncodingUTF8); + requestString = [[NSMutableString alloc] initWithString:authString]; + [authString release]; + + //We can only submit ten tracks at a time + for (i = 0; ([_tracks count] > 0) && (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]; + } + + //Create and send the request + NSMutableURLRequest *request = [[NSURLRequest requestWithURL:_postURL cachePolicy:NSURLRequestReloadIgnoringCacheData timeoutInterval:30] mutableCopy]; + [request setHTTPMethod:@"POST"]; + [request setHTTPBody:[requestString dataUsingEncoding:NSUTF8StringEncoding]]; + _currentStatus = AudioscrobblerSubmittingTracksStatus; + //[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 submitTracks]; + } + [self submitTrack:@"Immigrant Song" artist:@"Led Zeppelin" album:@"How The West Was Won" length:221]; + [self submitTrack:@"Comfortably Numb" artist:@"Pink Floyd" album:@"The Wall" length:384]; + [self submitTracks]; + } +} + #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 +214,65 @@ 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]; + } + + if (_currentStatus == AudioscrobblerRequestingHandshakeStatus) { + if ([lines count] < 2) { + //We have a protocol error + } + if ([responseAction isEqualToString:@"UPTODATE"]) { + 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 + } + //Something + } else if (([responseAction length] > 5) && [[responseAction substringToIndex:5] isEqualToString:@"UPDATE"]) { + //Something plus update action + } else if (([responseAction length] > 5) && [[responseAction substringToIndex:5] isEqualToString:@"FAILED"]) { + //We have a error + } else if ([responseAction isEqualToString:@"BADUSER"]) { + //We have a bad user + } else { + //We have a protocol + } + } else if (_currentStatus == AudioscrobblerSubmittingTracksStatus) { + if ([responseAction isEqualToString:@"OK"]) { + } else if ([responseAction isEqualToString:@"BADAUTH"]) { + //Bad auth + } else if (([responseAction length] > 5) && [[responseAction substringToIndex:5] isEqualToString:@"FAILED"]) { + //Failed + } + NSLog(string); + } + + //Handle the final INTERVAL response + if (([responseAction length] > 9) && [[responseAction substringToIndex:7] isEqualToString:@"INTERVAL"]) { + int seconds = [[[lines objectAtIndex:[lines count] - 1] substringFromIndex:9] intValue]; + ITDebugLog(@"Audioscrobbler: INTERVAL %i", seconds); + _delayDate = [NSDate dateWithTimeIntervalSinceNow:seconds]; + } else { + //We have a protocol error + } + [string release]; + [_responseData release]; +} + +-(NSURLRequest *)connection:(NSURLConnection *)connection willSendRequest:(NSURLRequest *)request redirectResponse:(NSURLResponse *)redirectResponse +{ + ITDebugLog(@"Audioscrobbler: Sending URL request."); + NSLog(@"Sending request."); + _responseData = [[NSMutableData alloc] init]; + return request; } @end