sttp.client4.FileHelpers.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of core_native0.4_3 Show documentation

core

The newest version!

package sttp.client4

import sttp.client4.internal._

import java.io.{File, FileOutputStream, IOException, InputStream}
import java.nio.file.AccessDeniedException

object FileHelpers {
  private[client4] def saveFile(file: File, is: InputStream): File = {
    if (!file.exists()) {
      if (file.getParentFile != null) {
        file.getParentFile.mkdirs()
      }
      try
        file.createNewFile()
      catch {
        case e: AccessDeniedException => throw new IOException("Permission denied", e) // aligns SN bahavior with Java
      }
    }

    val os = new FileOutputStream(file)

    transfer(is, os)
    file
  }

  private[client4] def getFilePath[T, R](response: GenericResponseAs[T, R]): Option[SttpFile] =
    response match {
      case MappedResponseAs(raw, g, _) => getFilePath(raw)
      case rfm: ResponseAsFromMetadata[T, _] =>
        rfm.conditions
          .flatMap(c => getFilePath(c.responseAs))
          .headOption
      case ResponseAsFile(file) => Some(file)
      case _                    => None
    }
}