com.xml.guard.utils.KtFileParser.kt Maven / Gradle / Ivy
package com.xml.guard.utils
import java.io.File
import java.util.regex.Pattern
private val classPattern by lazy { Pattern.compile("class\\s+([_a-zA-Z-0-9]+)[\\s(]") }
private val funPattern by lazy { Pattern.compile("(fun|fun\\s.*)\\s([._a-zA-Z0-9]+)\\s*\\(") }
private val fieldPattern by lazy { Pattern.compile("va[lr]\\s+([._a-zA-Z0-9]+)[\\s:]+") }
private val jvmFileNamePattern by lazy { Pattern.compile("@file\\s*:\\s*JvmName\\s*[(]\\s*\"(.+)\\s*\"[)]") }
private val docOrCommentPattern by lazy { Pattern.compile("(/[*][\\s\\S]*?[*]/)|(//.*\n)") }
/**
* User: ljx
* Date: 2023/6/24
* Time: 16:27
*
* 注意，如果声明的字符串里出现 //、 /*、 */、{、}、(、)、fun、val、var、class等字符时，极有可能出现未知异常
*/
class KtFileParser(file: File, filename: String) {
private val topClassNames = mutableListOf()
val topFunNames = mutableListOf()
val topFieldNames = mutableListOf()
var jvmName: String? = null
init {
if (file.isKt) {
try {
var content = file.readText()
val matcher = jvmFileNamePattern.matcher(content)
if (matcher.find()) {
jvmName = matcher.group(1).trim()
}
content = content.removeAllDocsAndComments()
content = content.removeAllBody()
topClassNames.addAll(content.findClassNames())
topFunNames.addAll(content.findFunNames())
topFieldNames.addAll(content.findFieldNames())
topClassNames.remove(filename)
} catch (t: Throwable) {
throw IllegalArgumentException("$file parser fail", t)
}
}
}
fun getTopClassOrFunOrFieldNames(): List {
val answers = mutableListOf()
answers.addAll(topClassNames)
answers.addAll(topFunNames)
answers.addAll(topFieldNames)
return answers;
}
}
private fun String.findClassNames(): MutableList {
val classNames = mutableListOf()
var index = 0
while (true) {
val matcher = classPattern.matcher(this)
if (matcher.find(index)) {
val classname = matcher.group(1)
classNames.add(classname)
index = matcher.end()
} else {
break
}
}
return classNames
}
private fun String.findFunNames(): List {
val funNames = mutableListOf()
var index = 0
while (true) {
val matcher = funPattern.matcher(this)
if (matcher.find(index)) {
val names = matcher.group(2).split('.') //拆分扩展方法
funNames.add(names.last())
index = matcher.end()
} else {
break
}
}
return funNames
}
private fun String.findFieldNames(): List {
val fieldNames = mutableListOf()
var index = 0
while (true) {
val matcher = fieldPattern.matcher(this)
if (matcher.find(index)) {
val names = matcher.group(1).split('.') //拆分扩展属性
fieldNames.add(names.last())
index = matcher.end()
} else {
break
}
}
return fieldNames
}
//移除所有的注释
private fun String.removeAllDocsAndComments(): String {
var content = this
var index = 0
while (true) {
val matcher = docOrCommentPattern.matcher(content)
if (matcher.find(index)) {
index = matcher.start()
content = content.removeRange(index, matcher.end())
} else {
break
}
}
return content
}
private fun String.removeAllBody(): String {
var content = this
var index = 0
while (true) {
val newIndex = content.indexOf('{', index)
if (newIndex == -1) break
content = content.removeBody(newIndex)
index = newIndex
}
return content
}
private fun String.removeBody(startIndex: Int): String {
if (startIndex < 0) return this
var endIndex = -1
var level = 0
val charArray = toCharArray()
for (i in startIndex until charArray.size) {
val char = charArray[i]
if (char == '{') {
level++
} else if (char == '}') {
level--
}
if (level == 0) {
endIndex = i
break
}
}
return removeRange(startIndex, endIndex + 1)
}