io.viper.common.ViperServer.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of io.viper.core Show documentation
A netty based HTTP server and toolkit.
The newest version!
package io.viper.common


import io.viper.core.server.file.StaticFileServerHandler
import org.jboss.netty.channel.{ChannelPipeline, ChannelPipelineFactory}
import java.util
import io.viper.core.server.router._
import collection.mutable.ListBuffer

class ViperServer(resourcePath: String) extends ChannelPipelineFactory with RestServer
{
  var resourceInstance: Class[_]  = this.getClass

  override def getPipeline: ChannelPipeline = {
    routes.clear()
    addRoutes
    addDefaultRoutes()
    buildPipeline
  }

  override def addRoutes {}

  private def addDefaultRoutes() {
    val provider = StaticFileContentInfoProviderFactory.create(resourceInstance, resourcePath)
    val handler = new StaticFileServerHandler(provider)
    get("/$path", handler)
    get("/", handler)
  }
}

object VirtualServer {
  def apply(hostname: String): VirtualServer = new VirtualServer(hostname)
  def apply(hostname: String, resourcePath: String) = new VirtualServer(hostname, resourcePath)
}

class VirtualServer(val hostname: String, resourcePath: String) extends ViperServer(resourcePath) with VirtualServerRunner {
  def this(hostname: String) {
    this(hostname, "res:///%s/".format(hostname))
  }

  private val initCode = new ListBuffer[() => Unit]

  override def addRoutes {
    for (proc <- initCode) proc()
  }

  def get(route: String, f:(util.Map[String, String]) => RouteResponse): VirtualServer = {
    initCode.append(() => { super.get(route)(f) })
    this
  }
  def put(route: String, f:(util.Map[String, String]) => RouteResponse): VirtualServer = {
    initCode.append(() => { super.put(route)(f) })
    this
  }
  def post(route: String, f:(util.Map[String, String]) => RouteResponse): VirtualServer = {
    initCode.append(() => { super.post(route)(f) })
    this
  }
  def delete(route: String, f:(util.Map[String, String]) => RouteResponse): VirtualServer = {
    initCode.append(() => { super.delete(route)(f) })
    this
  }

  def main(args: Array[String]) {
    val port = if (args.length > 0) args(0).toInt else 8080
    val hostRouterHandler = new HostRouterHandler
    if (port != 80) StaticFileContentInfoProviderFactory.enableCache(false)
    hostRouterHandler.putRoute("localhost", port, create)
    NestServer.create(port, hostRouterHandler)
    Thread.currentThread.join
  }

  def start {}
  def stop {}
  def create: ViperServer = new ViperServer(resourcePath)
}

trait VirtualServerRunner {
  def hostname: String
  def start
  def stop
  def create: ViperServer
}