liquibase.resource.ClassLoaderResourceAccessor Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of liquibase-core Show documentation
There is a newer version: 4.30.0
package liquibase.resource;

import liquibase.util.FileUtil;
import liquibase.util.StringUtils;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.net.*;
import java.util.*;

/**
 * An implementation of {@link liquibase.resource.ResourceAccessor} that wraps a class loader.
 */
public class ClassLoaderResourceAccessor extends AbstractResourceAccessor {

    private ClassLoader classLoader;
    public ClassLoaderResourceAccessor() {
        this.classLoader = getClass().getClassLoader();
        init(); //init needs to be called after classloader is set
    }

    public ClassLoaderResourceAccessor(ClassLoader classLoader) {
        this.classLoader = classLoader;
        init(); //init needs to be called after classloader is set
    }

    @Override
    public Set getResourcesAsStream(String path) throws IOException {
        Enumeration resources = classLoader.getResources(path);
        if (resources == null || !resources.hasMoreElements()) {
            return null;
        }
        Set seenUrls = new HashSet();
        Set returnSet = new HashSet();
        while (resources.hasMoreElements()) {
            URL url = resources.nextElement();
            if (seenUrls.contains(url.toExternalForm())) {
                continue;
            }
            seenUrls.add(url.toExternalForm());
            InputStream resourceAsStream = url.openStream();
            if (resourceAsStream != null) {
                returnSet.add(resourceAsStream);
            }
        }

        return returnSet;
    }

    @Override
    public Set list(String relativeTo, String path, boolean includeFiles, boolean includeDirectories, boolean recursive) throws IOException {
        path = convertToPath(relativeTo, path);

        URL fileUrl = classLoader.getResource(path);
        if (fileUrl == null) {
            return null;
        }

        if (!fileUrl.toExternalForm().startsWith("file:")) {
            if (fileUrl.toExternalForm().startsWith("jar:file:")
                    || fileUrl.toExternalForm().startsWith("wsjar:file:")
                    || fileUrl.toExternalForm().startsWith("zip:")) {

                String file = fileUrl.getFile();
                String splitPath = file.split("!")[0];
                if (splitPath.matches("file:\\/[A-Za-z]:\\/.*")) {
                    splitPath = splitPath.replaceFirst("file:\\/", "");
                } else {
                    splitPath = splitPath.replaceFirst("file:", "");
                }
                splitPath = URLDecoder.decode(splitPath, "UTF-8");
                File zipfile = new File(splitPath);


                File zipFileDir = FileUtil.unzip(zipfile);
                if (path.startsWith("classpath:")) {
                    path = path.replaceFirst("classpath:", "");
                }
                if (path.startsWith("classpath*:")) {
                    path = path.replaceFirst("classpath\\*:", "");
                }
                URI fileUri = new File(zipFileDir, path).toURI();
                fileUrl = fileUri.toURL();
            }
        }

        try {
            File file = new File(fileUrl.toURI());
            if (file.exists()) {
                Set returnSet = new HashSet();
                getContents(file, recursive, includeFiles, includeDirectories, path, returnSet);
                return returnSet;
            }
        } catch (URISyntaxException e) {
            //not a local file
        } catch (IllegalArgumentException e) {
            //not a local file
        }

        Enumeration resources = classLoader.getResources(path);
        if (resources == null || !resources.hasMoreElements()) {
            return null;
        }
        Set returnSet = new HashSet();
        while (resources.hasMoreElements()) {
            String url = resources.nextElement().toExternalForm();
            url = url.replaceFirst("^\\Q"+path+"\\E", "");
            returnSet.add(url);
        }
        return returnSet;
    }

    @Override
    public ClassLoader toClassLoader() {
        return classLoader;
    }

    @Override
    public String toString() {
        String description;
        if (classLoader instanceof URLClassLoader) {
            List urls = new ArrayList();
            for (URL url : ((URLClassLoader) classLoader).getURLs()) {
                urls.add(url.toExternalForm());
            }
            description = StringUtils.join(urls, ",");
        } else {
            description = classLoader.getClass().getName();
        }
        return getClass().getName()+"("+ description +")";

    }
}