Add integration with ktor-client

Author: koperagen

dataframe-jupyter/build.gradle.kts

@@ -8,13 +8,16 @@ group = "org.jetbrains.kotlinx"
 
 repositories {
     mavenCentral()
+    maven("https://packages.jetbrains.team/maven/p/kds/kotlin-ds-maven")
 }
 
 dependencies {
     compileOnly(project(":core"))
+    compileOnly(libs.jupyter.ktor.client)
     testImplementation(libs.junit)
     testImplementation(libs.serialization.json)
     testImplementation(project(":core"))
+    testImplementation(libs.jupyter.ktor.client)
     testImplementation(libs.kotestAssertions) {
         exclude("org.jetbrains.kotlin", "kotlin-stdlib-jdk8")
     }

dataframe-jupyter/src/main/kotlin/org/jetbrains/kotlinx/dataframe/jupyter/Integration.kt

@@ -267,6 +267,7 @@ internal class Integration(
         import("org.jetbrains.kotlinx.dataframe.jupyter.ImportDataSchema")
         import("org.jetbrains.kotlinx.dataframe.jupyter.importDataSchema")
         import("org.jetbrains.kotlinx.dataframe.jupyter.KotlinNotebookPluginUtils")
+        import("org.jetbrains.kotlinx.dataframe.jupyter.bodyAsDataFrame")
         import("java.net.URL")
         import("java.io.File")
         import("kotlinx.datetime.Instant")

dataframe-jupyter/src/main/kotlin/org/jetbrains/kotlinx/dataframe/jupyter/KtorClientIntegration.kt

@@ -0,0 +1,9 @@
+package org.jetbrains.kotlinx.dataframe.jupyter
+
+import org.jetbrains.kotlinx.dataframe.DataFrame
+import org.jetbrains.kotlinx.dataframe.io.readJsonStr
+import org.jetbrains.kotlinx.jupyter.ktor.client.core.NotebookHttpResponse
+
+public fun NotebookHttpResponse.bodyAsDataFrame(): DataFrame<*> {
+    return DataFrame.readJsonStr(bodyAsText())
+}

dataframe-jupyter/src/test/kotlin/org/jetbrains/kotlinx/dataframe/jupyter/KtorClientIntegrationTest.kt

@@ -0,0 +1,18 @@
+package org.jetbrains.kotlinx.dataframe.jupyter
+
+import org.jetbrains.kotlinx.jupyter.testkit.JupyterReplTestCase
+import org.jetbrains.kotlinx.jupyter.testkit.ReplProvider
+import org.junit.Test
+
+class KtorClientIntegrationTest : JupyterReplTestCase(ReplProvider.forLibrariesTesting(listOf("dataframe", "ktor-client"))) {
+    @Test
+    fun `extension function for NotebookHttpResponse is available`() {
+        execRaw("""%use dataframe""")
+        execRaw("""%use ktor-client""")
+        val test = """
+            fun test() {
+                http.get("https://raw.githubusercontent.com/Kotlin/dataframe/master/data/jetbrains.json").bodyAsDataFrame()
+            }""".trimIndent()
+        execRaw(test)
+    }
+}

gradle/libs.versions.toml

@@ -40,6 +40,7 @@ kotlinDatetime = "0.6.0"
 openapi = "2.1.20"
 kotlinLogging = "6.0.3"
 sl4j = "2.0.12"
+ktorClient = "0.2.0-dev-11"
 
 junit = "4.13.2"
 junit-jupiter = "5.10.2"
@@ -64,6 +65,7 @@ duckdb = "0.10.0"
 ksp-gradle = { group = "com.google.devtools.ksp", name = "symbol-processing-gradle-plugin", version.ref = "ksp" }
 ksp-api = { group = "com.google.devtools.ksp", name = "symbol-processing-api", version.ref = "ksp" }
 jupyter-api = { group = "org.jetbrains.kotlinx", name = "kotlin-jupyter-kernel", version.ref = "kotlinJupyter" }
+jupyter-ktor-client = { group = "org.jetbrains.kotlinx", name = "kotlin-jupyter-ktor-client", version.ref = "ktorClient"}
 
 kotlin-stdlib = { group = "org.jetbrains.kotlin", name = "kotlin-stdlib", version.ref = "kotlin" }
 kotlin-stdlib-jdk8 = { group = "org.jetbrains.kotlin", name = "kotlin-stdlib-jdk8", version.ref = "kotlin" }