Making MainMenu.nib 10.2+ only
[MenuTunes.git] / iTunesRemote.m
index 27d346e..74a3df8 100755 (executable)
@@ -14,7 +14,7 @@
 
 - (NSString *)information;
 {
-    return @"Default MenuTunes plugin to control iTunes.";
+    return @"Default MenuTunes plugin to control iTunes. Written by iThink Software.";
 }
 
 - (NSImage *)icon
 - (BOOL)begin
 {
     iTunesPSN = [self iTunesPSN];
-    
+
+    //We won't need this once we're pure AEs
+    asComponent = OpenDefaultComponent(kOSAComponentType, kAppleScriptSubtype);
+
     //Register for application termination in NSWorkspace
-    
-    NSLog(@"iTunes Plugin loaded");
+    [[[NSWorkspace sharedWorkspace] notificationCenter] addObserver:self selector:@selector(applicationLaunched:) name:NSWorkspaceDidLaunchApplicationNotification object:nil];
+    [[[NSWorkspace sharedWorkspace] notificationCenter] addObserver:self selector:@selector(applicationTerminated:) name:NSWorkspaceDidTerminateApplicationNotification object:nil];
+
     return YES;
 }
 
 - (BOOL)halt
 {
     iTunesPSN.highLongOfPSN = kNoProcess;
-    
+
+    //We won't need this once we're pure AEs
+    CloseComponent(asComponent);
+
     //Unregister for application termination in NSWorkspace
+    [[[NSWorkspace sharedWorkspace] notificationCenter] removeObserver:self];
+
     return YES;
 }
 
-- (NSArray *)sources
+- (BOOL)isAppRunning
 {
-    //This is probably unneeded
-    return nil;
+    NSArray *apps = [[NSWorkspace sharedWorkspace] launchedApplications];
+    int i,count = [apps count];
+
+    for (i = 0; i < count; i++) {
+        if ([[[apps objectAtIndex:i] objectForKey:@"NSApplicationName"]
+                isEqualToString:@"iTunes"]) {
+            return YES;
+        }
+    }
+    return NO;
 }
 
-- (int)currentSourceIndex
+- (PlayerState)playerState
 {
-    //This is probably unneeded
-    return nil;
+    NSString *result = [self runScriptAndReturnResult:@"get player state"];
+
+    if ([result isEqualToString:@"playing"]) {
+        return playing;
+    } else if ([result isEqualToString:@"paused"]) {
+        return paused;
+    } else if ([result isEqualToString:@"stopped"]) {
+        return stopped;
+    } else if ([result isEqualToString:@"rewinding"]) {
+        return rewinding;
+    } else if ([result isEqualToString:@"fast forwarding"]) {
+        return forwarding;
+    }
+
+    return stopped;
 }
 
-- (NSArray *)playlistsForCurrentSource
+- (NSArray *)playlists
 {
-    //This is probably unneeded
-    return nil;
+    int i;
+    int numPresets = [[self runScriptAndReturnResult:@"get number of playlists"] intValue];
+    NSMutableArray *presets = [[NSMutableArray alloc] init];
+    for (i = 1; i <= numPresets; i++) {
+        [presets addObject:[self runScriptAndReturnResult:[NSString stringWithFormat:@"get name of playlist %i", i]]];
+    }
+    return [presets autorelease];
 }
 
-- (NSString *)sourceTypeOfCurrentPlaylist
+- (int)numberOfSongsInPlaylistAtIndex:(int)index
+{
+    NSString *result = [self runScriptAndReturnResult:[NSString stringWithFormat:@"get number of tracks in playlist %i", index]];
+    return [result intValue];
+}
+
+- (NSString *)classOfPlaylistAtIndex:(int)index
 {
     //Not working yet. It returns the 4 character code instead of a name.
-    NSString *result;
-    result = [[ITAppleEventCenter sharedCenter]
-                sendTwoTierAEWithRequestedKey:@"pcls"
-                fromObjectByKey:@"pPla" eventClass:@"core" eventID:@"getd"
-                appPSN:[self iTunesPSN]];
-    return result;
+    int realResult = [[ITAppleEventCenter sharedCenter]
+                sendTwoTierAEWithRequestedKeyForNumber:@"pcls"
+                                               fromObjectByKey:@"pPla" eventClass:@"core" eventID:@"getd"
+                                                                                                         appPSN:[self iTunesPSN]];
+
+    //result = [self runScriptAndReturnResult:[NSString stringWithFormat:@"get class of playlist %i", index]];
+    if (realResult == 'cRTP') return @"radio tuner playlist";
+    else return @"playlist";
 }
 
 - (int)currentPlaylistIndex
     int result;
     result = [[ITAppleEventCenter sharedCenter]
                 sendTwoTierAEWithRequestedKeyForNumber:@"pidx"
-                fromObjectByKey:@"pPla" eventClass:@"core" eventID:@"getd"
-                appPSN:[self iTunesPSN]];
+                                                           fromObjectByKey:@"pPla" eventClass:@"core" eventID:@"getd"
+                                                                          appPSN:[self iTunesPSN]];
     return result;
 }
 
-- (NSString *)songTitleAtIndex
+- (NSString *)songTitleAtIndex:(int)index
 {
-    return nil; 
+    NSString *result = [self runScriptAndReturnResult:[NSString stringWithFormat:@"get name of track %i of current playlist", index]];
+    return result;
 }
 
 - (int)currentSongIndex
     int result;
     result = [[ITAppleEventCenter sharedCenter]
                 sendTwoTierAEWithRequestedKeyForNumber:@"pidx"
-                fromObjectByKey:@"pTrk" eventClass:@"core" eventID:@"getd"
-                appPSN:[self iTunesPSN]];
+                                                           fromObjectByKey:@"pTrk" eventClass:@"core" eventID:@"getd"
+                                                                          appPSN:[self iTunesPSN]];
     return result;
 }
 
 - (NSString *)currentSongTitle
 {
     return [[ITAppleEventCenter sharedCenter] sendTwoTierAEWithRequestedKey:@"pnam"
-                fromObjectByKey:@"pTrk" eventClass:@"core" eventID:@"getd"
-                appPSN:[self iTunesPSN]];
+                                                                                               fromObjectByKey:@"pTrk" eventClass:@"core" eventID:@"getd"
+                                                                                                           appPSN:[self iTunesPSN]];
 }
 
 - (NSString *)currentSongArtist
 {
     return [[ITAppleEventCenter sharedCenter] sendTwoTierAEWithRequestedKey:@"pArt"
-                fromObjectByKey:@"pTrk" eventClass:@"core" eventID:@"getd"
-                appPSN:[self iTunesPSN]];
+                                                                                               fromObjectByKey:@"pTrk" eventClass:@"core" eventID:@"getd"
+                                                                                                           appPSN:[self iTunesPSN]];
 }
 
 - (NSString *)currentSongAlbum
 {
     return [[ITAppleEventCenter sharedCenter] sendTwoTierAEWithRequestedKey:@"pAlb"
-                fromObjectByKey:@"pTrk" eventClass:@"core" eventID:@"getd"
-                appPSN:[self iTunesPSN]];
+                                                                                               fromObjectByKey:@"pTrk" eventClass:@"core" eventID:@"getd"
+                                                                                                           appPSN:[self iTunesPSN]];
 }
 
 - (NSString *)currentSongGenre
 {
     return [[ITAppleEventCenter sharedCenter] sendTwoTierAEWithRequestedKey:@"pGen"
-                fromObjectByKey:@"pTrk" eventClass:@"core" eventID:@"getd"
-                appPSN:[self iTunesPSN]];
+                                                                                               fromObjectByKey:@"pTrk" eventClass:@"core" eventID:@"getd"
+                                                                                                           appPSN:[self iTunesPSN]];
 }
 
 - (NSString *)currentSongLength
 {
-    return [[ITAppleEventCenter sharedCenter] sendTwoTierAEWithRequestedKey:@"pDur"
-                fromObjectByKey:@"pTrk" eventClass:@"core" eventID:@"getd"
-                appPSN:[self iTunesPSN]];
+    return [[ITAppleEventCenter sharedCenter] sendTwoTierAEWithRequestedKey:@"pTim"
+                                                                                               fromObjectByKey:@"pTrk" eventClass:@"core" eventID:@"getd"
+                                                                                                           appPSN:[self iTunesPSN]];
 }
 
 - (NSString *)currentSongRemaining
 {
     long duration = [[ITAppleEventCenter sharedCenter]
                         sendTwoTierAEWithRequestedKeyForNumber:@"pDur"
-                        fromObjectByKey:@"pTrk" eventClass:@"core" eventID:@"getd"
-                        appPSN:[self iTunesPSN]];
+                                                                         fromObjectByKey:@"pTrk" eventClass:@"core" eventID:@"getd"
+                                                                                        appPSN:[self iTunesPSN]];
     long current = [[ITAppleEventCenter sharedCenter]
                         sendAEWithRequestedKeyForNumber:@"pPos"
-                        eventClass:@"core" eventID:@"getd"
-                        appPSN:[self iTunesPSN]];
-    
+                                                                       eventClass:@"core" eventID:@"getd"
+                                                                           appPSN:[self iTunesPSN]];
+
     return [[NSNumber numberWithLong:duration - current] stringValue];
 }
 
 - (NSArray *)eqPresets;
 {
-    return nil;
+    int i;
+    int numPresets = [[self runScriptAndReturnResult:@"get number of EQ presets"] intValue];
+    NSMutableArray *presets = [[NSMutableArray alloc] init];
+
+    for (i = 1; i <= numPresets; i++) {
+        [presets addObject:[self runScriptAndReturnResult:[NSString stringWithFormat:@"get name of EQ preset %i", i]]];
+    }
+
+    return [presets autorelease];
+}
+
+- (int)currentEQPresetIndex
+{
+    int result;
+    result = [[ITAppleEventCenter sharedCenter]
+                sendTwoTierAEWithRequestedKeyForNumber:@"pidx"
+                                                           fromObjectByKey:@"pEQP" eventClass:@"core" eventID:@"getd"
+                                                                          appPSN:[self iTunesPSN]];
+    return result;
 }
 
 - (BOOL)play
 {
     [[ITAppleEventCenter sharedCenter] sendAEWithEventClass:@"hook" eventID:@"Play"
-            appPSN:[self iTunesPSN]];
+                                                                                  appPSN:[self iTunesPSN]];
     return YES;
 }
 
 - (BOOL)pause
 {
     [[ITAppleEventCenter sharedCenter] sendAEWithEventClass:@"hook" eventID:@"Paus"
-            appPSN:[self iTunesPSN]];
+                                                                                  appPSN:[self iTunesPSN]];
     return YES;
 }
 
 - (BOOL)goToNextSong
 {
     [[ITAppleEventCenter sharedCenter] sendAEWithEventClass:@"hook" eventID:@"Next"
-            appPSN:[self iTunesPSN]];
+                                                                                  appPSN:[self iTunesPSN]];
     return YES;
 }
 
 - (BOOL)goToPreviousSong
 {
     [[ITAppleEventCenter sharedCenter] sendAEWithEventClass:@"hook" eventID:@"Prev"
-            appPSN:[self iTunesPSN]];
+                                                                                  appPSN:[self iTunesPSN]];
     return YES;
 }
 
-- (BOOL)goToNextPlaylist
+- (BOOL)fastForward
 {
-    //This is probably unneeded
-    return NO;
+    [[ITAppleEventCenter sharedCenter] sendAEWithEventClass:@"hook" eventID:@"Fast"
+                                                                                  appPSN:[self iTunesPSN]];
+    return YES;
 }
 
-- (BOOL)goToPreviousPlaylist
+- (BOOL)rewind
 {
-    //This is probably unneeded
-    return NO;
+    [[ITAppleEventCenter sharedCenter] sendAEWithEventClass:@"hook" eventID:@"Rwnd"
+                                                                                  appPSN:[self iTunesPSN]];
+    return YES;
 }
 
-- (BOOL)switchToSourceAtIndex:(int)index
-{
-    //This is probably unneeded
-    return NO;
-}
 
 - (BOOL)switchToPlaylistAtIndex:(int)index
 {
+    [self runScriptAndReturnResult:[NSString stringWithFormat:
+        @"play playlist %i", index]];
     return NO;
 }
 
 - (BOOL)switchToSongAtIndex:(int)index
 {
+    [self runScriptAndReturnResult:[NSString stringWithFormat:
+        @"play track %i of current playlist", index]];
     return NO;
 }
 
 - (BOOL)switchToEQAtIndex:(int)index
 {
+    [self runScriptAndReturnResult:[NSString stringWithFormat:
+        @"set current EQ preset to EQ preset %i", index]];
+    [self runScriptAndReturnResult:@"set EQ enabled to 1"];
     return NO;
 }
 
 {
     NSArray *apps = [[NSWorkspace sharedWorkspace] launchedApplications];
     ProcessSerialNumber number;
-    int i;
-    
+    int i, count = [apps count];
+
     number.highLongOfPSN = kNoProcess;
-    
-    for (i = 0; i < [apps count]; i++)
+
+    for (i = 0; i < count; i++)
     {
         NSDictionary *curApp = [apps objectAtIndex:i];
-        
+
         if ([[curApp objectForKey:@"NSApplicationName"] isEqualToString:@"iTunes"])
         {
             number.highLongOfPSN = [[curApp objectForKey:@"NSApplicationProcessSerialNumberHigh"] intValue];
     return number;
 }
 
+- (void)applicationLaunched:(NSNotification *)note
+{
+    NSDictionary *info = [note userInfo];
+
+    if ([[info objectForKey:@"NSApplicationName"] isEqualToString:@"iTunes"]) {
+        iTunesPSN.highLongOfPSN = [[info objectForKey:@"NSApplicationProcessSerialNumberHigh"] longValue];
+        iTunesPSN.lowLongOfPSN = [[info objectForKey:@"NSApplicationProcessSerialNumberLow"] longValue];
+
+        [[NSNotificationCenter defaultCenter] postNotificationName:@"ITMTRemoteAppDidLaunchNotification" object:nil];
+    }
+}
+
+- (void)applicationTerminated:(NSNotification *)note
+{
+    NSDictionary *info = [note userInfo];
+
+    if ([[info objectForKey:@"NSApplicationName"] isEqualToString:@"iTunes"]) {
+        iTunesPSN.highLongOfPSN = kNoProcess;
+        [[NSNotificationCenter defaultCenter] postNotificationName:@"ITMTRemoteAppDidTerminateNotification" object:nil];
+    }
+}
+
+//This is just temporary
+- (NSString *)runScriptAndReturnResult:(NSString *)script
+{
+    AEDesc scriptDesc, resultDesc;
+    Size length;
+    NSString *result;
+    Ptr buffer;
+
+    script = [NSString stringWithFormat:@"tell application \"iTunes\"\n%@\nend tell", script];
+
+    //NSLog(@"I say \"%@\"",script);
+
+    AECreateDesc(typeChar, [script cString], [script cStringLength],
+                         &scriptDesc);
+
+    OSADoScript(asComponent, &scriptDesc, kOSANullScript, typeChar, kOSAModeCanInteract, &resultDesc);
+
+    length = AEGetDescDataSize(&resultDesc);
+    buffer = malloc(length);
+
+    AEGetDescData(&resultDesc, buffer, length);
+    AEDisposeDesc(&scriptDesc);
+    AEDisposeDesc(&resultDesc);
+    result = [NSString stringWithCString:buffer length:length];
+    if ( (! [result isEqualToString:@""])      &&
+         ([result characterAtIndex:0] == '\"') &&
+         ([result characterAtIndex:[result length] - 1] == '\"') ) {
+        result = [result substringWithRange:NSMakeRange(1, [result length] - 2)];
+    }
+    free(buffer);
+    buffer = nil;
+    //NSLog(@"You say \"%@\"",result);
+    return result;
+}
+
 @end