Newer
Older
gitbucket_jkp / src / main / scala / util / FileUploadUtil.scala
  1. package util
  2.  
  3. import java.text.SimpleDateFormat
  4. import javax.servlet.http.HttpSession
  5. import util.Directory._
  6. import org.apache.commons.io.FileUtils
  7.  
  8. object FileUploadUtil {
  9.  
  10. def generateFileId: String =
  11. new SimpleDateFormat("yyyyMMddHHmmSSsss").format(new java.util.Date(System.currentTimeMillis))
  12.  
  13. def TemporaryDir(implicit session: HttpSession): java.io.File =
  14. new java.io.File(GitBucketHome, s"tmp/_upload/${session.getId}")
  15.  
  16. def getTemporaryFile(fileId: String)(implicit session: HttpSession): java.io.File =
  17. new java.io.File(TemporaryDir, fileId)
  18.  
  19. // def removeTemporaryFile(fileId: String)(implicit session: HttpSession): Unit =
  20. // getTemporaryFile(fileId).delete()
  21.  
  22. def removeTemporaryFiles()(implicit session: HttpSession): Unit =
  23. FileUtils.deleteDirectory(TemporaryDir)
  24.  
  25. def getUploadedFilename(fileId: String)(implicit session: HttpSession): Option[String] = {
  26. val filename = Option(session.getAttribute("upload_" + fileId).asInstanceOf[String])
  27. if(filename.isDefined){
  28. session.removeAttribute("upload_" + fileId)
  29. }
  30. filename
  31. }
  32.  
  33. }