rcumflex.circumflex-web.3.0-RC1.source-code.param.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of circumflex-web Show documentation

Lightweight framework for quick and robust application development using Scala programming language.

The newest version!

package pro.savant.circumflex
package web

import core._

import collection.Iterator
import collection.mutable.Map

/*!# Parameters

  The `param` object of is a convenient helper which is used to
  retrieve the parameters of current match or current request:

    * the parameters are first resolved from `MatchResult` objects found in context;

    * if no match result contain a parameter with specified name,
  then the result is searched in request parameters.

  In other words, match results always override request parameters.
  */
object param extends Map[String, String] with KeyValueCoercion {

  def +=(kv: (String, String)): this.type = this

  def -=(key: String): this.type = this

  def iterator: Iterator[(String, String)] = ctx.iterator.flatMap(p => p._2 match {
    case m: MatchResult => m.params.iterator
    case s: String => Seq(p._1 -> s).iterator
    case _ => Iterator.empty
  }) ++ requestOption.toSeq.flatMap(_.params).iterator

  def get(key: String): Option[String] = iterator.find(_._1 == key).map(_._2)

  override def default(key: String): String = ""

  /*! Use the `list` method to retrieve multi-value parameters from request. */
  def list(key: String): Seq[String] = iterator.filter(_._1 == key).map(_._2).toList
}

object paramList extends Map[String, Seq[String]] {

  def +=(kv: (String, Seq[String])) = this

  def -=(key: String) = this

  def get(key: String) = requestOption
      .map(r => r.params.list(key))
      .orElse(Some(Nil))

  def iterator = requestOption
      .map(r => r.params.keys.map(k => k -> r.params.list(k)))
      .getOrElse(Nil)
      .iterator

}