DefaultModule.java

/**
 * Copyright (c) 2013 Puppet Labs, Inc. and other contributors, as listed below.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Apache License, Version 2.0
 * which accompanies this distribution, and is available at
 * http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Contributors:
 *   Puppet Labs
 */
package com.puppetlabs.puppetdb.javaclient.impl;

import org.apache.http.client.HttpClient;
import org.apache.http.conn.scheme.Scheme;
import org.apache.http.conn.ssl.SSLSocketFactory;
import org.apache.http.impl.client.DefaultHttpClient;
import org.apache.http.params.BasicHttpParams;
import org.apache.http.params.CoreConnectionPNames;
import org.apache.http.params.HttpConnectionParams;
import org.apache.http.params.HttpParams;

import com.google.gson.Gson;
import com.google.inject.AbstractModule;
import com.google.inject.Provides;
import com.google.inject.Singleton;
import com.google.inject.name.Names;
import com.puppetlabs.puppetdb.javaclient.APIPreferences;
import com.puppetlabs.puppetdb.javaclient.HttpConnector;
import com.puppetlabs.puppetdb.javaclient.PuppetDBClient;

/**
 * Default Guice Injection module
 */
public class DefaultModule extends AbstractModule {
	private final APIPreferences preferences;

	/**
	 * Create a module with preferences.
	 * 
	 * @param preferences
	 *            The preferences
	 */
	public DefaultModule(APIPreferences preferences) {
		this.preferences = preferences;
	}

	@Override
	protected void configure() {
		bind(APIPreferences.class).toInstance(preferences);
		bind(Integer.class).annotatedWith(Names.named(CoreConnectionPNames.CONNECTION_TIMEOUT)).toInstance(preferences.getConnectTimeout());
		bind(Integer.class).annotatedWith(Names.named(CoreConnectionPNames.SO_TIMEOUT)).toInstance(preferences.getSoTimeout());
		bind(Gson.class).toProvider(GsonProvider.class);
		if(preferences.getCertPEM() != null)
			bind(SSLSocketFactory.class).toProvider(PEM_SSLSocketFactoryProvider.class).in(Singleton.class);
		else
			bind(SSLSocketFactory.class).toInstance(SSLSocketFactory.getSocketFactory());
		bind(HttpConnector.class).to(HttpComponentsConnector.class);
		bind(PuppetDBClient.class).to(PuppetDBClientImpl.class);
	}

	/**
	 * Provides a HttpClient that is configured with the preferences of this module and the
	 * injected <code>sslSocketFactory</code>.
	 * 
	 * @param sslSocketFactory
	 *            The injected SSL socket factory
	 * @return The new HttpClient instance
	 */
	@Provides
	public HttpClient provideHttpClient(SSLSocketFactory sslSocketFactory) {
		HttpParams params = new BasicHttpParams();
		HttpConnectionParams.setConnectionTimeout(params, preferences.getConnectTimeout());
		HttpConnectionParams.setSoTimeout(params, preferences.getSoTimeout());

		DefaultHttpClient httpClient = new DefaultHttpClient(params);
		if(preferences.getCertPEM() != null)
			httpClient.getConnectionManager().getSchemeRegistry().register(new Scheme("https", 443, sslSocketFactory));
		return httpClient;
	}
}