com.cuubez.visualizer.scanner.ClassScanner Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of cuubez-api-visualizer Show documentation
Cuubez API Visualization Framework is complete implementation for describing and visualizing RESTful web services.
The newest version!
/**
 *	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> scan(String applicationPath) throws IOException {

            List> classes = new ArrayList>();

            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);
            }
        }
    }

}