Use context receiver

.editorconfig

@@ -1,4 +1,5 @@
 root = true
 
 [*]
-indent_size = 2
+indent_size = 2
+indent_style = space

app-compose-desktop/build.gradle.kts

@@ -1,4 +1,5 @@
 import org.jetbrains.compose.desktop.application.dsl.TargetFormat
+import org.jetbrains.kotlin.gradle.tasks.KotlinCompile
 
 plugins {
   kotlin("jvm")
@@ -18,7 +19,7 @@ dependencies {
   // (in a separate module for demo project and in testMain).
   // With compose.desktop.common you will also lose @Preview functionality
   implementation(compose.desktop.currentOs)
-  testImplementation(project(":roborazzi-desktop"))
+  testImplementation(project(":roborazzi-compose-desktop"))
 
   testImplementation(kotlin("test"))
 }
@@ -34,3 +35,10 @@ compose.desktop {
     }
   }
 }
+
+tasks.withType<KotlinCompile>().configureEach {
+  kotlinOptions {
+    incremental = false
+    freeCompilerArgs += "-Xcontext-receivers"
+  }
+}

app-compose-desktop/src/main/kotlin/Main.kt

@@ -23,7 +23,10 @@ fun App() {
       onClick = {
         text = "Hello, Desktop!"
       }) {
-      Text(text)
+      Text(
+        style = MaterialTheme.typography.h2,
+        text = text
+      )
     }
   }
 }

app-compose-desktop/src/test/kotlin/MainTest.kt

@@ -16,11 +16,11 @@ class MainTest {
       val roborazziOptions = RoborazziOptions(
         compareOptions = RoborazziOptions.CompareOptions(changeThreshold = 0F)
       )
-      onNode(isRoot())
-        .captureToImage().captureRoboImage(roborazziOptions = roborazziOptions)
+      onNode(isRoot()).captureRoboImage(roborazziOptions = roborazziOptions)
+
       onNodeWithTag("button").performClick()
 
-      onNode(isRoot()).captureToImage().captureRoboImage(roborazziOptions = roborazziOptions)
+      onNode(isRoot()).captureRoboImage(roborazziOptions = roborazziOptions)
     }
   }
-}
+}

roborazzi-desktop/build.gradle → roborazzi-compose-desktop/build.gradle

@@ -1,3 +1,5 @@
+import org.jetbrains.kotlin.gradle.tasks.KotlinCompile
+
 plugins {
   id 'org.jetbrains.kotlin.multiplatform'
   id 'org.jetbrains.compose'
@@ -47,6 +49,9 @@ java {
   }
 }
 
-sourceSets {
-  main.java.srcDir 'src/generated/kotlin'
+tasks.withType(KotlinCompile).configureEach {
+    kotlinOptions {
+        incremental = false
+        freeCompilerArgs += "-Xcontext-receivers"
+    }
 }

roborazzi-desktop/src/commonMain/kotlin/io/github/takahirom/roborazzi/RoborazziDesktop.kt → roborazzi-compose-desktop/src/commonMain/kotlin/io/github/takahirom/roborazzi/RoborazziDesktop.kt

@@ -2,10 +2,55 @@ package io.github.takahirom.roborazzi
 
 import androidx.compose.ui.graphics.ImageBitmap
 import androidx.compose.ui.graphics.toAwtImage
+import androidx.compose.ui.test.DesktopComposeUiTest
+import androidx.compose.ui.test.ExperimentalTestApi
+import androidx.compose.ui.test.SemanticsNodeInteraction
 import com.github.takahirom.roborazzi.*
 import java.awt.image.BufferedImage
 import java.io.File
 
+context(DesktopComposeUiTest)
+@OptIn(ExperimentalTestApi::class)
+fun SemanticsNodeInteraction.captureRoboImage(
+  filePath: String = DefaultFileNameGenerator.generateFilePath("png"),
+  roborazziOptions: RoborazziOptions = provideRoborazziContext().options,
+) {
+  if (!roborazziEnabled()) {
+    return
+  }
+  captureRoboImage(
+    file = File(filePath),
+    roborazziOptions = roborazziOptions
+  )
+}
+
+context(DesktopComposeUiTest)
+@OptIn(ExperimentalTestApi::class)
+fun SemanticsNodeInteraction.captureRoboImage(
+  file: File,
+  roborazziOptions: RoborazziOptions
+) {
+  if (!roborazziEnabled()) {
+    return
+  }
+  val awtImage = this.captureToImage().toAwtImage()
+  val canvas = AwtRoboCanvas(
+    width = awtImage.width,
+    height = awtImage.height,
+    filled = true,
+    bufferedImageType = BufferedImage.TYPE_INT_ARGB
+  )
+  canvas.apply {
+    drawImage(awtImage)
+  }
+  processOutputImageAndReportWithDefaults(
+    canvas = canvas,
+    goldenFile = file,
+    roborazziOptions = roborazziOptions
+  )
+  canvas.release()
+}
+
 fun ImageBitmap.captureRoboImage(
   filePath: String = DefaultFileNameGenerator.generateFilePath("png"),
   roborazziOptions: RoborazziOptions = provideRoborazziContext().options,

settings.gradle

@@ -18,7 +18,7 @@ include ':app-without-compose'
 include ':roborazzi'
 include ':app-compose-desktop'
 include ':roborazzi-junit-rule'
-include ':roborazzi-desktop'
+include ':roborazzi-compose-desktop'
 include ':roborazzi-compose'
 include ':roborazzi-painter'