Added some codegen features
This commit is contained in:
1
.github/summary.yaml
vendored
Normal file
1
.github/summary.yaml
vendored
Normal file
@@ -0,0 +1 @@
|
|||||||
|
total: 1110
|
||||||
@@ -14,6 +14,10 @@ test {
|
|||||||
useJUnitPlatform()
|
useJUnitPlatform()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
ksp {
|
||||||
|
arg("summaryOutputDir", "${projectDir}/.github")
|
||||||
|
}
|
||||||
|
|
||||||
compileKotlin {
|
compileKotlin {
|
||||||
kotlinOptions {
|
kotlinOptions {
|
||||||
freeCompilerArgs += [
|
freeCompilerArgs += [
|
||||||
|
|||||||
@@ -7,8 +7,10 @@ import com.google.devtools.ksp.symbol.KSAnnotated
|
|||||||
import com.google.devtools.ksp.symbol.KSClassDeclaration
|
import com.google.devtools.ksp.symbol.KSClassDeclaration
|
||||||
import com.google.devtools.ksp.symbol.KSVisitorVoid
|
import com.google.devtools.ksp.symbol.KSVisitorVoid
|
||||||
import com.google.devtools.ksp.validate
|
import com.google.devtools.ksp.validate
|
||||||
|
import java.io.File
|
||||||
import java.io.Writer
|
import java.io.Writer
|
||||||
import java.util.*
|
import java.util.*
|
||||||
|
import kotlin.math.log
|
||||||
|
|
||||||
class ParserProcessor(
|
class ParserProcessor(
|
||||||
private val codeGenerator: CodeGenerator,
|
private val codeGenerator: CodeGenerator,
|
||||||
@@ -19,9 +21,7 @@ class ParserProcessor(
|
|||||||
private val sourceNamePattern = Regex("[A-Z_][A-Z0-9_]{3,}")
|
private val sourceNamePattern = Regex("[A-Z_][A-Z0-9_]{3,}")
|
||||||
|
|
||||||
override fun process(resolver: Resolver): List<KSAnnotated> {
|
override fun process(resolver: Resolver): List<KSAnnotated> {
|
||||||
val symbols =
|
val symbols = resolver.getSymbolsWithAnnotation("org.koitharu.kotatsu.parsers.MangaSourceParser")
|
||||||
resolver
|
|
||||||
.getSymbolsWithAnnotation("org.koitharu.kotatsu.parsers.MangaSourceParser")
|
|
||||||
val ret = symbols.filterNot { it.validate() }.toList()
|
val ret = symbols.filterNot { it.validate() }.toList()
|
||||||
if (!symbols.iterator().hasNext()) {
|
if (!symbols.iterator().hasNext()) {
|
||||||
return ret
|
return ret
|
||||||
@@ -49,11 +49,12 @@ class ParserProcessor(
|
|||||||
logger.warn(e.toString(), null)
|
logger.warn(e.toString(), null)
|
||||||
null
|
null
|
||||||
}
|
}
|
||||||
sourcesFile?.writer().use { sourcesWriter ->
|
val totalCount = sourcesFile?.writer().use { sourcesWriter ->
|
||||||
factoryFile?.writer().use { factoryWriter ->
|
factoryFile?.writer().use { factoryWriter ->
|
||||||
writeContent(sourcesWriter, factoryWriter, symbols)
|
writeContent(sourcesWriter, factoryWriter, symbols)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
writeSummary(totalCount)
|
||||||
return ret
|
return ret
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -61,9 +62,9 @@ class ParserProcessor(
|
|||||||
sourcesWriter: Writer?,
|
sourcesWriter: Writer?,
|
||||||
factoryWriter: Writer?,
|
factoryWriter: Writer?,
|
||||||
symbols: Sequence<KSAnnotated>,
|
symbols: Sequence<KSAnnotated>,
|
||||||
) {
|
): Int {
|
||||||
if (sourcesWriter == null && factoryWriter == null) {
|
if (sourcesWriter == null && factoryWriter == null) {
|
||||||
return
|
return 0
|
||||||
}
|
}
|
||||||
factoryWriter?.write(
|
factoryWriter?.write(
|
||||||
"""
|
"""
|
||||||
@@ -91,9 +92,10 @@ class ParserProcessor(
|
|||||||
)
|
)
|
||||||
|
|
||||||
val visitor = ParserVisitor(sourcesWriter, factoryWriter)
|
val visitor = ParserVisitor(sourcesWriter, factoryWriter)
|
||||||
symbols
|
val totalCount = symbols
|
||||||
.filter { it is KSClassDeclaration && it.validate() }
|
.filter { it is KSClassDeclaration && it.validate() }
|
||||||
.forEach { it.accept(visitor, Unit) }
|
.onEach { it.accept(visitor, Unit) }
|
||||||
|
.count()
|
||||||
|
|
||||||
factoryWriter?.write(
|
factoryWriter?.write(
|
||||||
"""
|
"""
|
||||||
@@ -112,6 +114,12 @@ class ParserProcessor(
|
|||||||
}
|
}
|
||||||
""".trimIndent(),
|
""".trimIndent(),
|
||||||
)
|
)
|
||||||
|
return totalCount
|
||||||
|
}
|
||||||
|
|
||||||
|
private fun writeSummary(totalCount: Int) {
|
||||||
|
val file = File(options["summaryOutputDir"] ?: return, "summary.yaml")
|
||||||
|
file.writeText("total: $totalCount")
|
||||||
}
|
}
|
||||||
|
|
||||||
private inner class ParserVisitor(
|
private inner class ParserVisitor(
|
||||||
|
|||||||
@@ -0,0 +1,40 @@
|
|||||||
|
package org.koitharu.kotatsu.parsers.util
|
||||||
|
|
||||||
|
import org.junit.jupiter.api.Test
|
||||||
|
import org.koitharu.kotatsu.parsers.MangaLoaderContextMock
|
||||||
|
import org.koitharu.kotatsu.parsers.model.MangaParserSource
|
||||||
|
import org.koitharu.kotatsu.parsers.newParser
|
||||||
|
import java.io.File
|
||||||
|
|
||||||
|
class IntentFilterGenerator {
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun generateIntentFilter() {
|
||||||
|
val output = File("out/test/resources/intent-filter.xml")
|
||||||
|
output.printWriter(Charsets.UTF_8).use { writer ->
|
||||||
|
writer.appendLine("<intent-filter android:autoVerify=\"false\">")
|
||||||
|
writer.appendTab().appendLine("<action android:name=\"android.intent.action.VIEW\" />")
|
||||||
|
writer.appendLine()
|
||||||
|
writer.appendTab().appendLine("<category android:name=\"android.intent.category.DEFAULT\" />")
|
||||||
|
writer.appendTab().appendLine("<category android:name=\"android.intent.category.BROWSABLE\" />")
|
||||||
|
writer.appendLine()
|
||||||
|
writer.appendTab().appendLine("<data android:scheme=\"http\" />")
|
||||||
|
writer.appendTab().appendLine("<data android:scheme=\"https\" />")
|
||||||
|
writer.appendLine()
|
||||||
|
for (source in MangaParserSource.entries) {
|
||||||
|
if (source == MangaParserSource.DUMMY) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
val parser = source.newParser(MangaLoaderContextMock)
|
||||||
|
parser.configKeyDomain.presetValues.forEach { domain ->
|
||||||
|
writer.appendTab().append("<data android:host=\"").append(domain).appendLine("\" />")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
writer.appendLine()
|
||||||
|
writer.appendLine("</intent-filter>")
|
||||||
|
}
|
||||||
|
println(output.absolutePath)
|
||||||
|
}
|
||||||
|
|
||||||
|
private fun Appendable.appendTab() = append('\t')
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user