com.ancientlightstudios.quarkus.kotlin.openapi.refactoring.ReplaceSimpleOfComponentsRefactoring.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of quarkus-kotlin-openapi-maven-plugin Show documentation

A Maven plugin to use the OpenAPI generator.

There is a newer version: 0.4.14

Show newest version

package com.ancientlightstudios.quarkus.kotlin.openapi.refactoring

import com.ancientlightstudios.quarkus.kotlin.openapi.inspection.inspect
import com.ancientlightstudios.quarkus.kotlin.openapi.models.transformable.components.AllOfComponent
import com.ancientlightstudios.quarkus.kotlin.openapi.models.transformable.components.SomeOfComponent

// checks for schemas with a single one-item *Of component replaces it with an allOf component
class ReplaceSimpleOfComponentsRefactoring : SpecRefactoring {

    override fun RefactoringContext.perform() {
        spec.inspect {
            schemas {
                // is there a single *Of component?
                val someOfComponents = schema.components.filterIsInstance()
                val someOfComponent = when (someOfComponents.size) {
                    1 -> someOfComponents.first()
                    else -> return@schemas
                }

                // is it already a allOf component?
                if (someOfComponent is AllOfComponent) {
                    return@schemas
                }

                // can't replace it, without changing the meaning
                if (someOfComponent.schemas.size > 1) {
                    return@schemas
                }

                val newComponents = schema.components.toMutableList()
                newComponents.remove(someOfComponent)
                newComponents.add(AllOfComponent(someOfComponent.schemas))
                schema.components = newComponents
            }
        }
    }

}