xitrum.handler.outbound.OPTIONSResponse.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of xitrum_2.13 Show documentation

xitrum

The newest version!

package xitrum.handler.outbound

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

import xitrum.Config
import xitrum.etag.NotModified
import xitrum.handler.{AccessLog, HandlerEnv}

@Sharable
class OPTIONSResponse extends ChannelOutboundHandlerAdapter {
  override def write(ctx: ChannelHandlerContext, msg: Object, promise: ChannelPromise): Unit = {
    if (!msg.isInstanceOf[HandlerEnv]) {
      ctx.write(msg, promise)
      return
    }

    val env      = msg.asInstanceOf[HandlerEnv]
    val request  = env.request
    val response = env.response
    val pathInfo = env.pathInfo

    if (request.method == OPTIONS) {
      AccessLog.logOPTIONS(request)

      if (pathInfo == null) {
        // Static files/resources
        if (response.status != NOT_FOUND) response.setStatus(NO_CONTENT)
      } else {
        // Dynamic resources
        if (Config.routes.tryAllMethods(pathInfo).nonEmpty)
          response.setStatus(NO_CONTENT)
        else
          response.setStatus(NOT_FOUND)
      }

      NotModified.setClientCacheAggressively(response)
      response.content.clear()
    }

    ctx.write(env, promise)
  }
}