org.scalatra.test.uploadables.scala Maven / Gradle / Ivy

Go to download
package org.scalatra.test

import java.io.{InputStream, FileInputStream, ByteArrayOutputStream, File}
import annotation.tailrec

abstract class Uploadable {
  def content: Array[Byte]
  def fileName: String
  def contentType: String
  def contentLength: Long
}

case class FilePart(
  file: File,
  contentType: String = "application/octet-stream")
  extends Uploadable
{

  lazy val content  = {
    val fin = new FileInputStream(file)
    val bos = new ByteArrayOutputStream()
    val buf = new Array[Byte](4096)

    @tailrec
    def copyStream() {
      val bytesRead = fin.read(buf)
      if (bytesRead > 0) {
        bos.write(buf, 0, bytesRead)
        copyStream()
      }
    }

    try {
      copyStream()
      bos.toByteArray
    } finally {
      fin.close()
    }
  }

  def contentLength = file.length()

  def fileName = file.getName
}

case class BytesPart(
  fileName: String,
  content: Array[Byte],
  contentType: String = "application/octet-stream")
  extends Uploadable
{

  def contentLength = content.length.toLong
}