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