Java tutorial
/* 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 net.sourceforge.subsonic.domain.CoverArtScheme; import net.sourceforge.subsonic.domain.MediaFile; import net.sourceforge.subsonic.domain.Player; import net.sourceforge.subsonic.domain.UserSettings; import net.sourceforge.subsonic.service.MediaFileService; import net.sourceforge.subsonic.service.RatingService; import net.sourceforge.subsonic.service.PlayerService; import net.sourceforge.subsonic.service.SecurityService; import net.sourceforge.subsonic.service.SettingsService; import org.apache.commons.lang.StringUtils; import org.springframework.web.bind.ServletRequestUtils; import org.springframework.web.servlet.ModelAndView; import org.springframework.web.servlet.mvc.ParameterizableViewController; import org.springframework.web.servlet.view.RedirectView; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import java.io.File; import java.io.IOException; import java.util.ArrayList; import java.util.HashMap; import java.util.HashSet; import java.util.LinkedList; import java.util.List; import java.util.Map; import java.util.Set; /** * Controller for the main page. * * @author Sindre Mehus */ public class MainController extends ParameterizableViewController { private SecurityService securityService; private PlayerService playerService; private SettingsService settingsService; private RatingService ratingService; private MediaFileService mediaFileService; @Override protected ModelAndView handleRequestInternal(HttpServletRequest request, HttpServletResponse response) throws Exception { Map<String, Object> map = new HashMap<String, Object>(); Player player = playerService.getPlayer(request, response); List<MediaFile> mediaFiles = getMediaFiles(request); MediaFile dir = mediaFiles.get(0); if (dir.isFile()) { dir = mediaFileService.getParentOf(dir); } // Redirect if root directory. if (mediaFileService.isRoot(dir)) { return new ModelAndView(new RedirectView("home.view?")); } List<MediaFile> children = mediaFiles.size() == 1 ? mediaFileService.getChildrenOf(dir, true, true, true) : getMultiFolderChildren(mediaFiles); UserSettings userSettings = settingsService.getUserSettings(securityService.getCurrentUsername(request)); map.put("dir", dir); map.put("ancestors", getAncestors(dir)); map.put("children", children); map.put("artist", guessArtist(children)); map.put("album", guessAlbum(children)); map.put("player", player); map.put("user", securityService.getCurrentUser(request)); map.put("multipleArtists", isMultipleArtists(children)); map.put("visibility", userSettings.getMainVisibility()); map.put("updateNowPlaying", request.getParameter("updateNowPlaying") != null); map.put("partyMode", userSettings.isPartyModeEnabled()); map.put("brand", settingsService.getBrand()); try { MediaFile parent = mediaFileService.getParentOf(dir); map.put("parent", parent); map.put("navigateUpAllowed", !mediaFileService.isRoot(parent)); } catch (SecurityException x) { // Happens if Podcast directory is outside music folder. } String username = securityService.getCurrentUsername(request); Integer userRating = ratingService.getRatingForUser(username, dir); Double averageRating = ratingService.getAverageRating(dir); if (userRating == null) { userRating = 0; } if (averageRating == null) { averageRating = 0.0D; } map.put("userRating", 10 * userRating); map.put("averageRating", Math.round(10.0D * averageRating)); CoverArtScheme scheme = player.getCoverArtScheme(); if (scheme != CoverArtScheme.OFF) { List<File> coverArts = getCoverArts(dir, children); int size = coverArts.size() > 1 ? scheme.getSize() : scheme.getSize() * 2; map.put("coverArts", coverArts); map.put("coverArtSize", size); if (coverArts.isEmpty() && dir.isAlbum()) { map.put("showGenericCoverArt", true); } } setPreviousAndNextAlbums(dir, map); ModelAndView result = super.handleRequestInternal(request, response); result.addObject("model", map); return result; } private List<MediaFile> getMediaFiles(HttpServletRequest request) { List<MediaFile> mediaFiles = new ArrayList<MediaFile>(); for (String path : ServletRequestUtils.getStringParameters(request, "path")) { MediaFile mediaFile = mediaFileService.getMediaFile(path); if (mediaFile != null) { mediaFiles.add(mediaFile); } } for (int id : ServletRequestUtils.getIntParameters(request, "id")) { MediaFile mediaFile = mediaFileService.getMediaFile(id); if (mediaFile != null) { mediaFiles.add(mediaFile); } } return mediaFiles; } private String guessArtist(List<MediaFile> children) { for (MediaFile child : children) { if (child.isFile() && child.getArtist() != null) { return child.getArtist(); } } return null; } private String guessAlbum(List<MediaFile> children) { for (MediaFile child : children) { if (child.isFile() && child.getArtist() != null) { return child.getAlbumName(); } } return null; } private List<File> getCoverArts(MediaFile dir, List<MediaFile> children) throws IOException { int limit = settingsService.getCoverArtLimit(); if (limit == 0) { limit = Integer.MAX_VALUE; } List<File> coverArts = new ArrayList<File>(); if (dir.isAlbum()) { File coverArt = mediaFileService.getCoverArt(dir); if (coverArt != null) { coverArts.add(coverArt); } } else { for (MediaFile child : children) { if (child.isDirectory()) { File coverArt = mediaFileService.getCoverArt(child); if (coverArt != null) { coverArts.add(coverArt); if (coverArts.size() > limit) { break; } } } } } return coverArts; } private List<MediaFile> getMultiFolderChildren(List<MediaFile> mediaFiles) throws IOException { List<MediaFile> result = new ArrayList<MediaFile>(); for (MediaFile mediaFile : mediaFiles) { if (mediaFile.isFile()) { mediaFile = mediaFileService.getParentOf(mediaFile); } result.addAll(mediaFileService.getChildrenOf(mediaFile, true, true, true)); } return result; } private List<MediaFile> getAncestors(MediaFile dir) throws IOException { LinkedList<MediaFile> result = new LinkedList<MediaFile>(); try { MediaFile parent = mediaFileService.getParentOf(dir); while (parent != null && !mediaFileService.isRoot(parent)) { result.addFirst(parent); parent = mediaFileService.getParentOf(parent); } } catch (SecurityException x) { // Happens if Podcast directory is outside music folder. } return result; } private void setPreviousAndNextAlbums(MediaFile dir, Map<String, Object> map) throws IOException { MediaFile parent = mediaFileService.getParentOf(dir); if (dir.isAlbum() && !mediaFileService.isRoot(parent)) { List<MediaFile> sieblings = mediaFileService.getChildrenOf(parent, false, true, true); int index = sieblings.indexOf(dir); if (index > 0) { map.put("previousAlbum", sieblings.get(index - 1)); } if (index < sieblings.size() - 1) { map.put("nextAlbum", sieblings.get(index + 1)); } } } private boolean isMultipleArtists(List<MediaFile> children) { // Collect unique artist names. Set<String> artists = new HashSet<String>(); for (MediaFile child : children) { if (child.getArtist() != null) { artists.add(child.getArtist().toLowerCase()); } } // If zero or one artist, it is definitely not multiple artists. if (artists.size() < 2) { return false; } // Fuzzily compare artist names, allowing for some differences in spelling, whitespace etc. List<String> artistList = new ArrayList<String>(artists); for (String artist : artistList) { if (StringUtils.getLevenshteinDistance(artist, artistList.get(0)) > 3) { return true; } } return false; } public void setSecurityService(SecurityService securityService) { this.securityService = securityService; } public void setPlayerService(PlayerService playerService) { this.playerService = playerService; } public void setSettingsService(SettingsService settingsService) { this.settingsService = settingsService; } public void setRatingService(RatingService ratingService) { this.ratingService = ratingService; } public void setMediaFileService(MediaFileService mediaFileService) { this.mediaFileService = mediaFileService; } }