package cask.model

import java.io.{InputStream, OutputStream, OutputStreamWriter}

import cask.internal.Util


trait Response{
  def data: Response.Data
  def statusCode: Int
  def headers: Seq[(String, String)]
  def cookies: Seq[Cookie]
}
object Response{
  def apply(data: Data,
            statusCode: Int = 200,
            headers: Seq[(String, String)] = Nil,
            cookies: Seq[Cookie] = Nil) = Simple(data, statusCode, headers, cookies)
  case class Simple(data: Data,
                    statusCode: Int = 200,
                    headers: Seq[(String, String)] = Nil,
                    cookies: Seq[Cookie] = Nil) extends Response

  implicit def dataResponse[T](t: T)(implicit c: T => Data) = Response(t)
  trait Data{
    def write(out: OutputStream): Unit
  }
  object Data{
    implicit class StringData(s: String) extends Data{
      def write(out: OutputStream) = out.write(s.getBytes)
    }
    implicit class BytesData(b: Array[Byte]) extends Data{
      def write(out: OutputStream) = out.write(b)
    }
    implicit class StreamData(b: InputStream) extends Data{
      def write(out: OutputStream) = Util.transferTo(b, out)
    }
    implicit def JsonResponse[T: upickle.default.Writer](t: T) = new Data{
      def write(out: OutputStream) = implicitly[upickle.default.Writer[T]].write(
        new ujson.BaseRenderer(new OutputStreamWriter(out)),
        t
      )
    }
  }
}
case class Redirect(url: String)  extends Response{
  override def data = ""

  override def statusCode = 301

  override def headers = Seq("Location" -> url)

  override def cookies = Nil
}
case class Abort(code: Int) extends Response {
  override def data = ""

  override def statusCode = code

  override def headers = Nil

  override def cookies = Nil
}

case class StaticFile(path: String) extends Response {
  val relPath = java.nio.file.Paths.get(path)
  val (data0, statusCode0) =
    if (java.nio.file.Files.exists(relPath) && java.nio.file.Files.isRegularFile(relPath)){
      (java.nio.file.Files.newInputStream(relPath): Response.Data, 200)
    }else{
      ("": Response.Data, 404)
    }
  override def data = data0

  override def statusCode = statusCode0

  override def headers = Nil

  override def cookies = Nil
}

case class StaticResource(path: String, resourceRoot: ClassLoader) extends Response {
  val relPath = java.nio.file.Paths.get(path)
  val (data0, statusCode0) = resourceRoot.getResourceAsStream(path) match{
    case null => ("": Response.Data, 404)
    case res => (res: Response.Data, 200)
  }

  override def data = data0

  override def statusCode = statusCode0

  override def headers = Nil

  override def cookies = Nil
}