scalariform.lexer.XmlLexer.scala Maven / Gradle / Ivy
The newest version!
package scalariform.lexer
import scala.annotation.{ switch, tailrec }
import scalariform.lexer.Tokens._
import scalariform.utils.Utils
trait XmlLexer extends Lexer {
import ScalaLexer._
import CharConstants._
private def xmlMode: XmlMode = modeStack.head.asInstanceOf[XmlMode]
abstract sealed trait TagState
case object InStartTag extends TagState
case object InEndTag extends TagState
case object Normal extends TagState
private def tagMode = xmlMode.isTagMode
private def tagMode_=(isTagMode: Boolean) {
xmlMode.isTagMode = isTagMode
}
protected def isXmlMode = modeStack.head.isInstanceOf[XmlMode]
class XmlMode(private var tagNestLevel: Int = 0, var isTagMode: Boolean = false, var tagState: TagState = Normal) extends LexerMode {
def nestTag() { tagNestLevel += 1 }
def unnestTag(): Int = {
tagNestLevel -= 1
tagNestLevel
}
def nestingLevel = tagNestLevel
}
private def moreXmlToCome: Boolean = {
var offset = 0
while (ch(offset) != SU && isSpace(ch(offset)))
offset += 1
ch(offset) == '<' && isNameStart(ch(offset + 1))
}
protected def fetchXmlToken() {
(ch: @switch) match {
case '<' ⇒ {
if (ch(1) == '/') {
nextChar()
nextChar()
token(XML_END_OPEN)
xmlMode.isTagMode = true
xmlMode.tagState = InEndTag
} else if (ch(1) == '!') {
if (ch(2) == '-') {
getXmlComment()
if (xmlMode.nestingLevel == 0 && !moreXmlToCome)
modeStack.pop()
} else if (ch(2) == '[') {
getXmlCDATA()
if (xmlMode.nestingLevel == 0 && !moreXmlToCome)
modeStack.pop()
} else {
if (forgiveLexerErrors) {
munch("') {
nextChar()
nextChar()
token(XML_EMPTY_CLOSE)
xmlMode.isTagMode = false
xmlMode.tagState = Normal
if (xmlMode.nestingLevel == 0 && !moreXmlToCome)
modeStack.pop()
} else
getXmlCharData()
} else
getXmlCharData()
}
case '>' ⇒
if (tagMode) {
nextChar()
token(XML_TAG_CLOSE)
xmlMode.isTagMode = false
xmlMode.tagState match {
case InStartTag ⇒ xmlMode.nestTag()
case InEndTag ⇒ {
val nestingLevel = xmlMode.unnestTag()
if (nestingLevel == 0 && !moreXmlToCome)
modeStack.pop()
}
case Normal ⇒ throw new AssertionError("shouldn't reach here")
}
} else
getXmlCharData()
case '=' ⇒
if (tagMode) {
nextChar()
token(XML_ATTR_EQ)
} else {
getXmlCharData()
}
case '\'' ⇒
if (tagMode) {
getXmlAttributeValue('\'')
} else {
getXmlCharData()
}
case '"' ⇒
if (tagMode) {
getXmlAttributeValue('"')
} else {
getXmlCharData()
}
case '{' ⇒
if (ch(1) != '{')
switchToScalaModeAndFetchToken
else
getXmlCharData() // TODO: tagMode?
case SU ⇒ token(EOF)
case _ ⇒
if (tagMode && isNameStart(ch)) {
getXmlName()
} else if (tagMode && isSpace(ch)) {
getXmlSpace()
} else {
getXmlCharData()
// throw new ScalaLexerException("illegal character in xml: " + Character.valueOf(ch))
// TODO
}
}
}
private def getXmlCDATA() {
munch("")) {
munch("]]>")
continue = false
} else if (ch == SU)
if (forgiveLexerErrors) continue = false else throw new ScalaLexerException("Malformed XML CDATA")
else
nextChar()
}
token(XML_CDATA)
}
private def getXmlComment() {
munch("