de.sciss.freesound.lucre.impl.PreviewsCacheImpl.scala Maven / Gradle / Ivy

Go to download
/*
 *  PreviewsCacheImpl.scala
 *  (ScalaFreesound)
 *
 *  Copyright (c) 2010-2020 Hanns Holger Rutz. All rights reserved.
 *
 *	This software is published under the GNU Affero General Public License v3+
 *
 *
 *  For further information, please contact Hanns Holger Rutz at
 *  [email protected]
 */

package de.sciss.freesound.lucre.impl

import java.net.URI

import de.sciss.file._
import de.sciss.filecache
import de.sciss.filecache.{TxnConsumer, TxnProducer}
import de.sciss.freesound.lucre.PreviewsCache
import de.sciss.freesound.{Client, Freesound, Sound}
import de.sciss.lucre.TxnLike
import de.sciss.lucre.Txn.peer
import de.sciss.serial.{DataInput, DataOutput, ConstFormat}

import scala.concurrent.{ExecutionContext, Future}

object PreviewsCacheImpl {
  implicit private object uriFormat extends ConstFormat[URI] {
    def read (in         : DataInput ): URI  = new URI(in.readUTF())
    def write(v: URI, out: DataOutput): Unit = out.writeUTF(v.toString)
  }

  def apply(dir: File, capacity: filecache.Limit)(implicit tx: TxnLike, client: Client): PreviewsCache = {
    val config            = filecache.Config[URI, File]()
    config.capacity       = capacity
    config.evict          = (_ /* uri */, f) => if (!f.delete()) f.deleteOnExit()
    config.space          = (_ /* uri */, f) => f.length()
    config.accept         = (_ /* uri */, f) => f.length() > 0L
    config.folder         = dir
    config.fileExtension  = "freesound"
    config.folder.mkdirs()

    val prod: TxnProducer[URI, File] = TxnProducer(config)
    val cons = TxnConsumer(prod) { uri =>
      val uriS    = uri.getPath
      val name    = uriS.substring(uriS.lastIndexOf('/') + 1)
      val out     = config.folder / name
      val proc    = Freesound.downloadUriToFile(uri, out = out)
      implicit val exec: ExecutionContext = config.executionContext
      proc.transform[File]((_: Unit) => out, { e: Throwable => config.evict(uri, out); e })
    }
    new Impl(cons, config.executionContext)
  }

  private final class Impl(cons: TxnConsumer[URI, File], val executionContext: ExecutionContext)
    extends PreviewsCache {

    private def key(sound: Sound): URI = sound.previewUri(ogg = true, hq = false)

    def acquire(sound: Sound)(implicit tx: TxnLike): Future[File] =
      cons.acquire(key(sound))

    def release(sound: Sound)(implicit tx: TxnLike): Unit =
      cons.release(key(sound))
  }
}