From b4423e8bba42ae15abccc4916d8ca56450e21d13 Mon Sep 17 00:00:00 2001
From: Anton Belyi <anton.8938575@gmail.com>
Date: Mon, 1 May 2023 19:19:17 +0300
Subject: [PATCH] SNRK-69: added web interface

---
 snark-ktor/build.gradle.kts                   |   2 +
 snark-ktor/snark-ktor.iml                     |  12 ++
 .../space/kscience/snark/ktor/WebInterface.kt | 107 ++++++++++++++++++
 3 files changed, 121 insertions(+)
 create mode 100644 snark-ktor/snark-ktor.iml
 create mode 100644 snark-ktor/src/main/kotlin/space/kscience/snark/ktor/WebInterface.kt

diff --git a/snark-ktor/build.gradle.kts b/snark-ktor/build.gradle.kts
index cb51636..1a40f79 100644
--- a/snark-ktor/build.gradle.kts
+++ b/snark-ktor/build.gradle.kts
@@ -12,6 +12,8 @@ dependencies {
     api("io.ktor:ktor-server-core:$ktorVersion")
     api("io.ktor:ktor-server-html-builder:$ktorVersion")
     api("io.ktor:ktor-server-host-common:$ktorVersion")
+    implementation("io.ktor:ktor-server-netty:2.3.0")
+    implementation(project(mapOf("path" to ":snark-storage-driver")))
 
     testApi("io.ktor:ktor-server-tests:$ktorVersion")
 }
\ No newline at end of file
diff --git a/snark-ktor/snark-ktor.iml b/snark-ktor/snark-ktor.iml
new file mode 100644
index 0000000..3200c88
--- /dev/null
+++ b/snark-ktor/snark-ktor.iml
@@ -0,0 +1,12 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module type="JAVA_MODULE" version="4">
+  <component name="NewModuleRootManager" inherit-compiler-output="true">
+    <exclude-output />
+    <content url="file://$MODULE_DIR$">
+      <sourceFolder url="file://$MODULE_DIR$/src/main/kotlin" isTestSource="false" />
+    </content>
+    <orderEntry type="inheritedJdk" />
+    <orderEntry type="sourceFolder" forTests="false" />
+    <orderEntry type="library" name="KotlinJavaRuntime" level="project" />
+  </component>
+</module>
\ No newline at end of file
diff --git a/snark-ktor/src/main/kotlin/space/kscience/snark/ktor/WebInterface.kt b/snark-ktor/src/main/kotlin/space/kscience/snark/ktor/WebInterface.kt
new file mode 100644
index 0000000..2765170
--- /dev/null
+++ b/snark-ktor/src/main/kotlin/space/kscience/snark/ktor/WebInterface.kt
@@ -0,0 +1,107 @@
+package space.kscience.snark.ktor
+
+import io.ktor.http.*
+import io.ktor.http.content.*
+import io.ktor.server.engine.*
+import io.ktor.server.netty.*
+import io.ktor.server.application.*
+import io.ktor.server.html.*
+import io.ktor.server.request.*
+import io.ktor.server.response.*
+import io.ktor.server.http.content.*
+import kotlinx.html.*
+import io.ktor.server.routing.*
+import java.nio.file.Path
+import space.kscience.snark.storage.Directory
+import space.kscience.snark.storage.local.localStorage
+import kotlin.io.path.createTempDirectory
+import kotlin.io.path.isDirectory
+import kotlin.io.path.listDirectoryEntries
+import kotlin.io.path.name
+import space.kscience.snark.storage.unzip.unzip
+
+public interface DataHolder {
+
+    fun init() : Directory
+    fun represent(): String
+}
+class LocalDataHolder: DataHolder {
+    private var source: Path? = null
+    private var response: String = ""
+    override fun init(): Directory {
+        source = createTempDirectory()
+        return localStorage(source!!)
+    }
+    private fun buildResponse(path: Path) {
+        for (entry in path.listDirectoryEntries()) {
+            if (entry.isDirectory()) {
+                buildResponse(entry)
+            } else {
+                response += source!!.relativize(entry).toString() + "\n"
+            }
+        }
+    }
+    override fun represent() : String =
+        if (source == null) {
+            "No data was loaded!"
+        } else {
+            response = "List of files:\n"
+            buildResponse(source!!)
+            response
+        }
+}
+fun main() {
+    val dataHolder: DataHolder = LocalDataHolder()
+    embeddedServer(Netty, 9090) {
+        routing {
+            get("/") {
+                call.respondHtml(HttpStatusCode.OK) {
+                    head {
+                        title {
+                            +"SNARK"
+                        }
+                    }
+                    body {
+                        h1 {
+                            +"SNARK"
+                        }
+                    }
+                    body {
+                        postForm (action = "/upload", encType = FormEncType.multipartFormData)  {
+                            label {
+                                +"Choose zip archive: "
+                            }
+                            input (name = "file", type = InputType.file) {}
+                            button {
+                                +"Upload file"
+                            }
+                        }
+                        a("/data") {
+                            +"Show data\n"
+                        }
+                    }
+                }
+            }
+            post("/upload") {
+                val multipartData = call.receiveMultipart()
+                val tmp = createTempFile(suffix=".zip")
+                multipartData.forEachPart { part ->
+                    when (part) {
+                        is PartData.FileItem -> {
+                            val fileBytes = part.streamProvider().readBytes()
+                            tmp.writeBytes(fileBytes)
+                        }
+                        else -> {
+                        }
+                    }
+                    part.dispose()
+                }
+                unzip(tmp.toPath().toString(), dataHolder.init())
+                call.respondText("File is successfully uploaded")
+            }
+            get("/data") {
+                call.respondText(dataHolder.represent())
+            }
+        }
+    }.start(wait = true)
+}
\ No newline at end of file