io.vrap.codegen.languages.python.PyExtensions.kt Maven / Gradle / Ivy

Go to download
/**
 *  Copyright 2021 Michael van Tellingen
 */
package io.vrap.codegen.languages.python

import io.vrap.rmf.raml.model.resources.Resource
import io.vrap.rmf.raml.model.types.DescriptionFacet
import io.vrap.rmf.raml.model.util.StringCaseFormat

fun DescriptionFacet.toDocString(): String {
    val description = this.description
    return if (description?.value.isNullOrBlank()) {
        ""
    } else description.value
        .lines()
        .joinToString(prefix = "\"\"\"", postfix = "\n\"\"\"", separator = "\n")
}

fun DescriptionFacet.toLineComment(): String {
    val description = this.description
    return if (description?.value.isNullOrBlank()) {
        ""
    } else description.value
        .lines()
        .joinToString(separator = "\n")
        .trimMargin()
        .prependIndent("#: ")
}

fun Resource.pyServiceModuleName(): String {
    val parts = fullUri.template.split("/")
    if (parts.size > 2) {
        return parts[2].snakeCase()
    } else {
        return ""
    }
}

fun Resource.pyServiceClassName(): String {
    val name = this.pyServiceModuleName()
    return StringCaseFormat.UPPER_CAMEL_CASE.apply(name) + "Service"
}