com.simiacryptus.skyenet.webui.servlet.SessionShareServlet.kt Maven / Gradle / Ivy
package com.simiacryptus.skyenet.webui.servlet
import com.simiacryptus.util.JsonUtil
import com.simiacryptus.skyenet.core.platform.ApplicationServices
import com.simiacryptus.skyenet.core.platform.ApplicationServices.authenticationManager
import com.simiacryptus.skyenet.core.platform.ApplicationServices.authorizationManager
import com.simiacryptus.skyenet.core.platform.ApplicationServices.cloud
import com.simiacryptus.skyenet.core.platform.ApplicationServicesConfig.dataStorageRoot
import com.simiacryptus.skyenet.core.platform.AuthorizationInterface.OperationType
import com.simiacryptus.skyenet.core.platform.StorageInterface
import com.simiacryptus.skyenet.core.platform.StorageInterface.Companion.long64
import com.simiacryptus.skyenet.core.platform.User
import com.simiacryptus.skyenet.core.util.Selenium
import com.simiacryptus.skyenet.webui.application.ApplicationServer
import com.simiacryptus.skyenet.webui.application.ApplicationServer.Companion.getCookie
import com.simiacryptus.skyenet.util.Selenium2S3
import jakarta.servlet.http.HttpServlet
import jakarta.servlet.http.HttpServletRequest
import jakarta.servlet.http.HttpServletResponse
import org.apache.http.client.HttpClient
import org.apache.http.impl.client.HttpClients
import java.net.URI
import kotlin.reflect.jvm.javaType
import kotlin.reflect.typeOf
import com.google.zxing.BarcodeFormat
import com.google.zxing.qrcode.QRCodeWriter
import com.google.zxing.client.j2se.MatrixToImageWriter
import java.io.ByteArrayOutputStream
import java.util.Base64
class SessionShareServlet(
private val server: ApplicationServer,
) : HttpServlet() {
private fun generateQRCodeDataURL(url: String): String {
val qrCodeWriter = QRCodeWriter()
val bitMatrix = qrCodeWriter.encode(url, BarcodeFormat.QR_CODE, 200, 200)
val outputStream = ByteArrayOutputStream()
MatrixToImageWriter.writeToStream(bitMatrix, "PNG", outputStream)
val imageBytes = outputStream.toByteArray()
val base64Image = Base64.getEncoder().encodeToString(imageBytes)
return "data:image/png;base64,$base64Image"
}
override fun doGet(req: HttpServletRequest, resp: HttpServletResponse) {
val user = authenticationManager.getUser(req.getCookie())
val cookies = req.cookies
if (!req.parameterMap.containsKey("url")) {
resp.status = HttpServletResponse.SC_BAD_REQUEST
resp.writer.write("Url is required")
return
}
val url = req.getParameter("url")
val host = URI(url).host
val appName = url.split("/").dropLast(1).last()
val sessionID = url.split("#").lastOrNull() ?: throw IllegalArgumentException("No session id in url: $url")
require(acceptHost(user, host)) { "Invalid url: $url" }
val storageInterface = ApplicationServices.dataStorageFactory(dataStorageRoot)
val session = StorageInterface.parseSessionID(sessionID)
val pool = ApplicationServices.clientManager.getPool(session, user)
val infoFile = storageInterface.getDataDir(user, session).resolve("info.json").apply { parentFile.mkdirs() }
val json = if (infoFile.exists()) JsonUtil.fromJson>(
infoFile.readText(),
typeOf>().javaType
) else mapOf()
val sessionSettings = (json as? Map)?.toMutableMap() ?: mutableMapOf()
val previousShare = sessionSettings["shareId"]
var shareURL = url(appName, previousShare ?: long64())
var qrCodeDataURL = generateQRCodeDataURL(shareURL)
when {
null != previousShare && validateUrl(shareURL) -> {
log.info("Reusing shareId: $previousShare")
resp.contentType = "text/html"
resp.status = HttpServletResponse.SC_OK
//language=HTML
resp.writer.write(
"""
|
| Save Session
|
|
|
|
| Sharing URL
|
|
| Sharing URL
|
| 
|
|""".trimMargin()
)
}
!authorizationManager.isAuthorized(server.javaClass, user, OperationType.Share) -> {
resp.status = HttpServletResponse.SC_FORBIDDEN
resp.writer.write("Forbidden")
return
}
else -> {
val shareId = long64()
currentlyProcessing.add(shareId)
pool.submit {
try {
log.info("Generating shareId: $shareId")
sessionSettings["shareId"] = shareId
infoFile.writeText(JsonUtil.toJson(sessionSettings))
val selenium2S3: Selenium = ApplicationServices.seleniumFactory?.invoke(pool, cookies) ?: throw IllegalStateException("Selenium not configured")
if (selenium2S3 is Selenium2S3) {
selenium2S3.loadImages = req.getParameter("loadImages")?.toBoolean() ?: false
}
selenium2S3.save(
url = URI(url).toURL(),
saveRoot = "$appName/$shareId",
currentFilename = "index.html",
)
log.info("Saved session $sessionID to $appName/$shareId")
} catch (e: Throwable) {
log.error("Error saving session $sessionID to $appName/$shareId", e)
} finally {
currentlyProcessing.remove(shareId)
}
}
resp.contentType = "text/html"
resp.status = HttpServletResponse.SC_OK
//language=HTML
shareURL = url(appName, shareId)
qrCodeDataURL = generateQRCodeDataURL(shareURL)
resp.writer.write(
"""
|
|
| Saving Session
|
|
|
|
| Saving Session... This page will soon be ready!
|
|
| To monitor progress, you can use the session threads page
|
|
""".trimMargin()
)
}
}
}
private fun url(appName: String, shareId: String) =
"""${cloud!!.shareBase}/$appName/$shareId/index.html"""
private fun acceptHost(user: User?, host: String?): Boolean {
return when (host) {
"localhost" -> true
domain -> true
else -> authorizationManager.isAuthorized(server.javaClass, user, OperationType.Admin)
}
}
companion object {
private val log = org.slf4j.LoggerFactory.getLogger(SessionShareServlet::class.java)
private val currentlyProcessing = mutableSetOf()
fun validateUrl(previousShare: String): Boolean = when {
currentlyProcessing.contains(previousShare) -> true
else -> HttpClients.createSystem().use { httpClient: HttpClient ->
val responseEntity = httpClient.execute(org.apache.http.client.methods.HttpGet(previousShare))
return responseEntity.statusLine.statusCode == 200
}
}
var domain = System.getProperty("domain", "apps.simiacrypt.us")
}
}