com.liferay.mobile.android.http.client.OkHttpClientImpl.java Source code

Introduction

Here is the source code for com.liferay.mobile.android.http.client.OkHttpClientImpl.java

Source

/**
 * Copyright (c) 2000-present Liferay, Inc. All rights reserved.
 *
 * This library is free software; you can redistribute it and/or modify it under
 * the terms of the GNU Lesser General Public License as published by the Free
 * Software Foundation; either version 2.1 of the License, or (at your option)
 * any later version.
 *
 * 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.liferay.mobile.android.http.client;

import com.liferay.mobile.android.auth.Authentication;
import com.liferay.mobile.android.callback.Callback;
import com.liferay.mobile.android.http.Method;
import com.liferay.mobile.android.http.Request;
import com.liferay.mobile.android.http.Response;
import com.liferay.mobile.android.http.file.InputStreamBody;
import com.liferay.mobile.android.http.file.UploadData;

import com.squareup.okhttp.Authenticator;
import com.squareup.okhttp.Call;
import com.squareup.okhttp.HttpUrl;
import com.squareup.okhttp.MediaType;
import com.squareup.okhttp.MultipartBuilder;
import com.squareup.okhttp.OkHttpClient;
import com.squareup.okhttp.Request.Builder;
import com.squareup.okhttp.RequestBody;

import java.io.IOException;

import java.util.Iterator;
import java.util.Map;
import java.util.concurrent.TimeUnit;

import org.json.JSONObject;

/**
 * @author Bruno Farache
 */
public class OkHttpClientImpl implements HttpClient {

    public OkHttpClientImpl() {
        client = new OkHttpClient();
    }

    @Override
    public void cancel(Object tag) {
        client.cancel(tag);
    }

    @Override
    public String encodeURL(String url) {
        return HttpUrl.parse(url).toString();
    }

    @Override
    public Response send(Request request) throws Exception {
        Builder builder = new Builder();
        Method method = request.getMethod();

        if (method == Method.POST) {
            String body = (String) request.getBody();

            if (body != null) {
                MediaType type = MediaType.parse("application/json; charset=utf-8");

                builder.post(RequestBody.create(type, body));
            }
        } else if (method == Method.HEAD) {
            builder.head();
        }

        return send(builder, request);
    }

    @Override
    public Response upload(Request request) throws Exception {
        Builder builder = new Builder();
        builder.post(getUploadBody(request));

        return send(builder, request);
    }

    protected void addHeaders(Builder builder, Request request) {
        Map<String, String> headers = request.getHeaders();

        if (headers != null) {
            for (Map.Entry<String, String> header : headers.entrySet()) {
                builder.addHeader(header.getKey(), header.getValue());
            }
        }
    }

    protected void authenticate(OkHttpClient client, Request request) throws Exception {

        Authentication authentication = request.getAuthentication();

        if (authentication != null) {
            if (authentication instanceof Authenticator) {
                client.setAuthenticator((Authenticator) authentication);
            } else {
                authentication.authenticate(request);
            }
        }
    }

    protected OkHttpClient getClient(int connectionTimeout) {
        OkHttpClient clone = client.clone();

        clone.setConnectTimeout(connectionTimeout, TimeUnit.MILLISECONDS);
        clone.setReadTimeout(connectionTimeout, TimeUnit.MILLISECONDS);
        clone.setWriteTimeout(connectionTimeout, TimeUnit.MILLISECONDS);

        clone.setFollowRedirects(false);

        return clone;
    }

    protected RequestBody getUploadBody(Request request) {
        JSONObject body = (JSONObject) request.getBody();
        Object tag = request.getTag();

        MultipartBuilder builder = new MultipartBuilder().type(MultipartBuilder.FORM);

        Iterator<String> it = body.keys();

        while (it.hasNext()) {
            String key = it.next();
            Object value = body.opt(key);

            if (value instanceof UploadData) {
                UploadData data = (UploadData) value;
                RequestBody requestBody = new InputStreamBody(data, tag);
                builder.addFormDataPart(key, data.getFileName(), requestBody);
            } else {
                builder.addFormDataPart(key, value.toString());
            }
        }

        return builder.build();
    }

    protected Response send(Builder builder, final Request request) throws Exception {

        builder = builder.url(request.getURL());
        builder.tag(request.getTag());

        OkHttpClient client = getClient(request.getConnectionTimeout());

        authenticate(client, request);
        addHeaders(builder, request);

        Call call = client.newCall(builder.build());

        final Callback callback = request.getCallback();

        if (callback == null) {
            return new Response(call.execute());
        } else {
            sendAsync(call, callback);
            return null;
        }
    }

    protected void sendAsync(Call call, final Callback callback) {
        call.enqueue(new com.squareup.okhttp.Callback() {

            @Override
            public void onFailure(com.squareup.okhttp.Request request, IOException ioe) {

                callback.doFailure(ioe);
            }

            @Override
            public void onResponse(com.squareup.okhttp.Response response) throws IOException {

                callback.inBackground(new Response(response));
            }

        });
    }

    protected OkHttpClient client;

}