X-Git-Url: http://git.ithinksw.org/MenuTunes.git/blobdiff_plain/5e4dec969868a410da088273791f228def3b36a7..95bfc9d0fa830da8615ed00808ff347df2f50eb5:/MenuController.m

diff --git a/MenuController.m b/MenuController.m
index 50ffb98..36b112b 100755
--- a/MenuController.m
+++ b/MenuController.m
@@ -63,7 +63,7 @@
     NS_HANDLER
         [[MainController sharedController] networkError:localException];
     NS_ENDHANDLER
-    
+	
     ITDebugLog(@"Reset menu if required.");
     
     //Kill the old submenu items
@@ -444,7 +444,7 @@
                     keyEquivalent:@""];
             [tempItem setSubmenu:_upcomingSongsMenu];
             [tempItem setTag:2];
-            if (_playingRadio || !_currentPlaylist) {
+            if (_playingRadio || _currentPlaylist < 1) {
                 [tempItem setEnabled:NO];
             }
         } else if ([nextObject isEqualToString:@"artists"]) {
@@ -609,8 +609,17 @@
     if (_currentPlaylist && !_playingRadio) {
         if (numSongs > 0) {
             int i;
-            for (i = _currentTrack + 1; i <= _currentTrack + numSongsInAdvance; i++) {
-                if (i <= numSongs) {
+            for (i = _currentTrack + 1; i <= _currentTrack + numSongsInAdvance && i <= numSongs; i++) {
+				BOOL enabled;
+				
+				//Check if the song at this index is enabled for playback. If it isn't, skip over it
+				NS_DURING
+					enabled = [[[MainController sharedController] currentRemote] songEnabledAtIndex:i];
+				NS_HANDLER
+					[[MainController sharedController] networkError:localException];
+				NS_ENDHANDLER
+				
+                if (enabled) {
                     NSString *curSong = nil;
                     NS_DURING
                         curSong = [[[MainController sharedController] currentRemote] songTitleAtIndex:i];
@@ -623,8 +632,8 @@
                     [songItem setTag:i];
                     [songItem setTarget:self];
                 } else {
-                    break;
-                }
+					numSongsInAdvance++;
+				}
             }
         }
         
@@ -700,6 +709,15 @@
 			[tempItem setTag:p + [nextNode index] + 1];
 			[tempItem setTarget:self];
 		}
+		
+		PlaylistNode *root = node;
+		while ([root type] == ITMTPlaylistNode || [root type] == ITMTFolderNode) {
+			root = [root parent];
+		}
+		
+		if ([root index] == [[[MainController sharedController] currentRemote] currentSourceIndex] && [nextNode index] == _currentPlaylist) {
+			[tempItem setState:NSOnState];
+		}
 	}
 }
 
@@ -746,7 +764,7 @@
         }
 		
 		//Add other sources as needed (shared music, iPods, CDs)
-        for (i = 2; i < [playlists count]; i++) {
+        for (i = [playlists count] - 1; i > 1 ; i--) {
             PlaylistNode *nextSource = [playlists objectAtIndex:i];
             if ([nextSource type] != ITMTRemoteRadioSource) {
                 NSString *name = [nextSource name];
@@ -765,15 +783,11 @@
             }
         }
     }
-    ITDebugLog(@"Checking the current source.");
 	NS_DURING
 	if (_currentPlaylist != -1) {
 		if ( (source == ITMTRemoteSharedLibrarySource) || (source == ITMTRemoteiPodSource) || (source == ITMTRemoteGenericDeviceSource) || (source == ITMTRemoteCDSource) ) {
 			tempItem = [playlistsMenu itemAtIndex:[playlistsMenu numberOfItems] + [indices indexOfObject:[NSNumber numberWithInt:[[[MainController sharedController] currentRemote] currentSourceIndex]]] - [indices count]];
 			[tempItem setState:NSOnState];
-			[[[tempItem submenu] itemAtIndex:_currentPlaylist - 1] setState:NSOnState];
-		} else if (source == ITMTRemoteLibrarySource && _currentPlaylist) {
-			[[playlistsMenu itemAtIndex:_currentPlaylist - 2] setState:NSOnState];
 		}
 	}
 	NS_HANDLER