xitrum.handler.inbound.WebJarsServer.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of xitrum_2.11 Show documentation
xitrum
There is a newer version: 3.28.18
package xitrum.handler.inbound

import scala.util.control.NonFatal

import io.netty.channel.{ChannelHandler, SimpleChannelInboundHandler, ChannelHandlerContext}
import io.netty.handler.codec.http.{HttpMethod, HttpResponseStatus}
import ChannelHandler.Sharable
import HttpMethod._
import HttpResponseStatus._

import xitrum.handler.HandlerEnv
import xitrum.handler.outbound.{XSendFile, XSendResource}
import xitrum.etag.NotModified
import xitrum.util.PathSanitizer

/**
 * Routes /webjars/xxx URL to resources in classpath: http://www.webjars.org/contributing
 * See DefaultHttpChannelInitializer, this handler is put after XSendResource.
 */
@Sharable
class WebJarsServer extends SimpleChannelInboundHandler[HandlerEnv] {
  override def channelRead0(ctx: ChannelHandlerContext, env: HandlerEnv) {
    val request = env.request
    if (request.method != GET && request.method != HEAD && request.method != OPTIONS) {
      ctx.fireChannelRead(env)
      return
    }

    val pathInfo = request.uri.split('?')(0)
    if (!pathInfo.startsWith("/webjars/")) {
      ctx.fireChannelRead(env)
      return
    }

    val response = env.response
    PathSanitizer.sanitize(pathInfo) match {
      case None =>
        XSendFile.set404Page(response, fromController = false)
        ctx.channel.writeAndFlush(env)

      case Some(path) =>
        val resourcePath = "META-INF/resources" + pathInfo

        // Check resource existence, null means the resource does not exist
        val is = Thread.currentThread.getContextClassLoader.getResourceAsStream(resourcePath)
        if (is == null) {
          // Not found, this may be dynamic path (action)
          ctx.fireChannelRead(env)
        } else {
          // Make sure that this is a file, not a directory
          // getResourceAsStream("dir/") => is.available() will be 0
          // getResourceAsStream("dir")  => is.available() will cause NullPointerException
          try {
            if (is.available() > 0) {
              response.setStatus(OK)
              NotModified.setClientCacheAggressively(response)
              XSendResource.setHeader(response, resourcePath, fromController = false)
              ctx.channel.writeAndFlush(env)
            } else {
              ctx.fireChannelRead(env)
            }
          } catch {
            case NonFatal(e) => ctx.fireChannelRead(env)
          } finally {
            // getResourceAsStream("dir") => is.close() will cause NullPointerException
            try { is.close() } catch { case NonFatal(e) => }
          }
        }
    }
  }
}