/* 
 * Mutability Detector
 *
 * Copyright 2009 Graham Allan
 * 
 * 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 org.mutabilitydetector;

import static java.lang.String.format;
import static java.lang.Thread.currentThread;

import java.io.IOException;
import java.io.InputStream;

import org.mutabilitydetector.IAnalysisSession.AnalysisError;
import org.mutabilitydetector.checkers.IMutabilityChecker;
import org.objectweb.asm.ClassReader;

import com.google.classpath.ClassPath;

public class CheckerRunner {

	private ClassReader cr;
	private final ClassPath classpath;

	public CheckerRunner(ClassPath classpath) {
		this.classpath = classpath;
	}
	
	public static CheckerRunner createWithClasspath(ClassPath classpath) {
		return new CheckerRunner(classpath);
	}

	public static CheckerRunner createWithCurrentClasspath() {
		return new CheckerRunner(null);
	}

	public void run(IMutabilityChecker checker, Class<?> toCheck) {
		try {
			cr = new ClassReader(toCheck.getName());
		} catch (IOException e) {
			throw new RuntimeException(e);
		}
		cr.accept(checker, 0);
	}

	public void run(IAnalysisSession analysisSession, IMutabilityChecker checker, String dottedClassPath) {
		try {
			try {
				Class<?> toCheck = currentThread().getContextClassLoader().loadClass(dottedClassPath);
				cr = new ClassReader(toCheck.getName());
				cr.accept(checker, 0);
			} catch (Throwable e) {
				// Has to catch NoClassDefFoundError
				analyseAsStream(checker, dottedClassPath);
			}
		} catch (Throwable e) {
			handleException(analysisSession, checker, dottedClassPath, e);
		}
	}

	private void analyseAsStream(IMutabilityChecker checker, String dottedClassPath) throws IOException {
		String slashedClassPath = dottedClassPath.replace(".", "/").concat(".class");
		InputStream classStream = classpath.getResourceAsStream(slashedClassPath);
		cr = new ClassReader(classStream);
		cr.accept(checker, 0);
	}

	private void handleException(IAnalysisSession analysisSession, IMutabilityChecker checker, String dottedClassPath, Throwable e) {
		String errorDescription = format("It is likely that the class %s has dependencies outwith the given class path.", dottedClassPath);
		AnalysisError error = new AnalysisError(dottedClassPath, getNameOfChecker(checker), errorDescription);
		analysisSession.addAnalysisError(error);
	}

	private String getNameOfChecker(IMutabilityChecker checker) {
		String checkerName = checker.getClass().getName();
		checkerName = checkerName.substring(checkerName.lastIndexOf(".") + 1);
		return checkerName;

	}
}

@@ -692,54 +692,8 @@
 mat;
-%0Aimport static java.lang.Thread.currentThread;
 %0A%0Aim
@@ -868,16 +868,74 @@
 hecker;%0A
+import org.mutabilitydetector.cli.URLFallbackClassLoader;%0A
 import o
@@ -1751,34 +1751,23 @@
 k = 
-currentThread().getContext
+new URLFallback
 Clas
@@ -1780,12 +1780,11 @@
 r().
-load
+get
 Clas
@@ -2091,24 +2091,26 @@
 e);%0A%09%09%7D%0A%09%7D%0A%0A
+%0A%0A
 %09private voi
@@ -2419,24 +2419,24 @@
 er, 0);%0A%09%7D%0A%0A
-
 %09private voi
@@ -2550,24 +2550,47 @@
 owable e) %7B%0A
+%09%09e.printStackTrace();%0A
 %09%09String err

Use the new URLFallbackClassLoader.