github-repos/mirai
1
0
Fork 0
mirror of https://github.com/mamoe/mirai.git synced 2025-01-10 18:40:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

Support indenting for codegen

Browse Source
This commit is contained in:
Him188 2020-06-24 02:04:45 +08:00
parent 6b4124895f
commit 9f0465c91b
1 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
backend/codegen/src/main/kotlin/net/mamoe/mirai/console/codegen/Codegen.kt
View File

@ -52,15 +52,19 @@ class CodegenScope : MutableList<Replacer> by mutableListOf() {
@RegionCodegenDsl
operator fun RegionCodegen.invoke(ktTypes: Collection<KtType>) {
add(Replacer {
it.replace(Regex("""//// region $regionName CODEGEN ////([\s\S]*?)//// endregion $regionName CODEGEN ////""")) {
val code = CodegenScope().apply { (this@invoke as Codegen).invoke(*ktTypes.toTypedArray()) }.applyTo("")
add(Replacer { content ->
content.replace(Regex("""//// region $regionName CODEGEN ////([\s\S]*?)( *)//// endregion $regionName CODEGEN ////""")) { result ->
val indent = result.groups[2]!!.value
val indentedCode = CodegenScope()
.apply { (this@invoke as Codegen).invoke(*ktTypes.toTypedArray()) }
.applyTo("")
.mapLine { "${indent}$it" }
"""
|//// region $regionName CODEGEN ////
|
|$code
|$indentedCode
|
|//// endregion $regionName CODEGEN ////
|${indent}//// endregion $regionName CODEGEN ////
""".trimMargin()
}
})
@ -70,6 +74,8 @@ class CodegenScope : MutableList<Replacer> by mutableListOf() {
annotation class CodegenDsl
}
internal fun String.mapLine(mapper: (String) -> CharSequence) = this.lines().joinToString("\n", transform = mapper)
@DslMarker
annotation class RegionCodegenDsl