org.sisioh.trinity.example.ApplicationContext.scala Maven / Gradle / Ivy

Go to download
package org.sisioh.trinity.example

import com.twitter.util.FuturePool
import java.util.concurrent.Executors
import org.jboss.netty.handler.codec.http.HttpResponseStatus
import org.sisioh.trinity.application.TrinityApplication
import org.sisioh.trinity.domain.config.Config
import org.sisioh.trinity.domain.controller.GlobalSettings
import org.sisioh.trinity.domain.http.TrinityResponseBuilder
import org.sisioh.trinity.domain.routing.FutureAction
import org.sisioh.trinity.view.freemarker.FreeMarkerEngineContext
import org.sisioh.trinity.view.scalate.ScalateEngineContext
import org.sisioh.trinity.view.thymeleaf.ThymeleafEngineContext
import org.sisioh.trinity.view.velocity.VelocityEngineContext
import scala.concurrent.ExecutionContext

class UnauthorizedException extends Exception

trait ApplicationContext {

  implicit val config = Config.fromFile()

  val globalSettings = new GlobalSettings {
    override def error = Some {
      FutureAction {
        request =>
          request.error match {
            case Some(e: ArithmeticException) =>
              TrinityResponseBuilder().withStatus(HttpResponseStatus.valueOf(500)).withPlain("whoops, divide by zero!").toFinagleResponseFuture
            case Some(e: UnauthorizedException) =>
              TrinityResponseBuilder().withStatus(HttpResponseStatus.valueOf(401)).withPlain("Not Authorized!").toFinagleResponseFuture
            case Some(e) =>
              TrinityResponseBuilder().withStatus(HttpResponseStatus.valueOf(415)).withPlain("Unsupported Media Type!").toFinagleResponseFuture
            case _ =>
              TrinityResponseBuilder().withStatus(HttpResponseStatus.valueOf(500)).withPlain("Something went wrong!").toFinagleResponseFuture
          }
      }
    }

    override def notFound = Some {
      FutureAction {
        request =>
          TrinityResponseBuilder().withStatus(HttpResponseStatus.valueOf(404)).withPlain("not found yo").toFinagleResponseFuture
      }
    }

  }

  implicit val application = TrinityApplication(config, Some(globalSettings))

  // Thread Pool
  val threadPool = Executors.newFixedThreadPool(10)
  implicit val futurePool = FuturePool(threadPool)
  implicit val executor = ExecutionContext.fromExecutor(threadPool)

  implicit val scalate = ScalateEngineContext()
  implicit val thymeleaf = ThymeleafEngineContext()
  implicit val velocity = VelocityEngineContext()
  implicit val freemarker = FreeMarkerEngineContext(getClass)
}