override fun preprocessModel()

in codegen/smithy-aws-swift-codegen/src/main/kotlin/software/amazon/smithy/aws/swift/codegen/customization/sts/DisabledAuth.kt [26:36]

• 11 lines of code
• 3 McCabe index (conditional complexity)

    override fun preprocessModel(model: Model, settings: SwiftSettings): Model {
        val currentServiceId = model.expectShape<ServiceShape>(settings.service).id.toString()
        val disabledAuthOperations = disabledAuth[currentServiceId]!!
        return ModelTransformer.create().mapShapes(model) {
            if (disabledAuthOperations.contains(it.id.toString()) && it is OperationShape) {
                it.toBuilder().addTrait(AuthTrait(setOf(), SourceLocation.NONE)).build()
            } else {
                it
            }
        }
    }