mirror of
https://github.com/gitbucket/gitbucket.git
synced 2025-11-06 13:35:50 +01:00
33 lines
1011 B
Scala
33 lines
1011 B
Scala
package app
|
|
|
|
import _root_.util.{Keys, FileUtil}
|
|
import util.ControlUtil._
|
|
import org.scalatra._
|
|
import org.scalatra.servlet.{MultipartConfig, FileUploadSupport}
|
|
import org.apache.commons.io.FileUtils
|
|
|
|
/**
|
|
* Provides Ajax based file upload functionality.
|
|
*
|
|
* This servlet saves uploaded file as temporary file and returns the unique id.
|
|
* You can get uploaded file using [[app.FileUploadControllerBase#getTemporaryFile()]] with this id.
|
|
*/
|
|
class FileUploadController extends ScalatraServlet
|
|
with FileUploadSupport with FlashMapSupport with FileUploadControllerBase {
|
|
|
|
configureMultipartHandling(MultipartConfig(maxFileSize = Some(3 * 1024 * 1024)))
|
|
|
|
post("/image"){
|
|
fileParams.get("file") match {
|
|
case Some(file) if(FileUtil.isImage(file.name)) => defining(generateFileId){ fileId =>
|
|
FileUtils.writeByteArrayToFile(getTemporaryFile(fileId), file.get)
|
|
session += Keys.Session.Upload(fileId) -> file.name
|
|
Ok(fileId)
|
|
}
|
|
case None => BadRequest
|
|
}
|
|
}
|
|
|
|
}
|
|
|