io.micronaut.ktor.env.MicronautKtorEnvironmentConfig.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of micronaut-ktor Show documentation

Includes extensions to Micronaut for Kotlin

There is a newer version: 4.5.0

/*
 * Copyright 2017-2020 original authors
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package io.micronaut.ktor.env

import io.ktor.config.ApplicationConfig
import io.ktor.config.ApplicationConfigValue
import io.ktor.config.ApplicationConfigurationException
import io.micronaut.context.env.Environment
import io.micronaut.core.type.Argument
import java.util.*

/**
 * Implementation of the ApplicationConfig interface for Micronaut Ktor.
 *
 * @author graemerocher
 * @since 1.0
 */
class MicronautKtorEnvironmentConfig(val env : Environment, private val prefix : String? = "") : ApplicationConfig {

    override fun config(path: String): ApplicationConfig {
        if (env.containsProperties(path)) {
            return MicronautKtorEnvironmentConfig(env, path)
        } else {
            throw ApplicationConfigurationException("No configuration found for path: $path")
        }
    }

    override fun configList(path: String): List {
        return Collections.singletonList(config(path))
    }

    override fun property(path: String): ApplicationConfigValue {
        val fullPath = if(prefix == null) path else "$prefix.$path"
        if (env.containsProperty(fullPath)) {
            return KtorApplicationConfigValue(fullPath, env)
        } else {
            throw ApplicationConfigurationException("No configuration found for path: $path")
        }
    }

    override fun propertyOrNull(path: String): ApplicationConfigValue? {
        val fullPath = "$prefix.$path"
        return if (env.containsProperty(fullPath)) {
            KtorApplicationConfigValue(fullPath, env)
        } else {
            null
        }
    }

    @Suppress("UNCHECKED_CAST")
    class KtorApplicationConfigValue(private val prop : String, private val env: Environment) : ApplicationConfigValue {
        override fun getList(): List {
            val requiredProperty = env.getProperty(prop, Argument.of(List::class.java, String::class.java))
            return requiredProperty as List
        }

        override fun getString(): String {
            return env.getRequiredProperty(prop, String::class.java)
        }
    }
}