1 files changed, 247 insertions, 0 deletions
diff --git a/subsonic-main/src/main/java/net/sourceforge/subsonic/controller/WapController.java b/subsonic-main/src/main/java/net/sourceforge/subsonic/controller/WapController.java
new file mode 100644
index 00000000..02509687
--- /dev/null
+++ b/subsonic-main/src/main/java/net/sourceforge/subsonic/controller/WapController.java
@@ -0,0 +1,247 @@
+/*
+ This file is part of Subsonic.
+
+ Subsonic is free software: you can redistribute it and/or modify
+ it under the terms of the GNU General Public License as published by
+ the Free Software Foundation, either version 3 of the License, or
+ (at your option) any later version.
+
+ Subsonic is distributed in the hope that it will be useful,
+ but WITHOUT ANY WARRANTY; without even the implied warranty of
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ GNU General Public License for more details.
+
+ You should have received a copy of the GNU General Public License
+ along with Subsonic.  If not, see <http://www.gnu.org/licenses/>.
+
+ Copyright 2009 (C) Sindre Mehus
+ */
+package net.sourceforge.subsonic.controller;
+
+import java.io.IOException;
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.SortedMap;
+import java.util.SortedSet;
+
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+
+import org.springframework.web.bind.ServletRequestUtils;
+import org.springframework.web.servlet.ModelAndView;
+import org.springframework.web.servlet.mvc.multiaction.MultiActionController;
+
+import net.sourceforge.subsonic.domain.MediaFile;
+import net.sourceforge.subsonic.domain.MusicFolder;
+import net.sourceforge.subsonic.domain.MusicIndex;
+import net.sourceforge.subsonic.domain.Player;
+import net.sourceforge.subsonic.domain.PlayQueue;
+import net.sourceforge.subsonic.domain.RandomSearchCriteria;
+import net.sourceforge.subsonic.domain.SearchCriteria;
+import net.sourceforge.subsonic.domain.SearchResult;
+import net.sourceforge.subsonic.domain.User;
+import net.sourceforge.subsonic.service.SearchService;
+import net.sourceforge.subsonic.service.MediaFileService;
+import net.sourceforge.subsonic.service.MusicIndexService;
+import net.sourceforge.subsonic.service.PlayerService;
+import net.sourceforge.subsonic.service.PlaylistService;
+import net.sourceforge.subsonic.service.SecurityService;
+import net.sourceforge.subsonic.service.SettingsService;
+
+/**
+ * Multi-controller used for wap pages.
+ *
+ * @author Sindre Mehus
+ */
+public class WapController extends MultiActionController {
+
+    private SettingsService settingsService;
+    private PlayerService playerService;
+    private PlaylistService playlistService;
+    private SecurityService securityService;
+    private MusicIndexService musicIndexService;
+    private MediaFileService mediaFileService;
+    private SearchService searchService;
+
+    public ModelAndView index(HttpServletRequest request, HttpServletResponse response) throws Exception {
+        return wap(request, response);
+    }
+
+    public ModelAndView wap(HttpServletRequest request, HttpServletResponse response) throws Exception {
+        Map<String, Object> map = new HashMap<String, Object>();
+        List<MusicFolder> folders = settingsService.getAllMusicFolders();
+
+        if (folders.isEmpty()) {
+            map.put("noMusic", true);
+        } else {
+
+            SortedMap<MusicIndex, SortedSet<MusicIndex.Artist>> allArtists = musicIndexService.getIndexedArtists(folders);
+
+            // If an index is given as parameter, only show music files for this index.
+            String index = request.getParameter("index");
+            if (index != null) {
+                SortedSet<MusicIndex.Artist> artists = allArtists.get(new MusicIndex(index));
+                if (artists == null) {
+                    map.put("noMusic", true);
+                } else {
+                    map.put("artists", artists);
+                }
+            }
+
+            // Otherwise, list all indexes.
+            else {
+                map.put("indexes", allArtists.keySet());
+            }
+        }
+
+        return new ModelAndView("wap/index", "model", map);
+    }
+
+    public ModelAndView browse(HttpServletRequest request, HttpServletResponse response) throws Exception {
+        String path = request.getParameter("path");
+        MediaFile parent = mediaFileService.getMediaFile(path);
+
+        // Create array of file(s) to display.
+        List<MediaFile> children;
+        if (parent.isDirectory()) {
+            children = mediaFileService.getChildrenOf(parent, true, true, true);
+        } else {
+            children = new ArrayList<MediaFile>();
+            children.add(parent);
+        }
+
+        Map<String, Object> map = new HashMap<String, Object>();
+        map.put("parent", parent);
+        map.put("children", children);
+        map.put("user", securityService.getCurrentUser(request));
+
+        return new ModelAndView("wap/browse", "model", map);
+    }
+
+    public ModelAndView playlist(HttpServletRequest request, HttpServletResponse response) throws Exception {
+        // Create array of players to control. If the "player" attribute is set for this session,
+        // only the player with this ID is controlled.  Otherwise, all players are controlled.
+        List<Player> players = playerService.getAllPlayers();
+
+        String playerId = (String) request.getSession().getAttribute("player");
+        if (playerId != null) {
+            Player player = playerService.getPlayerById(playerId);
+            if (player != null) {
+                players = Arrays.asList(player);
+            }
+        }
+
+        Map<String, Object> map = new HashMap<String, Object>();
+
+        for (Player player : players) {
+            PlayQueue playQueue = player.getPlayQueue();
+            map.put("playlist", playQueue);
+
+            if (request.getParameter("play") != null) {
+                MediaFile file = mediaFileService.getMediaFile(request.getParameter("play"));
+                playQueue.addFiles(false, file);
+            } else if (request.getParameter("add") != null) {
+                MediaFile file = mediaFileService.getMediaFile(request.getParameter("add"));
+                playQueue.addFiles(true, file);
+            } else if (request.getParameter("skip") != null) {
+                playQueue.setIndex(Integer.parseInt(request.getParameter("skip")));
+            } else if (request.getParameter("clear") != null) {
+                playQueue.clear();
+            } else if (request.getParameter("load") != null) {
+                List<MediaFile> songs = playlistService.getFilesInPlaylist(ServletRequestUtils.getIntParameter(request, "id"));
+                playQueue.addFiles(false, songs);
+            } else if (request.getParameter("random") != null) {
+                List<MediaFile> randomFiles = searchService.getRandomSongs(new RandomSearchCriteria(20, null, null, null, null));
+                playQueue.addFiles(false, randomFiles);
+            }
+        }
+
+        map.put("players", players);
+        return new ModelAndView("wap/playlist", "model", map);
+    }
+
+    public ModelAndView loadPlaylist(HttpServletRequest request, HttpServletResponse response) throws Exception {
+        Map<String, Object> map = new HashMap<String, Object>();
+        map.put("playlists", playlistService.getReadablePlaylistsForUser(securityService.getCurrentUsername(request)));
+        return new ModelAndView("wap/loadPlaylist", "model", map);
+    }
+
+    public ModelAndView search(HttpServletRequest request, HttpServletResponse response) throws Exception {
+        return new ModelAndView("wap/search");
+    }
+
+    public ModelAndView searchResult(HttpServletRequest request, HttpServletResponse response) throws Exception {
+        String query = request.getParameter("query");
+
+        Map<String, Object> map = new HashMap<String, Object>();
+        map.put("hits", search(query));
+
+        return new ModelAndView("wap/searchResult", "model", map);
+    }
+
+    public ModelAndView settings(HttpServletRequest request, HttpServletResponse response) throws Exception {
+        String playerId = (String) request.getSession().getAttribute("player");
+
+        List<Player> allPlayers = playerService.getAllPlayers();
+        User user = securityService.getCurrentUser(request);
+        List<Player> players = new ArrayList<Player>();
+        Map<String, Object> map = new HashMap<String, Object>();
+
+        for (Player player : allPlayers) {
+            // Only display authorized players.
+            if (user.isAdminRole() || user.getUsername().equals(player.getUsername())) {
+                players.add(player);
+            }
+
+        }
+        map.put("playerId", playerId);
+        map.put("players", players);
+        return new ModelAndView("wap/settings", "model", map);
+    }
+
+    public ModelAndView selectPlayer(HttpServletRequest request, HttpServletResponse response) throws Exception {
+        request.getSession().setAttribute("player", request.getParameter("playerId"));
+        return settings(request, response);
+    }
+
+    private List<MediaFile> search(String query) throws IOException {
+        SearchCriteria criteria = new SearchCriteria();
+        criteria.setQuery(query);
+        criteria.setOffset(0);
+        criteria.setCount(50);
+
+        SearchResult result = searchService.search(criteria, SearchService.IndexType.SONG);
+        return result.getMediaFiles();
+    }
+
+    public void setSettingsService(SettingsService settingsService) {
+        this.settingsService = settingsService;
+    }
+
+    public void setPlayerService(PlayerService playerService) {
+        this.playerService = playerService;
+    }
+
+    public void setPlaylistService(PlaylistService playlistService) {
+        this.playlistService = playlistService;
+    }
+
+    public void setSecurityService(SecurityService securityService) {
+        this.securityService = securityService;
+    }
+
+    public void setMusicIndexService(MusicIndexService musicIndexService) {
+        this.musicIndexService = musicIndexService;
+    }
+
+    public void setMediaFileService(MediaFileService mediaFileService) {
+        this.mediaFileService = mediaFileService;
+    }
+
+    public void setSearchService(SearchService searchService) {
+        this.searchService = searchService;
+    }
+}