Application: gvSIG desktop » gvSIG scripting

package org.gvsig.scripting.impl;

import java.io.File;

import java.io.IOException;

import java.io.InputStream;

import java.net.URL;

import java.net.URLClassLoader;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.Enumeration;

import java.util.List;

import org.apache.commons.lang3.StringUtils;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class MutableURLClassLoader extends URLClassLoader {

    protected static final Logger logger = LoggerFactory.getLogger(MutableURLClassLoader.class);

    public MutableURLClassLoader(List<URL> urls, ClassLoader parent) {

        super(urls.toArray(new URL[urls.size()]),parent);

    }

    public void addUrls(List<URL> urls) {

        List<URL> currentsURLs = new ArrayList<>();

        currentsURLs.addAll(Arrays.asList(this.getURLs()));

        for (URL url : urls) {

            if( ! currentsURLs.contains(url) ) {

                this.addURL(url);

                currentsURLs.add(url);

            }

        }

    }

    public boolean needRecompile(String name, URL url) {

        if( url!=null && name.endsWith("$py.class") ) {

            if( StringUtils.equalsIgnoreCase(url.getProtocol(),"file") ) {

                String path = url.getPath();

                File binary = new File(path);

                File source = new File(StringUtils.left(path, path.length()-9)+".py");

                if( !binary.exists() || org.apache.commons.io.FileUtils.isFileNewer(source, binary) ) {

                    return true;

                }

            }

        }

        return false;

    }

    @Override

    public URL getResource(String name) {

        URL x = super.getResource(name); 

        if( needRecompile(name, x) ) {

            logger.trace("getResoure("+name+") -> null (need recompile).");

            return null;

        }

        logger.trace("getResoure("+name+") -> "+x);

        return x;

    }

    @Override

    public InputStream getResourceAsStream(String name) {

        URL url = super.getResource(name); 

        if( needRecompile(name, url) ) {

            logger.trace("getResourceAsStream("+name+") -> null (need recompile).");

            return null;

        }

        InputStream x = super.getResourceAsStream(name); 

        logger.trace("getResourceAsStream("+name+") -> "+(x==null? "null":"(stream)"));

        return x;

    }

    @Override

    public Enumeration<URL> getResources(String name) throws IOException {

        Enumeration<URL> urls = super.getResources(name); 

        logger.info("getResources("+name+") -> "+(urls==null? "null":urls));

        return urls;

    }

    public String getResourcePath(String name) {

        if( name.startsWith("__pyclasspath__/") ) {

            name = name.substring(16);

        }

        URL url = this.getResource(name);

        if( url == null ) {

            return name;

        }

        File f = org.apache.commons.io.FileUtils.toFile(url);

        return f.getAbsolutePath();

    }    

}