/*
* Copyright 2010-2011, Social Music Discovery project
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
* * Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* * Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* * Neither the name of Social Music Discovery project nor the
* names of its contributors may be used to endorse or promote products
* derived from this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED. IN NO EVENT SHALL SOCIAL MUSIC DISCOVERY PROJECT BE LIABLE FOR ANY
* DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
package org.socialmusicdiscovery.server.plugins.mediaimport.lastfm;
import com.sun.jersey.api.client.Client;
import org.codehaus.jettison.json.JSONArray;
import org.codehaus.jettison.json.JSONException;
import org.codehaus.jettison.json.JSONObject;
import org.socialmusicdiscovery.server.business.logic.InternetImageProvider;
import org.socialmusicdiscovery.server.business.model.SMDIdentity;
import org.socialmusicdiscovery.server.business.model.core.TrackEntity;
import org.socialmusicdiscovery.server.business.service.browse.*;
import javax.ws.rs.core.MediaType;
import java.io.UnsupportedEncodingException;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
/**
* Browse service that browses LastFM tracks
*/
public class LastFMTrackBrowseService extends AbstractLastFMBrowseService implements BrowseService<LastFMTrack> {
@Override
public Integer findChildrenCount(Collection<String> criteriaList) {
Result<LastFMTrack> result = findChildren(criteriaList,new ArrayList<String>(), null, null, false);
return result.getCount();
}
@Override
public Result<LastFMTrack> findChildren(Collection<String> criteriaList, Collection<String> sortCriteriaList, Integer firstItem, Integer maxItems, Boolean childCounters) {
String currentId = "";
for (String criteria : criteriaList) {
if(criteria.contains(":") && !criteria.startsWith(MenuLevelFolder.TYPE+":") && !criteria.startsWith(MenuLevelImageFolder.TYPE+":")) {
currentId = criteria;
}
}
TrackEntity entity = null;
if (criteriaList.size() == 1 && currentId.startsWith("Track:")) {
BrowseService browseService = browseServiceManager.getBrowseService("Track");
if (browseService != null) {
ResultItem currentItem = browseService.findById(currentId.substring(currentId.indexOf(":") + 1));
entity = (TrackEntity) currentItem.getItem();
}
}
Result<LastFMTrack> result = new Result<LastFMTrack>();
if (entity != null) {
try {
JSONObject object = Client.create().resource(getLastFmUrl("track.search&track=" + urlEncode(entity.getRecording().getWorks().iterator().next().getName()))).accept(MediaType.APPLICATION_JSON).get(JSONObject.class);
List<ResultItem<LastFMTrack>> tracks = new ArrayList<ResultItem<LastFMTrack>>();
JSONArray array = object.getJSONObject("results").getJSONObject("trackmatches").getJSONArray("track");
result.setCount(array.length());
for (int i = 0; i < array.length(); i++) {
if((firstItem==null || i>=firstItem) && (maxItems==null || maxItems>tracks.size())) {
tracks.add(createFromJSON(array.getJSONObject(i), null));
}
}
result.setItems(tracks);
} catch (UnsupportedEncodingException e) {
e.printStackTrace();
} catch (JSONException e) {
e.printStackTrace();
}
} else if (currentId.startsWith(LastFMAlbum.class.getSimpleName() + ":")) {
try {
String artistId = currentId.substring(12);
JSONObject object = Client.create().resource(getLastFmUrl("album.getinfo" + createQueryFromId(artistId))).accept(MediaType.APPLICATION_JSON).get(JSONObject.class);
List<ResultItem<LastFMTrack>> tracks = new ArrayList<ResultItem<LastFMTrack>>();
if (object.getJSONObject("album").optJSONObject("tracks")!=null && object.getJSONObject("album").getJSONObject("tracks").has("track")) {
JSONArray array = object.getJSONObject("album").getJSONObject("tracks").getJSONArray("track");
result.setCount(array.length());
for (int i = 0; i < array.length(); i++) {
if((firstItem==null || i>=firstItem) && (maxItems==null || maxItems>tracks.size())) {
tracks.add(createFromJSON(array.getJSONObject(i),object.getJSONObject("album")));
}
}
}
result.setCount(tracks.size());
result.setItems(tracks);
} catch (JSONException e) {
e.printStackTrace();
}
}
return result;
}
@Override
public ResultItem<LastFMTrack> findById(String id) {
try {
JSONObject object = Client.create().resource(getLastFmUrl("track.getinfo" + createQueryFromId(id))).accept(MediaType.APPLICATION_JSON).get(JSONObject.class);
return createFromJSON(object.getJSONObject("track"), null);
} catch (JSONException e) {
e.printStackTrace();
return null;
}
}
private ResultItem<LastFMTrack> createFromJSON(JSONObject json, JSONObject jsonAlbum) throws JSONException {
try {
String id;
if (json.has("mbid") && json.getString("mbid").length() > 0) {
id = "mbid:" + json.getString("mbid");
} else {
String artist;
if(json.optJSONObject("artist")==null) {
artist = json.getString("artist");
}else {
artist = json.getJSONObject("artist").getString("name");
}
id = "artist:" + urlEncode(artist) + ":track:" + urlEncode(json.getString("name"));
}
String name = json.getString("name");
String image = null;
JSONArray images = json.optJSONArray("image");
if(images!=null && images.length()>0) {
image = images.getJSONObject(images.length()-1).getString("#text");
if(image.length()==0) {
image = null;
}
}
if(image==null && json.optJSONObject("album")!=null && json.getJSONObject("album").optJSONArray("image")!=null) {
images = json.getJSONObject("album").getJSONArray("image");
image = images.getJSONObject(images.length()-1).getString("#text");
if(image.length()==0) {
image = null;
}
}
if(image==null && jsonAlbum!=null && jsonAlbum.optJSONArray("image")!=null) {
images = jsonAlbum.getJSONArray("image");
image = images.getJSONObject(images.length()-1).getString("#text");
if(image.length()==0) {
image = null;
}
}
LastFMTrack track = new LastFMTrack(id, null, name, image);
ResultItem<LastFMTrack> item = new ResultItem<LastFMTrack>(track, false, false);
item.setImage(getImage(track));
item.setType(track.getClass().getSimpleName());
return item;
} catch (UnsupportedEncodingException e) {
throw new JSONException(e);
}
}
@Override
public String getObjectType() {
return LastFMTrack.class.getSimpleName();
}
@Override
protected <T extends SMDIdentity> ResultItem.ResultItemImage getImage(T item) {
if(((LastFMTrack)item).getImage()!=null) {
return new ResultItem.ResultItemImage(InternetImageProvider.PROVIDER_ID, null, ((LastFMTrack)item).getImage());
}
return null;
}
}