com.simiacryptus.util.describe.YamlDescriber.kt Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of joe-penai Show documentation
Show all versions of joe-penai Show documentation
A Java client for OpenAI's API
package com.simiacryptus.util.describe
import com.fasterxml.jackson.module.kotlin.isKotlinClass
import com.google.common.reflect.TypeToken
import com.simiacryptus.util.describe.DescriptorUtil.componentType
import com.simiacryptus.util.describe.DescriptorUtil.isArray
import com.simiacryptus.util.describe.TypeDescriber.Companion.primitives
import java.lang.reflect.*
import java.util.*
import kotlin.reflect.KFunction
import kotlin.reflect.KParameter
import kotlin.reflect.KType
import kotlin.reflect.KVisibility
import kotlin.reflect.full.functions
import kotlin.reflect.full.memberProperties
import kotlin.reflect.jvm.javaType
open class YamlDescriber : TypeDescriber {
override fun describe(
rawType: Class,
stackMax: Int,
): String {
if (isAbbreviated(rawType) || stackMax <= 0) return """
|type: object
|class: ${rawType.name}
""".trimMargin()
val propertiesYaml = if (rawType.isKotlinClass()) {
rawType.kotlin.memberProperties.filter { it.visibility == KVisibility.PUBLIC }.map {
val description =
DescriptorUtil.getAllAnnotations(rawType, it).find { x -> x is Description } as? Description
if (description != null) {
"""
|${it.name}:
| description: ${description.value.trim()}
| ${toYaml(it.returnType.javaType, stackMax - 1).replace("\n", "\n ")}
""".trimMargin().trim()
} else {
"""
|${it.name}:
| ${toYaml(it.returnType.javaType, stackMax - 1).replace("\n", "\n ")}
""".trimMargin().trim()
}
}.toTypedArray()
} else {
rawType.declaredFields.filter { Modifier.isPublic(it.modifiers) }.map {
val description =
it.annotations.find { x -> x is Description } as? Description
return@map if (description != null) """
|${it.name}:
| description: ${description.value.trim()}
| ${toYaml(it.genericType, stackMax - 1).replace("\n", "\n ")}
""".trimIndent()
else
"""
|${it.name}:
| ${toYaml(it.genericType, stackMax - 1).replace("\n", "\n ")}
""".trimMargin().trim()
}.toTypedArray()
}
val methodsYaml = if (rawType.isKotlinClass()) {
rawType.kotlin.functions.filter {
it.visibility == KVisibility.PUBLIC
&& !methodBlacklist.contains(it.name)
&& !it.isOperator && !it.isInfix && !it.isAbstract
}.map {
"""
|${it.name}:
| ${describe(it, stackMax - 1).replace("\n", "\n ")}
""".trimMargin().trim()
}.toTypedArray()
} else {
if(includeMethods) {
rawType.methods
.filter {
Modifier.isPublic(it.modifiers) && !it.isSynthetic && !it.name.contains("$") && !methodBlacklist.contains(
it.name
)
}
.map {
"""
|${it.name}:
| ${describe(it, stackMax - 1).replace("\n", "\n ")}
""".trimMargin().trim()
}.toTypedArray()
} else {
arrayOf()
}
}
if (propertiesYaml.isEmpty() && methodsYaml.isEmpty()) return """
|type: object
|class: ${rawType.name}
""".trimMargin()
if (propertiesYaml.isEmpty()) return """
|type: object
|class: ${rawType.name}
|methods:
| ${methodsYaml.joinToString("\n").replace("\n", "\n ")}
""".trimMargin()
if (methodsYaml.isEmpty()) return """
|type: object
|class: ${rawType.name}
|properties:
| ${propertiesYaml.joinToString("\n").replace("\n", "\n ")}
""".trimMargin()
return """
|type: object
|class: ${rawType.name}
|properties:
| ${propertiesYaml.joinToString("\n").replace("\n", "\n ")}
|methods:
| ${methodsYaml.joinToString("\n").replace("\n", "\n ")}
""".trimMargin()
}
open val includeMethods: Boolean = true
override val methodBlacklist = setOf("equals", "hashCode", "copy", "toString", "valueOf", "wait", "notify", "notifyAll", "getClass", "invokeMethod")
override fun describe(self: Method, stackMax: Int): String {
if (stackMax <= 0) return "..."
val parameterYaml = self.parameters.map { toYaml(it, stackMax - 1) }.toTypedArray().joinToString("\n").trim()
val returnTypeYaml = toYaml(self.genericReturnType, stackMax - 1).trim()
val description = self.annotations.find { x -> x is Description } as? Description
val responseYaml = """
|responses:
| application/json:
| schema:
| ${returnTypeYaml.replace("\n", "\n ")}
""".trimMargin().trim()
val buffer = StringBuffer()
buffer.append("operationId: ${self.name}\n")
if (description != null) {
buffer.append("description: ${description.value.trim()}\n")
}
if (parameterYaml.isNotBlank()) {
buffer.append("parameters:\n ${parameterYaml.replace("\n", "\n ")}\n")
}
buffer.append("$responseYaml\n")
return buffer.toString()
}
private fun describe(self: KFunction<*>, stackMax: Int): String {
if (stackMax <= 0) return "..."
val parameterYaml = self.parameters.filter { it.name != null }
.map { toYaml(it, stackMax - 1) }.toTypedArray().joinToString("\n").trim()
val returnTypeYaml = toYaml(self.returnType, stackMax - 1).trim()
val description = self.annotations.find { x -> x is Description } as? Description
val responseYaml = """
|responses:
| application/json:
| schema:
| ${returnTypeYaml.replace("\n", "\n ")}
""".trimMargin().trim()
return if (description != null) {
"""
|operationId: ${self.name}
|description: ${description.value.trim()}
|parameters:
| ${parameterYaml.replace("\n", "\n ")}
|$responseYaml
""".trimMargin()
} else {
"""
|operationId: ${self.name}
|parameters:
| ${parameterYaml.replace("\n", "\n ")}
|$responseYaml
""".trimMargin()
}
}
private fun toYaml(self: Parameter, stackMax: Int): String {
if (stackMax <= 0) return "..."
val description = self.getAnnotation(Description::class.java)?.value?.trim()
return if (description != null) {
"""
|- name: ${self.name}
| description: ${description.replace("\n", "\\n")}
| ${toYaml(self.parameterizedType, stackMax - 1).replace("\n", "\n ")}
|""".trimMargin().trim()
} else {
"""
|- name: ${self.name}
| ${toYaml(self.parameterizedType, stackMax - 1).replace("\n", "\n ")}
|""".trimMargin().trim()
}
}
private fun toYaml(self: KParameter, stackMax: Int): String {
if (stackMax <= 0) return "..."
val description = (self.annotations.find { it is Description } as? Description)?.value?.trim()
return if (description != null) {
"""
|- name: ${self.name}
| description: ${description.replace("\n", "\\n")}
| ${toYaml(self.type, stackMax - 1).replace("\n", "\n ")}
|""".trimMargin().trim()
} else {
"""
|- name: ${self.name}
| ${toYaml(self.type, stackMax - 1).replace("\n", "\n ")}
|""".trimMargin().trim()
}
}
private fun toYaml(self: Type, stackMax: Int): String {
if (isAbbreviated(self) || stackMax <= 0) return """
|type: object
|class: ${self.typeName}
""".trimMargin()
val typeName = self.typeName.substringAfterLast('.').replace('$', '.').lowercase(Locale.getDefault())
return if (typeName in primitives) {
"type: $typeName"
} else if (self is ParameterizedType && List::class.java.isAssignableFrom(self.rawType as Class<*>)) {
"""
|type: array
|items:
| ${toYaml(self.actualTypeArguments[0], stackMax - 1).replace("\n", "\n ")}
|""".trimMargin()
} else if (self is ParameterizedType && Map::class.java.isAssignableFrom(self.rawType as Class<*>)) {
"""
|type: map
|keys:
| ${toYaml(self.actualTypeArguments[0], stackMax - 1).replace("\n", "\n ")}
|values:
| ${toYaml(self.actualTypeArguments[1], stackMax - 1).replace("\n", "\n ")}
|""".trimMargin()
} else if (self.isArray) {
"""
|type: array
|items:
| ${toYaml(self.componentType!!, stackMax - 1).replace("\n", "\n ")}
|""".trimMargin()
} else {
describe(TypeToken.of(self).rawType, stackMax)
}
}
private fun toYaml(self: KType, stackMax: Int): String {
if (isAbbreviated(self.javaType) || stackMax <= 0) return """
|type: object
|class: ${self.toString()}
""".trimMargin()
val typeName = self.toString().substringAfterLast('.').replace('$', '.').lowercase(Locale.getDefault())
return if (typeName in primitives) {
"type: $typeName"
} else if (self is ParameterizedType && List::class.java.isAssignableFrom(self.rawType as Class<*>)) {
"""
|type: array
|items:
| ${toYaml(self.actualTypeArguments[0], stackMax - 1).replace("\n", "\n ")}
|""".trimMargin()
} else if (self is ParameterizedType && Map::class.java.isAssignableFrom(self.rawType as Class<*>)) {
"""
|type: map
|keys:
| ${toYaml(self.actualTypeArguments[0], stackMax - 1).replace("\n", "\n ")}
|values:
| ${toYaml(self.actualTypeArguments[1], stackMax - 1).replace("\n", "\n ")}
|""".trimMargin()
} else if (self.javaType.isArray) {
"""
|type: array
|items:
| ${toYaml(self.javaType.componentType!!, stackMax - 1).replace("\n", "\n ")}
|""".trimMargin()
} else {
describe(TypeToken.of(self.javaType).rawType, stackMax)
}
}
} © 2015 - 2025 Weber Informatics LLC | Privacy Policy