Add Schematic Download

src/main/kotlin/de/steamwar/routes/Schematic.kt

@@ -22,7 +22,9 @@ package de.steamwar.routes
 import de.steamwar.plugins.SWAuthPrincipal
 import de.steamwar.plugins.SWPermissionCheck
 import de.steamwar.sql.NodeData
+import de.steamwar.sql.NodeDownload
 import de.steamwar.sql.NodeMember
+import de.steamwar.sql.SWException
 import de.steamwar.sql.SchematicNode
 import io.ktor.http.*
 import io.ktor.server.application.*
@@ -32,6 +34,9 @@ import io.ktor.server.response.*
 import io.ktor.server.routing.*
 import kotlinx.serialization.Serializable
 import java.security.MessageDigest
+import java.time.Duration
+import java.time.Instant
+import java.time.temporal.ChronoUnit
 import java.util.*
 
 @Serializable
@@ -67,6 +72,44 @@ data class SchematicCode(val id: Int, val code: String, val expires: Long)
 data class UploadSchematic(val name: String, val content: String)
 
 fun Route.configureSchematic() {
+    get("/download/{code}") {
+        val code = call.parameters["code"] ?: run {
+            call.respond(HttpStatusCode.BadRequest)
+            return@get
+        }
+
+        val dl = NodeDownload.get(code) ?: run {
+            call.respond(HttpStatusCode.NotFound)
+            return@get
+        }
+
+        dl.delete()
+
+        if(dl.timestamp.toInstant().plus(Duration.of(5, ChronoUnit.MINUTES)).isBefore(Instant.now())) {
+            call.respond(HttpStatusCode.Gone)
+            return@get
+        }
+
+        val node = SchematicNode.getSchematicNode(dl.nodeId) ?: run {
+            call.respond(HttpStatusCode.NotFound)
+            return@get
+        }
+
+        val user = call.principal<SWAuthPrincipal>()?.user
+        if(user != null && !node.accessibleByUser(user)) {
+            call.respond(HttpStatusCode.Forbidden)
+            SWException.log("User ${user.userName} tried to download schematic ${node.name} without permission", user.id.toString())
+            return@get
+        }
+
+        val data = NodeData.get(node) ?: run {
+            call.respond(HttpStatusCode.InternalServerError)
+            return@get
+        }
+
+        call.response.header("Content-Disposition", "attachment; filename=\"${node.name}.${if (data.nodeFormat) "schem" else "schematic"}\"")
+        call.respondBytes(data.schemData().readAllBytes(), contentType = ContentType.Application.OctetStream, status = HttpStatusCode.OK)
+    }
     route("/schem") {
         install(SWPermissionCheck)
         get {