com.cuubez.visualizer.scanner.ClassScanner.java Source code

Introduction

Here is the source code for com.cuubez.visualizer.scanner.ClassScanner.java

Source

/**
 *   Copyright [2013] [www.cuubez.com]
 *   Licensed under the Apache License, Version 2.0 (the "License");
 *   you may not use this file except in compliance with the License.
 *   You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 *   Unless required by applicable law or agreed to in writing, software
 *   distributed under the License is distributed on an "AS IS" BASIS,
 *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *   See the License for the specific language governing permissions and
 *   limitations under the License.
 */
package com.cuubez.visualizer.scanner;

import com.cuubez.visualizer.scanner.filter.*;
import com.cuubez.visualizer.scanner.reader.ClassFileReader;
import com.cuubez.visualizer.scanner.reader.FileReader;
import com.cuubez.visualizer.scanner.reader.JarFileReader;
import javassist.bytecode.ClassFile;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import java.io.*;
import java.net.URL;
import java.util.ArrayList;
import java.util.List;

public class ClassScanner extends FileScanner {

    private static Log log = LogFactory.getLog(ClassScanner.class);

    public List<Class<?>> scan(String applicationPath) throws IOException {

        List<Class<?>> classes = new ArrayList<Class<?>>();

        URL resource = findResources(applicationPath);
        FileReader itr = getFileReader(resource, new ClassFileFilter());

        InputStream is;
        while ((is = itr.next()) != null) {
            // make a data input stream
            DataInputStream dstream = new DataInputStream(new BufferedInputStream(is));

            try {
                // get java-assist class file
                ClassFile classFile = new ClassFile(dstream);
                classes.add(Class.forName(classFile.getName()));

            } catch (ClassNotFoundException e) {
                log.error(e);
            } finally {
                dstream.close();
                is.close();
            }
        }

        return classes;
    }

    private FileReader getFileReader(URL url, com.cuubez.visualizer.scanner.filter.FileFilter filter)
            throws IOException {
        String urlString = url.toString();
        if (urlString.endsWith("!/")) {
            urlString = urlString.substring(4);
            urlString = urlString.substring(0, urlString.length() - 2);
            url = new URL(urlString);
        }

        if (!urlString.endsWith("/")) {
            return new JarFileReader(url.openStream(), filter);
        } else {

            if (!url.getProtocol().equals("file")) {
                throw new IOException("Unable to understand protocol: " + url.getProtocol());
            }

            File f = new File(url.getPath());
            if (f.isDirectory()) {
                return new ClassFileReader(f, filter);
            } else {
                return new JarFileReader(url.openStream(), filter);
            }
        }
    }

}