Added basic upload functionality

1 files changed, 15 insertions, 7 deletions

diff --git a/src/main/kotlin/App.kt b/src/main/kotlin/App.kt
index 993ac9d..79c4e83 100644
--- a/src/main/kotlin/App.kt
+++ b/src/main/kotlin/App.kt
@@ -1,13 +1,12 @@
 package space.anity
 
-import io.javalin.Javalin
-import io.javalin.NotFoundResponse
-import java.io.File
-import java.nio.file.Files
-import java.nio.file.Paths
+import io.javalin.*
+import io.javalin.core.util.*
+import java.io.*
+import java.nio.file.*
 
 fun main(args: Array<String>) {
-    val app = Javalin.create().start(7000)
+    val app = Javalin.create().enableStaticFiles("../resources/").start(7000)
     val fileHome = "files"
 
     app.get("/") { ctx ->
@@ -25,9 +24,18 @@ fun main(args: Array<String>) {
             }
             ctx.result(files)
         } catch (_: java.nio.file.NoSuchFileException) {
-            throw NotFoundResponse("File or directory does not exist")
+            throw NotFoundResponse("Error: File or directory does not exist.")
         }
 
         //File("test").writeText(ctx.splat(0)!!)
     }
+
+    app.post("/upload") { ctx ->
+        ctx.uploadedFiles("files").forEach { (contentType, content, name, extension) ->
+            if (ctx.queryParam("dir") !== null)
+                FileUtil.streamToFile(content, "files/${ctx.queryParam("dir")}/$name")
+            else
+                throw BadRequestResponse("Error: Please enter a filename.")
+        }
+    }
 }