scouter.util.scan.Scanner Maven / Gradle / Ivy
/*
* Copyright 2015 the original author or authors.
* @https://github.com/scouter-project/scouter
*
* 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 scouter.util.scan;
import java.io.File;
import java.io.IOException;
import java.net.URL;
import java.util.Enumeration;
import java.util.HashSet;
import java.util.Iterator;
import java.util.Set;
import java.util.TreeSet;
import java.util.jar.JarEntry;
import java.util.jar.JarFile;
import scouter.util.StringUtil;
public class Scanner {
private String prefix;
public Scanner(String prefix) {
if (StringUtil.isEmpty(prefix))
this.prefix = null;
else
this.prefix = prefix.replace('.', '/');
}
public Set process() {
return process(Thread.currentThread().getContextClassLoader());
}
public Set process(ClassLoader loader) {
Set mainSet = new TreeSet();
if (this.prefix == null)
return mainSet;
try {
Set files = getRoot(loader);
Iterator itr = files.iterator();
while (itr.hasNext()) {
Set classes = listUp(itr.next());
mainSet.addAll(classes);
}
} catch (IOException e) {
e.printStackTrace();
}
return mainSet;
}
public Set getRoot() throws IOException {
Set files = new HashSet();
if (this.prefix != null) {
Enumeration en = Thread.currentThread().getContextClassLoader().getResources(prefix);
while (en.hasMoreElements()) {
File file = parse(en.nextElement());
files.add(file);
}
}
return files;
}
public Set getRoot(ClassLoader loader) throws IOException {
Set files = new HashSet();
if (this.prefix != null) {
Enumeration en = loader.getResources(prefix);
while (en.hasMoreElements()) {
File file = parse(en.nextElement());
files.add(file);
}
}
return files;
}
private File parse(URL res) {
String file = res.getFile();
int x = file.indexOf("!");
if (x > 0)
return new File(file.substring(file.indexOf("/"), x));
else
return new File(file.substring(file.indexOf("/"), file.length() - prefix.length()));
}
public Set listUp(File root) {
Set classes = new HashSet();
if (this.prefix != null) {
if (root.isDirectory()) {
listUp(classes, new File(root, prefix), root.getAbsolutePath());
} else {
try {
listUp(classes, new JarFile(root));
} catch (IOException e) {
e.printStackTrace();
}
}
}
return classes;
}
public static Set getClassesInJar(String jarName) {
Set classes = new HashSet();
try {
if (jarName == null)
return classes;
JarFile file = new JarFile(jarName);
Enumeration en = file.entries();
while (en.hasMoreElements()) {
JarEntry entry = en.nextElement();
if (entry.getName().toLowerCase().endsWith(".class")) {
classes.add(getClassName(entry.getName()));
}
}
} catch (Exception e) {
}
return classes;
}
public void listUp(Set classes, JarFile file) {
if (this.prefix == null)
return;
Enumeration en = file.entries();
while (en.hasMoreElements()) {
JarEntry entry = en.nextElement();
if (entry.getName().startsWith(prefix) == false) {
continue;
}
if (entry.getName().toLowerCase().endsWith(".class")) {
classes.add(getClassName(entry.getName()));
}
}
}
private void listUp(Set classes, File file, String root) {
if (file.isDirectory() == false)
return;
File[] sub = file.listFiles();
for (int i = 0; i < sub.length; i++) {
if (sub[i].getName().toLowerCase().endsWith(".class")) {
String name = sub[i].getAbsolutePath().substring(root.length());
name = name.replace('\\', '/');
if (name.startsWith("/"))
name = name.substring(1);
classes.add(getClassName(name));
}
if (sub[i].getName().startsWith("."))
continue;
if (sub[i].isDirectory()) {
listUp(classes, sub[i], root);
}
}
}
public static String getClassName(String name) {
return name.substring(0, name.length() - 6).replace('/', '.');
}
public static String cutOutLast(String name, String seperator) {
int x = name.lastIndexOf(seperator);
if (x < 0)
return name;
return name.substring(0, x);
}
}