org.beangle.commons.text.escape.XmlEscaper.scala Maven / Gradle / Ivy

Go to download
/*
 * Copyright (C) 2005, The Beangle Software.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published
 * by the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program.  If not, see .
 */

package org.beangle.commons.text.escape

object XmlEscaper {
  private val targets = Map('<' -> "<", '>' -> ">")

  def escape(str: String): String = {
    val ln = str.length
    var sb: StringBuilder = null
    val keys = targets.keySet
    (0 until ln).find(i => keys.contains(str.charAt(i))) match {
      case Some(firstMatchIdx) =>
        val sb = new StringBuilder(str.substring(0, firstMatchIdx))
        (0 until ln) foreach { i =>
          val c = str.charAt(i)
          sb.append(targets.getOrElse(c, c))
        }
        sb.mkString
      case None => str
    }
  }
}