Adding registration hooks. Could use a bit more work. #ifdefed out by default, and...
[MenuTunes.git] / 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 - (void)updateMenu;
17 - (void)rebuildUpcomingSongsMenu;
18 - (void)rebuildPlaylistMenu;
19 - (void)rebuildEQPresetsMenu;
20 - (void)setupHotKeys;
21 - (NSString *)runScriptAndReturnResult:(NSString *)script;
22 - (void)timerUpdate;
23 - (void)sendAEWithEventClass:(AEEventClass)eventClass andEventID:(AEEventID)eventID;
24
25 @end
26
27 @implementation MenuTunes
28
29 /*************************************************************************/
30 #pragma mark -
31 #pragma mark INITIALIZATION METHODS
32 /*************************************************************************/
33
34 - (void)applicationDidFinishLaunching:(NSNotification *)note
35 {
36     asComponent = OpenDefaultComponent(kOSAComponentType, kAppleScriptSubtype);
37
38     [self registerDefaultsIfNeeded];
39     
40     menu = [[NSMenu alloc] initWithTitle:@""];
41     iTunesPSN = [self iTunesPSN]; //Get PSN of iTunes if it's running
42     
43     if (!((iTunesPSN.highLongOfPSN == kNoProcess) && (iTunesPSN.lowLongOfPSN == 0)))
44     {
45         [self rebuildMenu];
46         refreshTimer = [NSTimer scheduledTimerWithTimeInterval:3.5
47                                                     target:self
48                                                     selector:@selector(timerUpdate)
49                                                     userInfo:nil
50                                                     repeats:YES];
51     }
52     else
53     {
54         menu = [[NSMenu alloc] initWithTitle:@""];
55         [[menu addItemWithTitle:@"Open iTunes" action:@selector(openiTunes:) keyEquivalent:@""] setTarget:self];
56         [[menu addItemWithTitle:@"Preferences" action:@selector(showPreferences:) keyEquivalent:@""] setTarget:self];
57         [[menu addItemWithTitle:@"Quit" action:@selector(quitMenuTunes:) keyEquivalent:@""] setTarget:self];
58         [[[NSWorkspace sharedWorkspace] notificationCenter] addObserver:self selector:@selector(iTunesLaunched:) name:NSWorkspaceDidLaunchApplicationNotification object:nil];
59         refreshTimer = nil;
60     }
61
62     statusItem = [[ITStatusItem alloc] initWithStatusBar:[NSStatusBar systemStatusBar]
63                                               withLength:NSSquareStatusItemLength];
64     
65     [statusItem setImage:[NSImage imageNamed:@"menu"]];
66     [statusItem setAlternateImage:[NSImage imageNamed:@"selected_image"]];
67     [statusItem setMenu:menu];
68     // Below line of code is for creating builds for Beta Testers
69     // [statusItem setToolTip:@[NSString stringWithFormat:@"This Nontransferable Beta (Built on %s) of iThink Software's MenuTunes is Registered to: Beta Tester (betatester@somedomain.com).",__DATE__]];
70 }
71
72
73 /*************************************************************************/
74 #pragma mark -
75 #pragma mark INSTANCE METHODS
76 /*************************************************************************/
77
78 - (void)registerDefaultsIfNeeded
79 {
80     NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults];
81     if (![defaults objectForKey:@"menu"]) {
82         bool found = NO;
83         NSMutableDictionary *loginwindow;
84         NSMutableArray *loginarray;
85         int i;
86         
87         [defaults setObject:
88             [NSArray arrayWithObjects:
89                 @"Play/Pause",
90                 @"Next Track",
91                 @"Previous Track",
92                 @"Fast Forward",
93                 @"Rewind",
94                 @"<separator>",
95                 @"Upcoming Songs",
96                 @"Playlists",
97                 @"<separator>",
98                 @"PreferencesÉ",
99                 @"Quit",
100                 @"<separator>",
101                 @"Current Track Info",
102                 nil] forKey:@"menu"];
103         
104         [defaults synchronize];
105         loginwindow = [[defaults persistentDomainForName:@"loginwindow"] mutableCopy];
106         loginarray = [loginwindow objectForKey:@"AutoLaunchedApplicationDictionary"];
107         
108         for (i = 0; i < [loginarray count]; i++) {
109             NSDictionary *tempDict = [loginarray objectAtIndex:i];
110             if ([[[tempDict objectForKey:@"Path"] lastPathComponent] isEqualToString:[[[NSBundle mainBundle] bundlePath] lastPathComponent]]) {
111                 found = YES;
112             }
113         }
114         
115         if (!found) {
116             if (NSRunInformationalAlertPanel(@"Auto-launch MenuTunes", @"Would you like MenuTunes to automatically launch at login?", @"Yes", @"No", nil) == NSOKButton) {
117                 AEDesc scriptDesc, resultDesc;
118                 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]];
119                 
120                 AECreateDesc(typeChar, [script cString], [script cStringLength], 
121             &scriptDesc);
122                 
123                 OSADoScript(asComponent, &scriptDesc, kOSANullScript, typeChar, kOSAModeCanInteract, &resultDesc);
124                 
125                 AEDisposeDesc(&scriptDesc);
126                 AEDisposeDesc(&resultDesc);
127             }
128         }
129     }
130     
131     if (![defaults integerForKey:@"SongsInAdvance"])
132     {
133         [defaults setInteger:5 forKey:@"SongsInAdvance"];
134     }
135     
136     if (![defaults objectForKey:@"showName"]) {
137         [defaults setBool:YES forKey:@"showName"];
138     }
139     
140     if (![defaults objectForKey:@"showArtist"]) {
141         [defaults setBool:YES forKey:@"showArtist"];
142     }
143     
144     if (![defaults objectForKey:@"showAlbum"]) {
145         [defaults setBool:NO forKey:@"showAlbum"];
146     }
147     
148     if (![defaults objectForKey:@"showTime"]) {
149         [defaults setBool:NO forKey:@"showTime"];
150     }
151 }
152
153 //Recreate the status item menu
154 - (void)rebuildMenu
155 {
156     NSArray *myMenu = [[NSUserDefaults standardUserDefaults] arrayForKey:@"menu"];
157     int i;
158     
159     trackInfoIndex = -1;
160     didHaveAlbumName = ([[self runScriptAndReturnResult:@"return album of current track"] length] > 0);
161     didHaveArtistName = ([[self runScriptAndReturnResult:@"return artist of current track"] length] > 0);
162     
163     
164     while ([menu numberOfItems] > 0) {
165         [menu removeItemAtIndex:0];
166     }
167     
168     playPauseMenuItem = nil;
169     upcomingSongsItem = nil;
170     playlistItem = nil;
171     [playlistMenu release];
172     playlistMenu = nil;
173     eqItem = nil;
174     [eqMenu release];
175     eqMenu = nil;
176     
177     for (i = 0; i < [myMenu count]; i++) {
178         NSString *item = [myMenu objectAtIndex:i];
179         if ([item isEqualToString:@"Play/Pause"]) {
180             playPauseMenuItem = [menu addItemWithTitle:@"Play"
181                                                 action:@selector(playPause:)
182                                          keyEquivalent:@""];
183             [playPauseMenuItem setTarget:self];
184         } else if ([item isEqualToString:@"Next Track"]) {
185             [[menu addItemWithTitle:@"Next Track"
186                              action:@selector(nextSong:)
187                       keyEquivalent:@""] setTarget:self];
188         } else if ([item isEqualToString:@"Previous Track"]) {
189             [[menu addItemWithTitle:@"Previous Track"
190                              action:@selector(prevSong:)
191                       keyEquivalent:@""] setTarget:self];
192         } else if ([item isEqualToString:@"Fast Forward"]) {
193             [[menu addItemWithTitle:@"Fast Forward"
194                              action:@selector(fastForward:)
195                       keyEquivalent:@""] setTarget:self];
196         } else if ([item isEqualToString:@"Rewind"]) {
197             [[menu addItemWithTitle:@"Rewind"
198                              action:@selector(rewind:)
199                       keyEquivalent:@""] setTarget:self];
200         } else if ([item isEqualToString:@"Upcoming Songs"]) {
201             upcomingSongsItem = [menu addItemWithTitle:@"Upcoming Songs"
202                                                 action:nil
203                                          keyEquivalent:@""];
204         } else if ([item isEqualToString:@"Playlists"]) {
205             playlistItem = [menu addItemWithTitle:@"Playlists"
206                                            action:nil
207                                     keyEquivalent:@""];
208         } else if ([item isEqualToString:@"EQ Presets"]) {
209             eqItem = [menu addItemWithTitle:@"EQ Presets"
210                                      action:nil
211                               keyEquivalent:@""];
212         } else if ([item isEqualToString:@"PreferencesÉ"]) {
213             [[menu addItemWithTitle:@"PreferencesÉ"
214                              action:@selector(showPreferences:)
215                       keyEquivalent:@""] setTarget:self];
216         } else if ([item isEqualToString:@"Quit"]) {
217             [[menu addItemWithTitle:@"Quit"
218                              action:@selector(quitMenuTunes:)
219                       keyEquivalent:@""] setTarget:self];
220         } else if ([item isEqualToString:@"Current Track Info"]) {
221             trackInfoIndex = [menu numberOfItems];
222             [menu addItemWithTitle:@"No Song"
223                             action:nil
224                      keyEquivalent:@""];
225         } else if ([item isEqualToString:@"<separator>"]) {
226             [menu addItem:[NSMenuItem separatorItem]];
227         }
228     }
229     
230     curTrackIndex = -1; //Force update of everything
231     [self timerUpdate]; //Updates dynamic info in the menu
232     
233     [self clearHotKeys];
234     [self setupHotKeys];
235 }
236
237 //Updates the menu with current player state, song, and upcoming songs
238 - (void)updateMenu
239 {
240     NSMenuItem *menuItem;
241     NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults];
242     
243     if ((iTunesPSN.highLongOfPSN == kNoProcess) && (iTunesPSN.lowLongOfPSN == 0)) {
244         return;
245     }
246     
247     if (upcomingSongsItem) {
248         [self rebuildUpcomingSongsMenu];
249     }
250     
251     if (playlistItem) {
252         [self rebuildPlaylistMenu];
253     }
254     
255     if (eqItem) {
256         [self rebuildEQPresetsMenu];
257     }
258     
259     if (trackInfoIndex > -1)
260     {
261         NSString *curSongName, *curAlbumName = @"", *curArtistName = @"";
262         curSongName = [self runScriptAndReturnResult:@"return name of current track"];
263         
264         if ([defaults boolForKey:@"showAlbum"]) {
265             curAlbumName = [self runScriptAndReturnResult:@"return album of current track"];
266         }
267         
268         if ([defaults boolForKey:@"showArtist"]) {
269             curArtistName = [self runScriptAndReturnResult:@"return artist of current track"];
270         }
271         
272         if ([curSongName length] > 0) {
273             int index = [menu indexOfItemWithTitle:@"Now Playing"];
274             if (index > -1) {
275                 if ([defaults boolForKey:@"showName"]) {
276                     [menu removeItemAtIndex:index + 1];
277                 }
278                 if (didHaveAlbumName) {
279                     [menu removeItemAtIndex:index + 1];
280                 }
281                 if (didHaveArtistName) {
282                     [menu removeItemAtIndex:index + 1];
283                 }
284                 if ([defaults boolForKey:@"showTime"]) {
285                     [menu removeItemAtIndex:index + 1];
286                 }
287             }
288             if (!isPlayingRadio) {
289                 if ([defaults boolForKey:@"showTime"]) {
290                     menuItem = [[NSMenuItem alloc] initWithTitle:[NSString stringWithFormat:@"  %@", [self runScriptAndReturnResult:@"return time of current track"]]
291                                                         action:nil
292                                                         keyEquivalent:@""];
293                     [menu insertItem:menuItem atIndex:trackInfoIndex + 1];
294                     [menuItem release];
295                 }
296                 
297                 if ([curArtistName length] > 0) {
298                     menuItem = [[NSMenuItem alloc] initWithTitle:[NSString stringWithFormat:@"  %@", curArtistName]
299                                                         action:nil
300                                                         keyEquivalent:@""];
301                     [menu insertItem:menuItem atIndex:trackInfoIndex + 1];
302                     [menuItem release];
303                 }
304                 
305                 if ([curAlbumName length] > 0) {
306                     menuItem = [[NSMenuItem alloc] initWithTitle:[NSString stringWithFormat:@"  %@", curAlbumName]
307                                                         action:nil
308                                                         keyEquivalent:@""];
309                     [menu insertItem:menuItem atIndex:trackInfoIndex + 1];
310                     [menuItem release];
311                 }
312             }
313             
314             if ([defaults boolForKey:@"showName"]) {
315                 menuItem = [[NSMenuItem alloc] initWithTitle:[NSString stringWithFormat:@"  %@", curSongName]
316                                                     action:nil
317                                                     keyEquivalent:@""];
318                 [menu insertItem:menuItem atIndex:trackInfoIndex + 1];
319                 [menuItem release];
320             }
321             
322             if (index == -1) {
323                 menuItem = [[NSMenuItem alloc] initWithTitle:@"Now Playing" action:nil keyEquivalent:@""];
324                 [menu removeItemAtIndex:[menu indexOfItemWithTitle:@"No Song"]];
325                 [menu insertItem:menuItem atIndex:trackInfoIndex];
326                 [menuItem release];
327             }
328         } else if ([menu indexOfItemWithTitle:@"No Song"] == -1) {
329             [menu removeItemAtIndex:trackInfoIndex];
330             
331             if ([defaults boolForKey:@"showName"] == YES) {
332                 [menu removeItemAtIndex:trackInfoIndex];
333             }
334             
335             if ([defaults boolForKey:@"showTime"] == YES) {
336                 [menu removeItemAtIndex:trackInfoIndex];
337             }
338             
339             if (didHaveArtistName && [defaults boolForKey:@"showArtist"]) {
340                 [menu removeItemAtIndex:trackInfoIndex];
341             }
342             
343             if (didHaveAlbumName && [defaults boolForKey:@"showAlbum"]) {
344                 [menu removeItemAtIndex:trackInfoIndex];
345             }
346             
347             menuItem = [[NSMenuItem alloc] initWithTitle:@"No Song" action:nil keyEquivalent:@""];
348             [menu insertItem:menuItem atIndex:trackInfoIndex];
349             [menuItem release];
350         }
351         
352         if ([defaults boolForKey:@"showArtist"]) {
353             didHaveAlbumName = (([curArtistName length] > 0) ? YES : NO);
354         }
355             
356         if ([defaults boolForKey:@"showAlbum"]) {
357             didHaveArtistName = (([curAlbumName length] > 0) ? YES : NO);
358         }
359     }
360 }
361
362 //Rebuild the upcoming songs submenu. Can be improved a lot.
363 - (void)rebuildUpcomingSongsMenu
364 {
365     int numSongs = [[self runScriptAndReturnResult:@"return number of tracks in current playlist"] intValue];
366     int numSongsInAdvance = [[NSUserDefaults standardUserDefaults] integerForKey:@"SongsInAdvance"];
367     if (!isPlayingRadio) {
368         if (numSongs > 0) {
369             int curTrack = [[self runScriptAndReturnResult:@"return index of current track"] intValue];
370             int i;
371             
372             [upcomingSongsMenu release];
373             upcomingSongsMenu = [[NSMenu alloc] initWithTitle:@""];
374             [upcomingSongsItem setSubmenu:upcomingSongsMenu];
375             [upcomingSongsItem setEnabled:YES];
376             
377             for (i = curTrack + 1; i <= curTrack + numSongsInAdvance; i++) {
378                 if (i <= numSongs) {
379                     NSString *curSong = [self runScriptAndReturnResult:[NSString stringWithFormat:@"return name of track %i of current playlist", i]];
380                     NSMenuItem *songItem;
381                     songItem = [[NSMenuItem alloc] initWithTitle:curSong action:@selector(playTrack:) keyEquivalent:@""];
382                     [songItem setTarget:self];
383                     [songItem setRepresentedObject:[NSNumber numberWithInt:i]];
384                     [upcomingSongsMenu addItem:songItem];
385                     [songItem release];
386                 } else {
387                     break;
388                 }
389             }
390         }
391     } else {
392         [upcomingSongsItem setSubmenu:nil];
393         [upcomingSongsItem setEnabled:NO];
394     }
395 }
396
397 - (void)rebuildPlaylistMenu
398 {
399     int numPlaylists = [[self runScriptAndReturnResult:@"return number of playlists"] intValue];
400     int i, curPlaylist = [[self runScriptAndReturnResult:@"return index of current playlist"] intValue];
401     
402     if (isPlayingRadio)
403     {
404         curPlaylist = 0;
405     }
406     
407     if (playlistMenu && (numPlaylists == [playlistMenu numberOfItems]))
408         return;
409     
410     [playlistMenu release];
411     playlistMenu = [[NSMenu alloc] initWithTitle:@""];
412     
413     for (i = 1; i <= numPlaylists; i++) {
414         NSString *playlistName = [self runScriptAndReturnResult:[NSString stringWithFormat:@"return name of playlist %i", i]];
415         NSMenuItem *tempItem;
416         tempItem = [[NSMenuItem alloc] initWithTitle:playlistName action:@selector(selectPlaylist:) keyEquivalent:@""];
417         [tempItem setTarget:self];
418         [tempItem setRepresentedObject:[NSNumber numberWithInt:i]];
419         [playlistMenu addItem:tempItem];
420         [tempItem release];
421     }
422     [playlistItem setSubmenu:playlistMenu];
423     
424     if (curPlaylist) {
425         [[playlistMenu itemAtIndex:curPlaylist - 1] setState:NSOnState];
426     }
427 }
428
429 //Build a menu with the list of all available EQ presets
430 - (void)rebuildEQPresetsMenu
431 {
432     int numSets = [[self runScriptAndReturnResult:@"return number of EQ presets"] intValue];
433     int i;
434     
435     if (eqMenu && (numSets == [eqMenu numberOfItems]))
436         return;
437     
438     [eqMenu release];
439     eqMenu = [[NSMenu alloc] initWithTitle:@""];
440     
441     for (i = 1; i <= numSets; i++) {
442         NSString *setName = [self runScriptAndReturnResult:[NSString stringWithFormat:@"return name of EQ preset %i", i]];
443         NSMenuItem *tempItem;
444         tempItem = [[NSMenuItem alloc] initWithTitle:setName action:@selector(selectEQPreset:) keyEquivalent:@""];
445         [tempItem setTarget:self];
446         [tempItem setRepresentedObject:[NSNumber numberWithInt:i]];
447         [eqMenu addItem:tempItem];
448         [tempItem release];
449     }
450     [eqItem setSubmenu:eqMenu];
451     
452     [[eqMenu itemAtIndex:[[self runScriptAndReturnResult:@"return index of current EQ preset"] intValue] - 1] setState:NSOnState];
453 }
454
455 - (void)clearHotKeys
456 {
457     [[HotKeyCenter sharedCenter] removeHotKey:@"PlayPause"];
458     [[HotKeyCenter sharedCenter] removeHotKey:@"NextTrack"];
459     [[HotKeyCenter sharedCenter] removeHotKey:@"PrevTrack"];
460     [[HotKeyCenter sharedCenter] removeHotKey:@"TrackInfo"];
461     [[HotKeyCenter sharedCenter] removeHotKey:@"UpcomingSongs"];
462 }
463
464 - (void)setupHotKeys
465 {
466     NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults];
467     
468     if ([defaults objectForKey:@"PlayPause"] != nil) {
469         [[HotKeyCenter sharedCenter] addHotKey:@"PlayPause"
470                 combo:[defaults keyComboForKey:@"PlayPause"]
471                 target:self action:@selector(playPause:)];
472     }
473     
474     if ([defaults objectForKey:@"NextTrack"] != nil) {
475         [[HotKeyCenter sharedCenter] addHotKey:@"NextTrack"
476                 combo:[defaults keyComboForKey:@"NextTrack"]
477                 target:self action:@selector(nextSong:)];
478     }
479     
480     if ([defaults objectForKey:@"PrevTrack"] != nil) {
481         [[HotKeyCenter sharedCenter] addHotKey:@"PrevTrack"
482                 combo:[defaults keyComboForKey:@"PrevTrack"]
483                 target:self action:@selector(prevSong:)];
484     }
485     
486     if ([defaults objectForKey:@"TrackInfo"] != nil) {
487         [[HotKeyCenter sharedCenter] addHotKey:@"TrackInfo"
488                 combo:[defaults keyComboForKey:@"TrackInfo"]
489                 target:self action:@selector(showCurrentTrackInfo)];
490     }
491     
492     if ([defaults objectForKey:@"UpcomingSongs"] != nil) {
493         [[HotKeyCenter sharedCenter] addHotKey:@"UpcomingSongs"
494                combo:[defaults keyComboForKey:@"UpcomingSongs"]
495                target:self action:@selector(showUpcomingSongs)];
496     }
497 }
498
499 //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.
500 - (NSString *)runScriptAndReturnResult:(NSString *)script
501 {
502     AEDesc scriptDesc, resultDesc;
503     Size length;
504     NSString *result;
505     Ptr buffer;
506     
507     script = [NSString stringWithFormat:@"tell application \"iTunes\"\n%@\nend tell", script];
508     
509     AECreateDesc(typeChar, [script cString], [script cStringLength], 
510 &scriptDesc);
511     
512     OSADoScript(asComponent, &scriptDesc, kOSANullScript, typeChar, kOSAModeCanInteract, &resultDesc);
513     
514     length = AEGetDescDataSize(&resultDesc);
515     buffer = malloc(length);
516     
517     AEGetDescData(&resultDesc, buffer, length);
518     AEDisposeDesc(&scriptDesc);
519     AEDisposeDesc(&resultDesc);
520     result = [NSString stringWithCString:buffer length:length];
521     if ( (! [result isEqualToString:@""])      &&
522          ([result characterAtIndex:0] == '\"') &&
523          ([result characterAtIndex:[result length] - 1] == '\"') ) {
524         result = [result substringWithRange:NSMakeRange(1, [result length] - 2)];
525     }
526     free(buffer);
527     buffer = nil;
528     return result;
529 }
530
531 //Called when the timer fires.
532 - (void)timerUpdate
533 {
534     int pid;
535     if (GetProcessPID(&iTunesPSN, &pid) == noErr) {
536         int trackPlayingIndex = [[self runScriptAndReturnResult:@"return index of current track"] intValue];
537         
538         if (trackPlayingIndex != curTrackIndex) {
539             bool wasPlayingRadio = isPlayingRadio;
540             isPlayingRadio = [[self runScriptAndReturnResult:@"return class of current playlist"] isEqualToString:@"radio tuner playlist"];
541             if (isPlayingRadio && !wasPlayingRadio) {
542                 int i;
543                 for (i = 0; i < [playlistMenu numberOfItems]; i++)
544                 {
545                     [[playlistMenu itemAtIndex:i] setState:NSOffState];
546                 }
547             }
548             if (wasPlayingRadio) {
549                 NSMenuItem *temp = [[NSMenuItem alloc] initWithTitle:@"" action:NULL keyEquivalent:@""];
550                 [menu insertItem:temp atIndex:trackInfoIndex + 1];
551                 [temp release];
552             }
553             [self updateMenu];
554             curTrackIndex = trackPlayingIndex;
555         }
556         else
557         {
558             int playlist = [[self runScriptAndReturnResult:@"return index of current playlist"] intValue];
559             if (playlist != curPlaylistIndex) {
560                 bool wasPlayingRadio = isPlayingRadio;
561                 isPlayingRadio = [[self runScriptAndReturnResult:@"return class of current playlist"] isEqualToString:@"radio tuner playlist"];
562                 if (isPlayingRadio && !wasPlayingRadio) {
563                     int i;
564                     for (i = 0; i < [playlistMenu numberOfItems]; i++)
565                     {
566                         [[playlistMenu itemAtIndex:i] setState:NSOffState];
567                     }
568                 }
569                 if (wasPlayingRadio) {
570                     NSMenuItem *temp = [[NSMenuItem alloc] initWithTitle:@"" action:NULL keyEquivalent:@""];
571                     [menu insertItem:temp atIndex:trackInfoIndex + 1];
572                     [temp release];
573                 }
574                 [self updateMenu];
575                 curTrackIndex = trackPlayingIndex;
576                 curPlaylistIndex = playlist;
577             }
578         }
579         //Update Play/Pause menu item
580         if (playPauseMenuItem){
581             if ([[self runScriptAndReturnResult:@"return player state"] isEqualToString:@"playing"]) {
582                 [playPauseMenuItem setTitle:@"Pause"];
583             } else {
584                 [playPauseMenuItem setTitle:@"Play"];
585             }
586         }
587     } else {
588         [menu release];
589         menu = [[NSMenu alloc] initWithTitle:@""];
590         [[menu addItemWithTitle:@"Open iTunes" action:@selector(openiTunes:) keyEquivalent:@""] setTarget:self];
591         [[menu addItemWithTitle:@"Preferences" action:@selector(showPreferences:) keyEquivalent:@""] setTarget:self];
592         [[menu addItemWithTitle:@"Quit" action:@selector(quitMenuTunes:) keyEquivalent:@""] setTarget:self];
593         [statusItem setMenu:menu];
594         
595         [[[NSWorkspace sharedWorkspace] notificationCenter] addObserver:self selector:@selector(iTunesLaunched:) name:NSWorkspaceDidLaunchApplicationNotification object:nil];
596         [refreshTimer invalidate];
597         refreshTimer = nil;
598         [self clearHotKeys];
599     }
600 }
601
602 - (void)iTunesLaunched:(NSNotification *)note
603 {
604     NSDictionary *info = [note userInfo];
605     
606     iTunesPSN.highLongOfPSN = [[info objectForKey:@"NSApplicationProcessSerialNumberHigh"] longValue];
607     iTunesPSN.lowLongOfPSN = [[info objectForKey:@"NSApplicationProcessSerialNumberLow"] longValue];
608     
609     //Restart the timer
610     refreshTimer = [NSTimer scheduledTimerWithTimeInterval:3.5 target:self selector:@selector(timerUpdate) userInfo:nil repeats:YES]; 
611     
612     [self rebuildMenu]; //Rebuild the menu since no songs will be playing
613     [statusItem setMenu:menu]; //Set the menu back to the main one
614     [[[NSWorkspace sharedWorkspace] notificationCenter] removeObserver:self];
615 }
616
617 //Return the PSN of iTunes, if it's running
618 - (ProcessSerialNumber)iTunesPSN
619 {
620     ProcessSerialNumber procNum;
621     procNum.highLongOfPSN = kNoProcess;
622     procNum.lowLongOfPSN = 0;
623     
624     while ( (GetNextProcess(&procNum) == noErr) ) {
625         CFStringRef procName;
626         if ( (CopyProcessName(&procNum, &procName) == noErr) ) {
627             if ([(NSString *)procName isEqualToString:@"iTunes"]) {
628                 return procNum;
629             }
630             CFRelease(procName);
631         }
632     }
633     return procNum;
634 }
635
636 //Send an AppleEvent with a given event ID
637 - (void)sendAEWithEventClass:(AEEventClass)eventClass 
638 andEventID:(AEEventID)eventID
639 {
640     OSType iTunesType = 'hook';
641     AppleEvent event, reply;
642     
643     AEBuildAppleEvent(eventClass, eventID, typeApplSignature, &iTunesType, sizeof(iTunesType), kAutoGenerateReturnID, kAnyTransactionID, &event, nil, "");
644     
645     AESend(&event, &reply, kAENoReply, kAENormalPriority, kAEDefaultTimeout, nil, nil);
646     AEDisposeDesc(&event);
647     AEDisposeDesc(&reply);
648 }
649
650 //
651 // Selectors - called from status item menu
652 //
653
654 - (void)playTrack:(id)sender
655 {
656     [self runScriptAndReturnResult:[NSString stringWithFormat:@"play track %i of current playlist", [[sender representedObject] intValue]]];
657     [self updateMenu];
658 }
659
660 - (void)selectPlaylist:(id)sender
661 {
662     int playlist = [[sender representedObject] intValue];
663     if (!isPlayingRadio) {
664         int curPlaylist = [[self runScriptAndReturnResult:@"return index of current playlist"] intValue];
665         [[playlistMenu itemAtIndex:curPlaylist - 1] setState:NSOffState];
666     }
667     [self runScriptAndReturnResult:[NSString stringWithFormat:@"play playlist %i", playlist]];
668     [[playlistMenu itemAtIndex:playlist - 1] setState:NSOnState];
669 }
670
671 - (void)selectEQPreset:(id)sender
672 {
673     int curSet = [[self runScriptAndReturnResult:@"return index of current EQ preset"] intValue];
674     int item = [[sender representedObject] intValue];
675     [self runScriptAndReturnResult:[NSString stringWithFormat:@"set current EQ preset to EQ preset %i", item]];
676     [self runScriptAndReturnResult:@"set EQ enabled to 1"];
677     [[eqMenu itemAtIndex:curSet - 1] setState:NSOffState];
678     [[eqMenu itemAtIndex:item - 1] setState:NSOnState];
679 }
680
681 - (void)playPause:(id)sender
682 {
683     NSString *state = [self runScriptAndReturnResult:@"return player state"];
684     if ([state isEqualToString:@"playing"]) {
685         [self sendAEWithEventClass:'hook' andEventID:'Paus'];
686         [playPauseMenuItem setTitle:@"Play"];
687     } else if ([state isEqualToString:@"fast forwarding"] || [state 
688 isEqualToString:@"rewinding"]) {
689         [self sendAEWithEventClass:'hook' andEventID:'Paus'];
690         [self sendAEWithEventClass:'hook' andEventID:'Play'];
691     } else {
692         [self sendAEWithEventClass:'hook' andEventID:'Play'];
693         [playPauseMenuItem setTitle:@"Pause"];
694     }
695 }
696
697 - (void)nextSong:(id)sender
698 {
699     [self sendAEWithEventClass:'hook' andEventID:'Next'];
700 }
701
702 - (void)prevSong:(id)sender
703 {
704     [self sendAEWithEventClass:'hook' andEventID:'Prev'];
705 }
706
707 - (void)fastForward:(id)sender
708 {
709     [self sendAEWithEventClass:'hook' andEventID:'Fast'];
710 }
711
712 - (void)rewind:(id)sender
713 {
714     [self sendAEWithEventClass:'hook' andEventID:'Rwnd'];
715 }
716
717 - (void)quitMenuTunes:(id)sender
718 {
719     [NSApp terminate:self];
720 }
721
722 - (void)openiTunes:(id)sender
723 {
724     [[NSWorkspace sharedWorkspace] launchApplication:@"iTunes"];
725 }
726
727 - (void)showPreferences:(id)sender
728 {
729     if (!prefsController) {
730         prefsController = [[PreferencesController alloc] initWithMenuTunes:self];
731         [self clearHotKeys];
732     }
733 }
734
735
736 - (void)closePreferences
737 {
738     if (!((iTunesPSN.highLongOfPSN == kNoProcess) && (iTunesPSN.lowLongOfPSN == 0))) {
739         [self setupHotKeys];
740     }
741     [prefsController release];
742     prefsController = nil;
743 }
744
745 //
746 //
747 // Show Current Track Info And Show Upcoming Songs Floaters
748 //
749 //
750
751 - (void)showCurrentTrackInfo
752 {
753     NSString *trackName = [self runScriptAndReturnResult:@"return name of current track"];
754     if (!statusController && [trackName length]) {
755         NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults];
756         NSString *stringToShow = @"";
757         int lines = 1;
758         
759         if ([defaults boolForKey:@"showName"]) {
760             if ([defaults boolForKey:@"showArtist"]) {
761                 NSString *trackArtist = [self runScriptAndReturnResult:@"return artist of current track"];
762                 trackName = [NSString stringWithFormat:@"%@ - %@", trackArtist, trackName];
763             }
764             stringToShow = [stringToShow stringByAppendingString:trackName];
765             stringToShow = [stringToShow stringByAppendingString:@"\n"];
766             if ([trackName length] > 38) {
767                 lines++;
768             }
769             lines++;
770         }
771         
772         if ([defaults boolForKey:@"showAlbum"]) {
773             NSString *trackAlbum = [self runScriptAndReturnResult:@"return album of current track"];
774             if ([trackAlbum length]) {
775                 stringToShow = [stringToShow stringByAppendingString:trackAlbum];
776                 stringToShow = [stringToShow stringByAppendingString:@"\n"];
777                 lines++;
778             }
779         }
780         
781         if ([defaults boolForKey:@"showTime"]) {
782             NSString *trackTime = [self runScriptAndReturnResult:@"return time of current track"];
783             if ([trackTime length]) {
784                 stringToShow = [NSString stringWithFormat:@"%@Total Time: %@\n", stringToShow, trackTime];
785                 lines++;
786             }
787         }
788         
789         {
790             int trackTimeLeft = [[self runScriptAndReturnResult:@"return (duration of current track) - player position"] intValue];
791             int minutes = trackTimeLeft / 60, seconds = trackTimeLeft % 60;
792             if (seconds < 10) {
793                 stringToShow = [stringToShow stringByAppendingString:
794                             [NSString stringWithFormat:@"Time Remaining: %i:0%i", minutes, seconds]];
795             } else {
796                 stringToShow = [stringToShow stringByAppendingString:
797                             [NSString stringWithFormat:@"Time Remaining: %i:%i", minutes, seconds]];
798             }
799         }
800         
801         statusController = [[StatusWindowController alloc] init];
802         [statusController setTrackInfo:stringToShow lines:lines];
803         [NSTimer scheduledTimerWithTimeInterval:3.0
804                                     target:self
805                                     selector:@selector(fadeAndCloseStatusWindow)
806                                     userInfo:nil
807                                     repeats:NO];
808     }
809 }
810
811 - (void)showUpcomingSongs
812 {
813     if (!statusController) {
814         int numSongs = [[self runScriptAndReturnResult:@"return number of tracks in current playlist"] intValue];
815         
816         if (numSongs > 0) {
817             int numSongsInAdvance = [[NSUserDefaults standardUserDefaults] integerForKey:@"SongsInAdvance"];
818             int curTrack = [[self runScriptAndReturnResult:@"return index of current track"] intValue];
819             int i;
820             NSString *songs = @"";
821             
822             statusController = [[StatusWindowController alloc] init];
823             for (i = curTrack + 1; i <= curTrack + numSongsInAdvance; i++) {
824                 if (i <= numSongs) {
825                     NSString *curSong = [self runScriptAndReturnResult:
826                         [NSString stringWithFormat:@"return name of track %i of current playlist", i]];
827                     songs = [songs stringByAppendingString:curSong];
828                     songs = [songs stringByAppendingString:@"\n"];
829                 }
830             }
831             [statusController setUpcomingSongs:songs numSongs:numSongsInAdvance];
832             [NSTimer scheduledTimerWithTimeInterval:3.0
833                                              target:self
834                                            selector:@selector(fadeAndCloseStatusWindow)
835                                            userInfo:nil
836                                             repeats:NO];
837         }
838     }
839 }
840
841 - (void)fadeAndCloseStatusWindow
842 {
843     [statusController fadeWindowOut];
844     [statusController release];
845     statusController = nil;
846 }
847
848 /*************************************************************************/
849 #pragma mark -
850 #pragma mark NSApplication DELEGATE METHODS
851 /*************************************************************************/
852
853 - (void)applicationWillTerminate:(NSNotification *)note
854 {
855     [self clearHotKeys];
856     [[NSStatusBar systemStatusBar] removeStatusItem:statusItem];
857 }
858
859
860 /*************************************************************************/
861 #pragma mark -
862 #pragma mark DEALLOCATION METHODS
863 /*************************************************************************/
864
865 - (void)dealloc
866 {
867     if (refreshTimer) {
868         [refreshTimer invalidate];
869         refreshTimer = nil;
870     }
871     CloseComponent(asComponent);
872     [statusItem release];
873     [menu release];
874 //  [view release];
875     [super dealloc];
876 }
877
878
879 @end