package org.drools.jboss.integration;

 import  org.jboss.arquillian.container.test.api.Deployment;

 import  org.jboss.shrinkwrap.api.ShrinkWrap;

 import  org.jboss.shrinkwrap.api.asset.Asset;

 import  org.jboss.shrinkwrap.api.spec.WebArchive;

 import org.slf4j.Logger;

 import org.slf4j.LoggerFactory;

import java.io.InputStream;

import java.util.Enumeration;

import java.util.jar.JarEntry;

import java.util.jar.JarFile;

import static junit.framework.TestCase.assertNotNull;

public class FullDistributionTest {

    private static final Logger logger = LoggerFactory.getLogger(FullDistributionTest.class);

    public static final String DIST_FILE = "distFile";

    public static WebArchive createDeploymentForDistribution(String distFile) throws Exception {

        WebArchive webArchive = ShrinkWrap.create(WebArchive.class);

        JarFile jar = new JarFile(distFile);

        JarEntry entry;

        Enumeration<JarEntry> entries = jar.entries();

        while (entries.hasMoreElements() && (entry = entries.nextElement()) != null) {

            if (!entry.isDirectory()) {

                webArchive.add(getAsset(jar, entry), entry.getName());

            }

        }

        webArchive.addClass(FullDistributionTest.class);

        System.out.println("Generated web archive");

        System.out.println("*******************************");

        System.out.println(webArchive.toString(true));

        System.out.println("*******************************");

        return webArchive;

    }

    public static Asset getAsset(final JarFile jar, final JarEntry entry) {

        return new Asset() {

            @Override

            public InputStream openStream() {

                try {

                    return jar.getInputStream(entry);

                } catch (Exception e) {

                    logger.error("asset generation failed. entry: " + entry, e);

                    //don't worry It's just a test

                    return null;

                }

            }

        };

    }

    @Deployment

    public static WebArchive createDeployment() {

        String distFile = System.getProperty(DIST_FILE);

        logger.info("Configured distribution file is: " + distFile);

        assertNotNull("Distribution file was not configured", distFile);

        WebArchive distWar = null;

        try {

            distWar = createDeploymentForDistribution(distFile);

        } catch (Exception e) {

            logger.error("WebArchive creation failed, fistFile: "+ distFile, e);

        }

        assertNotNull("WebArchive couldn't be created.", distWar);

        return distWar;

    }

}