-/*************************************************************************/
-#pragma mark -
-#pragma mark INITIALIZATION METHODS
-/*************************************************************************/
-
-- (void)applicationDidFinishLaunching:(NSNotification *)note
-{
- asComponent = OpenDefaultComponent(kOSAComponentType, kAppleScriptSubtype);
-
- [self registerDefaultsIfNeeded];
-
- menu = [[NSMenu alloc] initWithTitle:@""];
- iTunesPSN = [self iTunesPSN]; //Get PSN of iTunes if it's running
-
- if (!((iTunesPSN.highLongOfPSN == kNoProcess) && (iTunesPSN.lowLongOfPSN == 0)))
- {
- [self rebuildMenu];
- refreshTimer = [NSTimer scheduledTimerWithTimeInterval:3.5
- target:self
- selector:@selector(timerUpdate)
- userInfo:nil
- repeats:YES];
- }
- else
- {
- menu = [[NSMenu alloc] initWithTitle:@""];
- [[menu addItemWithTitle:@"Open iTunes" action:@selector(openiTunes:) keyEquivalent:@""] setTarget:self];
- [[menu addItemWithTitle:@"Preferences" action:@selector(showPreferences:) keyEquivalent:@""] setTarget:self];
- [[menu addItemWithTitle:@"Quit" action:@selector(quitMenuTunes:) keyEquivalent:@""] setTarget:self];
- [[[NSWorkspace sharedWorkspace] notificationCenter] addObserver:self selector:@selector(iTunesLaunched:) name:NSWorkspaceDidLaunchApplicationNotification object:nil];
- refreshTimer = nil;
- }
-
- statusItem = [[NSStatusBar systemStatusBar] statusItemWithLength:NSSquareStatusItemLength];
- [statusItem setImage:[NSImage imageNamed:@"menu.tiff"]];
- [statusItem setHighlightMode:YES];
- [statusItem setMenu:menu];
- [statusItem retain];
- view = [[MenuTunesView alloc] initWithFrame:[[statusItem view] frame]];
- //[statusItem setView:view];
-}
-
-
-/*************************************************************************/
-#pragma mark -
-#pragma mark INSTANCE METHODS
-/*************************************************************************/
-
-- (void)registerDefaultsIfNeeded
-{
- if (![[NSUserDefaults standardUserDefaults] objectForKey:@"menu"]) {
- [[NSUserDefaults standardUserDefaults] setObject:
- [NSArray arrayWithObjects:
- @"Play/Pause",
- @"Next Track",
- @"Previous Track",
- @"Fast Forward",
- @"Rewind",
- @"<separator>",
- @"Upcoming Songs",
- @"Playlists",
- @"<separator>",
- @"PreferencesÉ",
- @"Quit",
- @"<separator>",
- @"Current Track Info",
- nil] forKey:@"menu"];
- }
-}
-
-//Recreate the status item menu
-- (void)rebuildMenu
-{
- NSArray *myMenu = [[NSUserDefaults standardUserDefaults] arrayForKey:@"menu"];
- int i;
-
- trackInfoIndex = -1;
- didHaveAlbumName = ([[self runScriptAndReturnResult:@"return album of current track"] length] > 0);
-
- while ([menu numberOfItems] > 0) {
- [menu removeItemAtIndex:0];
- }
-
- playPauseMenuItem = nil;
- upcomingSongsItem = nil;
- playlistItem = nil;
- [playlistMenu release];
- playlistMenu = nil;
- eqItem = nil;
- [eqMenu release];
- eqMenu = nil;
-
- for (i = 0; i < [myMenu count]; i++) {
- NSString *item = [myMenu objectAtIndex:i];
- if ([item isEqualToString:@"Play/Pause"]) {
- playPauseMenuItem = [menu addItemWithTitle:@"Play"
- action:@selector(playPause:)
- keyEquivalent:@""];
- [playPauseMenuItem setTarget:self];
- } else if ([item isEqualToString:@"Next Track"]) {
- [[menu addItemWithTitle:@"Next Track"
- action:@selector(nextSong:)
- keyEquivalent:@""] setTarget:self];
- } else if ([item isEqualToString:@"Previous Track"]) {
- [[menu addItemWithTitle:@"Previous Track"
- action:@selector(prevSong:)
- keyEquivalent:@""] setTarget:self];
- } else if ([item isEqualToString:@"Fast Forward"]) {
- [[menu addItemWithTitle:@"Fast Forward"
- action:@selector(fastForward:)
- keyEquivalent:@""] setTarget:self];
- } else if ([item isEqualToString:@"Rewind"]) {
- [[menu addItemWithTitle:@"Rewind"
- action:@selector(rewind:)
- keyEquivalent:@""] setTarget:self];
- } else if ([item isEqualToString:@"Upcoming Songs"]) {
- upcomingSongsItem = [menu addItemWithTitle:@"Upcoming Songs"
- action:nil
- keyEquivalent:@""];
- } else if ([item isEqualToString:@"Playlists"]) {
- playlistItem = [menu addItemWithTitle:@"Playlists"
- action:nil
- keyEquivalent:@""];
- } else if ([item isEqualToString:@"EQ Presets"]) {
- eqItem = [menu addItemWithTitle:@"EQ Presets"
- action:nil
- keyEquivalent:@""];
- } else if ([item isEqualToString:@"PreferencesÉ"]) {
- [[menu addItemWithTitle:@"PreferencesÉ"
- action:@selector(showPreferences:)
- keyEquivalent:@""] setTarget:self];
- } else if ([item isEqualToString:@"Quit"]) {
- [[menu addItemWithTitle:@"Quit"
- action:@selector(quitMenuTunes:)
- keyEquivalent:@""] setTarget:self];
- } else if ([item isEqualToString:@"Current Track Info"]) {
- trackInfoIndex = [menu numberOfItems];
- [menu addItemWithTitle:@"No Song"
- action:nil
- keyEquivalent:@""];
- } else if ([item isEqualToString:@"<separator>"]) {
- [menu addItem:[NSMenuItem separatorItem]];
- }
- }
-
- curTrackIndex = -1; //Force update of everything
- [self timerUpdate]; //Updates dynamic info in the menu
-
- [self clearHotKeys];
- [self setupHotKeys];
-}
-
-//Updates the menu with current player state, song, and upcoming songs
-- (void)updateMenu
-{
- NSString *curSongName, *curAlbumName;
- NSMenuItem *menuItem;
-
- if ((iTunesPSN.highLongOfPSN == kNoProcess) && (iTunesPSN.lowLongOfPSN == 0)) {
- return;
- }
-
- //Get the current track name and album.
- curSongName = [self runScriptAndReturnResult:@"return name of current track"];
- curAlbumName = [self runScriptAndReturnResult:@"return album of current track"];
-
- if (upcomingSongsItem) {
- [self rebuildUpcomingSongsMenu];
- }
- if (playlistItem) {
- [self rebuildPlaylistMenu];
- }
- if (eqItem) {
- [self rebuildEQPresetsMenu];
- }
-
- if ([curSongName length] > 0) {
- int index = [menu indexOfItemWithTitle:@"Now Playing"];
-
- if (index > -1) {
- [menu removeItemAtIndex:index + 1];
-
- if (didHaveAlbumName) {
- [menu removeItemAtIndex:index + 1];
- }
- }
-
- if ([curAlbumName length] > 0) {
- menuItem = [[NSMenuItem alloc] initWithTitle:[NSString stringWithFormat:@" %@", curAlbumName]
- action:nil
- keyEquivalent:@""];
- [menu insertItem:menuItem atIndex:trackInfoIndex + 1];
- [menuItem release];
- }
-
- menuItem = [[NSMenuItem alloc] initWithTitle:[NSString stringWithFormat:@" %@", curSongName]
- action:nil
- keyEquivalent:@""];
- [menu insertItem:menuItem atIndex:trackInfoIndex + 1];
- [menuItem release];
-
- if (index == -1) {
- menuItem = [[NSMenuItem alloc] initWithTitle:@"Now Playing" action:nil keyEquivalent:@""];
- [menu removeItemAtIndex:[menu indexOfItemWithTitle:@"No Song"]];
- [menu insertItem:menuItem atIndex:trackInfoIndex];
- [menuItem release];
- }
-
- } else if ([menu indexOfItemWithTitle:@"No Song"] == -1) {
- [menu removeItemAtIndex:trackInfoIndex];
- [menu removeItemAtIndex:trackInfoIndex];
-
- if (didHaveAlbumName) {
- [menu removeItemAtIndex:trackInfoIndex];
- }
-
- menuItem = [[NSMenuItem alloc] initWithTitle:@"No Song" action:nil keyEquivalent:@""];
- [menu insertItem:menuItem atIndex:trackInfoIndex];
- [menuItem release];
- }
-
- didHaveAlbumName = (([curAlbumName length] > 0) ? YES : NO);
-}
-
-//Rebuild the upcoming songs submenu. Can be improved a lot.
-- (void)rebuildUpcomingSongsMenu
-{
- int numSongs = [[self runScriptAndReturnResult:@"return number of tracks in current playlist"] intValue];
- int numSongsInAdvance = [[NSUserDefaults standardUserDefaults] integerForKey:@"SongsInAdvance"];
-
- if (numSongs > 0) {
- int curTrack = [[self runScriptAndReturnResult:@"return index of current track"] intValue];
- int i;
-
- [upcomingSongsMenu release];
- upcomingSongsMenu = [[NSMenu alloc] initWithTitle:@""];
-
- for (i = curTrack + 1; i <= curTrack + numSongsInAdvance; i++) {
- if (i <= numSongs) {
- NSString *curSong = [self runScriptAndReturnResult:[NSString stringWithFormat:@"return name of track %i of current playlist", i]];
- NSMenuItem *songItem;
- songItem = [[NSMenuItem alloc] initWithTitle:curSong action:@selector(playTrack:) keyEquivalent:@""];
- [songItem setTarget:self];
- [songItem setRepresentedObject:[NSNumber numberWithInt:i]];
- [upcomingSongsMenu addItem:songItem];
- [songItem release];
- } else {
- [upcomingSongsMenu addItemWithTitle:@"End of playlist." action:nil keyEquivalent:@""];
- break;
- }
- }
- [upcomingSongsItem setSubmenu:upcomingSongsMenu];
- [upcomingSongsItem setEnabled:YES];
- }
-}
-
-- (void)rebuildPlaylistMenu
-{
- int numPlaylists = [[self runScriptAndReturnResult:@"return number of playlists"] intValue];
- int i, curPlaylist = [[self runScriptAndReturnResult:@"return index of current playlist"] intValue];
-
- if (playlistMenu && (numPlaylists == [playlistMenu numberOfItems]))
- return;
-
- [playlistMenu release];
- playlistMenu = [[NSMenu alloc] initWithTitle:@""];
-
- for (i = 1; i <= numPlaylists; i++) {
- NSString *playlistName = [self runScriptAndReturnResult:[NSString stringWithFormat:@"return name of playlist %i", i]];
- NSMenuItem *tempItem;
- tempItem = [[NSMenuItem alloc] initWithTitle:playlistName action:@selector(selectPlaylist:) keyEquivalent:@""];
- [tempItem setTarget:self];
- [tempItem setRepresentedObject:[NSNumber numberWithInt:i]];
- [playlistMenu addItem:tempItem];
- [tempItem release];
- }
- [playlistItem setSubmenu:playlistMenu];
-
- if (curPlaylist) {
- [[playlistMenu itemAtIndex:curPlaylist - 1] setState:NSOnState];
- }
-}
-
-//Build a menu with the list of all available EQ presets
-- (void)rebuildEQPresetsMenu
-{
- int numSets = [[self runScriptAndReturnResult:@"return number of EQ presets"] intValue];
- int i;
-
- if (eqMenu && (numSets == [eqMenu numberOfItems]))
- return;
-
- [eqMenu release];
- eqMenu = [[NSMenu alloc] initWithTitle:@""];
-
- for (i = 1; i <= numSets; i++) {
- NSString *setName = [self runScriptAndReturnResult:[NSString stringWithFormat:@"return name of EQ preset %i", i]];
- NSMenuItem *tempItem;
- tempItem = [[NSMenuItem alloc] initWithTitle:setName action:@selector(selectEQPreset:) keyEquivalent:@""];
- [tempItem setTarget:self];
- [tempItem setRepresentedObject:[NSNumber numberWithInt:i]];
- [eqMenu addItem:tempItem];
- [tempItem release];
- }
- [eqItem setSubmenu:eqMenu];
-
- [[eqMenu itemAtIndex:[[self runScriptAndReturnResult:@"return index of current EQ preset"] intValue] - 1] setState:NSOnState];
-}
-
-- (void)clearHotKeys
-{
- [[HotKeyCenter sharedCenter] removeHotKey:@"PlayPause"];
- [[HotKeyCenter sharedCenter] removeHotKey:@"NextTrack"];
- [[HotKeyCenter sharedCenter] removeHotKey:@"PrevTrack"];
- [[HotKeyCenter sharedCenter] removeHotKey:@"TrackInfo"];
- [[HotKeyCenter sharedCenter] removeHotKey:@"UpcomingSongs"];
-}
-
-- (void)setupHotKeys
-{
- NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults];
-
- if ([defaults objectForKey:@"PlayPause"] != nil) {
- [[HotKeyCenter sharedCenter] addHotKey:@"PlayPause"
- combo:[defaults keyComboForKey:@"PlayPause"]
- target:self action:@selector(playPause:)];
- }
-
- if ([defaults objectForKey:@"NextTrack"] != nil) {
- [[HotKeyCenter sharedCenter] addHotKey:@"NextTrack"
- combo:[defaults keyComboForKey:@"NextTrack"]
- target:self action:@selector(nextSong:)];
- }
-
- if ([defaults objectForKey:@"PrevTrack"] != nil) {
- [[HotKeyCenter sharedCenter] addHotKey:@"PrevTrack"
- combo:[defaults keyComboForKey:@"PrevTrack"]
- target:self action:@selector(prevSong:)];
- }
-
- if ([defaults objectForKey:@"TrackInfo"] != nil) {
- [[HotKeyCenter sharedCenter] addHotKey:@"TrackInfo"
- combo:[defaults keyComboForKey:@"TrackInfo"]
- target:self action:@selector(showCurrentTrackInfo)];
- }
-
- if ([defaults objectForKey:@"UpcomingSongs"] != nil) {
- [[HotKeyCenter sharedCenter] addHotKey:@"UpcomingSongs"
- combo:[defaults keyComboForKey:@"UpcomingSongs"]
- target:self action:@selector(showUpcomingSongs)];
- }
-}
-
-//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.
-- (NSString *)runScriptAndReturnResult:(NSString *)script
-{
- AEDesc scriptDesc, resultDesc;
- Size length;
- NSString *result;
- Ptr buffer;
-
- script = [NSString stringWithFormat:@"tell application \"iTunes\"\n%@\nend tell", 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;
- return result;
-}
-
-//Called when the timer fires.
-- (void)timerUpdate
-{
- int pid;
-
- if (GetProcessPID(&iTunesPSN, &pid) == noErr) {
- int trackPlayingIndex = [[self runScriptAndReturnResult:@"return index of current track"] intValue];
-
- if (trackPlayingIndex != curTrackIndex) {
- [self updateMenu];
- curTrackIndex = trackPlayingIndex;
- }
-
- //Update Play/Pause menu item
- if (playPauseMenuItem){
- if ([[self runScriptAndReturnResult:@"return player state"] isEqualToString:@"playing"]) {
- [playPauseMenuItem setTitle:@"Pause"];
- } else {
- [playPauseMenuItem setTitle:@"Play"];
- }
- }
- } else {
- [menu release];
- menu = [[NSMenu alloc] initWithTitle:@""];
- [[menu addItemWithTitle:@"Open iTunes" action:@selector(openiTunes:) keyEquivalent:@""] setTarget:self];
- [[menu addItemWithTitle:@"Preferences" action:@selector(showPreferences:) keyEquivalent:@""] setTarget:self];
- [[menu addItemWithTitle:@"Quit" action:@selector(quitMenuTunes:) keyEquivalent:@""] setTarget:self];
- [statusItem setMenu:menu];
-
- [[[NSWorkspace sharedWorkspace] notificationCenter] addObserver:self selector:@selector(iTunesLaunched:) name:NSWorkspaceDidLaunchApplicationNotification object:nil];
- [refreshTimer invalidate];
- refreshTimer = nil;
- [self clearHotKeys];
- }
-}
-
-- (void)iTunesLaunched:(NSNotification *)note
-{
- NSDictionary *info = [note userInfo];
-
- iTunesPSN.highLongOfPSN = [[info objectForKey:@"NSApplicationProcessSerialNumberHigh"] longValue];
- iTunesPSN.lowLongOfPSN = [[info objectForKey:@"NSApplicationProcessSerialNumberLow"] longValue];
-
- //Restart the timer
- refreshTimer = [NSTimer scheduledTimerWithTimeInterval:3.5 target:self selector:@selector(timerUpdate) userInfo:nil repeats:YES];
-
- [self rebuildMenu]; //Rebuild the menu since no songs will be playing
- [statusItem setMenu:menu]; //Set the menu back to the main one
- [[[NSWorkspace sharedWorkspace] notificationCenter] removeObserver:self];
-}
-
-//Return the PSN of iTunes, if it's running
-- (ProcessSerialNumber)iTunesPSN
-{
- ProcessSerialNumber procNum;
- procNum.highLongOfPSN = kNoProcess;
- procNum.lowLongOfPSN = 0;
-
- while ( (GetNextProcess(&procNum) == noErr) ) {
- CFStringRef procName;
- if ( (CopyProcessName(&procNum, &procName) == noErr) ) {
- if ([(NSString *)procName isEqualToString:@"iTunes"]) {
- return procNum;
- }
- [(NSString *)procName release];
- }
- }
- return procNum;
-}
-
-//Send an AppleEvent with a given event ID
-- (void)sendAEWithEventClass:(AEEventClass)eventClass
-andEventID:(AEEventID)eventID
-{
- OSType iTunesType = 'hook';
- AppleEvent event, reply;
-
- AEBuildAppleEvent(eventClass, eventID, typeApplSignature, &iTunesType, sizeof(iTunesType), kAutoGenerateReturnID, kAnyTransactionID, &event, nil, "");
-
- AESend(&event, &reply, kAENoReply, kAENormalPriority, kAEDefaultTimeout, nil, nil);
- AEDisposeDesc(&event);
- AEDisposeDesc(&reply);
-}
-
-//
-// Selectors - called from status item menu
-//
-
-- (void)playTrack:(id)sender
-{
- [self runScriptAndReturnResult:[NSString stringWithFormat:@"play track %i of current playlist", [[sender representedObject] intValue]]];
- [self updateMenu];
-}
-
-- (void)selectPlaylist:(id)sender
-{
- int playlist = [[sender representedObject] intValue];
- [self runScriptAndReturnResult:[NSString stringWithFormat:@"play playlist %i", playlist]];
- [[playlistMenu itemAtIndex:playlist - 1] setState:NSOnState];
- [self updateMenu];
-}
-
-- (void)selectEQPreset:(id)sender
-{
- int curSet = [[self runScriptAndReturnResult:@"return index of current EQ preset"] intValue];
- int item = [[sender representedObject] intValue];
- [self runScriptAndReturnResult:[NSString stringWithFormat:@"set current EQ preset to EQ preset %i", item]];
- [self runScriptAndReturnResult:@"set EQ enabled to 1"];
- [[eqMenu itemAtIndex:curSet - 1] setState:NSOffState];
- [[eqMenu itemAtIndex:item - 1] setState:NSOnState];
-}
-
-- (void)playPause:(id)sender
-{
- NSString *state = [self runScriptAndReturnResult:@"return player state"];
- if ([state isEqualToString:@"playing"]) {
- [self sendAEWithEventClass:'hook' andEventID:'Paus'];
- [playPauseMenuItem setTitle:@"Play"];
- } else if ([state isEqualToString:@"fast forwarding"] || [state
-isEqualToString:@"rewinding"]) {
- [self sendAEWithEventClass:'hook' andEventID:'Paus'];
- [self sendAEWithEventClass:'hook' andEventID:'Play'];
- } else {
- [self sendAEWithEventClass:'hook' andEventID:'Play'];
- [playPauseMenuItem setTitle:@"Pause"];
- }
-}
-
-- (void)nextSong:(id)sender
-{
- [self sendAEWithEventClass:'hook' andEventID:'Next'];
-}
-
-- (void)prevSong:(id)sender