com.anatawa12.javaStabGen.Annotations.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of java-stab-gen Show documentation
a stab in java of class file generator. a tool for workaround of KT-24309.
The newest version!
package com.anatawa12.javaStabGen

import com.squareup.javapoet.AnnotationSpec
import com.squareup.javapoet.ClassName
import com.squareup.javapoet.CodeBlock
import org.objectweb.asm.Type
import org.objectweb.asm.tree.AnnotationNode
import org.objectweb.asm.tree.ClassNode

internal fun List.toSpecs(classNode: ClassNode): List =
    mapNotNull { it.toSpec(classNode) }

internal fun AnnotationNode.toSpec(classNode: ClassNode): AnnotationSpec? {
    val typeName = typeNameFromDescriptor(desc, classNode) as? ClassName ?: return null
    val builder = AnnotationSpec.builder(typeName)

    for (list in values.orEmpty().asSequence().chunked(2)) {
        val name = list[0] as String
        val value = list[1]
        builder.addMember(name, value.asAnnotationCode(classNode))
    }

    return builder.build()
}

internal fun Any.asAnnotationCode(classNode: ClassNode): CodeBlock? {
    return when (this) {
        is ByteArray -> asArrayBodyCode()
        is BooleanArray -> asArrayBodyCode()
        is ShortArray -> asArrayBodyCode()
        is CharArray -> asArrayBodyCode()
        is IntArray -> asArrayBodyCode()
        is LongArray -> asArrayBodyCode()
        is FloatArray -> asArrayBodyCode()
        is DoubleArray -> asArrayBodyCode()
        is Byte -> asCode()
        is Boolean -> asCode()
        is Short -> asCode()
        is Char -> asCode()
        is Int -> asCode()
        is Long -> asCode()
        is Float -> asCode()
        is Double -> asCode()
        is String -> asCode()
        is Type -> asAnnotationCode(classNode)
        is AnnotationNode -> asArrayBodyCode(classNode)
        is List<*> -> asArrayBodyCode(classNode)
        is Array<*> -> {
            val desc = this[0] as String
            val name = this[1] as String
            val typeName = typeNameFromDescriptor(desc, classNode) ?: return null
            CodeBlock.of("\$T.\$N", typeName, name)
        }
        else -> error("invalid value type: ${this::class}")
    }
}

internal fun AnnotationNode.asArrayBodyCode(classNode: ClassNode): CodeBlock? = CodeBlock.of("\$L", toSpec(classNode))

internal fun List<*>.asArrayBodyCode(classNode: ClassNode) = CodeBlock.builder().add("{")
    .apply { forEach { add("\$L, ", it!!.asAnnotationCode(classNode)) } }.add("}").build()

internal fun ByteArray.asArrayBodyCode() = CodeBlock.builder().add("{")
    .apply { forEach { add("\$L, ", it.asCode()) } }.add("}").build()

internal fun BooleanArray.asArrayBodyCode() = CodeBlock.builder().add("{")
    .apply { forEach { add("\$L, ", it.asCode()) } }.add("}").build()

internal fun ShortArray.asArrayBodyCode() = CodeBlock.builder().add("{")
    .apply { forEach { add("\$L, ", it.asCode()) } }.add("}").build()

internal fun CharArray.asArrayBodyCode() = CodeBlock.builder().add("{")
    .apply { forEach { add("\$L, ", it.asCode()) } }.add("}").build()

internal fun IntArray.asArrayBodyCode() = CodeBlock.builder().add("{")
    .apply { forEach { add("\$L, ", it.asCode()) } }.add("}").build()

internal fun LongArray.asArrayBodyCode() = CodeBlock.builder().add("{")
    .apply { forEach { add("\$L, ", it.asCode()) } }.add("}").build()

internal fun FloatArray.asArrayBodyCode() = CodeBlock.builder().add("{")
    .apply { forEach { add("\$L, ", it.asCode()) } }.add("}").build()

internal fun DoubleArray.asArrayBodyCode() = CodeBlock.builder().add("{")
    .apply { forEach { add("\$L, ", it.asCode()) } }.add("}").build()

internal fun Byte.asCode() = CodeBlock.of("\$L", this)
internal fun Boolean.asCode() = CodeBlock.of("\$L", this)
internal fun Short.asCode() = CodeBlock.of("\$L", this)
internal fun Char.asCode() = CodeBlock.of("\$%L'", this.escape())
internal fun Int.asCode() = CodeBlock.of("\$L", this)
internal fun Long.asCode() = CodeBlock.of("\$L", this)
internal fun Float.asCode() = CodeBlock.of("\$L", this)
internal fun Double.asCode() = CodeBlock.of("\$L", this)
internal fun String.asCode() = CodeBlock.of("\$S", this)
internal fun Type.asAnnotationCode(classNode: ClassNode): CodeBlock? =
    asTypeName(classNode)?.let { CodeBlock.of("\$T.class", it) }

private fun Char.escape(): String = when (this) {
    '\b' -> "\\b" /* \u0008: backspace (BS) */
    '\t' -> "\\t" /* \u0009: horizontal tab (HT) */
    '\n' -> "\\n" /* \u000a: linefeed (LF) */
    '\u000c' -> "\\f" /* \u000c: form feed (FF) */
    '\r' -> "\\r" /* \u000d: carriage return (CR) */
    '\"' -> "\"" /* \u0022: double quote (") */
    '\'' -> "\\'" /* \u0027: single quote (') */
    '\\' -> "\\\\" /* \u005c: backslash (\) */
    else -> if (Character.isISOControl(this)) String.format("\\u%04x", this.toInt()) else this.toString()
}