| /* |
| * Copyright 2018 The Bazel Authors. All rights reserved. |
| * |
| * 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 |
| * |
| * http://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. |
| */ |
| @file:Suppress("UNCHECKED_CAST") |
| |
| package io.bazel.ruleskotlin.workers |
| |
| import java.util.stream.Stream |
| |
| class Context internal constructor( |
| val toolchain: KotlinToolchain, |
| private val flags: Map<Flag, String> |
| ) { |
| private val meta = mutableMapOf<Meta<*>, Any>() |
| |
| fun copyOfFlags(vararg fields: Flag): Map<Flag, String> = fields.mapNotNull { f -> flags[f]?.let { f to it } }.toMap() |
| |
| fun apply(vararg consumers: (Context) -> Unit) { |
| Stream.of(*consumers).forEach { it(this) } |
| } |
| |
| internal operator fun get(flag: Flag): String? = flags[flag] |
| operator fun <T : Any> get(key: Meta<T>): T? = meta[key] as T? |
| internal fun <T : Any> putIfAbsent(key: Meta<T>, value: T): T? = meta.putIfAbsent(key, value as Any) as T? |
| } |