edu.umd.cs.findbugs.test.service.ClassFileLocator Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of test-harness Show documentation
A test harness library for SpotBugs plugin developers to test on JUnit4
The newest version!
/**
 * Find Security Bugs
 * Copyright (c) Philippe Arteau, All rights reserved.
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 3.0 of the License, or (at your option) any later version.
 *
 * This library 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 GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library.
 */
package edu.umd.cs.findbugs.test.service;

import static org.junit.jupiter.api.Assertions.assertNotNull;
import static org.junit.jupiter.api.Assertions.fail;

import java.net.URISyntaxException;
import java.net.URL;

public class ClassFileLocator {
    private static final String PREFIX = "file:";

    /**
     * @param path
     *            class name
     * @return Full path to the class file base on class name.
     */
    public String getClassFilePath(String path) {
        ClassLoader cl = getClass().getClassLoader();
        URL url = cl.getResource(path + ".class");
        if (url != null) {
            return getFilenameFromUrl(url);
        }
        url = cl.getResource(path);
        assertNotNull(url, "No class found for the path = " + path);
        return getFilenameFromUrl(url);
    }

    public String getJspFilePath(String path) {
        ClassLoader cl = getClass().getClassLoader();

        //This is subject to change base on the JSP compiler implementation
        String generatedClassName = path.replace("_", "_005f").replace(".jsp", "_jsp");
        URL url = cl.getResource("jsp/" + generatedClassName + ".class");
        if (url == null) {
            url = cl.getResource("org/apache/jsp/" + generatedClassName + ".class");
        }

        assertNotNull(url, "No jsp file found for the path = " + path);
        return getFilenameFromUrl(url);
    }

    public String getJarFilePath(String path) {
        ClassLoader cl = getClass().getClassLoader();
        URL url = cl.getResource(path);
        assertNotNull(url, "No jar found for the path = " + path);
        return getFilenameFromUrl(url);
    }

    private String getFilenameFromUrl(URL url) {
        String filename;
        try {
            filename = url.toURI().getPath();
        } catch (final URISyntaxException e) {
            fail("Failed to get file path = " + url);
            return null;
        }

        if (filename.startsWith(PREFIX)) {
            filename = filename.substring(PREFIX.length());
        }
        return filename;
    }
}