commonMain.aws.smithy.kotlin.runtime.http.auth.EndpointAuth.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of http-auth-aws-jvm Show documentation

AWS-related HTTP Auth APIs

The newest version!

/*
 * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 * SPDX-License-Identifier: Apache-2.0
 */
package aws.smithy.kotlin.runtime.http.auth

import aws.smithy.kotlin.runtime.InternalApi
import aws.smithy.kotlin.runtime.auth.AuthOption
import aws.smithy.kotlin.runtime.collections.merge
import aws.smithy.kotlin.runtime.collections.toMutableAttributes

/**
 * Merge the list of modeled auth options with the auth schemes from the resolved endpoint context.
 */
@InternalApi
public fun mergeAuthOptions(modeled: List, endpointOptions: List): List {
    // merge the two lists, preferring the priority order from endpoints
    val modeledById = modeled.associateBy(AuthOption::schemeId)
    val merged = mutableListOf()
    endpointOptions.forEach {
        val modeledOption = modeledById[it.schemeId]
        val option = if (modeledOption != null && !modeledOption.attributes.isEmpty) {
            val attrs = modeledOption.attributes.toMutableAttributes()
            attrs.merge(it.attributes)
            AuthOption(it.schemeId, attrs)
        } else {
            it
        }

        merged.add(option)
    }

    // tack on auth options that only exist in modeled list
    val mergedById = merged.map(AuthOption::schemeId).toSet()
    val modelOnlyOptions = modeled.filterNot { mergedById.contains(it.schemeId) }
    merged.addAll(modelOnlyOptions)

    return merged
}