it.tidalwave.bluemarine2.metadata.musicbrainz.impl.DefaultMusicBrainzMetadataProvider Maven / Gradle / Ivy

Go to download
/*
 * #%L
 * *********************************************************************************************************************
 *
 * blueMarine2 - Semantic Media Center
 * http://bluemarine2.tidalwave.it - git clone https://[email protected]/tidalwave/bluemarine2-src.git
 * %%
 * Copyright (C) 2015 - 2017 Tidalwave s.a.s. (http://tidalwave.it)
 * %%
 *
 * *********************************************************************************************************************
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
 * the License. You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
 * an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations under the License.
 *
 * *********************************************************************************************************************
 *
 * $Id$
 *
 * *********************************************************************************************************************
 * #L%
 */
package it.tidalwave.bluemarine2.metadata.musicbrainz.impl;

import javax.annotation.Nonnull;
import java.util.Arrays;
import java.util.stream.Stream;
import java.io.IOException;
import org.springframework.http.ResponseEntity;
import org.musicbrainz.ns.mmd_2.Metadata;
import org.musicbrainz.ns.mmd_2.ReleaseGroupList;
import org.musicbrainz.ns.mmd_2.ReleaseList;
import it.tidalwave.bluemarine2.rest.CachingRestClientSupport;
import it.tidalwave.bluemarine2.rest.RestResponse;
import it.tidalwave.bluemarine2.metadata.musicbrainz.MusicBrainzMetadataProvider;
import lombok.Getter;
import lombok.Setter;
import lombok.extern.slf4j.Slf4j;
import static java.util.stream.Collectors.*;
import static it.tidalwave.bluemarine2.metadata.cddb.impl.MusicBrainzUtilities.*;

/***********************************************************************************************************************
 *
 * @author  Fabrizio Giudici ([email protected])
 * @version $Id: $
 *
 **********************************************************************************************************************/
@Slf4j
public class DefaultMusicBrainzMetadataProvider extends CachingRestClientSupport implements MusicBrainzMetadataProvider
  {
    private static final String URL_RELEASE_GROUP = "http://%s/ws/2/release-group/?query=release:%s";

    private static final String URL_DISCID = "http://%s/ws/2/discid/?toc=%s%s";

    private static final String URL_RESOURCE = "http://%s/ws/2/%s/%s%s";

    @Getter @Setter
    private String host = "musicbrainz.org";

    /*******************************************************************************************************************
     *
     *
     *
     ******************************************************************************************************************/
    public DefaultMusicBrainzMetadataProvider()
      {
        // sometimes MusicBrainz returns HTTP 500, but it's usually transitory
        setRetryStatusCodes(Arrays.asList(500, 503));
        setMaxRetry(10);
        setThrottleLimit(1500);
      }

    /*******************************************************************************************************************
     *
     * {@inheritDoc}
     *
     ******************************************************************************************************************/
    @Override @Nonnull
    public RestResponse findReleaseGroupByTitle (final @Nonnull String title,
                                                                   final @Nonnull String ... includes)
      throws IOException, InterruptedException
      {
        log.debug("findReleaseGroupByTitle({})", title);
        final ResponseEntity response = request(String.format(URL_RELEASE_GROUP, host, escape(title)));
        return MusicBrainzResponse.of(response, Metadata::getReleaseGroupList);
      }

    /*******************************************************************************************************************
     *
     * {@inheritDoc}
     *
     ******************************************************************************************************************/
    @Override @Nonnull
    public RestResponse findReleaseListByToc (final @Nonnull String toc,
                                                           final @Nonnull String ... includes)
      throws IOException, InterruptedException
      {
        log.debug("findReleaseListByToc({}, {})", toc, includes);
        final ResponseEntity response = request(String.format(URL_DISCID, host, toc, includesToString("&", includes)));
        return MusicBrainzResponse.of(response, Metadata::getReleaseList);
      }

    /*******************************************************************************************************************
     *
     * {@inheritDoc}
     *
     ******************************************************************************************************************/
    @Override @Nonnull
    public  RestResponse getResource (final @Nonnull ResourceType resourceType,
                                            final @Nonnull String id,
                                            final @Nonnull String ... includes)
      throws IOException, InterruptedException
      {
        log.debug("getResource({}. {}, {})", resourceType, id, includes);
        final String url = String.format(URL_RESOURCE, host, resourceType.getName(), id, includesToString("?", includes));
        return MusicBrainzResponse.of(request(url), resourceType.getResultProvider());
      }

    /*******************************************************************************************************************
     *
     *
     ******************************************************************************************************************/
    @Nonnull
    private static String includesToString (final @Nonnull String separator, final @Nonnull String ... includes)
      {
        return Stream.of(includes).collect(joining("%2b", separator + "inc=", ""));
      }
  }