com.infotel.seleniumrobot.grid.utils.ResourceFolder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of seleniumRobot-grid Show documentation
Selenium grid extension for mobile testing
There is a newer version: 5.0.11
package com.infotel.seleniumrobot.grid.utils;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.URISyntaxException;
import java.net.URL;
import java.net.URLDecoder;
import java.util.Enumeration;
import java.util.jar.JarEntry;
import java.util.jar.JarFile;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

import org.apache.log4j.Logger;
import org.zeroturnaround.zip.NameMapper;
import org.zeroturnaround.zip.ZipUtil;

import com.google.common.io.Resources;
import com.infotel.seleniumrobot.grid.exceptions.ResourcePackagingException;

/**
 * @author Alexey Nikolaenko [email protected]
 *         Date: 08/10/2015
 */
public class ResourceFolder {

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

    private String path;
    private String zipRoot;

    public ResourceFolder(String path) {
        this.path = path;
        zipRoot = path;
    }

    public File toZip() {

        try {
        	URL dirURL = null;
        	try {
        		dirURL = Resources.getResource(path);
        	} catch (IllegalArgumentException e) {
        		File dirFile = new File(path);
        		if (dirFile.isDirectory()) {
        			dirURL = dirFile.toURI().toURL();
        		}
        		zipRoot = dirFile.getName();
        	}
            if (dirURL == null) {
                throw new ResourcePackagingException("Failed to get resource at " + path);
            }

            File destZip = File.createTempFile("resources_", ".zip");

            if ("file".equals(dirURL.getProtocol())) {
                mapFolderFromResources(dirURL, destZip);
            } else {
                mapFolderFromJar(dirURL, destZip);
            }

            return destZip;
        } catch (Exception e) {
            throw new ResourcePackagingException(e);
        }
    }

    private void mapFolderFromResources(URL dirURL, File destZip) throws URISyntaxException {
        File srcFolder = new File(dirURL.toURI());
        ZipUtil.pack(srcFolder, destZip, pathKeeper());
    }

    private void mapFolderFromJar(URL dirURL, File destZip) throws IOException {
    	String jarPath = substringJarPath(dirURL.getPath());
        try (
                ZipOutputStream zip = new ZipOutputStream(new FileOutputStream(destZip));
        		JarFile jar = new JarFile(URLDecoder.decode(jarPath, "UTF-8"))) {

            byte[] buffer = new byte[1024];
            int len;

            Enumeration entries = jar.entries();
            while (entries.hasMoreElements()) {
                JarEntry jarEntry = entries.nextElement();
                String name = jarEntry.getName();

                if (name.startsWith(path)) {
                    logger.debug("Zipping entry " + name);

                    ZipEntry zipEntry = jar.getEntry(name);
                    zip.putNextEntry(zipEntry);

                    URL resource = Resources.getResource(zipEntry.getName());
                    try (InputStream inputStream = resource.openStream()) {
                        while ((len = inputStream.read(buffer)) > 0) {
                            zip.write(buffer, 0, len);
                        }
                    }
                    zip.closeEntry();
                }
            }
        }
    }

    /**
     * Example resource folder path may look like file:/C:/testware/target/ARTIFACT-1.0.0-SNAPSHOT.jar!/img
     * 
     * Method cuts file: as first 5 symbols,
     * then cut any path after jar!..
     *
     * @param resourceFolderPath
     * @return jar path
     */
    private String substringJarPath(String resourceFolderPath) {
        return resourceFolderPath.substring(5, resourceFolderPath.indexOf("!"));
    }

    private NameMapper pathKeeper() {
        return new NameMapper() {
            @Override
            public String map(String name) {
                return zipRoot + "/" + name;
            }
        };
    }
}