org.evosuite.utils.HtmlAnalyzer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of evosuite-client Show documentation
The newest version!
/**
 * Copyright (C) 2010-2018 Gordon Fraser, Andrea Arcuri and EvoSuite
 * contributors
 *
 * This file is part of EvoSuite.
 *
 * EvoSuite 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.
 *
 * EvoSuite 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 Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with EvoSuite. If not, see .
 */
package org.evosuite.utils;

import java.io.File;
import java.io.Serializable;
import java.util.Arrays;
import java.util.Collection;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

import org.apache.commons.io.FileUtils;
import org.evosuite.Properties;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class HtmlAnalyzer implements Serializable {

	private static final long serialVersionUID = -5634296119340294425L;

	private static final Logger logger = LoggerFactory.getLogger(HtmlAnalyzer.class);

	private Set files;

	/**
	 * 
	 * getClassContent
	 * 
	 * 
	 * @param fullClassName
	 *            a {@link java.lang.String} object.
	 * @return a {@link java.lang.Iterable} object.
	 */
	public Iterable getClassContent(String fullClassName) {
		if (files == null) {
			initFiles();
		}
		// String className = fullClassName.substring(fullClassName
		// .lastIndexOf('.') + 1);
		String className = getClassName(fullClassName);
		logger.debug("Looking for content of class " + fullClassName + " in "
		        + files.size() + " files.");
		logger.debug("Files: " + files);
		for (File f : files) {
			String name = getContainingClassName(f);
			if (name.endsWith(className) && !name.endsWith("Test" + className)) {
				List linesFromFile = FileIOUtils.readFile(f);
				logger.debug("Got file " + f + "for class " + fullClassName);
				return linesFromFile;
			}
		}
		String msg = "No source found for " + fullClassName;
		logger.debug(msg);
		return Arrays.asList(msg);
	}

	/**
	 * 
	 * getContainingClassName
	 * 
	 * 
	 * @param f
	 *            a {@link java.io.File} object.
	 * @return a {@link java.lang.String} object.
	 */
	public static String getContainingClassName(File f) {
		String name = f.getAbsolutePath();
		String sep = System.getProperty("file.separator");
		name = name.replace(sep, ".");
		if (name.endsWith(".java")) {
			name = name.substring(0, name.length() - 5);
		}
		int i = name.lastIndexOf(Properties.PROJECT_PREFIX);
		int j = name.lastIndexOf(sep);
		if (i > j) {

		} else if (i < 0) {
			name = "";
		} else {
			name = name.substring(i);
		}
		return name;
	}

	private void initFiles() {
		File startDirectory = new File(".");
		String[] extensions = { "java" };
		Collection javaFiles = FileUtils.listFiles(startDirectory, extensions, true);
		files = new HashSet(javaFiles);
	}

	private String getClassName(String name) {
		if (name.contains("$")) {
			name = name.substring(0, name.indexOf('$'));
		}
		return name;
	}
}