ai.platon.pulsar.persist.tools.WebPageCodeGenerator.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of pulsar-persist Show documentation
The newest version!
package ai.platon.pulsar.persist.tools

import ai.platon.pulsar.persist.WebPage
import ai.platon.pulsar.persist.experimental.WebAsset
import org.apache.commons.lang3.StringUtils
import kotlin.reflect.KType
import kotlin.reflect.full.declaredMemberFunctions

class WebPageCodeGenerator {
    
    fun generateJavaInterface() {
        val properties = WebPage::class.declaredMemberFunctions
            .filter { it.isOpen }
            .filter { it.name.startsWith("get") }
            .map {
                convertReturnType(it.returnType) + " " +
                it.name + "();"
            }

        val clazz = """
            |interface WebAsset {
            |    ${properties.joinToString("\n    ")}
            |}
        """.trimMargin()

        println(clazz)
    }
    
    fun generateJavaClassWithWebPageImpl() {
        val getters = WebAsset::class.declaredMemberFunctions
            .filter { it.isOpen }
            .filter { it.name.startsWith("get") }
            .map {
                """    public ${convertReturnType(it.returnType).trim()} ${it.name}() { return impl.${it.name}(); }"""
            }
        val setters = WebPage::class.declaredMemberFunctions
            .filter { it.isOpen }
            .filter { it.name.startsWith("set") }
            .map {
                """    public void ${it.name}(${it.parameters.map { it.name }}) { impl.${it.name}(${it.parameters.map { it.name }}); }"""
            }
        val properties = getters.zip(setters).map { it.first + "\n" + it.second }

        val clazz = """
            |class WebPage implements WebAsset {
            |    private final WebPageImpl impl;
            |    ${properties.joinToString("\n")}
            |}
        """.trimMargin()
        
        println(clazz)
    }
    
    fun generateJavaImmutableClass() {
        val properties = WebPage::class.declaredMemberFunctions
            .filter { it.isOpen }
            .filter { it.name.startsWith("get") }
            .map {
                convertReturnType(it.returnType) + " " +
                        it.name + "() { /** IMPLEMENTATION */ };"
            }

        val clazz = """
            |class WebPage {
            |    ${properties.joinToString("\n    ")}
            |}
        """.trimMargin()

        println(clazz)
    }

    fun generateKotlinImmutableClass() {
        val properties = WebPage::class.declaredMemberFunctions
            .filter { it.isOpen }
            .filter { it.name.startsWith("get") }
            .map {
                "val " + StringUtils.uncapitalize(it.name.substringAfter("get")) +
                        ": " + it.returnType.toString().filter { it != '!' }
                            .replace("kotlin.", "")
                            .replace("java.time.", "")
            }

        val clazz = """
            |class Asset {
            |    ${properties.joinToString("\n    ")}
            |}
        """.trimMargin()

        println(clazz)
    }

    fun generateKotlinMutableClass() {
        val properties = WebPage::class.declaredMemberFunctions
            .filter { it.isOpen }
            .filter { it.name.startsWith("get") }
            .map {
                "var " + StringUtils.uncapitalize(it.name.substringAfter("get")) +
                        ": " + convertReturnType(it.returnType)
            }

        val clazz = """
            |class MutableWebPage {
            |    ${properties.joinToString("\n    ")}
            |}
        """.trimMargin()

        println(clazz)
    }

    private fun convertReturnType(type: KType): String {
        var s = type.toString().filter { it != '!' }
            .replace("kotlin.", "")
            .replace("java.nio.", "")
            .replace("collections.(Mutable)", "")
            .replace("java.time.", "")

        if (!s.contains("<")) {
            s = s.substringAfterLast(".")
        }

        return s
    }
}

fun main() {
    val generator = WebPageCodeGenerator()
    generator.generateJavaClassWithWebPageImpl()
//    generator.generateJavaInterface()
//    generator.generateJavaImmutableClass()
//    generator.generateKotlinImmutableClass()
//    generator.generateKotlinMutableClass()
}