Added some codegen features

master
Koitharu 2 years ago
parent f410df40f1
commit c58d35288b
Signed by: Koitharu
GPG Key ID: 676DEE768C17A9D7

@ -0,0 +1 @@
total: 1110

@ -14,6 +14,10 @@ test {
useJUnitPlatform()
}
ksp {
arg("summaryOutputDir", "${projectDir}/.github")
}
compileKotlin {
kotlinOptions {
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.KSVisitorVoid
import com.google.devtools.ksp.validate
import java.io.File
import java.io.Writer
import java.util.*
import kotlin.math.log
class ParserProcessor(
private val codeGenerator: CodeGenerator,
@ -19,9 +21,7 @@ class ParserProcessor(
private val sourceNamePattern = Regex("[A-Z_][A-Z0-9_]{3,}")
override fun process(resolver: Resolver): List<KSAnnotated> {
val symbols =
resolver
.getSymbolsWithAnnotation("org.koitharu.kotatsu.parsers.MangaSourceParser")
val symbols = resolver.getSymbolsWithAnnotation("org.koitharu.kotatsu.parsers.MangaSourceParser")
val ret = symbols.filterNot { it.validate() }.toList()
if (!symbols.iterator().hasNext()) {
return ret
@ -49,11 +49,12 @@ class ParserProcessor(
logger.warn(e.toString(), null)
null
}
sourcesFile?.writer().use { sourcesWriter ->
val totalCount = sourcesFile?.writer().use { sourcesWriter ->
factoryFile?.writer().use { factoryWriter ->
writeContent(sourcesWriter, factoryWriter, symbols)
}
}
writeSummary(totalCount)
return ret
}
@ -61,9 +62,9 @@ class ParserProcessor(
sourcesWriter: Writer?,
factoryWriter: Writer?,
symbols: Sequence<KSAnnotated>,
) {
): Int {
if (sourcesWriter == null && factoryWriter == null) {
return
return 0
}
factoryWriter?.write(
"""
@ -91,9 +92,10 @@ class ParserProcessor(
)
val visitor = ParserVisitor(sourcesWriter, factoryWriter)
symbols
val totalCount = symbols
.filter { it is KSClassDeclaration && it.validate() }
.forEach { it.accept(visitor, Unit) }
.onEach { it.accept(visitor, Unit) }
.count()
factoryWriter?.write(
"""
@ -112,6 +114,12 @@ class ParserProcessor(
}
""".trimIndent(),
)
return totalCount
}
private fun writeSummary(totalCount: Int) {
val file = File(options["summaryOutputDir"] ?: return, "summary.yaml")
file.writeText("total: $totalCount")
}
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')
}
Loading…
Cancel
Save