Prepare kotlin-power-assert files for git merge with Kotlin

plugins/power-assert/power-assert.cli/src/com/bnorm/power/PowerAssertCommandLineProcessor.kt

@@ -0,0 +1,49 @@
+/*
+ * Copyright (C) 2020 Brian Norman
+ *
+ * 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.
+ */
+
+package com.bnorm.power
+
+import com.google.auto.service.AutoService
+import org.jetbrains.kotlin.compiler.plugin.AbstractCliOption
+import org.jetbrains.kotlin.compiler.plugin.CliOption
+import org.jetbrains.kotlin.compiler.plugin.CommandLineProcessor
+import org.jetbrains.kotlin.config.CompilerConfiguration
+
+@AutoService(CommandLineProcessor::class)
+class PowerAssertCommandLineProcessor : CommandLineProcessor {
+  override val pluginId: String = "com.bnorm.kotlin-power-assert"
+
+  override val pluginOptions: Collection<CliOption> = listOf(
+    CliOption(
+      optionName = "function",
+      valueDescription = "function full-qualified name",
+      description = "fully qualified path of function to intercept",
+      required = false, // TODO required for Kotlin/JS
+      allowMultipleOccurrences = true,
+    ),
+  )
+
+  override fun processOption(
+    option: AbstractCliOption,
+    value: String,
+    configuration: CompilerConfiguration,
+  ) {
+    return when (option.optionName) {
+      "function" -> configuration.add(KEY_FUNCTIONS, value)
+      else -> error("Unexpected config option ${option.optionName}")
+    }
+  }
+}

plugins/power-assert/power-assert.cli/src/com/bnorm/power/PowerAssertCompilerPluginRegistrar.kt

@@ -0,0 +1,46 @@
+/*
+ * Copyright (C) 2020 Brian Norman
+ *
+ * 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.
+ */
+
+package com.bnorm.power
+
+import com.google.auto.service.AutoService
+import org.jetbrains.kotlin.backend.common.extensions.IrGenerationExtension
+import org.jetbrains.kotlin.cli.common.CLIConfigurationKeys
+import org.jetbrains.kotlin.cli.common.messages.MessageCollector
+import org.jetbrains.kotlin.compiler.plugin.CompilerPluginRegistrar
+import org.jetbrains.kotlin.config.CompilerConfiguration
+import org.jetbrains.kotlin.config.CompilerConfigurationKey
+import org.jetbrains.kotlin.name.FqName
+
+val KEY_FUNCTIONS = CompilerConfigurationKey<List<String>>("fully-qualified function names")
+
+@AutoService(CompilerPluginRegistrar::class)
+class PowerAssertCompilerPluginRegistrar(
+  private val functions: Set<FqName>,
+) : CompilerPluginRegistrar() {
+  @Suppress("unused")
+  constructor() : this(emptySet()) // Used by service loader
+
+  override val supportsK2: Boolean = true
+
+  override fun ExtensionStorage.registerExtensions(configuration: CompilerConfiguration) {
+    val functions = configuration[KEY_FUNCTIONS]?.map { FqName(it) } ?: functions
+    if (functions.isEmpty()) return
+
+    val messageCollector = configuration.get(CLIConfigurationKeys.MESSAGE_COLLECTOR_KEY, MessageCollector.NONE)
+    IrGenerationExtension.registerExtension(PowerAssertIrGenerationExtension(messageCollector, functions.toSet()))
+  }
+}