BundleBuilder.java

/*
 * Copyright 2008,2009 Toni Menzel.
 * Copyright 2008 Alin Dreghiciu.
 *
 * Licensed  under the  Apache License,  Version 2.0  (the "License");
 * you may not use  this file  except in  compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed  under the  License is distributed on an "AS IS" BASIS,
 * WITHOUT  WARRANTIES OR CONDITIONS  OF ANY KIND, either  express  or
 * implied.
 *
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.ops4j.pax.url.dir.internal.bundle;

import java.io.IOException;
import java.io.InputStream;
import java.io.PipedInputStream;
import java.io.PipedOutputStream;
import java.util.Properties;
import java.util.jar.JarOutputStream;
import org.osgi.framework.Constants;
import org.ops4j.lang.NullArgumentException;
import org.ops4j.pax.swissbox.bnd.BndUtils;
import org.ops4j.pax.url.dir.internal.ResourceLocator;

/**
 * Responsible for creating the on-the fly bundle.
 *
 * @author Toni Menzel (toni@okidokiteam.com)
 * @author Alin Dreghiciu (adreghiciu@gmail.com)
 * @since May 29, 2008
 */
public class BundleBuilder
{

    private ResourceLocator m_resourceLocator;

    private Properties m_refs;

    /**
     * Constructor.
     *
     * @param ref             name of test class
     * @param resourceLocator locator that gathers all resources that have to be inside the test probe
     */
    public BundleBuilder( final Properties ref,
                          final ResourceLocator resourceLocator )
    {
        NullArgumentException.validateNotNull( ref, "ref" );
        NullArgumentException.validateNotNull( resourceLocator, "resourceLocator" );

        m_resourceLocator = resourceLocator;
        m_refs = ref;

    }

    /**
     * Builds an osgi bundle out of settings given while creating this instance.
     * Output is being asynchronously in a new thread when reading from the InputStream returned by this method.
     *
     * @return an inputstream that must be flushed in order to actually invoke the bundle build process.
     */
    public InputStream build()
    {
        try
        {
            // 1. create a basic jar with all classes in it..
            final PipedOutputStream pout = new PipedOutputStream();
            PipedInputStream fis = new PipedInputStream( pout );
            new Thread()
            {

                public void run()
                {
                    JarOutputStream jos;
                    try
                    {
                        jos = new DuplicateAwareJarOutputStream( pout );
                        m_resourceLocator.write( jos );
                        jos.close();
                    }

                    catch( IOException e )
                    {
                        //throw new RuntimeException( e );
                    }
                    finally
                    {
                        try
                        {
                            pout.close();
                        }
                        catch( Exception e )
                        {
                            //  throw new TestExecutionException( "Cannot close builder stream ??", e );
                        }
                    }
                }
            }.start();

            // TODO set args on BndUtils
            if( m_refs.getProperty( Constants.BUNDLE_SYMBOLICNAME ) == null )
            {
                m_refs.setProperty( Constants.BUNDLE_SYMBOLICNAME, "BuiltByDirUrlHandler" );
            }
            InputStream result = BndUtils.createBundle( fis, m_refs, m_resourceLocator.toString() );
            fis.close();
            pout.close();
            return result;
        }
        catch( IOException e )
        {
            throw new RuntimeException( e );
        }
    }


}