com.skillw.pouvoir.util.plugin.PluginUtils.kt Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of Pouvoir Show documentation
Show all versions of Pouvoir Show documentation
Bukkit Script Engine Plugin.
package com.skillw.pouvoir.util.plugin
import com.skillw.pouvoir.util.silent
import java.io.File
import java.io.IOException
import java.net.JarURLConnection
import java.net.URISyntaxException
import java.net.URL
import java.net.URLDecoder
import java.util.*
import java.util.jar.JarEntry
import java.util.jar.JarFile
object PluginUtils {
@JvmStatic
fun getClasses(clazz: Class<*>): List> {
val classes: MutableList> = ArrayList()
val url = clazz.protectionDomain.codeSource.location
silent {
val src: File = try {
File(url.toURI())
} catch (e: URISyntaxException) {
File(url.path)
}
JarFile(src).stream().filter { entry: JarEntry ->
entry.name.endsWith(".class")
}.forEach { entry: JarEntry ->
val className =
entry.name.replace('/', '.').substring(0, entry.name.length - 6)
silent { classes.add(Class.forName(className, false, clazz.classLoader)) }
}
}
return classes
}
/**
* 从包package中获取所有的Class
*
* @param pack
* @return
*/
@JvmStatic
fun getClasses(pack: String): Set> {
// 第一个class类的集合
val classes: MutableSet> = LinkedHashSet()
// 是否循环迭代
val recursive = true
// 获取包的名字 并进行替换
var packageName = pack
val packageDirName = packageName.replace('.', '/')
// 定义一个枚举的集合 并进行循环来处理这个目录下的things
val dirs: Enumeration
try {
dirs = Thread.currentThread().contextClassLoader.getResources(packageDirName)
// 循环迭代下去
while (dirs.hasMoreElements()) {
// 获取下一个元素
val url: URL = dirs.nextElement()
// 得到协议的名称
val protocol: String = url.protocol
// 如果是以文件的形式保存在服务器上
if ("file" == protocol) {
System.err.println("file类型的扫描")
// 获取包的物理路径
val filePath = URLDecoder.decode(url.file, "UTF-8")
// 以文件的方式扫描整个包下的文件 并添加到集合中
findAndAddClassesInPackageByFile(packageName, filePath, recursive, classes)
} else if ("jar" == protocol) {
// 如果是jar包文件
// 定义一个JarFile
// System.err.println("jar类型的扫描");
var jar: JarFile
try {
// 获取jar
jar = (url.openConnection() as JarURLConnection).jarFile
// 从此jar包 得到一个枚举类
val entries = jar.entries()
// 同样的进行循环迭代
while (entries.hasMoreElements()) {
// 获取jar里的一个实体 可以是目录 和一些jar包里的其他文件 如META-INF等文件
val entry = entries.nextElement()
var name = entry.name
// 如果是以/开头的
if (name[0] == '/') {
// 获取后面的字符串
name = name.substring(1)
}
// 如果前半部分和定义的包名相同
if (name.startsWith(packageDirName)) {
val idx = name.lastIndexOf('/')
// 如果以"/"结尾 是一个包
if (idx != -1) {
// 获取包名 把"/"替换成"."
packageName = name.substring(0, idx).replace('/', '.')
}
// 如果可以迭代下去 并且是一个包
if (idx != -1 || recursive) {
// 如果是一个.class文件 而且不是目录
if (name.endsWith(".class") && !entry.isDirectory) {
// 去掉后面的".class" 获取真正的类名
val className = name.substring(packageName.length + 1, name.length - 6)
try {
// 添加到classes
classes.add(Class.forName("$packageName.$className"))
} catch (e: ClassNotFoundException) {
e.printStackTrace()
}
}
}
}
}
} catch (e: IOException) {
// log.error("在扫描用户定义视图时从jar包获取文件出错");
e.printStackTrace()
}
}
}
} catch (e: IOException) {
e.printStackTrace()
}
return classes
}
/**
* 以文件的形式来获取包下的所有Class
*
* @param packageName
* @param packagePath
* @param recursive
* @param classes
*/
@JvmStatic
fun findAndAddClassesInPackageByFile(
packageName: String, packagePath: String?, recursive: Boolean,
classes: MutableSet>,
) {
// 获取此包的目录 建立一个File
val dir = File(packagePath)
// 如果不存在或者 也不是目录就直接返回
if (!dir.exists() || !dir.isDirectory) {
// log.warn("用户定义包名 " + packageName + " 下没有任何文件");
return
}
// 如果存在 就获取包下的所有文件 包括目录
val dirfiles: Array = dir.listFiles { file ->
// 自定义过滤规则 如果可以循环(包含子目录) 或则是以.class结尾的文件(编译好的java类文件)
recursive && file.isDirectory || file.name.endsWith(".class")
}!!
// 循环所有文件
for (file in dirfiles) {
// 如果是目录 则继续扫描
if (file.isDirectory) {
findAndAddClassesInPackageByFile(
packageName + "." + file.name, file.absolutePath, recursive,
classes
)
} else {
// 如果是java类文件 去掉后面的.class 只留下类名
val className = file.name.substring(0, file.name.length - 6)
try {
// 添加到集合中去
// classes.add(Class.forName(packageName + '.' + className));
// 经过回复同学的提醒，这里用forName有一些不好，会触发static方法，没有使用classLoader的load干净
classes.add(
Thread.currentThread().contextClassLoader.loadClass("$packageName.$className")
)
} catch (e: ClassNotFoundException) {
// log.error("添加用户自定义视图类错误 找不到此类的.class文件");
e.printStackTrace()
}
}
}
}
// --------------------------------------------------------------------------------------------------------
// --------------------------------------------------------------------------------------------------------
@JvmStatic
fun getByInterface(clazz: Class<*>, classesAll: Set>): Set> {
val classes: MutableSet> = LinkedHashSet()
// 获取指定接口的实现类
if (!clazz.isInterface) {
try {
/** 循环判断路径下的所有类是否继承了指定类 并且排除父类自己 */
val iterator = classesAll.iterator()
while (iterator.hasNext()) {
val cls = iterator.next()
/**
* isAssignableFrom该方法的解析，请参考博客：
* http://blog.csdn.net/u010156024/article/details/44875195
*/
if (clazz.isAssignableFrom(cls)) {
if (clazz != cls) { // 自身并不加进去
classes.add(cls)
} else {
}
}
}
} catch (e: Exception) {
println("出现异常")
}
}
return classes
}
} © 2015 - 2024 Weber Informatics LLC | Privacy Policy