Add missing overloads for insert

Author: nikitinas

src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/add.kt

@@ -12,6 +12,7 @@ import org.jetbrains.kotlinx.dataframe.columns.ColumnAccessor
 import org.jetbrains.kotlinx.dataframe.columns.ColumnPath
 import org.jetbrains.kotlinx.dataframe.columns.ColumnReference
 import org.jetbrains.kotlinx.dataframe.impl.DataRowImpl
+import org.jetbrains.kotlinx.dataframe.impl.api.insertImpl
 import org.jetbrains.kotlinx.dataframe.impl.columns.resolveSingle
 import org.jetbrains.kotlinx.dataframe.index
 import kotlin.reflect.KProperty
@@ -60,7 +61,7 @@ public inline fun <reified R, T> DataFrame<T>.add(
 ): DataFrame<T> {
     val col = map(path.name(), infer, expression)
     if (path.size == 1) return this + col
-    return insert(path, col)
+    return insertImpl(path, col)
 }
 
 public fun <T> DataFrame<T>.add(body: AddDsl<T>.() -> Unit): DataFrame<T> {

src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/insert.kt

@@ -6,12 +6,10 @@ import org.jetbrains.kotlinx.dataframe.DataFrame
 import org.jetbrains.kotlinx.dataframe.RowExpression
 import org.jetbrains.kotlinx.dataframe.columns.ColumnAccessor
 import org.jetbrains.kotlinx.dataframe.columns.ColumnPath
-import org.jetbrains.kotlinx.dataframe.impl.api.ColumnToInsert
 import org.jetbrains.kotlinx.dataframe.impl.api.insertImpl
+import org.jetbrains.kotlinx.dataframe.impl.columnName
 import org.jetbrains.kotlinx.dataframe.impl.removeAt
-
-public fun <T> DataFrame<T>.insert(path: ColumnPath, column: AnyCol): DataFrame<T> =
-    insertImpl(this, listOf(ColumnToInsert(path, column)))
+import kotlin.reflect.KProperty
 
 public fun <T> DataFrame<T>.insert(column: AnyCol): InsertClause<T> = InsertClause(this, column)
 
@@ -27,19 +25,27 @@ public inline fun <T, reified R> DataFrame<T>.insert(
     noinline expression: RowExpression<T, R>
 ): InsertClause<T> = insert(column.name(), infer, expression)
 
+public inline fun <T, reified R> DataFrame<T>.insert(
+    column: KProperty<R>,
+    infer: Infer = Infer.Nulls,
+    noinline expression: RowExpression<T, R>
+): InsertClause<T> = insert(column.columnName, infer, expression)
+
 public data class InsertClause<T>(val df: DataFrame<T>, val column: AnyCol)
 
 public fun <T> InsertClause<T>.under(column: ColumnSelector<T, *>): DataFrame<T> = under(df.getColumnPath(column))
-public fun <T> InsertClause<T>.under(columnPath: ColumnPath): DataFrame<T> = df.insert(columnPath + column.name, column)
+public fun <T> InsertClause<T>.under(columnPath: ColumnPath): DataFrame<T> = df.insertImpl(columnPath + column.name, column)
 public fun <T> InsertClause<T>.under(column: ColumnAccessor<*>): DataFrame<T> = under(column.path())
+public fun <T> InsertClause<T>.under(column: KProperty<*>): DataFrame<T> = under(column.columnName)
 public fun <T> InsertClause<T>.under(column: String): DataFrame<T> = under(pathOf(column))
 
 public fun <T> InsertClause<T>.after(column: ColumnSelector<T, *>): DataFrame<T> = after(df.getColumnPath(column))
 public fun <T> InsertClause<T>.after(column: String): DataFrame<T> = df.add(this.column).move(this.column).after(column)
 public fun <T> InsertClause<T>.after(column: ColumnAccessor<*>): DataFrame<T> = after(column.path())
+public fun <T> InsertClause<T>.after(column: KProperty<*>): DataFrame<T> = after(column.columnName)
 public fun <T> InsertClause<T>.after(columnPath: ColumnPath): DataFrame<T> {
     val dstPath = ColumnPath(columnPath.removeAt(columnPath.size - 1) + column.name())
-    return df.insert(dstPath, column).move { dstPath }.after { columnPath }
+    return df.insertImpl(dstPath, column).move { dstPath }.after { columnPath }
 }
 
 public fun <T> InsertClause<T>.at(position: Int): DataFrame<T> = df.add(column).move(column).to(position)

src/main/kotlin/org/jetbrains/kotlinx/dataframe/impl/api/insert.kt

@@ -1,6 +1,7 @@
 package org.jetbrains.kotlinx.dataframe.impl.api
 
 import org.jetbrains.kotlinx.dataframe.AnyBaseColumn
+import org.jetbrains.kotlinx.dataframe.AnyCol
 import org.jetbrains.kotlinx.dataframe.DataColumn
 import org.jetbrains.kotlinx.dataframe.DataFrame
 import org.jetbrains.kotlinx.dataframe.api.cast
@@ -18,6 +19,10 @@ internal data class ColumnToInsert(
     val referenceNode: ReadonlyTreeNode<ReferenceData>? = null
 )
 
+@PublishedApi
+internal fun <T> DataFrame<T>.insertImpl(path: ColumnPath, column: AnyCol): DataFrame<T> =
+    insertImpl(this, listOf(ColumnToInsert(path, column)))
+
 @JvmName("insertImplT")
 internal fun <T> DataFrame<T>.insertImpl(columns: List<ColumnToInsert>) =
     insertImpl(this, columns)