DisseminationCopyHandler.scala

/*
 * Media Converter
 * Extracts media files from Fedora repository
 * Copyright (C) 2020  Memoriav
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */

package ch.memobase

import java.io.{ByteArrayOutputStream, FileOutputStream, IOException}
import java.nio.file.{Files, Path, Paths}

import ch.memobase.models.{Conversions, MediaFileType}

import scala.util.{Failure, Success, Try}

/**
 * Manages dissemination copies of media files
 *
 * @param audioDestPath Path to folder containing audio dissemination copies
 * @param imageDestPath Path to folder containing image dissemination copies
 * @param videoDestPath Path to folder containing video dissemination copies
 */
class DisseminationCopyHandler(audioDestPath: String, imageDestPath: String, videoDestPath: String, audioSnippetDuration: Int) {

  private def writeData(data: ByteArrayOutputStream, destFile: Path): Try[Path] = {
    Try(new FileOutputStream(destFile.toFile)) match {
      case Success(fos) =>
        try {
          data.writeTo(fos)
          Success(destFile)
        } catch {
          case e: IOException => Failure(e)
        } finally {
          fos.close()
          data.close()
        }
      case Failure(ex) => Failure(ex)
    }
  }


  /**
   * Creates dissemination copy of audio file
   *
   * @param data           binary data as [[java.io.ByteArrayOutputStream]] instance
   * @param destId         Filename of dissemination copy without extension
   * @param sourceFileType File type of input data
   * @return
   */
  def createAudioCopy(data: ByteArrayOutputStream, destId: String, sourceFileType: MediaFileType): Try[Path] = Try {
    val tempFilePath = Files.createTempFile("media-", "." + Conversions.getFileTypeExtension(sourceFileType).get)
    val destFile = Paths.get(audioDestPath, destId + ".mp4")
    val snippetFile = Paths.get(audioDestPath, destId + "-intro." + Conversions.getFileTypeExtension(sourceFileType).get)
    writeData(data, tempFilePath)
    MediaTransformations.audioToMp4(tempFilePath.toString, destFile.toString).get
    MediaTransformations.createAudioSnippet(tempFilePath.toString, snippetFile.toString, audioSnippetDuration)
    Files.delete(tempFilePath)
    destFile
  }

  /**
   * Creates dissemination copy of image file
   *
   * @param data           binary data as [[java.io.ByteArrayOutputStream]] instance
   * @param destId         Filename of dissemination copy without extension
   * @param sourceFileType File type of input data
   * @return
   */
  def createImageCopy(data: ByteArrayOutputStream, destId: String, sourceFileType: MediaFileType): Try[Path] = Try {
    val tempFilePath = Files.createTempFile("media-", "." + Conversions.getFileTypeExtension(sourceFileType).get)
    val destFile = Paths.get(imageDestPath, destId + ".jp2")
    writeData(data, tempFilePath)
    MediaTransformations.imageToJp2(tempFilePath.toString, destFile.toString).get
    Files.delete(tempFilePath)
    destFile
  }

  /**
   * Create dissemination copy of video file
   *
   * @param data           binary data as [[java.io.ByteArrayOutputStream]] instance
   * @param destId         Filename of dissemination copy without extension
   * @param sourceFileType File type of input data
   * @return
   */
  def createVideoCopy(data: ByteArrayOutputStream, destId: String, sourceFileType: MediaFileType): Try[Path] = {
    val destFile = Paths.get(videoDestPath, s"$destId.${Conversions.getFileTypeExtension(sourceFileType).get}")
    writeData(data, destFile)
  }

  /**
   * Deletes dissemination copy of audio file
   *
   * @param destId Filename of dissemination copy without extension
   * @return
   */
  def deleteAudioCopy(destId: String): Try[Boolean] =
    Try {
      Paths.get(audioDestPath, destId + ".mp4").toFile.delete()
      Paths.get(audioDestPath, destId + "-intro.mp3").toFile.delete()
    }

  /**
   * Deletes dissemination copy of image file
   *
   * @param destId Filename of dissemination copy without extension
   * @return
   */
  def deleteImageCopy(destId: String): Try[Boolean] =
    Try {
      Paths.get(imageDestPath, destId + ".jp2").toFile.delete()
    }

  /**
   * Deletes dissemination copy of video file
   *
   * @param destId   Filename of dissemination copy without extension
   * @param fileType File type of media
   * @return
   */
  def deleteVideoCopy(destId: String, fileType: MediaFileType): Try[Boolean] =
    Try {
      Paths.get(videoDestPath, s"$destId.${Conversions.getFileTypeExtension(fileType).get}").toFile.delete()
    }
}