OkConnectionFactory.java

package com.dozuki.ifixit.util;

import com.github.kevinsawicki.http.HttpRequest;
import com.squareup.okhttp.OkHttpClient;

import java.io.IOException;
import java.net.HttpURLConnection;
import java.net.Proxy;
import java.net.URL;

/**
 * A {@link HttpRequest.ConnectionFactory connection factory} which uses OkHttp.
 * <p/>
 * Call {@link HttpRequest#setConnectionFactory(HttpRequest.ConnectionFactory)} with an instance of
 * this class to enable.
 */
public class OkConnectionFactory implements HttpRequest.ConnectionFactory {
   private final OkHttpClient client;

   public OkConnectionFactory() {
      this(Utils.createOkHttpClient());
   }

   public OkConnectionFactory(OkHttpClient client) {
      if (client == null) {
         throw new NullPointerException("Client must not be null.");
      }
      this.client = client;
   }

   public HttpURLConnection create(URL url) throws IOException {
      return client.open(url);
   }

   public HttpURLConnection create(URL url, Proxy proxy) throws IOException {
      throw new UnsupportedOperationException(
       "Per-connection proxy is not supported. Use OkHttpClient's setProxy instead.");
   }
}