Allow schema to be a multiplatform artifact

treehouse-gradle/src/main/kotlin/app/cash/treehouse/gradle/TreehouseSchemaGeneratorPlugin.kt

@@ -4,6 +4,9 @@ import app.cash.treehouse.gradle.TreehouseSchemaGeneratorPlugin.Strategy.Compose
 import app.cash.treehouse.gradle.TreehouseSchemaGeneratorPlugin.Strategy.Widget
 import org.gradle.api.Plugin
 import org.gradle.api.Project
+import org.gradle.api.attributes.Usage
+import org.gradle.api.attributes.Usage.JAVA_RUNTIME
+import org.gradle.api.attributes.Usage.USAGE_ATTRIBUTE
 import org.gradle.api.tasks.JavaExec
 import org.jetbrains.kotlin.gradle.dsl.KotlinMultiplatformExtension
 import java.io.File
@@ -38,7 +41,11 @@ abstract class TreehouseSchemaGeneratorPlugin(
       TreehouseSchemaExtensionImpl::class.java,
     )
 
-    val configuration = project.configurations.create("treehouseSchema")
+    val configuration = project.configurations.create("treehouseSchema") {
+      // Ensure we get JVM artifacts from any multiplatform dependencies for use with JavaExec.
+      val runtimeUsage = project.objects.named(Usage::class.java, JAVA_RUNTIME)
+      it.attributes.attribute(USAGE_ATTRIBUTE, runtimeUsage)
+    }
     project.dependencies.add(
       configuration.name,
       "app.cash.treehouse:treehouse-schema-generator:$treehouseVersion",

treehouse-gradle/src/main/kotlin/app/cash/treehouse/gradle/TreehouseSchemaPlugin.kt

@@ -3,6 +3,8 @@ package app.cash.treehouse.gradle
 import org.gradle.api.Plugin
 import org.gradle.api.Project
 import org.jetbrains.kotlin.gradle.dsl.KotlinJvmProjectExtension
+import org.jetbrains.kotlin.gradle.dsl.KotlinMultiplatformExtension
+import org.jetbrains.kotlin.gradle.plugin.KotlinTarget
 
 class TreehouseSchemaPlugin : Plugin<Project> {
   override fun apply(project: Project) {
@@ -12,18 +14,28 @@ class TreehouseSchemaPlugin : Plugin<Project> {
       applied = true
 
       val kotlin = project.extensions.getByType(KotlinJvmProjectExtension::class.java)
-      kotlin.target.apply {
-        compilations.getByName("main") { compilation ->
-          compilation.dependencies {
-            api("app.cash.treehouse:treehouse-schema-annotations:$treehouseVersion")
-          }
-        }
-      }
+      kotlin.target.applySchemaAnnotationDependency()
+    }
+    project.plugins.withId("org.jetbrains.kotlin.multiplatform") {
+      applied = true
+
+      val kotlin = project.extensions.getByType(KotlinMultiplatformExtension::class.java)
+      kotlin.targets.all { it.applySchemaAnnotationDependency() }
+
+      // TODO check there is a JVM target in an afterEvaluate
     }
 
     project.afterEvaluate {
       check(applied) {
-        "Treehouse schema plugin requires the Kotlin JVM plugin to be applied."
+        "Treehouse schema plugin requires the Kotlin JVM or multiplatform plugin to be applied."
+      }
+    }
+  }
+
+  private fun KotlinTarget.applySchemaAnnotationDependency() {
+    compilations.getByName("main") { compilation ->
+      compilation.dependencies {
+        api("app.cash.treehouse:treehouse-schema-annotations:$treehouseVersion")
       }
     }
   }

treehouse-gradle/src/test/fixture/schema-jvm/build.gradle

@@ -0,0 +1,22 @@
+buildscript {
+  dependencies {
+    classpath "app.cash.treehouse:treehouse-gradle:$treehouseVersion"
+    classpath 'org.jetbrains.kotlin:kotlin-gradle-plugin:1.4.31'
+  }
+
+  repositories {
+    maven {
+      url "file://${rootDir.absolutePath}/../../../../../build/localMaven"
+    }
+    mavenCentral()
+  }
+}
+
+allprojects {
+  repositories {
+    maven {
+      url "file://${rootDir.absolutePath}/../../../../../build/localMaven"
+    }
+    mavenCentral()
+  }
+}

treehouse-gradle/src/test/fixture/schema-jvm/gradle.properties

@@ -0,0 +1 @@
+kotlin.mpp.stability.nowarn=true

treehouse-gradle/src/test/fixture/schema-jvm/schema/build.gradle

@@ -0,0 +1,2 @@
+apply plugin: 'org.jetbrains.kotlin.jvm'
+apply plugin: 'app.cash.treehouse.schema'

treehouse-gradle/src/test/fixture/schema-jvm/schema/src/main/kotlin/example/counter/schema.kt

@@ -0,0 +1,33 @@
+package example.counter
+
+import app.cash.treehouse.schema.Children
+import app.cash.treehouse.schema.Default
+import app.cash.treehouse.schema.Property
+import app.cash.treehouse.schema.Schema
+import app.cash.treehouse.schema.Widget
+
+@Schema([
+  CounterBox::class,
+  CounterText::class,
+  CounterButton::class,
+])
+interface Counter
+
+@Widget(1)
+data class CounterBox(
+  @Property(1) val orientation: Boolean, // TODO enum or whatever
+  @Children(1) val children: List<Any>,
+)
+
+@Widget(2)
+data class CounterText(
+  @Property(1) val text: String?,
+  @Property(2) @Default("\"black\"") val color: String,
+)
+
+@Widget(3)
+data class CounterButton(
+  @Property(1) val text: String?,
+  @Property(2) @Default("true") val enabled: Boolean,
+  @Property(3) val onClick: () -> Unit,
+)

treehouse-gradle/src/test/fixture/schema-jvm/schema/widget/build.gradle

@@ -0,0 +1,14 @@
+apply plugin: 'org.jetbrains.kotlin.multiplatform'
+apply plugin: 'app.cash.treehouse.schema.widget'
+
+kotlin {
+  jvm()
+}
+
+treehouse {
+  schema = 'example.counter.Counter'
+}
+
+dependencies {
+  treehouseSchema project(':schema')
+}

treehouse-gradle/src/test/fixture/schema-jvm/settings.gradle

@@ -0,0 +1,2 @@
+include ':schema'
+include ':schema:widget'

treehouse-gradle/src/test/fixture/schema-multiplatform/build.gradle

@@ -0,0 +1,22 @@
+buildscript {
+  dependencies {
+    classpath "app.cash.treehouse:treehouse-gradle:$treehouseVersion"
+    classpath 'org.jetbrains.kotlin:kotlin-gradle-plugin:1.4.31'
+  }
+
+  repositories {
+    maven {
+      url "file://${rootDir.absolutePath}/../../../../../build/localMaven"
+    }
+    mavenCentral()
+  }
+}
+
+allprojects {
+  repositories {
+    maven {
+      url "file://${rootDir.absolutePath}/../../../../../build/localMaven"
+    }
+    mavenCentral()
+  }
+}

treehouse-gradle/src/test/fixture/schema-multiplatform/gradle.properties

@@ -0,0 +1,2 @@
+kotlin.mpp.stability.nowarn=true
+kotlin.js.compiler=ir

treehouse-gradle/src/test/fixture/schema-multiplatform/schema/build.gradle

@@ -0,0 +1,9 @@
+apply plugin: 'org.jetbrains.kotlin.multiplatform'
+apply plugin: 'app.cash.treehouse.schema'
+
+kotlin {
+  js {
+    nodejs()
+  }
+  jvm()
+}

treehouse-gradle/src/test/fixture/schema-multiplatform/schema/src/commonMain/kotlin/example/counter/schema.kt

@@ -0,0 +1,33 @@
+package example.counter
+
+import app.cash.treehouse.schema.Children
+import app.cash.treehouse.schema.Default
+import app.cash.treehouse.schema.Property
+import app.cash.treehouse.schema.Schema
+import app.cash.treehouse.schema.Widget
+
+@Schema([
+  CounterBox::class,
+  CounterText::class,
+  CounterButton::class,
+])
+interface Counter
+
+@Widget(1)
+data class CounterBox(
+  @Property(1) val orientation: Boolean, // TODO enum or whatever
+  @Children(1) val children: List<Any>,
+)
+
+@Widget(2)
+data class CounterText(
+  @Property(1) val text: String?,
+  @Property(2) @Default("\"black\"") val color: String,
+)
+
+@Widget(3)
+data class CounterButton(
+  @Property(1) val text: String?,
+  @Property(2) @Default("true") val enabled: Boolean,
+  @Property(3) val onClick: () -> Unit,
+)

treehouse-gradle/src/test/fixture/schema-multiplatform/schema/widget/build.gradle

@@ -0,0 +1,17 @@
+apply plugin: 'org.jetbrains.kotlin.multiplatform'
+apply plugin: 'app.cash.treehouse.schema.widget'
+
+kotlin {
+  js {
+    nodejs()
+  }
+  jvm()
+}
+
+treehouse {
+  schema = 'example.counter.Counter'
+}
+
+dependencies {
+  treehouseSchema project(':schema')
+}

treehouse-gradle/src/test/fixture/schema-multiplatform/settings.gradle

@@ -0,0 +1,2 @@
+include ':schema'
+include ':schema:widget'

treehouse-gradle/src/test/kotlin/app/cash/treehouse/gradle/FixtureTest.kt

@@ -16,6 +16,8 @@ class FixtureTest(
     @Parameters(name = "{0}")
     fun parameters() = listOf(
       arrayOf("counter"),
+      arrayOf("schema-multiplatform"),
+      arrayOf("schema-jvm"),
     )
   }
 

treehouse-schema-annotations/build.gradle

@@ -3,6 +3,9 @@ apply plugin: 'com.vanniktech.maven.publish'
 apply plugin: 'org.jetbrains.dokka' // Must be applied here for publish plugin.
 
 kotlin {
+  js() {
+    nodejs()
+  }
   jvm()
   ios()
 }

treehouse-schema-generator/src/main/kotlin/app/cash/treehouse/schema/generator/main.kt

@@ -7,6 +7,7 @@ import app.cash.treehouse.schema.generator.TreehouseGenerator.Type.Compose
 import app.cash.treehouse.schema.generator.TreehouseGenerator.Type.Widget
 import app.cash.treehouse.schema.parser.parseSchema
 import com.github.ajalt.clikt.core.CliktCommand
+import com.github.ajalt.clikt.core.CliktError
 import com.github.ajalt.clikt.parameters.arguments.argument
 import com.github.ajalt.clikt.parameters.arguments.convert
 import com.github.ajalt.clikt.parameters.arguments.help
@@ -38,7 +39,11 @@ private class TreehouseGenerator : CliktCommand() {
     .help("Fully-qualified class name for the @Schema-annotated interface")
     .convert {
       // Replace with https://youtrack.jetbrains.com/issue/KT-10440 once it ships.
-      Class.forName(it).kotlin
+      try {
+        Class.forName(it).kotlin
+      } catch (e: ClassNotFoundException) {
+        throw CliktError("Unable to load class: $it", e)
+      }
     }
 
   override fun run() {