io.finch.endpoint.endpoint.scala Maven / Gradle / Ivy

Go to download
package io.finch
package endpoint

import cats.Applicative
import cats.effect.{Resource, Sync}
import cats.syntax.all._
import com.twitter.finagle.http.{Method => FinagleMethod}
import com.twitter.io.Buf
import shapeless.HNil

import java.io.InputStream

private[finch] class FromInputStream[F[_]](stream: Resource[F, InputStream])(implicit F: Sync[F]) extends Endpoint[F, Buf] {
  private def readLoop(left: Buf, stream: InputStream): F[Buf] = F.defer {
    for {
      buffer <- F.delay(new Array[Byte](1024))
      n <- F.blocking(stream.read(buffer))
      buf <- if (n == -1) F.pure(left) else readLoop(left.concat(Buf.ByteArray.Owned(buffer, 0, n)), stream)
    } yield buf
  }

  final def apply(input: Input): Endpoint.Result[F, Buf] =
    EndpointResult.Matched(
      input,
      Trace.empty,
      stream.use(s => readLoop(Buf.Empty, s).map(buf => Output.payload(buf)))
    )
}

private[finch] class Asset[F[_]](path: String)(implicit F: Applicative[F]) extends Endpoint[F, HNil] {
  final def apply(input: Input): Endpoint.Result[F, HNil] = {
    val req = input.request
    if (req.method != FinagleMethod.Get || req.path != path) EndpointResult.NotMatched[F]
    else
      EndpointResult.Matched(
        input.withRoute(Nil),
        Trace.fromRoute(input.route),
        F.pure(Output.HNil)
      )
  }

  final override def toString: String = s"GET /$path"
}