com.rjuarez.webapp.tools.ApiUrl.java Source code

Introduction

Here is the source code for com.rjuarez.webapp.tools.ApiUrl.java

Source

/*
 *      Copyright (c) 2004-2015 Stuart Boston
 *
 *      This file is part of TheMovieDB API.
 *
 *      TheMovieDB API 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
 *      any later version.
 *
 *      TheMovieDB API 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 TheMovieDB API.  If not, see <http://www.gnu.org/licenses/>.
 *
 */

package com.rjuarez.webapp.tools;

import java.io.UnsupportedEncodingException;
import java.net.MalformedURLException;
import java.net.URL;
import java.net.URLEncoder;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import org.apache.commons.lang3.builder.ToStringBuilder;
import org.apache.commons.lang3.builder.ToStringStyle;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class ApiUrl {

    private static final String SLASH = "/";
    private static final Logger LOG = LoggerFactory.getLogger(ApiUrl.class);
    // TheMovieDbApi API Base URL
    private static final String TMDB_API_BASE = "http://api.themoviedb.org/3/";
    // Parameter configuration
    private static final String DELIMITER_FIRST = "?";
    private static final String DELIMITER_SUBSEQUENT = "&";
    // Properties
    private final String apiKey;
    private final TheMovieDatabaseMethod method;
    private MethodSub submethod = MethodSub.NONE;
    private static final List<TheMovieDatabaseQueries> IGNORE_PARAMS = new ArrayList<>();

    static {
        IGNORE_PARAMS.add(TheMovieDatabaseQueries.ID);
        IGNORE_PARAMS.add(TheMovieDatabaseQueries.QUERY);
        IGNORE_PARAMS.add(TheMovieDatabaseQueries.SEASON_NUMBER);
        IGNORE_PARAMS.add(TheMovieDatabaseQueries.EPISODE_NUMBER);
    }

    public ApiUrl(final String apiKey, final TheMovieDatabaseMethod method) {
        this.apiKey = apiKey;
        this.method = method;
    }

    public ApiUrl subMethod(final MethodSub submethod) {
        if (submethod != MethodSub.NONE) {
            this.submethod = submethod;
        }
        return this;
    }

    public URL buildUrl() {
        return buildUrl(new TheMovieDatabaseParameters());
    }

    public URL buildUrl(final TheMovieDatabaseParameters params) {
        final StringBuilder urlString = new StringBuilder(TMDB_API_BASE);

        LOG.trace("Method: '{}', Sub-method: '{}', Params: {}", method.getValue(), submethod.getValue(),
                ToStringBuilder.reflectionToString(params, ToStringStyle.SHORT_PREFIX_STYLE));

        // Get the start of the URL, substituting TV for the season or episode
        // methods
        if (method == TheMovieDatabaseMethod.SEASON || method == TheMovieDatabaseMethod.EPISODE) {
            urlString.append(TheMovieDatabaseMethod.TV.getValue());
        } else {
            urlString.append(method.getValue());
        }

        // We have either a queury, or a ID request
        if (params.has(TheMovieDatabaseQueries.QUERY)) {
            urlString.append(queryProcessing(params));
        } else {
            urlString.append(idProcessing(params));
        }

        urlString.append(otherProcessing(params));

        try {
            LOG.trace("URL: {}", urlString.toString());
            return new URL(urlString.toString());
        } catch (final MalformedURLException ex) {
            LOG.warn("Failed to create URL {} - {}", urlString.toString(), ex.getMessage());
            return null;
        }
    }

    private StringBuilder queryProcessing(final TheMovieDatabaseParameters params) {
        final StringBuilder urlString = new StringBuilder();

        // Append the suffix of the API URL
        if (submethod != MethodSub.NONE) {
            urlString.append(SLASH).append(submethod.getValue());
        }

        // Append the key information
        urlString.append(DELIMITER_FIRST).append(TheMovieDatabaseQueries.API_KEY.getValue()).append(apiKey);

        // Append the search term
        urlString.append(DELIMITER_SUBSEQUENT);
        urlString.append(TheMovieDatabaseQueries.QUERY.getValue());

        final String query = (String) params.get(TheMovieDatabaseQueries.QUERY);

        try {
            urlString.append(URLEncoder.encode(query, "UTF-8"));
        } catch (final UnsupportedEncodingException ex) {
            LOG.trace("Unable to encode query: '{}' trying raw.", query, ex);
            // If we can't encode it, try it raw
            urlString.append(query);
        }

        return urlString;
    }

    private StringBuilder idProcessing(final TheMovieDatabaseParameters params) {
        final StringBuilder urlString = new StringBuilder();

        // Append the ID
        if (params.has(TheMovieDatabaseQueries.ID)) {
            urlString.append(SLASH).append(params.get(TheMovieDatabaseQueries.ID));
        }

        if (params.has(TheMovieDatabaseQueries.SEASON_NUMBER)) {
            urlString.append("/season/").append(params.get(TheMovieDatabaseQueries.SEASON_NUMBER));
        }

        if (params.has(TheMovieDatabaseQueries.EPISODE_NUMBER)) {
            urlString.append("/episode/").append(params.get(TheMovieDatabaseQueries.EPISODE_NUMBER));
        }

        if (submethod != MethodSub.NONE) {
            urlString.append(SLASH).append(submethod.getValue());
        }

        // Append the key information
        urlString.append(DELIMITER_FIRST).append(TheMovieDatabaseQueries.API_KEY.getValue()).append(apiKey);

        return urlString;
    }

    private StringBuilder otherProcessing(final TheMovieDatabaseParameters params) {
        final StringBuilder urlString = new StringBuilder();

        for (Map.Entry<TheMovieDatabaseQueries, String> argEntry : params.getEntries()) {
            // Skip the ID an QUERY params
            if (IGNORE_PARAMS.contains(argEntry.getKey())) {
                continue;
            }

            urlString.append(DELIMITER_SUBSEQUENT).append(argEntry.getKey().getValue()).append(argEntry.getValue());
        }
        return urlString;
    }
}