Migrate usages of AccessOverload APIs in inline functions to public alternatives

Author: koperagen

core/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/into.kt

@@ -8,10 +8,12 @@ import org.jetbrains.kotlinx.dataframe.annotations.AccessApiOverload
 import org.jetbrains.kotlinx.dataframe.annotations.Interpretable
 import org.jetbrains.kotlinx.dataframe.annotations.Refine
 import org.jetbrains.kotlinx.dataframe.columns.ColumnAccessor
+import org.jetbrains.kotlinx.dataframe.columns.ColumnPath
 import org.jetbrains.kotlinx.dataframe.impl.aggregation.internal
 import org.jetbrains.kotlinx.dataframe.impl.aggregation.withExpr
 import org.jetbrains.kotlinx.dataframe.impl.columnName
 import kotlin.reflect.KProperty
+import kotlin.reflect.KType
 import kotlin.reflect.typeOf
 
 // region GroupBy
@@ -29,7 +31,7 @@ public fun <T> GroupBy<T, *>.into(column: KProperty<AnyFrame>): DataFrame<T> = t
 public inline fun <T, G, reified V> GroupBy<T, G>.into(
     columnName: String? = null,
     noinline expression: RowExpression<G, V>,
-): DataFrame<G> = into(pathOf(columnName ?: groups.name()).cast(), expression)
+): DataFrame<G> = into(pathOf(columnName ?: groups.name()), expression, typeOf<V>())
 
 // @Hide
 @AccessApiOverload
@@ -44,6 +46,16 @@ public inline fun <T, G, reified V> GroupBy<T, G>.into(
     }
 }
 
+@PublishedApi
+internal fun <T, G, V> GroupBy<T, G>.into(
+    path: ColumnPath,
+    expression: RowExpression<G, V>,
+    type: KType,
+): DataFrame<G> =
+    aggregate {
+        internal().withExpr(type, path, expression)
+    }
+
 @AccessApiOverload
 public inline fun <T, G, reified V> GroupBy<T, G>.into(
     column: KProperty<V>,

core/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/toDataFrame.kt

@@ -31,7 +31,7 @@ public inline fun <reified T> Iterable<T>.toDataFrame(): DataFrame<T> =
         // or has no properties
         if (!T::class.canBeUnfolded) {
             // create a single `value` column
-            ValueProperty<T>::value from { it }
+            ValueProperty<T>::value.name from { it }
         } else {
             // otherwise creates columns based on properties
             properties()

core/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/with.kt

@@ -44,7 +44,7 @@ public inline fun <T, reified V> ReducedPivotGroupBy<T>.with(noinline expression
         val value = reducer(this)?.let {
             val value = expression(it, it)
             if (value is AnyColumnReference) {
-                it[value]
+                value.getValue(it)
             } else {
                 value
             }

core/src/main/kotlin/org/jetbrains/kotlinx/dataframe/impl/aggregation/aggregations.kt

@@ -95,7 +95,7 @@ internal fun <T, V> AggregateInternalDsl<T>.withExpr(type: KType, path: ColumnPa
     val values = df.rows().map {
         val value = expression(it, it)
         if (value is AnyColumnReference) {
-            it[value]
+            value.getValue(it)
         } else {
             value
         }

core/src/main/kotlin/org/jetbrains/kotlinx/dataframe/impl/aggregation/modes/aggregateBy.kt

@@ -112,7 +112,7 @@ internal inline fun <T, reified V : R, R> Aggregator<V & Any, R>.aggregateByOrNu
 ): DataRow<T>? =
     data.getOrNull(
         indexOfAggregationResult(
-            values = data.asSequence().map { it[column] },
+            values = data.asSequence().map { column.getValue(it) },
             valueType = typeOf<V>(),
         ),
     )

core/src/test/kotlin/org/jetbrains/kotlinx/dataframe/testSets/person/DataFrameTests.kt

@@ -1177,8 +1177,9 @@ class DataFrameTests : BaseTest() {
             .split { others }.intoRows()
             .add(sum) { name.length + other().length }
 
-        val matrix = src.pivot { other }.groupBy { name }.with { sum }
+        val matrix = src.pivot { other }.groupBy { name }.with { sum() }
         matrix.getColumnGroup(other.name()).ncol shouldBe names.size
+        matrix.getColumnGroup(other.name())["Alice"].type() shouldBe typeOf<List<Int>>()
     }
 
     @Test

core/src/test/kotlin/org/jetbrains/kotlinx/dataframe/testSets/person/PivotTests.kt

@@ -161,6 +161,13 @@ class PivotTests {
             .groupBy { name }
             .default("-")
             .values { value } shouldBe res
+        typed
+            .pivot { key }
+            .groupBy { name }
+            .default("-")
+            .with { value() } shouldBe res
+
+        // special case that looks very odd, but kept it for compatibility
         typed
             .pivot { key }
             .groupBy { name }
@@ -180,7 +187,7 @@ class PivotTests {
             .pivot(key)
             .groupBy(name)
             .default("-")
-            .with { value } shouldBe res
+            .with { value() } shouldBe res
         typed
             .groupBy { name }
             .pivot { key }