com.github.fabienbarbero.flickr.api.FlickrService Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of flickr-api Show documentation
A Java library to access Flickr API (works with Android too!)
There is a newer version: 1.2
/*
 * (C) Copyright 2014 Fabien Barbero.
 *
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the GNU Lesser General Public License
 * (LGPL) version 2.1 which accompanies this distribution, and is available at
 * http://www.gnu.org/licenses/lgpl-2.1.html
 *
 * This library 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
 * Lesser General Public License for more details.
 */
package com.github.fabienbarbero.flickr.api;

import com.github.scribejava.core.model.OAuthRequest;
import com.github.scribejava.core.model.Response;
import com.github.scribejava.core.model.Verb;
import org.apache.http.HttpEntity;
import org.apache.http.HttpHeaders;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.Map;

/**
 * @author Fabien Barbero
 */
public abstract class FlickrService
{

    public static final int MAX_PER_PAGE = Integer.MAX_VALUE;
    private static final String URL_PREFIX = "https://api.flickr.com/services/rest";
    private final OAuthHandler oauthHandler;

    FlickrService( OAuthHandler oauth )
    {
        this.oauthHandler = oauth;
    }

    final  T doGet( CommandArguments args, Class clazz )
            throws FlickrException
    {
        OAuthRequest request = new OAuthRequest( Verb.GET, URL_PREFIX );

        for ( Map.Entry param : args.getParameters().entrySet() ) {
            request.addQuerystringParameter( param.getKey(), String.valueOf( param.getValue() ) );
        }

        try {
            oauthHandler.signRequest( request );
            Response response = oauthHandler.execute( request );
            String body = response.getBody();

            return parseBody( args, clazz, body );

        } catch ( IOException ex ) {
            throw new FlickrException( "Error executing request", ex );
        }
    }

    final  T doPost( CommandArguments args, Class clazz )
            throws FlickrException
    {
        return doPost( args, clazz, URL_PREFIX );
    }

    final  T doPost( CommandArguments args, Class clazz, String url )
            throws FlickrException
    {
        try {
            OAuthRequest request = new OAuthRequest( Verb.POST, url );

            for ( Map.Entry param : args.getParameters().entrySet() ) {
                if ( param.getValue() instanceof String ) {
                    request.addQuerystringParameter( param.getKey(), ( String ) param.getValue() );
                }
            }

            oauthHandler.signRequest( request );

            HttpEntity multipart = args.getBody( request.getOauthParameters() );
            ByteArrayOutputStream baos = new ByteArrayOutputStream();
            multipart.writeTo( baos );

            request.setPayload( baos.toByteArray() );
            request.addHeader( HttpHeaders.CONTENT_TYPE, multipart.getContentType().getValue() );

            Response response = oauthHandler.execute( request );
            String body = response.getBody();

            return parseBody( args, clazz, body );

        } catch ( IOException ex ) {
            throw new FlickrException( "Error preparing multipart request", ex );
        }
    }

    private  T parseBody( CommandArguments args, Class clazz, String body )
            throws FlickrException
    {
        try {
            if ( Flickr.debug ) {
                try {
                    System.out.println( "Server response for method " + args.getMethod() + "\n" + body );
                } catch ( Exception ignored ) {
                }
            }

            T instance = clazz.newInstance();
            instance.read( body, args.getMethod() );

            return instance;

        } catch ( IllegalStateException | InstantiationException | IllegalAccessException ex ) {
            throw new FlickrException( "Server request error", ex );
        }
    }

}