4 // iThink Software, Copyright 2002
10 ¥ Radio mode makes things act oddly
11 ¥ Make preferences window pretty
13 - hot keys can't be set when NSBGOnly is on. The window is not key,
14 so the KeyBroadcaster does not pick up key combos
15 - going to need a different way of defining key combos
16 ¥ Optimize, this thing is big and slow :(
17 ¥ Apple Events! Apple Events! Apple Events!
19 ¥ I think I found a slight memory leak:
20 425 MenuTunes 7.8% 8:29.87 1 56 4827 215M+ 3.14M 135M- 599M+
24 #import "MenuTunesView.h"
25 #import "PreferencesController.h"
26 #import "HotKeyCenter.h"
27 #import "StatusWindowController.h"
29 @implementation MenuTunes
31 - (void)applicationDidFinishLaunching:(NSNotification *)note
33 if (![[NSUserDefaults standardUserDefaults] objectForKey:@"menu"])
35 [[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"];
38 menu = [[NSMenu alloc] initWithTitle:@""];
39 iTunesPSN = [self iTunesPSN]; //Get PSN of iTunes if it's running
41 if (!((iTunesPSN.highLongOfPSN == kNoProcess) && (iTunesPSN.lowLongOfPSN == 0)))
44 refreshTimer = [NSTimer scheduledTimerWithTimeInterval:3.5
45 target:self selector:@selector(timerUpdate) userInfo:nil repeats:YES];
49 menu = [[NSMenu alloc] initWithTitle:@""];
50 [[menu addItemWithTitle:@"Open iTunes" action:@selector(openiTunes:) keyEquivalent:@""] setTarget:self];
51 [[menu addItemWithTitle:@"Preferences" action:@selector(showPreferences:) keyEquivalent:@""] setTarget:self];
52 [[menu addItemWithTitle:@"Quit" action:@selector(quitMenuTunes:) keyEquivalent:@""] setTarget:self];
53 [[[NSWorkspace sharedWorkspace] notificationCenter] addObserver:self selector:@selector(iTunesLaunched:) name:NSWorkspaceDidLaunchApplicationNotification object:nil];
57 statusItem = [[NSStatusBar systemStatusBar] statusItemWithLength:NSSquareStatusItemLength];
58 [statusItem setImage:[NSImage imageNamed:@"menu.tiff"]];
59 [statusItem setHighlightMode:YES];
60 [statusItem setMenu:menu];
62 view = [[MenuTunesView alloc] initWithFrame:[[statusItem view] frame]];
63 //[statusItem setView:view];
66 - (void)applicationWillTerminate:(NSNotification *)note
69 [[NSStatusBar systemStatusBar] removeStatusItem:statusItem];
76 [refreshTimer invalidate];
84 //Recreate the status item menu
87 NSArray *myMenu = [[NSUserDefaults standardUserDefaults] arrayForKey:@"menu"];
91 didHaveAlbumName = (([[self runScriptAndReturnResult:@"return album of current track"] length] > 0) ? YES : NO);
93 while ([menu numberOfItems] > 0)
95 [menu removeItemAtIndex:0];
98 playPauseMenuItem = nil;
99 upcomingSongsItem = nil;
101 [playlistMenu release];
107 for (i = 0; i < [myMenu count]; i++)
109 NSString *item = [myMenu objectAtIndex:i];
110 if ([item isEqualToString:@"Play/Pause"])
112 playPauseMenuItem = [menu addItemWithTitle:@"Play" action:@selector(playPause:) keyEquivalent:@""];
113 [playPauseMenuItem setTarget:self];
115 else if ([item isEqualToString:@"Next Track"])
117 [[menu addItemWithTitle:@"Next Track" action:@selector(nextSong:) keyEquivalent:@""] setTarget:self];
119 else if ([item isEqualToString:@"Previous Track"])
121 [[menu addItemWithTitle:@"Previous Track" action:@selector(prevSong:) keyEquivalent:@""] setTarget:self];
123 else if ([item isEqualToString:@"Fast Forward"])
125 [[menu addItemWithTitle:@"Fast Forward" action:@selector(fastForward:) keyEquivalent:@""] setTarget:self];
127 else if ([item isEqualToString:@"Rewind"])
129 [[menu addItemWithTitle:@"Rewind" action:@selector(rewind:) keyEquivalent:@""] setTarget:self];
131 else if ([item isEqualToString:@"Upcoming Songs"])
133 upcomingSongsItem = [menu addItemWithTitle:@"Upcoming Songs" action:NULL keyEquivalent:@""];
135 else if ([item isEqualToString:@"Playlists"])
137 playlistItem = [menu addItemWithTitle:@"Playlists" action:NULL keyEquivalent:@""];
139 else if ([item isEqualToString:@"EQ Presets"])
141 eqItem = [menu addItemWithTitle:@"EQ Presets" action:NULL keyEquivalent:@""];
143 else if ([item isEqualToString:@"PreferencesÉ"])
145 [[menu addItemWithTitle:@"PreferencesÉ" action:@selector(showPreferences:) keyEquivalent:@""] setTarget:self];
147 else if ([item isEqualToString:@"Quit"])
149 [[menu addItemWithTitle:@"Quit" action:@selector(quitMenuTunes:) keyEquivalent:@""] setTarget:self];
151 else if ([item isEqualToString:@"Current Track Info"])
153 trackInfoIndex = [menu numberOfItems];
154 [menu addItemWithTitle:@"No Song" action:NULL keyEquivalent:@""];
156 else if ([item isEqualToString:@"<separator>"])
158 [menu addItem:[NSMenuItem separatorItem]];
162 curTrackIndex = -1; //Force update of everything
163 [self timerUpdate]; //Updates dynamic info in the menu
169 //Updates the menu with current player state, song, and upcoming songs
172 NSString *curSongName, *curAlbumName;
173 NSMenuItem *menuItem;
175 if ((iTunesPSN.highLongOfPSN == kNoProcess) && (iTunesPSN.lowLongOfPSN == 0))
180 //Get the current track name and album.
181 curSongName = [self runScriptAndReturnResult:@"return name of current track"];
182 curAlbumName = [self runScriptAndReturnResult:@"return album of current track"];
184 if (upcomingSongsItem)
186 [self rebuildUpcomingSongsMenu];
190 [self rebuildPlaylistMenu];
194 [self rebuildEQPresetsMenu];
197 if ([curSongName length] > 0)
199 int index = [menu indexOfItemWithTitle:@"Now Playing"];
203 [menu removeItemAtIndex:index + 1];
204 if (didHaveAlbumName)
206 [menu removeItemAtIndex:index + 1];
210 if ([curAlbumName length] > 0)
212 menuItem = [[NSMenuItem alloc] initWithTitle:[NSString stringWithFormat:@" %@", curAlbumName] action:NULL keyEquivalent:@""];
213 [menu insertItem:menuItem atIndex:trackInfoIndex + 1];
217 menuItem = [[NSMenuItem alloc] initWithTitle:[NSString stringWithFormat:@" %@", curSongName] action:NULL keyEquivalent:@""];
218 [menu insertItem:menuItem atIndex:trackInfoIndex + 1];
223 menuItem = [[NSMenuItem alloc] initWithTitle:@"Now Playing" action:NULL keyEquivalent:@""];
224 [menu removeItemAtIndex:[menu indexOfItemWithTitle:@"No Song"]];
225 [menu insertItem:menuItem atIndex:trackInfoIndex];
229 else if ([menu indexOfItemWithTitle:@"No Song"] == -1)
231 [menu removeItemAtIndex:trackInfoIndex];
232 [menu removeItemAtIndex:trackInfoIndex];
233 if (didHaveAlbumName)
235 [menu removeItemAtIndex:trackInfoIndex];
237 menuItem = [[NSMenuItem alloc] initWithTitle:@"No Song" action:NULL keyEquivalent:@""];
238 [menu insertItem:menuItem atIndex:trackInfoIndex];
242 didHaveAlbumName = (([curAlbumName length] > 0) ? YES : NO);
245 //Rebuild the upcoming songs submenu. Can be improved a lot.
246 - (void)rebuildUpcomingSongsMenu
248 int numSongs = [[self runScriptAndReturnResult:@"return number of tracks in current playlist"] intValue];
249 int numSongsInAdvance = [[NSUserDefaults standardUserDefaults] integerForKey:@"SongsInAdvance"];
253 int curTrack = [[self runScriptAndReturnResult:@"return index of current track"] intValue];
256 [upcomingSongsMenu release];
257 upcomingSongsMenu = [[NSMenu alloc] initWithTitle:@""];
259 for (i = curTrack + 1; i <= curTrack + numSongsInAdvance; i++)
263 NSString *curSong = [self runScriptAndReturnResult:[NSString stringWithFormat:@"return name of track %i of current playlist", i]];
264 NSMenuItem *songItem;
265 songItem = [[NSMenuItem alloc] initWithTitle:curSong action:@selector(playTrack:) keyEquivalent:@""];
266 [songItem setTarget:self];
267 [songItem setEnabled:YES];
268 [songItem setRepresentedObject:[NSNumber numberWithInt:i]];
269 [upcomingSongsMenu addItem:songItem];
274 [upcomingSongsMenu addItemWithTitle:@"End of playlist." action:NULL keyEquivalent:@""];
278 [upcomingSongsItem setSubmenu:upcomingSongsMenu];
279 [upcomingSongsItem setEnabled:YES];
283 - (void)rebuildPlaylistMenu
285 int numPlaylists = [[self runScriptAndReturnResult:@"return number of playlists"] intValue];
286 int i, curPlaylist = [[self runScriptAndReturnResult:@"return index of current playlist"] intValue];
288 if (playlistMenu && (numPlaylists == [playlistMenu numberOfItems]))
291 [playlistMenu release];
292 playlistMenu = [[NSMenu alloc] initWithTitle:@""];
294 for (i = 1; i <= numPlaylists; i++)
296 NSString *playlistName = [self runScriptAndReturnResult:[NSString stringWithFormat:@"return name of playlist %i", i]];
297 NSMenuItem *tempItem;
298 tempItem = [[NSMenuItem alloc] initWithTitle:playlistName action:@selector(selectPlaylist:) keyEquivalent:@""];
299 [tempItem setTarget:self];
300 [tempItem setRepresentedObject:[NSNumber numberWithInt:i]];
301 [playlistMenu addItem:tempItem];
304 [playlistItem setSubmenu:playlistMenu];
308 [[playlistMenu itemAtIndex:curPlaylist - 1] setState:NSOnState];
312 //Build a menu with the list of all available EQ presets
313 - (void)rebuildEQPresetsMenu
315 int numSets = [[self runScriptAndReturnResult:@"return number of EQ presets"] intValue];
318 if (eqMenu && (numSets == [eqMenu numberOfItems]))
322 eqMenu = [[NSMenu alloc] initWithTitle:@""];
324 for (i = 1; i <= numSets; i++)
326 NSString *setName = [self runScriptAndReturnResult:[NSString stringWithFormat:@"return name of EQ preset %i", i]];
327 NSMenuItem *tempItem;
328 tempItem = [[NSMenuItem alloc] initWithTitle:setName action:@selector(selectEQPreset:) keyEquivalent:@""];
329 [tempItem setTarget:self];
330 [tempItem setRepresentedObject:[NSNumber numberWithInt:i]];
331 [eqMenu addItem:tempItem];
334 [eqItem setSubmenu:eqMenu];
336 [[eqMenu itemAtIndex:[[self runScriptAndReturnResult:@"return index of current EQ preset"] intValue] - 1] setState:NSOnState];
341 [[HotKeyCenter sharedCenter] removeHotKey:@"PlayPause"];
342 [[HotKeyCenter sharedCenter] removeHotKey:@"NextTrack"];
343 [[HotKeyCenter sharedCenter] removeHotKey:@"PrevTrack"];
344 [[HotKeyCenter sharedCenter] removeHotKey:@"TrackInfo"];
345 [[HotKeyCenter sharedCenter] removeHotKey:@"UpcomingSongs"];
350 NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults];
352 if ([defaults objectForKey:@"PlayPause"] != nil)
354 [[HotKeyCenter sharedCenter] addHotKey:@"PlayPause"
355 combo:[defaults keyComboForKey:@"PlayPause"]
356 target:self action:@selector(playPause:)];
359 if ([defaults objectForKey:@"NextTrack"] != nil)
361 [[HotKeyCenter sharedCenter] addHotKey:@"NextTrack"
362 combo:[defaults keyComboForKey:@"NextTrack"]
363 target:self action:@selector(nextSong:)];
366 if ([defaults objectForKey:@"PrevTrack"] != nil)
368 [[HotKeyCenter sharedCenter] addHotKey:@"PrevTrack"
369 combo:[defaults keyComboForKey:@"PrevTrack"]
370 target:self action:@selector(prevSong:)];
373 if ([defaults objectForKey:@"TrackInfo"] != nil)
375 [[HotKeyCenter sharedCenter] addHotKey:@"TrackInfo"
376 combo:[defaults keyComboForKey:@"TrackInfo"]
377 target:self action:@selector(showCurrentTrackInfo)];
380 if ([defaults objectForKey:@"UpcomingSongs"] != nil)
382 [[HotKeyCenter sharedCenter] addHotKey:@"UpcomingSongs"
383 combo:[defaults keyComboForKey:@"UpcomingSongs"]
384 target:self action:@selector(showUpcomingSongs)];
388 //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.
389 - (NSString *)runScriptAndReturnResult:(NSString *)script
391 AEDesc scriptDesc, resultDesc;
396 script = [NSString stringWithFormat:@"tell application \"iTunes\"\n%@\nend tell", script];
398 AECreateDesc(typeChar, [script cString], [script cStringLength],
401 OSADoScript(OpenDefaultComponent(kOSAComponentType, kAppleScriptSubtype), &scriptDesc, kOSANullScript, typeChar, kOSAModeCanInteract, &resultDesc);
403 length = AEGetDescDataSize(&resultDesc);
404 buffer = malloc(length);
406 AEGetDescData(&resultDesc, buffer, length);
407 AEDisposeDesc(&scriptDesc);
408 AEDisposeDesc(&resultDesc);
409 result = [NSString stringWithCString:buffer length:length];
410 if (![result isEqualToString:@""] &&
411 ([result characterAtIndex:0] == '\"') &&
412 ([result characterAtIndex:[result length] - 1] == '\"'))
414 result = [result substringWithRange:NSMakeRange(1, [result length] - 2)];
421 //Called when the timer fires.
426 if (GetProcessPID(&iTunesPSN, &pid) == noErr)
428 int trackPlayingIndex = [[self runScriptAndReturnResult:@"return index of current track"] intValue];
430 if (trackPlayingIndex != curTrackIndex)
433 curTrackIndex = trackPlayingIndex;
436 //Update Play/Pause menu item
437 if (playPauseMenuItem)
439 if ([[self runScriptAndReturnResult:@"return player state"] isEqualToString:@"playing"])
441 [playPauseMenuItem setTitle:@"Pause"];
445 [playPauseMenuItem setTitle:@"Play"];
452 menu = [[NSMenu alloc] initWithTitle:@""];
453 [[menu addItemWithTitle:@"Open iTunes" action:@selector(openiTunes:) keyEquivalent:@""] setTarget:self];
454 [[menu addItemWithTitle:@"Preferences" action:@selector(showPreferences:) keyEquivalent:@""] setTarget:self];
455 [[menu addItemWithTitle:@"Quit" action:@selector(quitMenuTunes:) keyEquivalent:@""] setTarget:self];
456 [statusItem setMenu:menu];
458 [[[NSWorkspace sharedWorkspace] notificationCenter] addObserver:self selector:@selector(iTunesLaunched:) name:NSWorkspaceDidLaunchApplicationNotification object:nil];
459 [refreshTimer invalidate];
465 - (void)iTunesLaunched:(NSNotification *)note
467 NSDictionary *info = [note userInfo];
469 iTunesPSN.highLongOfPSN = [[info objectForKey:@"NSApplicationProcessSerialNumberHigh"] longValue];
470 iTunesPSN.lowLongOfPSN = [[info objectForKey:@"NSApplicationProcessSerialNumberLow"] longValue];
473 refreshTimer = [NSTimer scheduledTimerWithTimeInterval:3.5 target:self selector:@selector(timerUpdate) userInfo:nil repeats:YES];
475 [self rebuildMenu]; //Rebuild the menu since no songs will be playing
476 [statusItem setMenu:menu]; //Set the menu back to the main one
477 [[[NSWorkspace sharedWorkspace] notificationCenter] removeObserver:self];
480 //Return the PSN of iTunes, if it's running
481 - (ProcessSerialNumber)iTunesPSN
483 ProcessSerialNumber procNum;
484 procNum.highLongOfPSN = kNoProcess;
485 procNum.lowLongOfPSN = 0;
487 while ( (GetNextProcess(&procNum) == noErr) )
489 CFStringRef procName;
490 if ( (CopyProcessName(&procNum, &procName) == noErr) )
492 if ([(NSString *)procName isEqualToString:@"iTunes"])
496 [(NSString *)procName release];
502 //Send an AppleEvent with a given event ID
503 - (void)sendAEWithEventClass:(AEEventClass)eventClass
504 andEventID:(AEEventID)eventID
506 OSType iTunesType = 'hook';
507 AppleEvent event, reply;
509 AEBuildAppleEvent(eventClass, eventID, typeApplSignature, &iTunesType, sizeof(iTunesType), kAutoGenerateReturnID, kAnyTransactionID, &event, NULL, "");
511 AESend(&event, &reply, kAENoReply, kAENormalPriority, kAEDefaultTimeout, nil, nil);
512 AEDisposeDesc(&event);
513 AEDisposeDesc(&reply);
517 // Selectors - called from status item menu
520 - (void)playTrack:(id)sender
522 [self runScriptAndReturnResult:[NSString stringWithFormat:@"play track %i of current playlist", [[sender representedObject] intValue]]];
526 - (void)selectPlaylist:(id)sender
528 int playlist = [[sender representedObject] intValue];
529 [self runScriptAndReturnResult:[NSString stringWithFormat:@"play playlist %i", playlist]];
530 [[playlistMenu itemAtIndex:playlist - 1] setState:NSOnState];
534 - (void)selectEQPreset:(id)sender
536 int curSet = [[self runScriptAndReturnResult:@"return index of current EQ preset"] intValue];
537 int item = [[sender representedObject] intValue];
538 [self runScriptAndReturnResult:[NSString stringWithFormat:@"set current EQ preset to EQ preset %i", item]];
539 [self runScriptAndReturnResult:@"set EQ enabled to 1"];
540 [[eqMenu itemAtIndex:curSet - 1] setState:NSOffState];
541 [[eqMenu itemAtIndex:item - 1] setState:NSOnState];
544 - (void)playPause:(id)sender
546 NSString *state = [self runScriptAndReturnResult:@"return player state"];
547 if ([state isEqualToString:@"playing"])
549 [self sendAEWithEventClass:'hook' andEventID:'Paus'];
550 [playPauseMenuItem setTitle:@"Play"];
552 else if ([state isEqualToString:@"fast forwarding"] || [state
553 isEqualToString:@"rewinding"])
555 [self sendAEWithEventClass:'hook' andEventID:'Paus'];
556 [self sendAEWithEventClass:'hook' andEventID:'Play'];
560 [self sendAEWithEventClass:'hook' andEventID:'Play'];
561 [playPauseMenuItem setTitle:@"Pause"];
565 - (void)nextSong:(id)sender
567 [self sendAEWithEventClass:'hook' andEventID:'Next'];
570 - (void)prevSong:(id)sender
572 [self sendAEWithEventClass:'hook' andEventID:'Prev'];
575 - (void)fastForward:(id)sender
577 [self sendAEWithEventClass:'hook' andEventID:'Fast'];
580 - (void)rewind:(id)sender
582 [self sendAEWithEventClass:'hook' andEventID:'Rwnd'];
585 - (void)quitMenuTunes:(id)sender
587 [NSApp terminate:self];
590 - (void)openiTunes:(id)sender
592 [[NSWorkspace sharedWorkspace] launchApplication:@"iTunes"];
595 - (void)showPreferences:(id)sender
597 if (!prefsController)
599 prefsController = [[PreferencesController alloc] initWithMenuTunes:self];
605 - (void)closePreferences
607 if (!((iTunesPSN.highLongOfPSN == kNoProcess) && (iTunesPSN.lowLongOfPSN == 0)))
611 [prefsController release];
612 prefsController = nil;
617 // Show Current Track Info And Show Upcoming Songs Floaters
621 - (void)showCurrentTrackInfo
623 NSString *trackName = [self runScriptAndReturnResult:@"return name of current track"];
624 if (!statusController && [trackName length])
626 NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults];
627 NSString *stringToShow = @"";
630 if ([defaults boolForKey:@"showName"])
632 if ([defaults boolForKey:@"showArtist"])
634 NSString *trackArtist = [self runScriptAndReturnResult:@"return artist of current track"];
635 trackName = [NSString stringWithFormat:@"%@ - %@", trackArtist, trackName];
637 stringToShow = [stringToShow stringByAppendingString:trackName];
638 stringToShow = [stringToShow stringByAppendingString:@"\n"];
639 if ([trackName length] > 38)
646 if ([defaults boolForKey:@"showAlbum"])
648 NSString *trackAlbum = [self runScriptAndReturnResult:@"return album of current track"];
649 if ([trackAlbum length])
651 stringToShow = [stringToShow stringByAppendingString:trackAlbum];
652 stringToShow = [stringToShow stringByAppendingString:@"\n"];
657 if ([defaults boolForKey:@"showTime"])
659 NSString *trackTime = [self runScriptAndReturnResult:@"return time of current track"];
660 if ([trackTime length])
662 stringToShow = [NSString stringWithFormat:@"%@Total Time: %@\n", stringToShow, trackTime];
668 int trackTimeLeft = [[self runScriptAndReturnResult:@"return (duration of current track) - player position"] intValue];
669 int minutes = trackTimeLeft / 60, seconds = trackTimeLeft % 60;
672 stringToShow = [stringToShow stringByAppendingString:
673 [NSString stringWithFormat:@"Time Remaining: %i:0%i", minutes, seconds]];
677 stringToShow = [stringToShow stringByAppendingString:
678 [NSString stringWithFormat:@"Time Remaining: %i:%i", minutes, seconds]];
682 statusController = [[StatusWindowController alloc] init];
683 [statusController setTrackInfo:stringToShow lines:lines];
684 [NSTimer scheduledTimerWithTimeInterval:3.0 target:self selector:@selector(fadeAndCloseStatusWindow) userInfo:nil repeats:NO];
688 - (void)showUpcomingSongs
690 if (!statusController)
692 int numSongs = [[self runScriptAndReturnResult:@"return number of tracks in current playlist"] intValue];
696 int numSongsInAdvance = [[NSUserDefaults standardUserDefaults] integerForKey:@"SongsInAdvance"];
697 int curTrack = [[self runScriptAndReturnResult:@"return index of current track"] intValue];
699 NSString *songs = @"";
701 statusController = [[StatusWindowController alloc] init];
702 for (i = curTrack + 1; i <= curTrack + numSongsInAdvance; i++)
706 NSString *curSong = [self runScriptAndReturnResult:[NSString stringWithFormat:@"return name of track %i of current playlist", i]];
707 songs = [songs stringByAppendingString:curSong];
708 songs = [songs stringByAppendingString:@"\n"];
711 [statusController setUpcomingSongs:songs numSongs:numSongsInAdvance];
712 [NSTimer scheduledTimerWithTimeInterval:3.0 target:self selector:@selector(fadeAndCloseStatusWindow) userInfo:nil repeats:NO];
717 - (void)fadeAndCloseStatusWindow
719 [statusController fadeWindowOut];
720 [statusController release];
721 statusController = nil;