session-android/src/org/thoughtcrime/securesms/net/ContentProxySelector.java

package org.thoughtcrime.securesms.net;


import android.os.AsyncTask;

import org.thoughtcrime.securesms.linkpreview.LinkPreviewDomains;
import org.thoughtcrime.securesms.logging.Log;

import network.loki.messenger.BuildConfig;
import org.thoughtcrime.securesms.util.Util;

import java.io.IOException;
import java.net.InetSocketAddress;
import java.net.Proxy;
import java.net.ProxySelector;
import java.net.SocketAddress;
import java.net.SocketException;
import java.net.URI;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

public class ContentProxySelector extends ProxySelector {

  private static final String TAG = ContentProxySelector.class.getSimpleName();

  private static final Set<String> WHITELISTED_DOMAINS = new HashSet<>();
  static {
    WHITELISTED_DOMAINS.addAll(LinkPreviewDomains.LINKS);
    WHITELISTED_DOMAINS.addAll(LinkPreviewDomains.IMAGES);
  }

  private final List<Proxy> CONTENT = new ArrayList<Proxy>(1) {{
    add(new Proxy(Proxy.Type.HTTP, InetSocketAddress.createUnresolved(BuildConfig.CONTENT_PROXY_HOST,
                                                                      BuildConfig.CONTENT_PROXY_PORT)));
  }};

  @Override
  public List<Proxy> select(URI uri) {
    for (String domain : WHITELISTED_DOMAINS) {
      if (uri.getHost().endsWith(domain)) {
        return CONTENT;
      }
    }
    throw new IllegalArgumentException("Tried to proxy a non-whitelisted domain.");
  }

  @Override
  public void connectFailed(URI uri, SocketAddress address, IOException failure) {
    if (failure instanceof SocketException) {
      Log.d(TAG, "Socket exception. Likely a cancellation.");
    } else {
      Log.w(TAG, "Connection failed.", failure);
    }
  }
}