Merge pull request #1862 from Kotlin/texts_samples

Added texts samples saving

Author: AndreiKingsley

docs/StardustDocs/topics/schemas/DataSchemaGenerationMethods.md

@@ -192,7 +192,7 @@ Use [`cast`](cast.md) to apply the generated schema to a `DataFrame`:
 <!---FUN notebook_test_generate_docs_3-->
 
 ```kotlin
-df.cast<_DataFrameType1>().filter { orders.all { orderId >= 102 } }
+df.filter { orders.all { orderId >= 102 } }
 ```
 
 <!---END-->

samples/build.gradle.kts

@@ -95,6 +95,7 @@ korro {
                 include("toHTML.md")
                 include("guides/*.md")
                 include("concepts/*.md")
+                include("schemas/*.md")
                 include("operations/utils/*.md")
                 include("operations/multiple/*.md")
                 include("operations/column/*.md")
@@ -126,6 +127,12 @@ korro {
                 include("src/test/kotlin/org/jetbrains/kotlinx/dataframe/samples/io/*.kt")
             },
         )
+
+        outputs.from(
+            fileTree(project.layout.buildDirectory) {
+                include("korroOutputLines/*")
+            },
+        )
     }
 
     groupSamples {

samples/src/test/kotlin/org/jetbrains/kotlinx/dataframe/samples/DataFrameSampleHelper.kt

@@ -1,9 +1,11 @@
 package org.jetbrains.kotlinx.dataframe.samples
 
 import org.jetbrains.kotlinx.dataframe.DataColumn
+import org.jetbrains.kotlinx.dataframe.api.CodeString
 import org.jetbrains.kotlinx.dataframe.api.toDataFrame
 import org.jetbrains.kotlinx.dataframe.samples.api.TestBase
 import org.jetbrains.kotlinx.kandy.letsplot.samples.SampleHelper
+import java.io.File
 
 abstract class DataFrameSampleHelper(sampleName: String, subFolder: String = "samples") :
     SampleHelper(
@@ -14,6 +16,31 @@ abstract class DataFrameSampleHelper(sampleName: String, subFolder: String = "sa
     ),
     TestBase {
 
+    fun fqnTestName(): String = "${this::class.java.name}.${testName.methodName}"
+
+    private val korroOutputLinesDir: File = File("build/korroOutputLines")
+
+    fun String.saveSample(addOutputLine: Boolean = true) {
+        val fqnName = fqnTestName().replace("_dataframe", "")
+        val text = if (addOutputLine) {
+            "\nOutput:\n\n$this\n"
+        } else {
+            "\n$this\n"
+        }
+
+        korroOutputLinesDir.mkdirs()
+        File(korroOutputLinesDir, fqnName).writeText(text)
+    }
+
+    fun CodeString.saveSample(addOutputLine: Boolean = true, addCodeBlock: Boolean = true) {
+        val text = if (addCodeBlock) {
+            "```kotlin\n$value\n```"
+        } else {
+            value
+        }
+        text.saveSample(addOutputLine)
+    }
+
     fun DataColumn<*>.saveDfHtmlSample() {
         toDataFrame().saveDfHtmlSample()
     }