de.lhns.fs2.compress.ArchiveSingleFile.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of fs2-compress_sjs1_2.13 Show documentation

fs2-compress

The newest version!

package de.lhns.fs2.compress

import fs2.{Pipe, Stream}

class ArchiveSingleFileCompressor[F[_], Size[A] <: Option[A]] private (
    archiver: Archiver[F, Size],
    entry: ArchiveEntry[Size, Any]
) extends Compressor[F] {
  override def compress: Pipe[F, Byte, Byte] = { stream =>
    Stream
      .emit((entry, stream))
      .through(archiver.archive)
  }
}

object ArchiveSingleFileCompressor {
  def apply[F[_], Size[A] <: Option[A]](
      archiver: Archiver[F, Size],
      entry: ArchiveEntry[Size, Any]
  ): ArchiveSingleFileCompressor[F, Size] =
    new ArchiveSingleFileCompressor(archiver, entry)

  def forName[F[_]](archiver: Archiver[F, Option], name: String): ArchiveSingleFileCompressor[F, Option] =
    new ArchiveSingleFileCompressor(archiver, ArchiveEntry(name))

  def forName[F[_]](archiver: Archiver[F, Some], name: String, size: Long): ArchiveSingleFileCompressor[F, Some] =
    new ArchiveSingleFileCompressor(archiver, ArchiveEntry(name, Some(size)))
}

class ArchiveSingleFileDecompressor[F[_], Size[A] <: Option[A], Underlying] private (
    unarchiver: Unarchiver[F, Size, Underlying]
) extends Decompressor[F] {
  override def decompress: Pipe[F, Byte, Byte] = { stream =>
    stream
      .through(unarchiver.unarchive)
      .flatMap {
        case (entry, s) if entry.isDirectory => s.drain
        case (_, s) => Stream.emit(s)
      }
      .head
      .flatten
  }
}

object ArchiveSingleFileDecompressor {
  def apply[F[_], Size[A] <: Option[A], Underlying](
      unarchiver: Unarchiver[F, Size, Underlying]
  ): ArchiveSingleFileDecompressor[F, Size, Underlying] =
    new ArchiveSingleFileDecompressor(unarchiver)
}