X-Git-Url: http://git.ithinksw.org/MenuTunes.git/blobdiff_plain/71ca8f780ad1fb4238e5649e84accdfab6b4ad59..c1c078a84366c6e043d2fac0ab86c912e53bc804:/AudioscrobblerController.m

diff --git a/AudioscrobblerController.m b/AudioscrobblerController.m
index 113e4b7..47b4a8b 100644
--- a/AudioscrobblerController.m
+++ b/AudioscrobblerController.m
@@ -12,19 +12,17 @@
  */
 
 #import "AudioscrobblerController.h"
+#import "PreferencesController.h"
 #import <openssl/evp.h>
+#import <ITFoundation/ITDebug.h>
+
+#define AUDIOSCROBBLER_ID @"mtu"
+#define AUDIOSCROBBLER_VERSION [[[NSBundle mainBundle] infoDictionary] objectForKey:@"CFBundleVersion"]
 
 static AudioscrobblerController *_sharedController = nil;
 
 @implementation AudioscrobblerController
 
-+ (void)load
-{
-	NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
-	[[AudioscrobblerController sharedController] submitTrack:@"Stairway To Heaven" artist:@"Led Zeppelin" album:@"Led Zeppelin IV" length:483];
-	[pool release];
-}
-
 + (AudioscrobblerController *)sharedController
 {
 	if (!_sharedController) {
@@ -37,36 +35,66 @@ static AudioscrobblerController *_sharedController = nil;
 {
 	if ( (self = [super init]) ) {
 		_handshakeCompleted = NO;
-		_responseData = nil;
 		_md5Challenge = nil;
 		_postURL = nil;
 		
-		//Test variables
-		_md5Challenge = @"315EFDA9FDA6A24B421BE991511DEE90";
-		_postURL = [[NSURL alloc] initWithString:@"http://62.216.251.205:80/protocol_1.1"];
-		_handshakeCompleted = YES;
-		[[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(handleAudioscrobblerNotification:) name:nil object:self];
+		/*_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:@"AudioscrobblerHandshakeComplete" object:self];
 	}
 	return self;
 }
 
 - (void)dealloc
 {
+	[_lastStatus release];
 	[_md5Challenge release];
 	[_postURL release];
 	[_responseData release];
+	[_submitTracks release];
+	[_tracks release];
+	[_delayDate release];
 	[super dealloc];
 }
 
+- (NSString *)lastStatus
+{
+	return _lastStatus;
+}
+
 - (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=tst&v=%@&u=%@", version, user]];
-	NSURLConnection *connection;
+	[self attemptHandshake:NO];
+}
+
+- (void)attemptHandshake:(BOOL)force
+{
+	if (_handshakeCompleted && !force) {
+		return;
+	}
 	
-	_currentStatus = AudioscrobblerRequestingHandshakeStatus;
-	_responseData = [[NSMutableData alloc] init];
-	connection = [NSURLConnection connectionWithRequest:[NSURLRequest requestWithURL:url cachePolicy:NSURLRequestReloadIgnoringCacheData timeoutInterval:30] 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;
+	}
+	
+	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
@@ -75,87 +103,137 @@ static AudioscrobblerController *_sharedController = nil;
 }
 
 - (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];
+		[self attemptHandshake:NO];
 		return;
 	}
 	
-	//What we eventually want is a submission list that sends backlogs also
-	NSMutableURLRequest *request = [[NSURLRequest requestWithURL:_postURL cachePolicy:NSURLRequestReloadIgnoringCacheData timeoutInterval:30] mutableCopy];
-	NSString *responseHash, *requestBody;
+	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;
+	}
 	
-	char *pass = "waffles";
-	unsigned char *buffer, *buffer2, buffer3[16];
-	EVP_MD_CTX ctx;
 	int i;
+	NSMutableString *requestString;
+	NSString *authString, *responseHash = @"";
+	unsigned char *buffer;
+	EVP_MD_CTX ctx;
 	
-	buffer = malloc(EVP_MD_size(EVP_md5()));
-	//buffer3 = malloc(EVP_MD_size(EVP_md5()));
+	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_DigestUpdate(&ctx, [_md5Challenge UTF8String], strlen([_md5Challenge UTF8String]));
 	EVP_DigestFinal(&ctx, buffer, NULL);
 	
 	for (i = 0; i < 16; i++) {
-		char hex1, hex2;
-		hex1 = toascii(48+ (buffer[i] / 16));
-		if (hex1 > 57) {
-			hex1 = hex1 + 39;
-		}
-		hex2 = toascii(48 + (buffer[i] % 16));
-		if (hex2 > 57) {
-			hex2 = hex2 + 39;
-		}
-		
-		buffer3[i] = hex1;
-		buffer3[i + 1] = hex2;
+		responseHash = [responseHash stringByAppendingFormat:@"%0.2x", buffer[i]];
 	}
 	
-	NSLog(@"%s", buffer3);
-	
-	/*unsigned char *cat = strcat(buffer3, [[_md5Challenge lowercaseString] UTF8String]);
-	
+	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, buffer2, NULL);
+	EVP_DigestFinal(&ctx, buffer, NULL);
 	
+	responseHash = @"";
 	for (i = 0; i < 16; i++) {
-		char hex1, hex2;
-		hex1 = toascii(48+ (buffer2[i] / 16));
-		if (hex1 > 57) {
-			hex1 = hex1 + 39;
-		}
-		hex2 = toascii(48 + (buffer2[i] % 16));
-		if (hex2 > 57) {
-			hex2 = hex2 + 39;
-		}
-		buffer3[i] = hex1;
-		buffer3[i + 1] = hex2;
+		responseHash = [responseHash stringByAppendingFormat:@"%0.2x", buffer[i]];
 	}
-	NSLog(@"%s", buffer3);*/
+	free(buffer);
 	
-	if ([NSString respondsToSelector:@selector(stringWithCString:encoding:)]) {
-		responseHash = [NSString stringWithCString:buffer3 encoding:NSASCIIStringEncoding];
-	} else {
-		responseHash = [NSString stringWithCString:buffer3 length:strlen(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 *artistEscaped, *titleEscaped, *albumEscaped, *timeEscaped, *ampersand = @"&";
+		
+		//Escape each of the individual parameters we're sending
+		artistEscaped = (NSString *)CFURLCreateStringByAddingPercentEscapes(NULL, (CFStringRef)[nextTrack objectForKey:@"artist"], NULL, (CFStringRef)ampersand, kCFStringEncodingUTF8);
+		titleEscaped = (NSString *)CFURLCreateStringByAddingPercentEscapes(NULL, (CFStringRef)[nextTrack objectForKey:@"title"], NULL, (CFStringRef)ampersand, kCFStringEncodingUTF8);
+		albumEscaped = (NSString *)CFURLCreateStringByAddingPercentEscapes(NULL, (CFStringRef)[nextTrack objectForKey:@"album"], NULL, (CFStringRef)ampersand, kCFStringEncodingUTF8);
+		timeEscaped = (NSString *)CFURLCreateStringByAddingPercentEscapes(NULL, (CFStringRef)[nextTrack objectForKey:@"time"], NULL, (CFStringRef)ampersand, kCFStringEncodingUTF8);
+		
+		[requestString appendString:[NSString stringWithFormat:@"&a[%i]=%@&t[%i]=%@&b[%i]=%@&m[%i]=&l[%i]=%@&i[%i]=%@", i, artistEscaped,
+																														i, titleEscaped,
+																														i, albumEscaped,
+																														i,
+																														i, [nextTrack objectForKey:@"length"],
+																														i, timeEscaped]];
+		
+		//Release the escaped strings
+		[artistEscaped release];
+		[titleEscaped release];
+		[albumEscaped release];
+		[timeEscaped release];
+		
+		[_submitTracks addObject:nextTrack];
 	}
 	
-	requestBody = [NSString stringWithFormat:@"u=%@&s=%@&a[0]=%@&t[0]=%@&b[0]=%@&m[0]=&l[0]=%i&i[0]=%@", @"Tristrex", @"rawr", responseHash, title, album, length, [[NSDate date] descriptionWithCalendarFormat:@"%Y-%m-%d %H:%M:%S" timeZone:nil locale:nil]];
-	[request setHTTPMethod:@"POST"];
-	[request setHTTPBody:[requestBody dataUsingEncoding:NSUTF8StringEncoding]];
+	ITDebugLog(@"Audioscrobbler: Sending track submission request");
 	
-	_currentStatus = AudioscrobblerSubmittingTrackStatus;
+	//Create and send the request
+	NSMutableURLRequest *request = [[NSURLRequest requestWithURL:_postURL cachePolicy:NSURLRequestReloadIgnoringCacheData timeoutInterval:15] mutableCopy];
+	NSLog(@"Posting Audioscrobbler URL request: %@", requestString);
+	[request setHTTPMethod:@"POST"];
+	[request setHTTPBody:[requestString dataUsingEncoding:NSUTF8StringEncoding]];
+	_currentStatus = AudioscrobblerSubmittingTracksStatus;
 	_responseData = [[NSMutableData alloc] init];
 	[NSURLConnection connectionWithRequest:request delegate:self];
+	[requestString release];
 	[request release];
+	
+	//For now we're not going to cache results, as it is less of a headache
+	//[_tracks removeObjectsInArray:_submitTracks];
+	[_tracks removeAllObjects];
+	[_submitTracks removeAllObjects];
+	
+	//If we have tracks left, submit again after the interval seconds
 }
 
 - (void)handleAudioscrobblerNotification:(NSNotification *)note
 {
-	if ([[note name] isEqualToString:@"AudioscrobblerHandshakeComplete"]) {
-		[[AudioscrobblerController sharedController] submitTrack:@"Stairway To Heaven" artist:@"Led Zeppelin" album:@"Led Zeppelin IV" length:483];
+	if ([_tracks count] > 0) {
+		[self performSelector:@selector(submitTracks) withObject:nil afterDelay:2];
 	}
 }
 
@@ -163,7 +241,11 @@ static AudioscrobblerController *_sharedController = nil;
 
 - (void)connection:(NSURLConnection *)connection didFailWithError:(NSError *)error
 {
-	NSLog(@"Failed with an error: %@", error);
+	[_responseData release];
+	[_lastStatus release];
+	_lastStatus = [[NSString stringWithFormat:NSLocalizedString(@"audioscrobbler_error", @"Error - %@"), [error localizedDescription]] retain];
+	[[NSNotificationCenter defaultCenter] postNotificationName:@"AudioscrobblerStatusChanged" object:self userInfo:[NSDictionary dictionaryWithObject:_lastStatus forKey:@"StatusString"]];
+	ITDebugLog(@"Audioscrobbler: Connection error \"%@\"", error);
 }
 
 - (void)connection:(NSURLConnection *)connection didReceiveData:(NSData *)data
@@ -174,39 +256,85 @@ static AudioscrobblerController *_sharedController = nil;
 - (void)connectionDidFinishLoading:(NSURLConnection *)connection
 {
 	NSString *string = [[NSString alloc] initWithData:_responseData encoding:NSASCIIStringEncoding];
+	NSArray *lines = [string componentsSeparatedByString:@"\n"];
+	NSString *responseAction = nil, *key = nil, *comment = nil;
 	
+	if ([lines count] > 0) {
+		responseAction = [lines objectAtIndex:0];
+	}
+	ITDebugLog(@"Audioscrobbler: Response %@", string);
+	NSLog(@"Audioscrobbler: Response %@", string);
 	if (_currentStatus == AudioscrobblerRequestingHandshakeStatus) {
-		NSArray *lines = [string componentsSeparatedByString:@"\n"];
-		NSString *responseAction;
 		if ([lines count] < 2) {
-			//We have an error
+			//We have a protocol error
 		}
-		responseAction = [lines objectAtIndex:0];
-		if ([responseAction isEqualToString:@"UPTODATE"]) {
+		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;
-				NSLog(@"%@", _md5Challenge);
 				[[NSNotificationCenter defaultCenter] postNotificationName:@"AudioscrobblerHandshakeComplete" object:self];
+				key = @"audioscrobbler_handshake_complete";
+				comment = @"Handshake complete";
 			} else {
-				//We have an error
+				//We have a protocol error
 			}
-			//Something
-		} else if (([responseAction length] > 6) && [[responseAction substringToIndex:5] isEqualToString:@"UPDATE"]) {
-			//Something plus update action
-		} else if (([responseAction length] > 6) && [[responseAction substringToIndex:5] isEqualToString:@"FAILED"]) {
-			//We have an error
+		} else if (([responseAction length] > 5) && [[responseAction substringToIndex:5] isEqualToString:@"FAILED"]) {
+			ITDebugLog(@"Audioscrobbler: Handshake failed (%@)", [responseAction substringFromIndex:6]);
+			key = @"audioscrobbler_handshake_failed";
+			comment = @"Handshake failed";
+			//We have a error
 		} else if ([responseAction isEqualToString:@"BADUSER"]) {
-			//We have an error
+			ITDebugLog(@"Audioscrobbler: Bad user name");
+			key = @"audioscrobbler_bad_user";
+			comment = @"Handshake failed - invalid user name";
+			//We have a bad user
 		} else {
-			//We have an error
+			ITDebugLog(@"Audioscrobbler: Handshake failed, protocol error");
+			key = @"audioscrobbler_protocol_error";
+			comment = @"Internal 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];
+			}
+			key = @"audioscrobbler_submission_ok";
+			comment = @"Last track submission successful";
+		} else if ([responseAction isEqualToString:@"BADAUTH"]) {
+			ITDebugLog(@"Audioscrobbler: Bad password");
+			key = @"audioscrobbler_bad_password";
+			comment = @"Last track submission failed - invalid password";
+			//Bad auth
+		} else if (([responseAction length] > 5) && [[responseAction substringToIndex:5] isEqualToString:@"FAILED"]) {
+			ITDebugLog(@"Audioscrobbler: Submission failed (%@)", [responseAction substringFromIndex:6]);
+			NSLog(@"Audioscrobbler: Submission failed (%@)", [responseAction substringFromIndex:6]);
+			key = @"audioscrobbler_submission_failed";
+			comment = @"Last track submission failed - see console for error";
+			//Failed
 		}
-	} else if (_currentStatus == AudioscrobblerSubmittingTrackStatus) {
-		NSLog(string);
 	}
 	
+	//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
+	}
+	[_lastStatus release];
+	_lastStatus = [NSLocalizedString(key, comment) retain];
+	[[NSNotificationCenter defaultCenter] postNotificationName:@"AudioscrobblerStatusChanged" object:nil userInfo:[NSDictionary dictionaryWithObject:_lastStatus forKey:@"StatusString"]];
 	[string release];
+	[_responseData release];
 }
 
 @end