io.vrap.codegen.languages.javalang.client.builder.requests.JavaRequestBuilderResourceRenderer.kt Maven / Gradle / Ivy
package io.vrap.codegen.languages.javalang.client.builder.requests
import com.google.common.net.MediaType
import io.vrap.codegen.languages.extensions.*
import io.vrap.codegen.languages.java.base.JavaSubTemplates
import io.vrap.codegen.languages.java.base.extensions.*
import io.vrap.rmf.codegen.firstLowerCase
import io.vrap.rmf.codegen.io.TemplateFile
import io.vrap.rmf.codegen.rendering.ResourceRenderer
import io.vrap.rmf.codegen.rendering.utils.escapeAll
import io.vrap.rmf.codegen.rendering.utils.keepIndentation
import io.vrap.rmf.codegen.types.*
import io.vrap.rmf.raml.model.resources.Method
import io.vrap.rmf.raml.model.resources.Resource
import io.vrap.rmf.raml.model.resources.ResourceContainer
import io.vrap.rmf.raml.model.types.Annotation
import io.vrap.rmf.raml.model.types.AnyType
import io.vrap.rmf.raml.model.types.ObjectType
import io.vrap.rmf.raml.model.types.StringInstance
import java.util.*
class JavaRequestBuilderResourceRenderer constructor(override val vrapTypeProvider: VrapTypeProvider) : ResourceRenderer, JavaEObjectTypeExtensions {
override fun render(type: Resource): TemplateFile {
val vrapType = vrapTypeProvider.doSwitch(type).toJavaVType() as VrapObjectType
val resourceName : String = type.toResourceName()
val className : String = "${resourceName}RequestBuilder"
val implements = arrayListOf()
.plus(
when (val ex = type.getAnnotation("java-implements") ) {
is Annotation -> {
(ex.value as StringInstance).value.escapeAll()
}
else -> null
}
)
.filterNotNull()
val content : String = """
|package ${vrapType.`package`};
|
|import java.util.ArrayList;
|import java.util.List;
|import java.util.function.Function;
|import java.util.function.UnaryOperator;
|
|import io.vrap.rmf.base.client.ApiHttpClient;
|import io.vrap.rmf.base.client.ApiMethod;
|import io.vrap.rmf.base.client.Builder;
|import io.vrap.rmf.base.client.utils.Generated;
|
|<${JavaSubTemplates.generatedAnnotation}>${if (type.markDeprecated()) """
|@Deprecated""" else ""}
|public class $className ${if (implements.isNotEmpty()) "implements ${implements.joinToString(", ")}" else ""} {
|
| <${type.fields()}>
|
| <${type.constructor()}>
|
| <${type.methods()}>
|
| <${type.subResources()}>
|
| <${type.getAnnotation("java-mixin")?.value?.value?.let { (it as String).escapeAll()} ?: ""}>
|}
|
""".trimMargin().keepIndentation()
return TemplateFile(
relativePath = "${vrapType.`package`}.$className".replace(".", "/") + ".java",
content = content
)
}
private fun Resource.allFields(): List {
return listOf(NamedField(apiHttpClient,"apiHttpClient"))
.plus(
this.pathArguments().map {
NamedField(VrapScalarType("String"), it)
}
)
}
private fun Resource.fields() : String {
return this.allFields()
.map {
"private final ${it.type.toJavaVType().simpleName()} ${it.name}"
}
.joinToString(separator = ";\n",postfix = ";\n")
}
private fun Resource.constructor() : String {
val resourceName : String = this.toResourceName()
val className : String = "${resourceName}RequestBuilder"
val constructorArguments : String = this.allFields().map { "final ${it.type.toJavaVType().simpleName()} ${it.name}" }.joinToString(separator = ",")
val constructorAssignment : String = this.allFields().map { "this.${it.name} = ${it.name};" }.joinToString(separator = "\n")
return """
|public $className ($constructorArguments) {
| <$constructorAssignment>
|}
""".trimMargin()
}
private fun Resource.toResourceName(): String {
return this.fullUri.toParamName("By")
}
private fun Resource.methods() : String {
return this.methods.map { it.method() }.joinToString(separator = "\n\n")
}
private fun Method.method() : String {
if (this.bodies != null && this.bodies.isNotEmpty() && this.bodies[0].contentMediaType.`is`(MediaType.FORM_DATA)) {
val requestArguments = mutableListOf("apiHttpClient")
this.pathArguments().forEach { requestArguments.add(it) }
requestArguments.add("new ArrayList<>()".escapeAll())
return """
|public ${this.toRequestName()} ${this.method.name.lowercase(Locale.getDefault())}(${this.constructorArguments()}) {
| return new ${this.toRequestName()}(${this.requestArguments()});
|}
|
|public ${this.toRequestName()} ${this.method.name.lowercase(Locale.getDefault())}() {
| return new ${this.toRequestName()}(${requestArguments.joinToString(separator = ", ")});
|}
""".trimMargin()
}
val methodBodyVrapType = if (this.bodies != null && this.bodies.isNotEmpty()) this.bodies[0].type.toVrapType() else null
return """
|public ${this.toRequestName()} ${this.method.name.lowercase(Locale.getDefault())}(${this.constructorArguments()}) {
| return new ${this.toRequestName()}(${this.requestArguments()});
|}
|${if(this.methodName == "delete" && this.queryParameters.any { queryParameter -> queryParameter.name == "version" }) """public \ ${this.toRequestName()} ${this.method.name.lowercase(Locale.getDefault())}(TValue version) {
| return delete().withVersion(version);
|}""" else ""}
|${if(methodBodyVrapType is VrapObjectType && this.bodies[0].type.isFile().not()) """
|public ${this.toStringRequestName()} ${this.method.name.lowercase(Locale.getDefault())}(${this.stringConstructorArguments()}) {
| return new ${this.toStringRequestName()}(${this.requestArguments()});
|}
|public ${this.toRequestName()} ${this.method.name.lowercase(Locale.getDefault())}(${this.bodies[0].type.builderOp()}) {
| return ${this.method.name.lowercase(Locale.getDefault())}(op.apply(${methodBodyVrapType.`package`}.${methodBodyVrapType.simpleClassName}Builder.of()).build());
|}""" else ""}
""".trimMargin()
}
private fun AnyType.builderOp(): String {
val vrapType = this.toVrapType() as VrapObjectType
if (this is ObjectType && this.discriminator != null) {
return "Function<${vrapType.`package`}.${vrapType.simpleClassName}Builder, Builder> op".escapeAll()
}
return "UnaryOperator<${vrapType.`package`}.${vrapType.simpleClassName}Builder> op".escapeAll()
}
private fun Method.stringConstructorArguments(): String? {
val methodBodyVrapType = this.bodies[0].type.toVrapType() as VrapObjectType
return "final String ${methodBodyVrapType.simpleClassName.firstLowerCase()}"
}
private fun Method.constructorArguments(): String? {
return if(this.bodies != null && this.bodies.isNotEmpty()){
val methodBodyVrapType = this.bodies[0].type.toVrapType()
if(methodBodyVrapType is VrapObjectType) {
val methodBodyArgument =
"${methodBodyVrapType.`package`}.${methodBodyVrapType.simpleClassName} ${methodBodyVrapType.simpleClassName.firstLowerCase()}"
methodBodyArgument
} else if (this.bodies[0].contentMediaType.`is`(MediaType.FORM_DATA)) {
"List> formParams".escapeAll()
} else {
"Object obj"
}
}else {
""
}
}
private fun Method.requestArguments() : String {
val requestArguments = mutableListOf("apiHttpClient")
this.pathArguments().forEach { requestArguments.add(it) }
if(this.bodies != null && this.bodies.isNotEmpty()){
val vrapType = this.bodies[0].type.toVrapType()
if(vrapType is VrapObjectType) {
requestArguments.add(vrapType.simpleClassName.firstLowerCase())
} else if (this.bodies[0].contentMediaType.`is`(MediaType.FORM_DATA)) {
requestArguments.add("formParams")
} else {
requestArguments.add("obj")
}
}
return requestArguments.joinToString(separator = ", ")
}
private fun ResourceContainer.subResources() : String {
return this.resources
.filterNot { it.deprecated() }
.map {
val args = if (it.relativeUri.variables.isNullOrEmpty()){
""
}else {
it.relativeUri.variables.map { "String $it" }.joinToString(separator = " ,")
}
val subResourceArgs : String = listOf("apiHttpClient")
.plus(
it.pathArguments()
)
.joinToString(separator = ", ")
"""
|${if (it.markDeprecated()) "@Deprecated" else ""}
|public ${it.toResourceName()}RequestBuilder ${it.getMethodName()}($args) {
| return new ${it.toResourceName()}RequestBuilder($subResourceArgs);
|}
""".trimMargin()
}.joinToString(separator = "\n")
}
private fun Method.pathArguments() : List = this.resource().pathArguments()
private fun Resource.pathArguments(): List = this.fullUri.variables.toList()
}