com.netaporter.uri.encoding.UriEncoder.scala Maven / Gradle / Ivy

Go to download

package com.netaporter.uri.encoding

trait UriEncoder {
  def shouldEncode(ch: Char): Boolean
  def encodeChar(ch: Char): String

  def encode(s: String, charset: String) = {
    val chars = s.getBytes(charset).map(_.toChar)

    val encChars = chars.flatMap(ch => {
      if (shouldEncode(ch)) {
        encodeChar(ch).getBytes(charset)
      } else {
        Array(ch.toByte)
      }
    })

    new String(encChars, charset)
  }

  def +(other: UriEncoder) = ChainedUriEncoder(other :: this :: Nil)
}