updated quickstart.kt with more compiler plugin magic

Author: Jolanrensen

docs/StardustDocs/resources/api/tail/notebook_test_tail_1.html

@@ -459,6 +459,7 @@
 /*<!--*/
 call_DataFrame(function() { DataFrame.addTable({ cols: [{ name: "<span title=\"firstName: String\">firstName</span>", children: [], rightAlign: false, values: ["Alice","Bob","Charlie","Charlie","Bob","Alice","Charlie"] }, 
 { name: "<span title=\"lastName: String\">lastName</span>", children: [], rightAlign: false, values: ["Cooper","Dylan","Daniels","Chaplin","Marley","Wolf","Byrd"] }, 
+{ name: "<span title=\"name: DataRow<*>\">name</span>", children: [0, 1], rightAlign: false, values: ["<span class=\"formatted\" title=\"firstName: Alice\nlastName: Cooper\"><span class=\"structural\">{ </span><span class=\"structural\">firstName: </span>Alice<span class=\"structural\">, </span><span class=\"structural\">lastName: </span>Cooper<span class=\"structural\"> }</span></span>","<span class=\"formatted\" title=\"firstName: Bob\nlastName: Dylan\"><span class=\"structural\">{ </span><span class=\"structural\">firstName: </span>Bob<span class=\"structural\">, </span><span class=\"structural\">lastName: </span>Dylan<span class=\"structural\"> }</span></span>","<span class=\"formatted\" title=\"firstName: Charlie\nlastName: Daniels\"><span class=\"structural\">{ </span><span class=\"structural\">firstName: </span>Charlie<span class=\"structural\">, </span><span class=\"structural\">lastName: </span>Dan<span class=\"structural\">...</span><span class=\"structural\"> }</span></span>","<span class=\"formatted\" title=\"firstName: Charlie\nlastName: Chaplin\"><span class=\"structural\">{ </span><span class=\"structural\">firstName: </span>Charlie<span class=\"structural\">, </span><span class=\"structural\">lastName: </span>Cha<span class=\"structural\">...</span><span class=\"structural\"> }</span></span>","<span class=\"formatted\" title=\"firstName: Bob\nlastName: Marley\"><span class=\"structural\">{ </span><span class=\"structural\">firstName: </span>Bob<span class=\"structural\">, </span><span class=\"structural\">lastName: </span>Marley<span class=\"structural\"> }</span></span>","<span class=\"formatted\" title=\"firstName: Alice\nlastName: Wolf\"><span class=\"structural\">{ </span><span class=\"structural\">firstName: </span>Alice<span class=\"structural\">, </span><span class=\"structural\">lastName: </span>Wolf<span class=\"structural\"> }</span></span>","<span class=\"formatted\" title=\"firstName: Charlie\nlastName: Byrd\"><span class=\"structural\">{ </span><span class=\"structural\">firstName: </span>Charlie<span class=\"structural\">, </span><span class=\"structural\">lastName: </span>Byrd<span class=\"structural\"> }</span></span>"] }, 
 { name: "<span title=\"age: Int\">age</span>", children: [], rightAlign: true, values: ["<span class=\"formatted\" title=\"\"><span class=\"numbers\">15</span></span>","<span class=\"formatted\" title=\"\"><span class=\"numbers\">45</span></span>","<span class=\"formatted\" title=\"\"><span class=\"numbers\">20</span></span>","<span class=\"formatted\" title=\"\"><span class=\"numbers\">40</span></span>","<span class=\"formatted\" title=\"\"><span class=\"numbers\">30</span></span>","<span class=\"formatted\" title=\"\"><span class=\"numbers\">20</span></span>","<span class=\"formatted\" title=\"\"><span class=\"numbers\">30</span></span>"] }, 
 { name: "<span title=\"city: String?\">city</span>", children: [], rightAlign: false, values: ["London","Dubai","Moscow","Milan","Tokyo","<span class=\"formatted\" title=\"\"><span class=\"null\">null</span></span>","Moscow"] }, 
 { name: "<span title=\"weight: Int?\">weight</span>", children: [], rightAlign: true, values: ["<span class=\"formatted\" title=\"\"><span class=\"numbers\">54</span></span>","<span class=\"formatted\" title=\"\"><span class=\"numbers\">87</span></span>","<span class=\"formatted\" title=\"\"><span class=\"null\">null</span></span>","<span class=\"formatted\" title=\"\"><span class=\"null\">null</span></span>","<span class=\"formatted\" title=\"\"><span class=\"numbers\">68</span></span>","<span class=\"formatted\" title=\"\"><span class=\"numbers\">55</span></span>","<span class=\"formatted\" title=\"\"><span class=\"numbers\">90</span></span>"] }, 

docs/StardustDocs/resources/api/tail/notebook_test_tail_2.html

@@ -459,6 +459,7 @@
 /*<!--*/
 call_DataFrame(function() { DataFrame.addTable({ cols: [{ name: "<span title=\"firstName: String\">firstName</span>", children: [], rightAlign: false, values: ["Charlie","Charlie","Bob","Alice","Charlie"] }, 
 { name: "<span title=\"lastName: String\">lastName</span>", children: [], rightAlign: false, values: ["Daniels","Chaplin","Marley","Wolf","Byrd"] }, 
+{ name: "<span title=\"name: DataRow<*>\">name</span>", children: [0, 1], rightAlign: false, values: ["<span class=\"formatted\" title=\"firstName: Charlie\nlastName: Daniels\"><span class=\"structural\">{ </span><span class=\"structural\">firstName: </span>Charlie<span class=\"structural\">, </span><span class=\"structural\">lastName: </span>Dan<span class=\"structural\">...</span><span class=\"structural\"> }</span></span>","<span class=\"formatted\" title=\"firstName: Charlie\nlastName: Chaplin\"><span class=\"structural\">{ </span><span class=\"structural\">firstName: </span>Charlie<span class=\"structural\">, </span><span class=\"structural\">lastName: </span>Cha<span class=\"structural\">...</span><span class=\"structural\"> }</span></span>","<span class=\"formatted\" title=\"firstName: Bob\nlastName: Marley\"><span class=\"structural\">{ </span><span class=\"structural\">firstName: </span>Bob<span class=\"structural\">, </span><span class=\"structural\">lastName: </span>Marley<span class=\"structural\"> }</span></span>","<span class=\"formatted\" title=\"firstName: Alice\nlastName: Wolf\"><span class=\"structural\">{ </span><span class=\"structural\">firstName: </span>Alice<span class=\"structural\">, </span><span class=\"structural\">lastName: </span>Wolf<span class=\"structural\"> }</span></span>","<span class=\"formatted\" title=\"firstName: Charlie\nlastName: Byrd\"><span class=\"structural\">{ </span><span class=\"structural\">firstName: </span>Charlie<span class=\"structural\">, </span><span class=\"structural\">lastName: </span>Byrd<span class=\"structural\"> }</span></span>"] }, 
 { name: "<span title=\"age: Int\">age</span>", children: [], rightAlign: true, values: ["<span class=\"formatted\" title=\"\"><span class=\"numbers\">20</span></span>","<span class=\"formatted\" title=\"\"><span class=\"numbers\">40</span></span>","<span class=\"formatted\" title=\"\"><span class=\"numbers\">30</span></span>","<span class=\"formatted\" title=\"\"><span class=\"numbers\">20</span></span>","<span class=\"formatted\" title=\"\"><span class=\"numbers\">30</span></span>"] }, 
 { name: "<span title=\"city: String?\">city</span>", children: [], rightAlign: false, values: ["Moscow","Milan","Tokyo","<span class=\"formatted\" title=\"\"><span class=\"null\">null</span></span>","Moscow"] }, 
 { name: "<span title=\"weight: Int?\">weight</span>", children: [], rightAlign: true, values: ["<span class=\"formatted\" title=\"\"><span class=\"null\">null</span></span>","<span class=\"formatted\" title=\"\"><span class=\"null\">null</span></span>","<span class=\"formatted\" title=\"\"><span class=\"numbers\">68</span></span>","<span class=\"formatted\" title=\"\"><span class=\"numbers\">55</span></span>","<span class=\"formatted\" title=\"\"><span class=\"numbers\">90</span></span>"] }, 

docs/StardustDocs/resources/api/tail/notebook_test_tail_3.html

@@ -459,6 +459,7 @@
 /*<!--*/
 call_DataFrame(function() { DataFrame.addTable({ cols: [{ name: "<span title=\"firstName: String\">firstName</span>", children: [], rightAlign: false, values: ["Alice","Charlie"] }, 
 { name: "<span title=\"lastName: String\">lastName</span>", children: [], rightAlign: false, values: ["Wolf","Byrd"] }, 
+{ name: "<span title=\"name: DataRow<*>\">name</span>", children: [0, 1], rightAlign: false, values: ["<span class=\"formatted\" title=\"firstName: Alice\nlastName: Wolf\"><span class=\"structural\">{ </span><span class=\"structural\">firstName: </span>Alice<span class=\"structural\">, </span><span class=\"structural\">lastName: </span>Wolf<span class=\"structural\"> }</span></span>","<span class=\"formatted\" title=\"firstName: Charlie\nlastName: Byrd\"><span class=\"structural\">{ </span><span class=\"structural\">firstName: </span>Charlie<span class=\"structural\">, </span><span class=\"structural\">lastName: </span>Byrd<span class=\"structural\"> }</span></span>"] }, 
 { name: "<span title=\"age: Int\">age</span>", children: [], rightAlign: true, values: ["<span class=\"formatted\" title=\"\"><span class=\"numbers\">20</span></span>","<span class=\"formatted\" title=\"\"><span class=\"numbers\">30</span></span>"] }, 
 { name: "<span title=\"city: String?\">city</span>", children: [], rightAlign: false, values: ["<span class=\"formatted\" title=\"\"><span class=\"null\">null</span></span>","Moscow"] }, 
 { name: "<span title=\"weight: Int\">weight</span>", children: [], rightAlign: true, values: ["<span class=\"formatted\" title=\"\"><span class=\"numbers\">55</span></span>","<span class=\"formatted\" title=\"\"><span class=\"numbers\">90</span></span>"] }, 

samples/build.gradle.kts

@@ -16,6 +16,7 @@ import org.gradle.kotlin.dsl.runKtlintFormatOverTestSourceSet
 import org.gradle.kotlin.dsl.sourceSets
 import org.gradle.kotlin.dsl.test
 import org.gradle.kotlin.dsl.testImplementation
+import org.jetbrains.kotlin.gradle.tasks.KotlinCompile
 
 plugins {
     java
@@ -45,7 +46,7 @@ val dependentProjects = with(projects) {
     )
 }.map { project(it.path) }
 
-tasks.compileKotlin {
+tasks.withType<KotlinCompile> {
     dependentProjects.forEach {
         dependsOn("${it.path}:jar")
     }
@@ -60,7 +61,7 @@ val dependentProjectJarPaths = dependentProjects.map {
 }
 
 dependencies {
-    // implementation(projects.dataframe) // Must depend on jars for the compiler plugin to work!
+    runtimeOnly(projects.dataframe) // Must depend on jars for the compiler plugin to work!
     implementation(files(dependentProjectJarPaths))
 
     // include api() dependencies from dependent projects, as they are not included in the jars

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

@@ -11,7 +11,8 @@ abstract class DataFrameSampleHelper(sampleName: String, subFolder: String = "sa
         subFolder,
         "../docs/StardustDocs/images",
         "../docs/StardustDocs/resources",
-    ), TestBase {
+    ),
+    TestBase {
 
     fun DataColumn<*>.saveDfHtmlSample() {
         toDataFrame().saveDfHtmlSample()

samples/src/test/kotlin/org/jetbrains/kotlinx/dataframe/samples/api/Generate.kt

@@ -20,10 +20,10 @@ import org.junit.Test
 class Generate : DataFrameSampleHelper("generate_docs", "api") {
 
     @DataSchema
-    interface Orders {
-        val orderId: Int
-        val amount: Double
-    }
+    data class Orders(
+        val orderId: Int,
+        val amount: Double,
+    )
 
     private val ordersAlice = dataFrameOf(
         "orderId" to listOf(101, 102),
@@ -36,10 +36,10 @@ class Generate : DataFrameSampleHelper("generate_docs", "api") {
     ).cast<Orders>()
 
     @DataSchema
-    interface Customer {
-        val user: String
-        val orders: List<Orders>
-    }
+    data class Customer(
+        val user: String,
+        val orders: List<Orders>,
+    )
 
     private val df = dataFrameOf(
         "user" to listOf("Alice", "Bob"),
@@ -79,7 +79,7 @@ class Generate : DataFrameSampleHelper("generate_docs", "api") {
     @Test
     fun notebook_test_generate_docs_5() {
         // SampleStart
-        val customers: List<Customer> = df.toList()
+        val customers: List<Customer> = df.cast<Customer>().toList()
         // SampleEnd
     }
 
@@ -93,7 +93,7 @@ class Generate : DataFrameSampleHelper("generate_docs", "api") {
     @Test
     fun notebook_test_generate_docs_7() {
         // SampleStart
-        df
+        df.cast<Customer>()
             .add("ordersTotal") { orders.sumOf { it.amount } }
             .filter { user.startsWith("A") }
             .rename { user }.into("customer")

samples/src/test/kotlin/org/jetbrains/kotlinx/dataframe/samples/api/Modify.kt

@@ -3,12 +3,9 @@ package org.jetbrains.kotlinx.dataframe.samples.api
 import org.jetbrains.kotlinx.dataframe.DataColumn
 import org.jetbrains.kotlinx.dataframe.api.FormattingDsl
 import org.jetbrains.kotlinx.dataframe.api.and
-import org.jetbrains.kotlinx.dataframe.api.cast
 import org.jetbrains.kotlinx.dataframe.api.dataFrameOf
 import org.jetbrains.kotlinx.dataframe.api.format
 import org.jetbrains.kotlinx.dataframe.api.getColumnIndex
-import org.jetbrains.kotlinx.dataframe.api.group
-import org.jetbrains.kotlinx.dataframe.api.into
 import org.jetbrains.kotlinx.dataframe.api.linearBg
 import org.jetbrains.kotlinx.dataframe.api.max
 import org.jetbrains.kotlinx.dataframe.api.min
@@ -18,8 +15,6 @@ import org.jetbrains.kotlinx.dataframe.api.with
 import org.jetbrains.kotlinx.dataframe.samples.DataFrameSampleHelper
 import org.junit.Test
 
-
-
 @Suppress("ktlint:standard:argument-list-wrapping")
 class Modify : DataFrameSampleHelper("operations", "modify") {
 

samples/src/test/kotlin/org/jetbrains/kotlinx/dataframe/samples/api/TestBase.kt

@@ -3,7 +3,7 @@
 package org.jetbrains.kotlinx.dataframe.samples.api
 
 import io.kotest.matchers.shouldBe
-import org.jetbrains.kotlinx.dataframe.DataRow
+import org.jetbrains.kotlinx.dataframe.DataFrame
 import org.jetbrains.kotlinx.dataframe.annotations.DataSchema
 import org.jetbrains.kotlinx.dataframe.api.cast
 import org.jetbrains.kotlinx.dataframe.api.dataFrameOf
@@ -13,7 +13,7 @@ import org.jetbrains.kotlinx.dataframe.api.into
 @Suppress("ktlint:standard:argument-list-wrapping")
 interface TestBase {
 
-    val peopleDf
+    val peopleDf: DataFrame<Person>
         get() = dataFrameOf("firstName", "lastName", "age", "city", "weight", "isHappy")(
             "Alice", "Cooper", 15, "London", 54, true,
             "Bob", "Dylan", 45, "Dubai", 87, true,
@@ -22,7 +22,8 @@ interface TestBase {
             "Bob", "Marley", 30, "Tokyo", 68, true,
             "Alice", "Wolf", 20, null, 55, false,
             "Charlie", "Byrd", 30, "Moscow", 90, true,
-        ).group { firstName and lastName }.into("name").cast<Person>()
+        ).group("firstName", "lastName").into("name")
+            .cast<Person>(verify = false)
 
     @DataSchema
     interface Name {

samples/src/test/kotlin/org/jetbrains/kotlinx/dataframe/samples/api/collectionsInterop/AssociateSamples.kt

@@ -1,42 +1,14 @@
 package org.jetbrains.kotlinx.dataframe.samples.api.collectionsInterop
 
-import org.jetbrains.kotlinx.dataframe.DataRow
-import org.jetbrains.kotlinx.dataframe.annotations.DataSchema
 import org.jetbrains.kotlinx.dataframe.api.associate
-import org.jetbrains.kotlinx.dataframe.api.cast
-import org.jetbrains.kotlinx.dataframe.api.dataFrameOf
-import org.jetbrains.kotlinx.dataframe.api.group
-import org.jetbrains.kotlinx.dataframe.api.into
 import org.jetbrains.kotlinx.dataframe.samples.DataFrameSampleHelper
-import org.jetbrains.kotlinx.dataframe.samples.api.utils.age
-import org.jetbrains.kotlinx.dataframe.samples.api.utils.name
+import org.jetbrains.kotlinx.dataframe.samples.api.age
+import org.jetbrains.kotlinx.dataframe.samples.api.name
 import org.junit.Test
 
 class AssociateSamples : DataFrameSampleHelper("associate", "api/collectionsInterop") {
 
-    @DataSchema
-    interface Name {
-        val firstName: String
-        val lastName: String
-    }
-
-    @DataSchema
-    interface Person {
-        val age: Int
-        val city: String?
-        val name: DataRow<Name>
-        val weight: Int?
-        val isHappy: Boolean
-    }
-
-    private val df = dataFrameOf(
-        "firstName" to listOf("Alice", "Bob", "Charlie", "Charlie", "Bob", "Alice", "Charlie"),
-        "lastName" to listOf("Cooper", "Dylan", "Daniels", "Chaplin", "Marley", "Wolf", "Byrd"),
-        "age" to listOf(15, 45, 20, 40, 30, 20, 30),
-        "city" to listOf("London", "Dubai", "Moscow", "Milan", "Tokyo", null, "Moscow"),
-        "weight" to listOf(54, 87, null, null, 68, 55, 90),
-        "isHappy" to listOf(true, true, false, true, true, false, true),
-    ).group { firstName and lastName }.into("name").cast<Person>()
+    private val df = peopleDf
 
     @Test
     fun notebook_test_associate_1() {