skinny.routing.RichRoute.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of skinny-framework_2.11 Show documentation

skinny-framework

There is a newer version: 2.1.2

Show newest version

package skinny.routing

import skinny.controller.feature.SkinnyControllerCommonBase
import skinny.controller.ActionDefinition
import skinny.micro.Handler
import skinny.micro.constant.HttpMethod
import skinny.micro.routing.Route
import skinny.exception.RouteMetadataException

/**
 * Route wrapper.
 *
 * @param route route
 * @param method HTTP method
 * @param controller controller
 */
case class RichRoute(route: Route, method: HttpMethod, controller: SkinnyControllerCommonBase) {

  /**
   * Registers action name to the controller.
   *
   * @param name action name
   * @return route
   */
  def as(name: Symbol): Route = {
    val expectedMethod = route.metadata.get(Handler.RouteMetadataHttpMethodCacheKey)
      .map(_.asInstanceOf[HttpMethod]).getOrElse {
        throw new RouteMetadataException(
          s"Metadata attribute ${Handler.RouteMetadataHttpMethodCacheKey} of Route should be set."
        )
      }
    controller.addActionDefinition(ActionDefinition(
      name = name,
      method = expectedMethod,
      matcher = (method: HttpMethod, path: String) => method == expectedMethod && route.apply(path).isDefined
    ))

    route
  }

}