MenuTunes.m

   1 /*
   2 Things to do:
   3 ¥ Make preferences window pretty
   4 ¥ Optimize
   5 ¥ Apple Events! Apple Events! Apple Events!
   6 ¥ Manual and webpage
   7 ¥ Finish up registration frontend
   8 */
   9
  10 #import "MenuTunes.h"
  11 #import "PreferencesController.h"
  12 #import "HotKeyCenter.h"
  13 #import "StatusWindowController.h"
  14
  15 @interface MenuTunes(Private)
  16 - (ITMTRemote *)loadRemote;
  17 - (void)updateMenu;
  18 - (void)rebuildUpcomingSongsMenu;
  19 - (void)rebuildPlaylistMenu;
  20 - (void)rebuildEQPresetsMenu;
  21 - (void)setupHotKeys;
  22 - (NSString *)runScriptAndReturnResult:(NSString *)script;
  23 - (void)timerUpdate;
  24 - (void)sendAEWithEventClass:(AEEventClass)eventClass andEventID:(AEEventID)eventID;
  25 - (void)setKeyEquivalentForCode:(short)code andModifiers:(long)modifiers
  26         onItem:(NSMenuItem *)item;
  27
  28 @end
  29
  30 @implementation MenuTunes
  31
  32 /*************************************************************************/
  33 #pragma mark -
  34 #pragma mark INITIALIZATION METHODS
  35 /*************************************************************************/
  36
  37 - (id)init
  38 {
  39     if ( ( self = [super init] ) ) {
  40         remoteArray = [[NSMutableArray alloc] initWithCapacity:1];
  41     }
  42     return self;
  43 }
  44
  45 - (void)applicationDidFinishLaunching:(NSNotification *)note
  46 {
  47     currentRemote = [self loadRemote];
  48     [currentRemote begin];
  49
  50     asComponent = OpenDefaultComponent(kOSAComponentType, kAppleScriptSubtype);
  51
  52     [self registerDefaultsIfNeeded];
  53
  54     menu = [[NSMenu alloc] initWithTitle:@""];
  55     iTunesPSN = [self iTunesPSN]; //Get PSN of iTunes if it's running
  56
  57     if (!((iTunesPSN.highLongOfPSN == kNoProcess) && (iTunesPSN.lowLongOfPSN == 0)))
  58     {
  59         [self rebuildMenu];
  60         refreshTimer = [NSTimer scheduledTimerWithTimeInterval:3.5
  61                             target:self
  62                             selector:@selector(timerUpdate)
  63                             userInfo:nil
  64                             repeats:YES];
  65
  66         [[[NSWorkspace sharedWorkspace] notificationCenter] addObserver:self selector:@selector(iTunesTerminated:) name:NSWorkspaceDidTerminateApplicationNotification object:nil];
  67     }
  68     else
  69     {
  70         menu = [[NSMenu alloc] initWithTitle:@""];
  71         [[menu addItemWithTitle:@"Open iTunes" action:@selector(openiTunes:) keyEquivalent:@""] setTarget:self];
  72         [[menu addItemWithTitle:@"Preferences" action:@selector(showPreferences:) keyEquivalent:@""] setTarget:self];
  73         [[menu addItemWithTitle:@"Quit" action:@selector(quitMenuTunes:) keyEquivalent:@""] setTarget:self];
  74         [[[NSWorkspace sharedWorkspace] notificationCenter] addObserver:self selector:@selector(iTunesLaunched:) name:NSWorkspaceDidLaunchApplicationNotification object:nil];
  75         refreshTimer = nil;
  76     }
  77
  78     statusItem = [[ITStatusItem alloc] initWithStatusBar:[NSStatusBar systemStatusBar]
  79                                               withLength:NSSquareStatusItemLength];
  80
  81     [statusItem setImage:[NSImage imageNamed:@"menu"]];
  82     [statusItem setAlternateImage:[NSImage imageNamed:@"selected_image"]];
  83     [statusItem setMenu:menu];
  84     // Below line of code is for creating builds for Beta Testers
  85     // [statusItem setToolTip:@[NSString stringWithFormat:@"This Nontransferable Beta (Built on %s) of iThink Software's MenuTunes is Registered to: Beta Tester (betatester@somedomain.com).",__DATE__]];
  86 }
  87
  88 - (ITMTRemote *)loadRemote
  89 {
  90     NSString *folderPath = [[NSBundle mainBundle] builtInPlugInsPath];
  91
  92     if (folderPath) {
  93         NSArray      *bundlePathList = [NSBundle pathsForResourcesOfType:@"remote" inDirectory:folderPath];
  94         NSEnumerator *enumerator     = [bundlePathList objectEnumerator];
  95         NSString     *bundlePath;
  96
  97         while ( (bundlePath = [enumerator nextObject]) ) {
  98             NSBundle* remoteBundle = [NSBundle bundleWithPath:bundlePath];
  99
 100             if (remoteBundle) {
 101                 Class remoteClass = [remoteBundle principalClass];
 102
 103                 if ([remoteClass conformsToProtocol:@protocol(ITMTRemote)] &&
 104                     [remoteClass isKindOfClass:[NSObject class]]) {
 105
 106                     id remote = [remoteClass remote];
 107                     [remoteArray addObject:remote];
 108                 }
 109             }
 110         }
 111
 112 //      if ( [remoteArray count] > 0 ) {
 113 //          if ( [remoteArray count] > 1 ) {
 114 //              [remoteArray sortUsingSelector:@selector(sortAlpha:)];
 115 //          }
 116 //          [self loadModuleAccessUI]; //Comment out this line to disable remote visibility
 117 //      }
 118     }
 119     NSLog(@"%@", [remoteArray objectAtIndex:0]);
 120     return [remoteArray objectAtIndex:0];
 121 }
 122
 123
 124 /*************************************************************************/
 125 #pragma mark -
 126 #pragma mark INSTANCE METHODS
 127 /*************************************************************************/
 128
 129 - (void)registerDefaultsIfNeeded
 130 {
 131     NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults];
 132     if (![defaults objectForKey:@"menu"]) {
 133         bool found = NO;
 134         NSMutableDictionary *loginwindow;
 135         NSMutableArray *loginarray;
 136         int i;
 137
 138         [defaults setObject:
 139             [NSArray arrayWithObjects:
 140                 @"Play/Pause",
 141                 @"Next Track",
 142                 @"Previous Track",
 143                 @"Fast Forward",
 144                 @"Rewind",
 145                 @"<separator>",
 146                 @"Upcoming Songs",
 147                 @"Playlists",
 148                 @"<separator>",
 149                 @"PreferencesÉ",
 150                 @"Quit",
 151                 @"<separator>",
 152                 @"Current Track Info",
 153                 nil] forKey:@"menu"];
 154
 155         [defaults synchronize];
 156         loginwindow = [[defaults persistentDomainForName:@"loginwindow"] mutableCopy];
 157         loginarray = [loginwindow objectForKey:@"AutoLaunchedApplicationDictionary"];
 158
 159         for (i = 0; i < [loginarray count]; i++) {
 160             NSDictionary *tempDict = [loginarray objectAtIndex:i];
 161             if ([[[tempDict objectForKey:@"Path"] lastPathComponent] isEqualToString:[[[NSBundle mainBundle] bundlePath] lastPathComponent]]) {
 162                 found = YES;
 163             }
 164         }
 165
 166         if (!found) {
 167             if (NSRunInformationalAlertPanel(@"Auto-launch MenuTunes", @"Would you like MenuTunes to automatically launch at login?", @"Yes", @"No", nil) == NSOKButton) {
 168                 AEDesc scriptDesc, resultDesc;
 169                 NSString *script = [NSString stringWithFormat:@"tell application \"System Events\"\nmake new login item at end of login items with properties {path:\"%@\", kind:\"APPLICATION\"}\nend tell", [[NSBundle mainBundle] bundlePath]];
 170
 171                 AECreateDesc(typeChar, [script cString], [script cStringLength],
 172             &scriptDesc);
 173
 174                 OSADoScript(asComponent, &scriptDesc, kOSANullScript, typeChar, kOSAModeCanInteract, &resultDesc);
 175
 176                 AEDisposeDesc(&scriptDesc);
 177                 AEDisposeDesc(&resultDesc);
 178             }
 179         }
 180     }
 181
 182     if (![defaults integerForKey:@"SongsInAdvance"])
 183     {
 184         [defaults setInteger:5 forKey:@"SongsInAdvance"];
 185     }
 186
 187     if (![defaults objectForKey:@"showName"]) {
 188         [defaults setBool:YES forKey:@"showName"];
 189     }
 190
 191     if (![defaults objectForKey:@"showArtist"]) {
 192         [defaults setBool:YES forKey:@"showArtist"];
 193     }
 194
 195     if (![defaults objectForKey:@"showAlbum"]) {
 196         [defaults setBool:NO forKey:@"showAlbum"];
 197     }
 198
 199     if (![defaults objectForKey:@"showTime"]) {
 200         [defaults setBool:NO forKey:@"showTime"];
 201     }
 202 }
 203
 204 //Recreate the status item menu
 205 - (void)rebuildMenu
 206 {
 207     NSArray *myMenu = [[NSUserDefaults standardUserDefaults] arrayForKey:@"menu"];
 208     int i;
 209
 210     trackInfoIndex = -1;
 211     lastSongIndex = -1;
 212     didHaveAlbumName = ([[self runScriptAndReturnResult:@"return album of current track"] length] > 0);
 213     didHaveArtistName = ([[self runScriptAndReturnResult:@"return artist of current track"] length] > 0);
 214
 215
 216     while ([menu numberOfItems] > 0) {
 217         [menu removeItemAtIndex:0];
 218     }
 219
 220     playPauseMenuItem = nil;
 221     upcomingSongsItem = nil;
 222     playlistItem = nil;
 223     [playlistMenu release];
 224     playlistMenu = nil;
 225     eqItem = nil;
 226     [eqMenu release];
 227     eqMenu = nil;
 228
 229     for (i = 0; i < [myMenu count]; i++) {
 230         NSString *item = [myMenu objectAtIndex:i];
 231         if ([item isEqualToString:@"Play/Pause"]) {
 232             KeyCombo *tempCombo = [[NSUserDefaults standardUserDefaults] keyComboForKey:@"PlayPause"];
 233             playPauseMenuItem = [menu addItemWithTitle:@"Play"
 234                                     action:@selector(playPause:)
 235                                     keyEquivalent:@""];
 236             [playPauseMenuItem setTarget:self];
 237
 238             if (tempCombo)
 239             {
 240                 [self setKeyEquivalentForCode:[tempCombo keyCode]
 241                     andModifiers:[tempCombo modifiers] onItem:playPauseMenuItem];
 242                 [tempCombo release];
 243             }
 244         } else if ([item isEqualToString:@"Next Track"]) {
 245             KeyCombo *tempCombo = [[NSUserDefaults standardUserDefaults] keyComboForKey:@"NextTrack"];
 246             NSMenuItem *nextTrack = [menu addItemWithTitle:@"Next Track"
 247                                         action:@selector(nextSong:)
 248                                         keyEquivalent:@""];
 249
 250             [nextTrack setTarget:self];
 251             if (tempCombo)
 252             {
 253                 [self setKeyEquivalentForCode:[tempCombo keyCode]
 254                     andModifiers:[tempCombo modifiers] onItem:nextTrack];
 255                 [tempCombo release];
 256             }
 257         } else if ([item isEqualToString:@"Previous Track"]) {
 258             KeyCombo *tempCombo = [[NSUserDefaults standardUserDefaults] keyComboForKey:@"PrevTrack"];
 259             NSMenuItem *prevTrack = [menu addItemWithTitle:@"Previous Track"
 260                                         action:@selector(prevSong:)
 261                                         keyEquivalent:@""];
 262
 263             [prevTrack setTarget:self];
 264             if (tempCombo)
 265             {
 266                 [self setKeyEquivalentForCode:[tempCombo keyCode]
 267                     andModifiers:[tempCombo modifiers] onItem:prevTrack];
 268                 [tempCombo release];
 269             }
 270         } else if ([item isEqualToString:@"Fast Forward"]) {
 271             [[menu addItemWithTitle:@"Fast Forward"
 272                              action:@selector(fastForward:)
 273                       keyEquivalent:@""] setTarget:self];
 274         } else if ([item isEqualToString:@"Rewind"]) {
 275             [[menu addItemWithTitle:@"Rewind"
 276                              action:@selector(rewind:)
 277                       keyEquivalent:@""] setTarget:self];
 278         } else if ([item isEqualToString:@"Upcoming Songs"]) {
 279             upcomingSongsItem = [menu addItemWithTitle:@"Upcoming Songs"
 280                                                 action:nil
 281                                          keyEquivalent:@""];
 282         } else if ([item isEqualToString:@"Playlists"]) {
 283             playlistItem = [menu addItemWithTitle:@"Playlists"
 284                                            action:nil
 285                                     keyEquivalent:@""];
 286         } else if ([item isEqualToString:@"EQ Presets"]) {
 287             eqItem = [menu addItemWithTitle:@"EQ Presets"
 288                                      action:nil
 289                               keyEquivalent:@""];
 290         } else if ([item isEqualToString:@"PreferencesÉ"]) {
 291             [[menu addItemWithTitle:@"PreferencesÉ"
 292                              action:@selector(showPreferences:)
 293                       keyEquivalent:@""] setTarget:self];
 294         } else if ([item isEqualToString:@"Quit"]) {
 295             [[menu addItemWithTitle:@"Quit"
 296                              action:@selector(quitMenuTunes:)
 297                       keyEquivalent:@""] setTarget:self];
 298         } else if ([item isEqualToString:@"Current Track Info"]) {
 299             trackInfoIndex = [menu numberOfItems];
 300             [menu addItemWithTitle:@"No Song"
 301                             action:nil
 302                      keyEquivalent:@""];
 303         } else if ([item isEqualToString:@"<separator>"]) {
 304             [menu addItem:[NSMenuItem separatorItem]];
 305         }
 306     }
 307
 308     [self timerUpdate]; //Updates dynamic info in the menu
 309
 310     [self clearHotKeys];
 311     [self setupHotKeys];
 312 }
 313
 314 //Updates the menu with current player state, song, and upcoming songs
 315 - (void)updateMenu
 316 {
 317     NSMenuItem *menuItem;
 318     NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults];
 319
 320     if ((iTunesPSN.highLongOfPSN == kNoProcess) && (iTunesPSN.lowLongOfPSN == 0)) {
 321         return;
 322     }
 323
 324     if (upcomingSongsItem) {
 325         [self rebuildUpcomingSongsMenu];
 326     }
 327
 328     if (playlistItem) {
 329         [self rebuildPlaylistMenu];
 330     }
 331
 332     if (eqItem) {
 333         [self rebuildEQPresetsMenu];
 334     }
 335
 336     if (trackInfoIndex > -1)
 337     {
 338         NSString *curSongName, *curAlbumName = @"", *curArtistName = @"";
 339         curSongName = [currentRemote currentSongTitle];
 340
 341         if ([defaults boolForKey:@"showAlbum"]) {
 342             curAlbumName = [currentRemote currentSongAlbum];
 343         }
 344
 345         if ([defaults boolForKey:@"showArtist"]) {
 346             curArtistName = [currentRemote currentSongArtist];
 347         }
 348
 349         if ([curSongName length] > 0) {
 350             int index = [menu indexOfItemWithTitle:@"Now Playing"];
 351             if (index > -1) {
 352                 if ([defaults boolForKey:@"showName"]) {
 353                     [menu removeItemAtIndex:index + 1];
 354                 }
 355                 if (didHaveAlbumName && [defaults boolForKey:@"showAlbum"]) {
 356                     [menu removeItemAtIndex:index + 1];
 357                 }
 358                 if (didHaveArtistName && [defaults boolForKey:@"showArtist"]) {
 359                     [menu removeItemAtIndex:index + 1];
 360                 }
 361                 if ([defaults boolForKey:@"showTime"]) {
 362                     [menu removeItemAtIndex:index + 1];
 363                 }
 364             }
 365
 366             if (!isPlayingRadio) {
 367                 if ([defaults boolForKey:@"showTime"]) {
 368                     menuItem = [[NSMenuItem alloc] initWithTitle:[NSString stringWithFormat:@"  %@", [self runScriptAndReturnResult:@"return time of current track"]]
 369                                                         action:nil
 370                                                         keyEquivalent:@""];
 371                     [menu insertItem:menuItem atIndex:trackInfoIndex + 1];
 372                     [menuItem release];
 373                 }
 374
 375                 if ([curArtistName length] > 0) {
 376                     menuItem = [[NSMenuItem alloc] initWithTitle:[NSString stringWithFormat:@"  %@", curArtistName]
 377                                                         action:nil
 378                                                         keyEquivalent:@""];
 379                     [menu insertItem:menuItem atIndex:trackInfoIndex + 1];
 380                     [menuItem release];
 381                 }
 382
 383                 if ([curAlbumName length] > 0) {
 384                     menuItem = [[NSMenuItem alloc] initWithTitle:[NSString stringWithFormat:@"  %@", curAlbumName]
 385                                                         action:nil
 386                                                         keyEquivalent:@""];
 387                     [menu insertItem:menuItem atIndex:trackInfoIndex + 1];
 388                     [menuItem release];
 389                 }
 390             }
 391
 392             if ([defaults boolForKey:@"showName"]) {
 393                 menuItem = [[NSMenuItem alloc] initWithTitle:[NSString stringWithFormat:@"  %@", curSongName]
 394                                                     action:nil
 395                                                     keyEquivalent:@""];
 396                 [menu insertItem:menuItem atIndex:trackInfoIndex + 1];
 397                 [menuItem release];
 398             }
 399
 400             if (index == -1) {
 401                 menuItem = [[NSMenuItem alloc] initWithTitle:@"Now Playing" action:nil keyEquivalent:@""];
 402                 [menu removeItemAtIndex:[menu indexOfItemWithTitle:@"No Song"]];
 403                 [menu insertItem:menuItem atIndex:trackInfoIndex];
 404                 [menuItem release];
 405             }
 406         } else if ([menu indexOfItemWithTitle:@"No Song"] == -1) {
 407             [menu removeItemAtIndex:trackInfoIndex];
 408
 409             if ([defaults boolForKey:@"showName"] == YES) {
 410                 [menu removeItemAtIndex:trackInfoIndex];
 411             }
 412
 413             if ([defaults boolForKey:@"showTime"] == YES) {
 414                 [menu removeItemAtIndex:trackInfoIndex];
 415             }
 416
 417             if (didHaveArtistName && [defaults boolForKey:@"showArtist"]) {
 418                 [menu removeItemAtIndex:trackInfoIndex];
 419             }
 420
 421             if (didHaveAlbumName && [defaults boolForKey:@"showAlbum"]) {
 422                 [menu removeItemAtIndex:trackInfoIndex];
 423             }
 424
 425             menuItem = [[NSMenuItem alloc] initWithTitle:@"No Song" action:nil keyEquivalent:@""];
 426             [menu insertItem:menuItem atIndex:trackInfoIndex];
 427             [menuItem release];
 428         }
 429
 430         if ([defaults boolForKey:@"showArtist"]) {
 431             didHaveArtistName = (([curArtistName length] > 0) ? YES : NO);
 432         }
 433
 434         if ([defaults boolForKey:@"showAlbum"]) {
 435             didHaveAlbumName = (([curAlbumName length] > 0) ? YES : NO);
 436         }
 437     }
 438 }
 439
 440 //Rebuild the upcoming songs submenu. Can be improved a lot.
 441 - (void)rebuildUpcomingSongsMenu
 442 {
 443     int curIndex = [currentRemote currentPlaylistIndex];
 444     int numSongs = [currentRemote numberOfSongsInPlaylistAtIndex:curIndex];
 445     int numSongsInAdvance = [[NSUserDefaults standardUserDefaults] integerForKey:@"SongsInAdvance"];
 446     if (!isPlayingRadio) {
 447         if (numSongs > 0) {
 448             int curTrack = [currentRemote currentSongIndex];
 449             int i;
 450
 451             [upcomingSongsMenu release];
 452             upcomingSongsMenu = [[NSMenu alloc] initWithTitle:@""];
 453             [upcomingSongsItem setSubmenu:upcomingSongsMenu];
 454             [upcomingSongsItem setEnabled:YES];
 455
 456             for (i = curTrack + 1; i <= curTrack + numSongsInAdvance; i++) {
 457                 if (i <= numSongs) {
 458                     NSString *curSong = [currentRemote songTitleAtIndex:i];
 459                     NSMenuItem *songItem;
 460                     songItem = [[NSMenuItem alloc] initWithTitle:curSong action:@selector(playTrack:) keyEquivalent:@""];
 461                     [songItem setTarget:self];
 462                     [songItem setRepresentedObject:[NSNumber numberWithInt:i]];
 463                     [upcomingSongsMenu addItem:songItem];
 464                     [songItem release];
 465                 } else {
 466                     break;
 467                 }
 468             }
 469         }
 470     } else {
 471         [upcomingSongsItem setSubmenu:nil];
 472         [upcomingSongsItem setEnabled:NO];
 473     }
 474 }
 475
 476 - (void)rebuildPlaylistMenu
 477 {
 478     NSArray *playlists = [currentRemote playlists];
 479     int i, curPlaylist = [currentRemote currentPlaylistIndex];
 480
 481     if (isPlayingRadio)
 482     {
 483         curPlaylist = 0;
 484     }
 485
 486     if (playlistMenu && ([playlists count] == [playlistMenu numberOfItems]))
 487         return;
 488
 489     [playlistMenu release];
 490     playlistMenu = [[NSMenu alloc] initWithTitle:@""];
 491
 492     for (i = 1; i < [playlists count]; i++) {
 493         NSString *playlistName = [playlists objectAtIndex:i];
 494         NSMenuItem *tempItem;
 495         tempItem = [[NSMenuItem alloc] initWithTitle:playlistName action:@selector(selectPlaylist:) keyEquivalent:@""];
 496         [tempItem setTarget:self];
 497         [tempItem setRepresentedObject:[NSNumber numberWithInt:i]];
 498         [playlistMenu addItem:tempItem];
 499         [tempItem release];
 500     }
 501     [playlistItem setSubmenu:playlistMenu];
 502
 503     if (curPlaylist) {
 504         [[playlistMenu itemAtIndex:curPlaylist - 1] setState:NSOnState];
 505     }
 506 }
 507
 508 //Build a menu with the list of all available EQ presets
 509 - (void)rebuildEQPresetsMenu
 510 {
 511     NSArray *eqPresets = [currentRemote eqPresets];
 512     int i;
 513
 514     if (eqMenu && ([[currentRemote eqPresets] count] == [eqMenu numberOfItems]))
 515         return;
 516
 517     [eqMenu release];
 518     eqMenu = [[NSMenu alloc] initWithTitle:@""];
 519
 520     for (i = 0; i < [eqPresets count]; i++) {
 521         NSString *setName = [eqPresets objectAtIndex:i];
 522         NSMenuItem *tempItem;
 523         tempItem = [[NSMenuItem alloc] initWithTitle:setName action:@selector(selectEQPreset:) keyEquivalent:@""];
 524         [tempItem setTarget:self];
 525         [tempItem setRepresentedObject:[NSNumber numberWithInt:i]];
 526         [eqMenu addItem:tempItem];
 527         [tempItem release];
 528     }
 529     [eqItem setSubmenu:eqMenu];
 530
 531     [[eqMenu itemAtIndex:[currentRemote currentEQPresetIndex] - 1] setState:NSOnState];
 532 }
 533
 534 - (void)clearHotKeys
 535 {
 536     [[HotKeyCenter sharedCenter] removeHotKey:@"PlayPause"];
 537     [[HotKeyCenter sharedCenter] removeHotKey:@"NextTrack"];
 538     [[HotKeyCenter sharedCenter] removeHotKey:@"PrevTrack"];
 539     [[HotKeyCenter sharedCenter] removeHotKey:@"TrackInfo"];
 540     [[HotKeyCenter sharedCenter] removeHotKey:@"UpcomingSongs"];
 541 }
 542
 543 - (void)setupHotKeys
 544 {
 545     NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults];
 546
 547     if ([defaults objectForKey:@"PlayPause"] != nil) {
 548         [[HotKeyCenter sharedCenter] addHotKey:@"PlayPause"
 549                 combo:[defaults keyComboForKey:@"PlayPause"]
 550                 target:self action:@selector(playPause:)];
 551     }
 552
 553     if ([defaults objectForKey:@"NextTrack"] != nil) {
 554         [[HotKeyCenter sharedCenter] addHotKey:@"NextTrack"
 555                 combo:[defaults keyComboForKey:@"NextTrack"]
 556                 target:self action:@selector(nextSong:)];
 557     }
 558
 559     if ([defaults objectForKey:@"PrevTrack"] != nil) {
 560         [[HotKeyCenter sharedCenter] addHotKey:@"PrevTrack"
 561                 combo:[defaults keyComboForKey:@"PrevTrack"]
 562                 target:self action:@selector(prevSong:)];
 563     }
 564
 565     if ([defaults objectForKey:@"TrackInfo"] != nil) {
 566         [[HotKeyCenter sharedCenter] addHotKey:@"TrackInfo"
 567                 combo:[defaults keyComboForKey:@"TrackInfo"]
 568                 target:self action:@selector(showCurrentTrackInfo)];
 569     }
 570
 571     if ([defaults objectForKey:@"UpcomingSongs"] != nil) {
 572         [[HotKeyCenter sharedCenter] addHotKey:@"UpcomingSongs"
 573                combo:[defaults keyComboForKey:@"UpcomingSongs"]
 574                target:self action:@selector(showUpcomingSongs)];
 575     }
 576 }
 577
 578 //Runs an AppleScript and returns the result as an NSString after stripping quotes, if needed. It takes in script and automatically adds the tell iTunes and end tell statements.
 579 - (NSString *)runScriptAndReturnResult:(NSString *)script
 580 {
 581     AEDesc scriptDesc, resultDesc;
 582     Size length;
 583     NSString *result;
 584     Ptr buffer;
 585
 586     script = [NSString stringWithFormat:@"tell application \"iTunes\"\n%@\nend tell", script];
 587
 588     AECreateDesc(typeChar, [script cString], [script cStringLength],
 589 &scriptDesc);
 590
 591     OSADoScript(asComponent, &scriptDesc, kOSANullScript, typeChar, kOSAModeCanInteract, &resultDesc);
 592
 593     length = AEGetDescDataSize(&resultDesc);
 594     buffer = malloc(length);
 595
 596     AEGetDescData(&resultDesc, buffer, length);
 597     AEDisposeDesc(&scriptDesc);
 598     AEDisposeDesc(&resultDesc);
 599     result = [NSString stringWithCString:buffer length:length];
 600     if ( (! [result isEqualToString:@""])      &&
 601          ([result characterAtIndex:0] == '\"') &&
 602          ([result characterAtIndex:[result length] - 1] == '\"') ) {
 603         result = [result substringWithRange:NSMakeRange(1, [result length] - 2)];
 604     }
 605     free(buffer);
 606     buffer = nil;
 607     return result;
 608 }
 609
 610 //Called when the timer fires.
 611 - (void)timerUpdate
 612 {
 613     int trackPlayingIndex = [currentRemote currentSongIndex];
 614     int playlist = [currentRemote currentPlaylistIndex];
 615
 616     if (trackPlayingIndex != lastSongIndex) {
 617         bool wasPlayingRadio = isPlayingRadio;
 618         isPlayingRadio = [[currentRemote classOfPlaylistAtIndex:playlist] isEqualToString:@"radio tuner playlist"];
 619         if (isPlayingRadio && !wasPlayingRadio) {
 620             int i;
 621             for (i = 0; i < [playlistMenu numberOfItems]; i++)
 622             {
 623                 [[playlistMenu itemAtIndex:i] setState:NSOffState];
 624             }
 625         }
 626         if (wasPlayingRadio) {
 627             NSMenuItem *temp = [[NSMenuItem alloc] initWithTitle:@"" action:NULL keyEquivalent:@""];
 628             [menu insertItem:temp atIndex:trackInfoIndex + 1];
 629             [temp release];
 630         }
 631         [self updateMenu];
 632         lastSongIndex = trackPlayingIndex;
 633     }
 634     else
 635     {
 636         if (playlist != [currentRemote currentPlaylistIndex]) {
 637             bool wasPlayingRadio = isPlayingRadio;
 638             isPlayingRadio = [[currentRemote classOfPlaylistAtIndex:playlist] isEqualToString:@"radio tuner playlist"];
 639             if (isPlayingRadio && !wasPlayingRadio) {
 640                 int i;
 641                 for (i = 0; i < [playlistMenu numberOfItems]; i++)
 642                 {
 643                     [[playlistMenu itemAtIndex:i] setState:NSOffState];
 644                 }
 645             }
 646             if (wasPlayingRadio) {
 647                 NSMenuItem *temp = [[NSMenuItem alloc] initWithTitle:@"" action:NULL keyEquivalent:@""];
 648                 [menu insertItem:temp atIndex:trackInfoIndex + 1];
 649                 [temp release];
 650             }
 651             [self updateMenu];
 652             lastSongIndex = trackPlayingIndex;
 653         }
 654     }
 655     //Update Play/Pause menu item
 656     if (playPauseMenuItem){
 657         if ([[self runScriptAndReturnResult:@"return player state"] isEqualToString:@"playing"]) {
 658             [playPauseMenuItem setTitle:@"Pause"];
 659         } else {
 660             [playPauseMenuItem setTitle:@"Play"];
 661         }
 662     }
 663 }
 664
 665 - (void)iTunesLaunched:(NSNotification *)note
 666 {
 667     NSDictionary *info = [note userInfo];
 668
 669     iTunesPSN.highLongOfPSN = [[info objectForKey:@"NSApplicationProcessSerialNumberHigh"] longValue];
 670     iTunesPSN.lowLongOfPSN = [[info objectForKey:@"NSApplicationProcessSerialNumberLow"] longValue];
 671
 672     //Restart the timer
 673     refreshTimer = [NSTimer scheduledTimerWithTimeInterval:3.5 target:self selector:@selector(timerUpdate) userInfo:nil repeats:YES];
 674
 675     [self rebuildMenu]; //Rebuild the menu since no songs will be playing
 676     [statusItem setMenu:menu]; //Set the menu back to the main one
 677     [[[NSWorkspace sharedWorkspace] notificationCenter] removeObserver:self];
 678
 679     [[[NSWorkspace sharedWorkspace] notificationCenter] addObserver:self selector:@selector(iTunesTerminated:) name:NSWorkspaceDidTerminateApplicationNotification object:nil];
 680 }
 681
 682 - (void)iTunesTerminated:(NSNotification *)note
 683 {
 684     [[[NSWorkspace sharedWorkspace] notificationCenter] removeObserver:self];
 685
 686     [menu release];
 687     menu = [[NSMenu alloc] initWithTitle:@""];
 688     [[menu addItemWithTitle:@"Open iTunes" action:@selector(openiTunes:) keyEquivalent:@""] setTarget:self];
 689     [[menu addItemWithTitle:@"Preferences" action:@selector(showPreferences:) keyEquivalent:@""] setTarget:self];
 690     [[menu addItemWithTitle:@"Quit" action:@selector(quitMenuTunes:) keyEquivalent:@""] setTarget:self];
 691     [statusItem setMenu:menu];
 692
 693     [[[NSWorkspace sharedWorkspace] notificationCenter] addObserver:self selector:@selector(iTunesLaunched:) name:NSWorkspaceDidLaunchApplicationNotification object:nil];
 694     [refreshTimer invalidate];
 695     refreshTimer = nil;
 696     [self clearHotKeys];
 697 }
 698
 699 //Return the PSN of iTunes, if it's running
 700 - (ProcessSerialNumber)iTunesPSN
 701 {
 702     NSArray *apps = [[NSWorkspace sharedWorkspace] launchedApplications];
 703     ProcessSerialNumber number;
 704     int i;
 705
 706     number.highLongOfPSN = kNoProcess;
 707
 708     for (i = 0; i < [apps count]; i++)
 709     {
 710         NSDictionary *curApp = [apps objectAtIndex:i];
 711
 712         if ([[curApp objectForKey:@"NSApplicationName"] isEqualToString:@"iTunes"])
 713         {
 714             number.highLongOfPSN = [[curApp objectForKey:@"NSApplicationProcessSerialNumberHigh"] intValue];
 715             number.lowLongOfPSN = [[curApp objectForKey:@"NSApplicationProcessSerialNumberLow"] intValue];
 716         }
 717     }
 718     return number;
 719 }
 720
 721 //Send an AppleEvent with a given event ID
 722 - (void)sendAEWithEventClass:(AEEventClass)eventClass
 723 andEventID:(AEEventID)eventID
 724 {
 725     OSType iTunesType = 'hook';
 726     AppleEvent event, reply;
 727
 728     AEBuildAppleEvent(eventClass, eventID, typeApplSignature, &iTunesType, sizeof(iTunesType), kAutoGenerateReturnID, kAnyTransactionID, &event, nil, "");
 729
 730     AESend(&event, &reply, kAENoReply, kAENormalPriority, kAEDefaultTimeout, nil, nil);
 731     AEDisposeDesc(&event);
 732     AEDisposeDesc(&reply);
 733 }
 734
 735 //
 736 //
 737 // Selectors - called from status item menu
 738 //
 739 //
 740
 741 // Plugin dependent selectors
 742
 743 - (void)playTrack:(id)sender
 744 {
 745     [currentRemote switchToSongAtIndex:[[sender representedObject] intValue]];
 746     [self updateMenu];
 747 }
 748
 749 - (void)selectPlaylist:(id)sender
 750 {
 751     int playlist = [[sender representedObject] intValue];
 752     if (!isPlayingRadio) {
 753         int curPlaylist = [currentRemote currentPlaylistIndex];
 754         [[playlistMenu itemAtIndex:curPlaylist - 1] setState:NSOffState];
 755     }
 756     [currentRemote switchToPlaylistAtIndex:playlist];
 757     [[playlistMenu itemAtIndex:playlist - 1] setState:NSOnState];
 758     [self updateMenu];
 759 }
 760
 761 - (void)selectEQPreset:(id)sender
 762 {
 763     int curSet = [currentRemote currentEQPresetIndex];
 764     int item = [[sender representedObject] intValue];
 765     [currentRemote switchToEQAtIndex:item];
 766     [[eqMenu itemAtIndex:curSet - 1] setState:NSOffState];
 767     [[eqMenu itemAtIndex:item - 1] setState:NSOnState];
 768 }
 769
 770 - (void)playPause:(id)sender
 771 {
 772     NSString *state = [self runScriptAndReturnResult:@"return player state"];
 773     NSLog(@"%i", [currentRemote playerState]);
 774     if ([state isEqualToString:@"playing"]) {
 775         [currentRemote play];
 776         [playPauseMenuItem setTitle:@"Play"];
 777     } else if ([state isEqualToString:@"fast forwarding"] || [state
 778 isEqualToString:@"rewinding"]) {
 779         [currentRemote play];
 780         [currentRemote pause];
 781     } else {
 782         [currentRemote play];
 783         [playPauseMenuItem setTitle:@"Pause"];
 784     }
 785 }
 786
 787 - (void)nextSong:(id)sender
 788 {
 789     [currentRemote goToNextSong];
 790 }
 791
 792 - (void)prevSong:(id)sender
 793 {
 794     [currentRemote goToPreviousSong];
 795 }
 796
 797 - (void)fastForward:(id)sender
 798 {
 799     [currentRemote fastForward];
 800 }
 801
 802 - (void)rewind:(id)sender
 803 {
 804     [currentRemote rewind];
 805 }
 806
 807 //
 808 //
 809 // Plugin independent selectors
 810 //
 811 //
 812 - (void)quitMenuTunes:(id)sender
 813 {
 814     [NSApp terminate:self];
 815 }
 816
 817 //How is this going to work, now that we're pluginized?
 818 - (void)openiTunes:(id)sender
 819 {
 820     [[NSWorkspace sharedWorkspace] launchApplication:@"iTunes"];
 821 }
 822
 823 - (void)showPreferences:(id)sender
 824 {
 825     if (!prefsController) {
 826         prefsController = [[PreferencesController alloc] initWithMenuTunes:self];
 827         [self clearHotKeys];
 828     }
 829 }
 830
 831
 832 - (void)closePreferences
 833 {
 834     if (!((iTunesPSN.highLongOfPSN == kNoProcess) && (iTunesPSN.lowLongOfPSN == 0))) {
 835         [self setupHotKeys];
 836     }
 837     [prefsController release];
 838     prefsController = nil;
 839 }
 840
 841 //
 842 //
 843 // Show Current Track Info And Show Upcoming Songs Floaters
 844 //
 845 //
 846
 847 - (void)showCurrentTrackInfo
 848 {
 849     NSString *trackName = [self runScriptAndReturnResult:@"return name of current track"];
 850     if (!statusController && [trackName length]) {
 851         NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults];
 852         NSString *stringToShow = @"";
 853
 854         if ([defaults boolForKey:@"showName"]) {
 855             if ([defaults boolForKey:@"showArtist"]) {
 856                 NSString *trackArtist = [currentRemote currentSongArtist];
 857                 trackName = [NSString stringWithFormat:@"%@ - %@", trackArtist, trackName];
 858             }
 859             stringToShow = [stringToShow stringByAppendingString:trackName];
 860             stringToShow = [stringToShow stringByAppendingString:@"\n"];
 861         }
 862
 863         if ([defaults boolForKey:@"showAlbum"]) {
 864             NSString *trackAlbum = [currentRemote currentSongAlbum];
 865             if ([trackAlbum length]) {
 866                 stringToShow = [stringToShow stringByAppendingString:trackAlbum];
 867                 stringToShow = [stringToShow stringByAppendingString:@"\n"];
 868             }
 869         }
 870
 871         if ([defaults boolForKey:@"showTime"]) {
 872             NSString *trackTime = [currentRemote currentSongLength];
 873             NSLog(@"%@", trackTime);
 874             if ([trackTime length]) {
 875                 stringToShow = [NSString stringWithFormat:@"%@Total Time: %@\n", stringToShow, trackTime];
 876             }
 877         }
 878
 879         {
 880             int trackTimeLeft = [[currentRemote currentSongRemaining] intValue];
 881             int minutes = trackTimeLeft / 60, seconds = trackTimeLeft % 60;
 882             if (seconds < 10) {
 883                 stringToShow = [stringToShow stringByAppendingString:
 884                             [NSString stringWithFormat:@"Time Remaining: %i:0%i", minutes, seconds]];
 885             } else {
 886                 stringToShow = [stringToShow stringByAppendingString:
 887                             [NSString stringWithFormat:@"Time Remaining: %i:%i", minutes, seconds]];
 888             }
 889         }
 890
 891         statusController = [[StatusWindowController alloc] init];
 892         [statusController setTrackInfo:stringToShow];
 893         [NSTimer scheduledTimerWithTimeInterval:3.0
 894                                     target:self
 895                                     selector:@selector(fadeAndCloseStatusWindow)
 896                                     userInfo:nil
 897                                     repeats:NO];
 898     }
 899 }
 900
 901 - (void)showUpcomingSongs
 902 {
 903     int curPlaylist = [currentRemote currentPlaylistIndex];
 904     if (!statusController) {
 905         int numSongs = [currentRemote numberOfSongsInPlaylistAtIndex:curPlaylist];
 906
 907         if (numSongs > 0) {
 908             int numSongsInAdvance = [[NSUserDefaults standardUserDefaults] integerForKey:@"SongsInAdvance"];
 909             int curTrack = [currentRemote currentSongIndex];
 910             int i;
 911             NSString *songs = @"";
 912
 913             statusController = [[StatusWindowController alloc] init];
 914             for (i = curTrack + 1; i <= curTrack + numSongsInAdvance; i++) {
 915                 if (i <= numSongs) {
 916                     NSString *curSong = [currentRemote songTitleAtIndex:i];
 917                     songs = [songs stringByAppendingString:curSong];
 918                     songs = [songs stringByAppendingString:@"\n"];
 919                 }
 920             }
 921             [statusController setUpcomingSongs:songs];
 922             [NSTimer scheduledTimerWithTimeInterval:3.0
 923                         target:self
 924                         selector:@selector(fadeAndCloseStatusWindow)
 925                         userInfo:nil
 926                         repeats:NO];
 927         }
 928     }
 929 }
 930
 931 - (void)fadeAndCloseStatusWindow
 932 {
 933     [statusController fadeWindowOut];
 934     [statusController release];
 935     statusController = nil;
 936 }
 937
 938 - (void)setKeyEquivalentForCode:(short)code andModifiers:(long)modifiers
 939         onItem:(NSMenuItem *)item
 940 {
 941     unichar charcode = 'a';
 942     int i;
 943     long cocoaModifiers = 0;
 944     static long carbonToCocoa[6][2] =
 945     {
 946         { cmdKey, NSCommandKeyMask },
 947         { optionKey, NSAlternateKeyMask },
 948         { controlKey, NSControlKeyMask },
 949         { shiftKey, NSShiftKeyMask },
 950     };
 951
 952     for (i = 0; i < 6; i++)
 953     {
 954         if (modifiers & carbonToCocoa[i][0])
 955         {
 956             cocoaModifiers += carbonToCocoa[i][1];
 957         }
 958     }
 959     [item setKeyEquivalentModifierMask:cocoaModifiers];
 960
 961     //Missing key combos for some keys. Must find them later.
 962     switch (code)
 963     {
 964         case 36:
 965             charcode = '\r';
 966         break;
 967
 968         case 48:
 969             charcode = '\t';
 970         break;
 971
 972         //Space -- ARGH!
 973         case 49:
 974         {
 975             /*MenuRef menuRef = _NSGetCarbonMenu([item menu]);
 976             NSLog(@"%@", menuRef);
 977             SetMenuItemCommandKey(menuRef, 0, NO, 49);
 978             SetMenuItemModifiers(menuRef, 0, kMenuNoCommandModifier);
 979             SetMenuItemKeyGlyph(menuRef, 0, kMenuBlankGlyph);
 980             charcode = 'b';*/
 981         }
 982         break;
 983
 984         case 51:
 985             charcode = NSDeleteFunctionKey;
 986         break;
 987
 988         case 53:
 989             charcode = '\e';
 990         break;
 991
 992         case 71:
 993             charcode = '\e';
 994         break;
 995
 996         case 76:
 997             charcode = '\r';
 998         break;
 999
1000         case 96:
1001             charcode = NSF5FunctionKey;
1002         break;
1003
1004         case 97:
1005             charcode = NSF6FunctionKey;
1006         break;
1007
1008         case 98:
1009             charcode = NSF7FunctionKey;
1010         break;
1011
1012         case 99:
1013             charcode = NSF3FunctionKey;
1014         break;
1015
1016         case 100:
1017             charcode = NSF8FunctionKey;
1018         break;
1019
1020         case 101:
1021             charcode = NSF9FunctionKey;
1022         break;
1023
1024         case 103:
1025             charcode = NSF11FunctionKey;
1026         break;
1027
1028         case 105:
1029             charcode = NSF3FunctionKey;
1030         break;
1031
1032         case 107:
1033             charcode = NSF14FunctionKey;
1034         break;
1035
1036         case 109:
1037             charcode = NSF10FunctionKey;
1038         break;
1039
1040         case 111:
1041             charcode = NSF12FunctionKey;
1042         break;
1043
1044         case 113:
1045             charcode = NSF13FunctionKey;
1046         break;
1047
1048         case 114:
1049             charcode = NSInsertFunctionKey;
1050         break;
1051
1052         case 115:
1053             charcode = NSHomeFunctionKey;
1054         break;
1055
1056         case 116:
1057             charcode = NSPageUpFunctionKey;
1058         break;
1059
1060         case 117:
1061             charcode = NSDeleteFunctionKey;
1062         break;
1063
1064         case 118:
1065             charcode = NSF4FunctionKey;
1066         break;
1067
1068         case 119:
1069             charcode = NSEndFunctionKey;
1070         break;
1071
1072         case 120:
1073             charcode = NSF2FunctionKey;
1074         break;
1075
1076         case 121:
1077             charcode = NSPageDownFunctionKey;
1078         break;
1079
1080         case 122:
1081             charcode = NSF1FunctionKey;
1082         break;
1083
1084         case 123:
1085             charcode = NSLeftArrowFunctionKey;
1086         break;
1087
1088         case 124:
1089             charcode = NSRightArrowFunctionKey;
1090         break;
1091
1092         case 125:
1093             charcode = NSDownArrowFunctionKey;
1094         break;
1095
1096         case 126:
1097             charcode = NSUpArrowFunctionKey;
1098         break;
1099     }
1100
1101     if (charcode == 'a') {
1102         unsigned long state;
1103         long keyTrans;
1104         char charCode;
1105         Ptr kchr;
1106         state = 0;
1107         kchr = (Ptr) GetScriptVariable(smCurrentScript, smKCHRCache);
1108         keyTrans = KeyTranslate(kchr, code, &state);
1109         charCode = keyTrans;
1110         [item setKeyEquivalent:[NSString stringWithCString:&charCode length:1]];
1111     } else if (charcode != 'b') {
1112         [item setKeyEquivalent:[NSString stringWithCharacters:&charcode length:1]];
1113     }
1114 }
1115
1116 /*************************************************************************/
1117 #pragma mark -
1118 #pragma mark NSApplication DELEGATE METHODS
1119 /*************************************************************************/
1120
1121 - (void)applicationWillTerminate:(NSNotification *)note
1122 {
1123     [self clearHotKeys];
1124     [[NSStatusBar systemStatusBar] removeStatusItem:statusItem];
1125 }
1126
1127
1128 /*************************************************************************/
1129 #pragma mark -
1130 #pragma mark DEALLOCATION METHODS
1131 /*************************************************************************/
1132
1133 - (void)dealloc
1134 {
1135     if (refreshTimer) {
1136         [refreshTimer invalidate];
1137         refreshTimer = nil;
1138     }
1139     CloseComponent(asComponent);
1140     [currentRemote halt];
1141     [statusItem release];
1142     [menu release];
1143 //  [view release];
1144     [super dealloc];
1145 }
1146
1147 @end