55 lines
1.7 KiB
Kotlin
55 lines
1.7 KiB
Kotlin
package starter.routes
|
|
|
|
import org.http4k.core.Method
|
|
import org.http4k.core.Response
|
|
import org.http4k.core.body.form
|
|
import org.http4k.core.body.formAsMap
|
|
import org.http4k.core.with
|
|
import org.http4k.routing.bind
|
|
import starter.Project
|
|
import starter.ProjectZip
|
|
import starter.config.StarterConfig
|
|
import starter.extensions.attachment
|
|
import starter.extensions.badRequest
|
|
import starter.extensions.ok
|
|
import java.io.ByteArrayInputStream
|
|
|
|
class ZipRouteSupplier(
|
|
private val conf: StarterConfig,
|
|
private val projectZip: ProjectZip,
|
|
) : RouteSupplier {
|
|
|
|
override fun get() = "/" bind Method.POST to { req ->
|
|
|
|
val deps = conf.dependencies.filter {
|
|
req.form(it.name) != null
|
|
}
|
|
|
|
val inputKeys = conf.inputs.map { it.name }
|
|
val inputs = req.formAsMap()
|
|
.filter { it.key in inputKeys }
|
|
.map { (name, value) ->
|
|
conf.inputs.find { it.name == name }!!.copy(value = value.first())
|
|
}
|
|
|
|
val projectName = inputs.find { it.name == "name" }!!.value!!
|
|
val basePackage = inputs.find { it.name == "basePackage" }!!.value!!
|
|
|
|
if (basePackage.contains("/") || basePackage.contains("..")) {
|
|
Response.badRequest()
|
|
} else {
|
|
val repositories = deps.mapNotNull { it.repository }.toSet()
|
|
val project = Project(projectName, basePackage, inputs, deps, repositories)
|
|
val outputStream = projectZip.createZip(project)
|
|
|
|
Response.ok().with(
|
|
attachment(
|
|
value = ByteArrayInputStream(outputStream.toByteArray()),
|
|
name = "$projectName.zip",
|
|
contentType = "application/zip"
|
|
)
|
|
)
|
|
}
|
|
}
|
|
}
|