/*
 *
 * Paros and its related class files.
 * 
 * Paros is an HTTP/HTTPS proxy for assessing web application security.
 * Copyright (C) 2003-2004 Chinotec Technologies Company
 * 
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the Clarified Artistic License
 * as published by the Free Software Foundation.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * Clarified Artistic License for more details.
 * 
 * You should have received a copy of the Clarified Artistic License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 */
package org.parosproxy.paros.common;

import java.io.File;
import java.io.FilenameFilter;
import java.io.IOException;
import java.io.InputStream;
import java.lang.reflect.Modifier;
import java.net.MalformedURLException;
import java.net.URI;
import java.net.URISyntaxException;
import java.net.URL;
import java.net.URLClassLoader;
import java.util.Enumeration;
import java.util.Vector;
import java.util.jar.JarFile;
import java.util.zip.ZipEntry;

/**
 *
 * To change the template for this generated type comment go to
 * Window - Preferences - Java - Code Generation - Code and Comments
 */
public class DynamicLoader extends URLClassLoader {

    private String directory = "";
    private URLClassLoader loader = null;
    private Vector listClassName = new Vector();
    private String packageName = "";
    
    public DynamicLoader(String directory, String packageName) {
        super(new URL[0]);
        this.directory = directory;
        checkLocal(packageName);
        searchJars();
    }
    
    private void searchJars() {
        File dir = new File(directory);
        File[] listFile = dir.listFiles(new FilenameFilter() {
            public boolean accept(File dir, String fileName) {
                if (fileName.endsWith(".jar")) {
                    return true;
                }
                return false;
            }
        });
        
        for (int i=0; i<listFile.length; i++) {
            if (!listFile[i].isFile()) {
                continue;
            }
            
            parseJarFile(listFile[i]);
        }
        
    }

    /**
     * Check local jar (paros.jar) or related package if any target file is found.
     *
     */
    private void checkLocal(String packageName) {
        if (packageName == null || packageName.equals("")) {
            return;
        }
        
        String folder = packageName.replace('.', '/');
        URL local = DynamicLoader.class.getClassLoader().getResource(folder);
        if (local == null) {
            return;
        }
        String jarFile = null;
        if (local.getProtocol().equals("jar")) {
            jarFile = local.toString().substring("jar:".length());
            int pos = jarFile.indexOf("!");
            jarFile = jarFile.substring(0, pos);
            System.out.println(jarFile);
            try {
                parseJarFile(new File(new URI(jarFile)));
            } catch (URISyntaxException e) {
                e.printStackTrace();
            }
        } else {
            try {
                parseClassDir(new File(new URI(local.toString())), packageName);
            } catch (URISyntaxException e) {
                e.printStackTrace();
            }
        }
        //jar:file:/C:/eclipse/workspace/parosng_make/paros/paros.jar!/com/proofsecure/paros/
        //file:/C:/eclipse/workspace/parosng/output/com/proofsecure/paros/
    }
    
    
    public Vector getFilteredObject (Class classType) {
        String className = "";
        Class cls = null;
        Vector listClass = new Vector();
        for (int i=0; i<listClassName.size(); i++) {
            className = (String) listClassName.get(i);
            try {
                cls = loadClass(className);
                // abstract class or interface cannot be constructed.
                if (Modifier.isAbstract(cls.getModifiers()) || Modifier.isInterface(cls.getModifiers())) {
                    continue;
                }
                if (classType.isAssignableFrom(cls)) {
                    Object obj = cls.newInstance();
                    listClass.add(obj);
                }
            } catch (Throwable e) {
                //e.printStackTrace();
            }
        }
        
        return listClass;
    }

    private void parseClassDir(File file, String packageName) {
        File[] listFile = file.listFiles(new FilenameFilter() {
            public boolean accept(File file, String fileName) {
                if (fileName.endsWith(".class")) {
                    return true;
                }
                return false;
            }
        });
        
        for (int i=0; i<listFile.length; i++) {
            File entry = listFile[i];
            if (!entry.isFile()) {
                continue;
            }
            String fileName = entry.toString();
            String match = packageName.replace('.', File.separatorChar);
            int pos = fileName.indexOf(match);
            if (pos > 0) {
                String className = fileName.substring(pos).replaceAll("\\.class$","").replace(File.separatorChar, '.');
                listClassName.add(className);

            }
        }
    }
    
    private void parseJarFile(File file) {
        JarFile jarFile = null;
        ZipEntry entry = null;
        InputStream in = null;
        String className = "";
        try {
            jarFile = new JarFile(file);
            Enumeration entries = jarFile.entries();
            while (entries.hasMoreElements()) {
                entry = (ZipEntry) entries.nextElement();
                if (entry.isDirectory() || !entry.getName().endsWith(".class")) {
                    continue;
                }
                className = entry.toString().replaceAll("\\.class$","").replaceAll("/",".");
                listClassName.add(className);
            }
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            if (jarFile != null) {
                try {
                    jarFile.close();
                } catch (IOException e1) {
                }
            }
        }
        try {
            this.addURL(file.toURL());
        } catch (MalformedURLException e1) {
        }
    }
    
}
