com.qmetry.qaf.automation.http.UriProxySelector Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of qaf Show documentation
Functional test automation framework for web, mobile-web, mobile native and web-service
There is a newer version: 4.0.0-RC3
/*******************************************************************************
 * Copyright (c) 2019 Infostretch Corporation
 * 
 * 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:
 * 
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 * 
 * 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.
 ******************************************************************************/
package com.qmetry.qaf.automation.http;

import static com.qmetry.qaf.automation.core.ConfigurationManager.getBundle;

import java.io.IOException;
import java.net.InetSocketAddress;
import java.net.Proxy;
import java.net.ProxySelector;
import java.net.SocketAddress;
import java.net.URI;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.impl.LogFactoryImpl;

import com.qmetry.qaf.automation.util.StringUtil;

/**
 * @author chirag.jayswal
 *
 */
public class UriProxySelector extends ProxySelector {
	private static final Log logger = LogFactoryImpl.getLog(UriProxySelector.class);
	private static final String PROXY_SERVER_KEY = "proxy.server";
	private static final String PROXY_PORT_KEY = "proxy.port";
	private static final String PROXY_HOSTS_KEY = "host.to.proxy";

	private static final UriProxySelector INSTANCE = new UriProxySelector();
	private ProxySelector defaultSelector = null;
	private List hostsToProxy = null;
	private List proxies = null;

	private UriProxySelector() {
		defaultSelector = ProxySelector.getDefault();
		logger.info("Registed UriProxySelector");
	}

	public static UriProxySelector getInstance() {
		return INSTANCE;
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see java.net.ProxySelector#select(java.net.URI)
	 */
	@Override
	public List select(URI uri) {
		if (null == proxies) {
			// this is the first time when proxy select request happened!...
			intiProxySettings();
		}
		if (null != proxies && !proxies.isEmpty() && null != hostsToProxy) {
			for (String host : hostsToProxy) {
				if (uri.getHost().toUpperCase().contains(host.toUpperCase())) {
					return proxies;
				}
			}
		}
		return defaultSelector.select(uri);
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see java.net.ProxySelector#connectFailed(java.net.URI,
	 * java.net.SocketAddress, java.io.IOException)
	 */
	@Override
	public void connectFailed(URI uri, SocketAddress sa, IOException ioe) {
		defaultSelector.connectFailed(uri, sa, ioe);
	}

	@SuppressWarnings("unchecked")
	private void intiProxySettings() {
		try {
			hostsToProxy = getBundle().getList(PROXY_HOSTS_KEY);
			String proxyServer = getBundle().getString(PROXY_SERVER_KEY, "");
			int proxyPort = getBundle().getInt(PROXY_PORT_KEY, 80);
			if (StringUtil.isNotBlank(proxyServer)) {
				SocketAddress sa = new InetSocketAddress(proxyServer, proxyPort);
				proxies = Arrays.asList(new Proxy(Proxy.Type.HTTP, sa), new Proxy(Proxy.Type.SOCKS, sa));
				logger.info("proxy settings done using proxy server " + proxies + " for " + hostsToProxy);
			} else {
				logger.warn(
						"proxy ignored as missing proxyServer. To enable proxy make sure you have non blank value of property '"
								+ PROXY_SERVER_KEY + "'");
				proxies = new ArrayList();
			}
		} catch (Exception e) {
			logger.error("Unable to init proxy settings", e);
		}
	}
}