com.platform.middlewares.APIProxy.java Source code

Introduction

Here is the source code for com.platform.middlewares.APIProxy.java

Source

package com.platform.middlewares;

import android.util.Log;

import com.breadwallet.presenter.activities.MainActivity;
import com.platform.APIClient;
import com.platform.BRHTTPHelper;
import com.platform.interfaces.Middleware;

import org.apache.commons.io.IOUtils;

import java.io.IOException;
import java.util.Arrays;
import java.util.Enumeration;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import okhttp3.Headers;
import okhttp3.MediaType;
import okhttp3.Request;
import okhttp3.RequestBody;
import okhttp3.Response;
import okhttp3.ResponseBody;

/**
 * BreadWallet
 * <p/>
 * Created by Mihail Gutan on <mihail@breadwallet.com> 10/19/16.
 * Copyright (c) 2016 breadwallet LLC
 * <p/>
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 * <p/>
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 * <p/>
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */
public class APIProxy implements Middleware {
    public static final String TAG = APIProxy.class.getName();

    private APIClient apiInstance;
    private static final String MOUNT_POINT = "/_api";
    private final String SHOULD_VERIFY_HEADER = "x-should-verify";
    private final String SHOULD_AUTHENTICATE = "x-should-authenticate";
    private final String[] bannedSendHeaders = new String[] { SHOULD_VERIFY_HEADER, SHOULD_AUTHENTICATE,
            "connection", "authorization", "host", "user-agent" };

    private final String[] bannedReceiveHeaders = new String[] { "content-length", "content-encoding",
            "connection" };

    public APIProxy() {
        apiInstance = APIClient.getInstance(MainActivity.app);
    }

    @Override
    public boolean handle(String target, org.eclipse.jetty.server.Request baseRequest, HttpServletRequest request,
            HttpServletResponse response) {
        if (!target.startsWith(MOUNT_POINT))
            return false;
        Log.i(TAG, "handling: " + target + " " + baseRequest.getMethod());
        String path = target.substring(MOUNT_POINT.length());
        String queryString = baseRequest.getQueryString();
        if (queryString != null && queryString.length() > 0)
            path += "?" + queryString;
        boolean auth = false;
        Request req = mapToOkHttpRequest(baseRequest, path, request);
        String authHeader = baseRequest.getHeader(SHOULD_AUTHENTICATE);
        if (authHeader != null && authHeader.toLowerCase().equals("yes"))
            auth = true;
        Response res = apiInstance.sendRequest(req, auth, 0);

        if (res.code() == 599) {
            Log.e(TAG, "handle: code 599: " + target + " " + baseRequest.getMethod());
            return BRHTTPHelper.handleError(599, null, baseRequest, response);
        }
        ResponseBody body = res.body();
        String cType = body.contentType() == null ? null : body.contentType().toString();
        String resString = null;
        byte[] bodyBytes = new byte[0];
        try {
            bodyBytes = body.bytes();
            resString = new String(bodyBytes);
        } catch (IOException e) {
            e.printStackTrace();
        }

        response.setContentType(cType);
        Headers headers = res.headers();
        for (String s : headers.names()) {
            if (Arrays.asList(bannedReceiveHeaders).contains(s.toLowerCase()))
                continue;
            response.addHeader(s, res.header(s));
        }
        response.setContentLength(bodyBytes.length);
        if (res.isSuccessful()) {
            return BRHTTPHelper.handleSuccess(res.code(), bodyBytes, baseRequest, response, null);
        } else {
            Log.e(TAG, "RES IS NOT SUCCESSFUL: " + res.request().url() + ": " + res.code() + "(" + res.message()
                    + ")");
            return BRHTTPHelper.handleError(res.code(), null, baseRequest, response);
        }

    }

    private Request mapToOkHttpRequest(org.eclipse.jetty.server.Request baseRequest, String path,
            HttpServletRequest request) {
        Request req;
        Request.Builder builder = new Request.Builder().url(apiInstance.buildUrl(path));

        Enumeration<String> headerNames = baseRequest.getHeaderNames();
        while (headerNames.hasMoreElements()) {
            String hName = headerNames.nextElement();
            if (Arrays.asList(bannedSendHeaders).contains(hName.toLowerCase()))
                continue;
            builder.addHeader(hName, baseRequest.getHeader(hName));
        }

        byte[] bodyText = new byte[0];
        try {
            bodyText = IOUtils.toByteArray(request.getInputStream());
        } catch (IOException e) {
            e.printStackTrace();
        }
        String contentType = baseRequest.getContentType() == null ? null : baseRequest.getContentType();
        RequestBody reqBody = RequestBody.create(contentType == null ? null : MediaType.parse(contentType),
                bodyText);

        switch (baseRequest.getMethod()) {
        case "GET":
            builder.get();
            break;
        case "DELETE":
            builder.delete();
            break;
        case "POST":
            builder.post(reqBody);
            break;
        case "PUT":
            builder.put(reqBody);
            break;
        default:
            Log.e(TAG, "mapToOkHttpRequest: WARNING: method: " + baseRequest.getMethod());
            break;
        }

        req = builder.build();
        return req;
    }

}