/*******************************************************************************
* Copyright (c) 2010, 2013 Sonatype, Inc.
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the Eclipse Public License v1.0
* which accompanies this distribution, and is available at
* http://www.eclipse.org/legal/epl-v10.html
*
* Contributors:
* Stuart McCulloch (Sonatype, Inc.) - initial API and implementation
*******************************************************************************/
package org.eclipse.sisu.launch;
import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.lang.reflect.Proxy;
import java.util.Iterator;
import java.util.Map.Entry;
import javax.inject.Inject;
import org.eclipse.sisu.inject.BeanLocator;
import org.eclipse.sisu.inject.Logs;
import org.eclipse.sisu.wire.WireModule;
import com.google.inject.Binder;
import com.google.inject.Guice;
import com.google.inject.Injector;
import com.google.inject.Key;
import com.google.inject.Module;
import com.google.inject.util.Providers;
public final class SisuGuice
{
// ----------------------------------------------------------------------
// Implementation fields
// ----------------------------------------------------------------------
private static final ThreadLocal<BeanLocator> LOCATOR = new InheritableThreadLocal<BeanLocator>();
private static volatile BeanLocator latest;
// ----------------------------------------------------------------------
// Constructors
// ----------------------------------------------------------------------
private SisuGuice()
{
// static utility class, not allowed to create instances
}
// ----------------------------------------------------------------------
// Public methods
// ----------------------------------------------------------------------
@Inject
public static void setBeanLocator( final BeanLocator locator )
{
if ( null != locator )
{
LOCATOR.set( locator );
}
else
{
LOCATOR.remove();
}
latest = locator;
}
public static BeanLocator getBeanLocator()
{
final BeanLocator locator = LOCATOR.get();
return null != locator ? locator : latest;
}
public static <T> T lookup( final Key<T> key )
{
final BeanLocator locator = getBeanLocator();
if ( null != locator )
{
final Iterator<? extends Entry<?, T>> i = locator.locate( key ).iterator();
if ( i.hasNext() )
{
return i.next().getValue();
}
}
else
{
Logs.trace( "No BeanLocator found for thread {}", Thread.currentThread(), null );
}
return null;
}
public static void inject( final Object that )
{
final BeanLocator locator = getBeanLocator();
if ( null != locator )
{
Guice.createInjector( new WireModule()
{
@Override
public void configure( final Binder binder )
{
binder.bind( BeanLocator.class ).toProvider( Providers.of( locator ) );
binder.requestInjection( that );
}
} );
}
else
{
Logs.trace( "No BeanLocator found for thread {}", Thread.currentThread(), null );
}
}
public static Injector enhance( final Injector injector )
{
final Class<?>[] api = { Injector.class };
return (Injector) Proxy.newProxyInstance( api[0].getClassLoader(), api, new InvocationHandler()
{
@SuppressWarnings( { "rawtypes", "unchecked" } )
public Object invoke( final Object proxy, final Method method, final Object[] args )
throws Throwable
{
final String methodName = method.getName();
if ( "getInstance".equals( methodName ) )
{
final Key key = args[0] instanceof Key ? (Key) args[0] : Key.get( (Class) args[0] );
final Iterator<Entry> i = injector.getInstance( BeanLocator.class ).locate( key ).iterator();
return i.hasNext() ? i.next().getValue() : null;
}
if ( "injectMembers".equals( methodName ) )
{
Guice.createInjector( new WireModule( new Module()
{
public void configure( final Binder binder )
{
binder.bind( BeanLocator.class ).toProvider( injector.getProvider( BeanLocator.class ) );
binder.requestInjection( args[0] );
}
} ) );
return null;
}
return method.invoke( injector, args );
}
} );
}
}