Merge branch 'bnorm/kotlin-power-assert'

libraries/tools/kotlin-power-assert/src/main/kotlin/com/bnorm/power/PowerAssertGradleExtension.kt

@@ -0,0 +1,22 @@
+/*
+ * 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
+
+open class PowerAssertGradleExtension {
+  var functions: List<String> = listOf("kotlin.assert")
+  var excludedSourceSets: List<String> = listOf()
+}

libraries/tools/kotlin-power-assert/src/main/kotlin/com/bnorm/power/PowerAssertGradlePlugin.kt

@@ -0,0 +1,56 @@
+/*
+ * 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 org.gradle.api.Project
+import org.gradle.api.provider.Provider
+import org.jetbrains.kotlin.gradle.plugin.KotlinCompilation
+import org.jetbrains.kotlin.gradle.plugin.KotlinCompilerPluginSupportPlugin
+import org.jetbrains.kotlin.gradle.plugin.SubpluginArtifact
+import org.jetbrains.kotlin.gradle.plugin.SubpluginOption
+
+class PowerAssertGradlePlugin : KotlinCompilerPluginSupportPlugin {
+  override fun apply(target: Project): Unit = with(target) {
+    extensions.create("kotlinPowerAssert", PowerAssertGradleExtension::class.java)
+  }
+
+  override fun isApplicable(kotlinCompilation: KotlinCompilation<*>): Boolean {
+    val project = kotlinCompilation.target.project
+    val extension = project.extensions.getByType(PowerAssertGradleExtension::class.java)
+    return extension.excludedSourceSets.none { it == kotlinCompilation.defaultSourceSet.name }
+  }
+
+  override fun getCompilerPluginId(): String = "com.bnorm.kotlin-power-assert"
+
+  override fun getPluginArtifact(): SubpluginArtifact = SubpluginArtifact(
+    groupId = BuildConfig.PLUGIN_GROUP_ID,
+    artifactId = BuildConfig.PLUGIN_ARTIFACT_ID,
+    version = BuildConfig.PLUGIN_VERSION,
+  )
+
+  override fun applyToCompilation(
+    kotlinCompilation: KotlinCompilation<*>,
+  ): Provider<List<SubpluginOption>> {
+    val project = kotlinCompilation.target.project
+    val extension = project.extensions.getByType(PowerAssertGradleExtension::class.java)
+    return project.provider {
+      extension.functions.map {
+        SubpluginOption(key = "function", value = it)
+      }
+    }
+  }
+}

plugins/power-assert/power-assert.backend/src/com/bnorm/power/IrUtils.kt

@@ -0,0 +1,60 @@
+/*
+ * 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 org.jetbrains.kotlin.backend.common.lower.DeclarationIrBuilder
+import org.jetbrains.kotlin.descriptors.DescriptorVisibilities
+import org.jetbrains.kotlin.ir.builders.IrBlockBodyBuilder
+import org.jetbrains.kotlin.ir.builders.IrBuilderWithScope
+import org.jetbrains.kotlin.ir.builders.declarations.buildFun
+import org.jetbrains.kotlin.ir.builders.irBlockBody
+import org.jetbrains.kotlin.ir.builders.irString
+import org.jetbrains.kotlin.ir.builders.parent
+import org.jetbrains.kotlin.ir.declarations.IrDeclarationOrigin
+import org.jetbrains.kotlin.ir.expressions.IrFunctionExpression
+import org.jetbrains.kotlin.ir.expressions.IrStatementOrigin
+import org.jetbrains.kotlin.ir.expressions.impl.IrFunctionExpressionImpl
+import org.jetbrains.kotlin.ir.types.IrType
+import org.jetbrains.kotlin.name.Name
+
+fun IrBuilderWithScope.irString(builderAction: StringBuilder.() -> Unit) =
+  irString(buildString { builderAction() })
+
+fun IrBuilderWithScope.irLambda(
+  returnType: IrType,
+  lambdaType: IrType,
+  startOffset: Int = this.startOffset,
+  endOffset: Int = this.endOffset,
+  block: IrBlockBodyBuilder.() -> Unit,
+): IrFunctionExpression {
+  val scope = this
+  val lambda = context.irFactory.buildFun {
+    this.startOffset = startOffset
+    this.endOffset = endOffset
+    name = Name.special("<anonymous>")
+    this.returnType = returnType
+    visibility = DescriptorVisibilities.LOCAL
+    origin = IrDeclarationOrigin.LOCAL_FUNCTION_FOR_LAMBDA
+  }.apply {
+    val bodyBuilder = DeclarationIrBuilder(context, symbol, startOffset, endOffset)
+    body = bodyBuilder.irBlockBody {
+      block()
+    }
+    parent = scope.parent
+  }
+  return IrFunctionExpressionImpl(startOffset, endOffset, lambdaType, lambda, IrStatementOrigin.LAMBDA)
+}