mirror of https://github.com/status-im/NimYAML.git
491 lines
20 KiB
Nim
491 lines
20 KiB
Nim
# NimYAML - YAML implementation in Nim
|
|
# (c) Copyright 2015 Felix Krause
|
|
#
|
|
# See the file "copying.txt", included in this
|
|
# distribution, for details about the copyright.
|
|
|
|
type
|
|
DumperState = enum
|
|
dBlockExplicitMapKey, dBlockImplicitMapKey, dBlockMapValue,
|
|
dBlockInlineMap, dBlockSequenceItem, dFlowImplicitMapKey, dFlowMapValue,
|
|
dFlowExplicitMapKey, dFlowSequenceItem, dFlowMapStart,
|
|
dFlowSequenceStart
|
|
|
|
proc needsEscaping(scalar: string): bool {.raises: [].} =
|
|
scalar.len == 0 or
|
|
scalar.find({'{', '}', '[', ']', ',', '#', '-', ':', '?', '%', '"',
|
|
'\'', '\x0A', '\c'}) != -1
|
|
|
|
proc writeDoubleQuoted(scalar: string, s: Stream)
|
|
{.raises: [YamlPresenterOutputError].} =
|
|
try:
|
|
s.write('"')
|
|
for c in scalar:
|
|
if c == '"':
|
|
s.write('\\')
|
|
s.write(c)
|
|
s.write('"')
|
|
except:
|
|
var e = newException(YamlPresenterOutputError, "")
|
|
e.cause = getCurrentException()
|
|
raise e
|
|
|
|
template safeWrite(s: string or char) {.dirty.} =
|
|
try:
|
|
target.write(s)
|
|
except:
|
|
var e = newException(YamlPresenterOutputError, "")
|
|
e.cause = getCurrentException()
|
|
raise e
|
|
|
|
proc startItem(target: Stream, style: YamlPresentationStyle, indentation: int,
|
|
state: var DumperState, isObject: bool)
|
|
{.raises: [YamlPresenterOutputError].} =
|
|
try:
|
|
case state
|
|
of dBlockMapValue:
|
|
target.write('\x0A')
|
|
target.write(repeat(' ', indentation))
|
|
if isObject or style == ypsCanonical:
|
|
target.write("? ")
|
|
state = dBlockExplicitMapKey
|
|
else:
|
|
state = dBlockImplicitMapKey
|
|
of dBlockInlineMap:
|
|
state = dBlockImplicitMapKey
|
|
of dBlockExplicitMapKey:
|
|
target.write('\x0A')
|
|
target.write(repeat(' ', indentation))
|
|
target.write(": ")
|
|
state = dBlockMapValue
|
|
of dBlockImplicitMapKey:
|
|
target.write(": ")
|
|
state = dBlockMapValue
|
|
of dFlowExplicitMapKey:
|
|
if style != ypsMinimal:
|
|
target.write('\x0A')
|
|
target.write(repeat(' ', indentation))
|
|
target.write(": ")
|
|
state = dFlowMapValue
|
|
of dFlowMapValue:
|
|
if (isObject and style != ypsMinimal) or
|
|
style in [ypsJson, ypsCanonical]:
|
|
target.write(",\x0A" & repeat(' ', indentation))
|
|
if style == ypsJson:
|
|
state = dFlowImplicitMapKey
|
|
else:
|
|
target.write("? ")
|
|
state = dFlowExplicitMapKey
|
|
elif isObject and style == ypsMinimal:
|
|
target.write(", ? ")
|
|
state = dFlowExplicitMapKey
|
|
else:
|
|
target.write(", ")
|
|
state = dFlowImplicitMapKey
|
|
of dFlowMapStart:
|
|
if (isObject and style != ypsMinimal) or
|
|
style in [ypsJson, ypsCanonical]:
|
|
target.write("\x0A" & repeat(' ', indentation))
|
|
if style == ypsJson:
|
|
state = dFlowImplicitMapKey
|
|
else:
|
|
target.write("? ")
|
|
state = dFlowExplicitMapKey
|
|
else:
|
|
state = dFlowImplicitMapKey
|
|
of dFlowImplicitMapKey:
|
|
target.write(": ")
|
|
state = dFlowMapValue
|
|
of dBlockSequenceItem:
|
|
target.write('\x0A')
|
|
target.write(repeat(' ', indentation))
|
|
target.write("- ")
|
|
of dFlowSequenceStart:
|
|
case style
|
|
of ypsMinimal, ypsDefault:
|
|
discard
|
|
of ypsCanonical, ypsJson:
|
|
target.write('\x0A')
|
|
target.write(repeat(' ', indentation))
|
|
of ypsBlockOnly:
|
|
discard # can never happen
|
|
state = dFlowSequenceItem
|
|
of dFlowSequenceItem:
|
|
case style
|
|
of ypsMinimal, ypsDefault:
|
|
target.write(", ")
|
|
of ypsCanonical, ypsJson:
|
|
target.write(",\x0A")
|
|
target.write(repeat(' ', indentation))
|
|
of ypsBlockOnly:
|
|
discard # can never happen
|
|
except:
|
|
var e = newException(YamlPresenterOutputError, "")
|
|
e.cause = getCurrentException()
|
|
raise e
|
|
|
|
proc writeTagAndAnchor(target: Stream, tag: TagId, tagLib: YamlTagLibrary,
|
|
anchor: AnchorId) {.raises:[YamlPresenterOutputError].} =
|
|
try:
|
|
if tag notin [yTagQuestionMark, yTagExclamationMark]:
|
|
let tagUri = tagLib.uri(tag)
|
|
if tagUri.startsWith(tagLib.secondaryPrefix):
|
|
target.write("!!")
|
|
target.write(tagUri[18..^1])
|
|
target.write(' ')
|
|
elif tagUri.startsWith("!"):
|
|
target.write(tagUri)
|
|
target.write(' ')
|
|
else:
|
|
target.write("!<")
|
|
target.write(tagUri)
|
|
target.write("> ")
|
|
if anchor != yAnchorNone:
|
|
target.write("&")
|
|
# TODO: properly select an anchor
|
|
target.write(cast[byte]('a') + cast[byte](anchor))
|
|
target.write(' ')
|
|
except:
|
|
var e = newException(YamlPresenterOutputError, "")
|
|
e.cause = getCurrentException()
|
|
raise e
|
|
|
|
proc present*(s: YamlStream, target: Stream, tagLib: YamlTagLibrary,
|
|
style: YamlPresentationStyle = ypsDefault,
|
|
indentationStep: int = 2) =
|
|
var
|
|
cached = initQueue[YamlStreamEvent]()
|
|
cacheIterator = iterator(): YamlStreamEvent =
|
|
while true:
|
|
while cached.len > 0:
|
|
yield cached.dequeue()
|
|
try:
|
|
let item = s()
|
|
if finished(s):
|
|
break
|
|
cached.enqueue(item)
|
|
except:
|
|
var e = newException(YamlPresenterStreamError, "")
|
|
e.cause = getCurrentException()
|
|
raise e
|
|
indentation = 0
|
|
levels = newSeq[DumperState]()
|
|
|
|
for item in cacheIterator():
|
|
case item.kind
|
|
of yamlStartDocument:
|
|
if style != ypsJson:
|
|
# TODO: tag directives
|
|
try:
|
|
target.write("%YAML 1.2\x0A")
|
|
if tagLib.secondaryPrefix != yamlTagRepositoryPrefix:
|
|
target.write("%TAG !! " &
|
|
tagLib.secondaryPrefix & '\x0A')
|
|
target.write("--- ")
|
|
except:
|
|
var e = newException(YamlPresenterOutputError, "")
|
|
e.cause = getCurrentException()
|
|
raise e
|
|
of yamlScalar:
|
|
if levels.len == 0:
|
|
if style != ypsJson:
|
|
safeWrite('\x0A')
|
|
else:
|
|
startItem(target, style, indentation, levels[levels.high],
|
|
false)
|
|
if style != ypsJson:
|
|
writeTagAndAnchor(target,
|
|
item.scalarTag, tagLib, item.scalarAnchor)
|
|
|
|
if style == ypsJson:
|
|
if item.scalarTag in [yTagQuestionMark, yTagBoolean] and
|
|
item.scalarType in [yTypeBoolTrue, yTypeBoolFalse]:
|
|
if item.scalarType == yTypeBoolTrue:
|
|
safeWrite("true")
|
|
else:
|
|
safeWrite("false")
|
|
elif item.scalarTag in [yTagQuestionMark, yTagNull] and
|
|
item.scalarType == yTypeNull:
|
|
safeWrite("null")
|
|
elif item.scalarTag in [yTagQuestionMark, yTagFloat] and
|
|
item.scalarType in [yTypeFloatInf, yTypeFloatNaN]:
|
|
raise newException(YamlPresenterJsonError,
|
|
"Infinity and not-a-number values cannot be presented as JSON!")
|
|
else:
|
|
writeDoubleQuoted(item.scalarContent, target)
|
|
elif style == ypsCanonical or item.scalarContent.needsEscaping:
|
|
writeDoubleQuoted(item.scalarContent, target)
|
|
else:
|
|
safeWrite(item.scalarContent)
|
|
of yamlAlias:
|
|
if style == ypsJson:
|
|
raise newException(YamlPresenterJsonError,
|
|
"Alias not allowed in JSON output")
|
|
assert levels.len > 0
|
|
startItem(target, style, indentation, levels[levels.high], false)
|
|
try:
|
|
target.write('*')
|
|
target.write(cast[byte]('a') + cast[byte](item.aliasTarget))
|
|
except:
|
|
var e = newException(YamlPresenterOutputError, "")
|
|
e.cause = getCurrentException()
|
|
raise e
|
|
of yamlStartSequence:
|
|
var nextState: DumperState
|
|
case style
|
|
of ypsDefault:
|
|
var length = 0
|
|
while true:
|
|
try:
|
|
let next = s()
|
|
assert (not finished(s))
|
|
cached.enqueue(next)
|
|
case next.kind
|
|
of yamlScalar:
|
|
length += 2 + next.scalarContent.len
|
|
of yamlAlias:
|
|
length += 6
|
|
of yamlEndSequence:
|
|
break
|
|
else:
|
|
length = int.high
|
|
break
|
|
except:
|
|
var e = newException(YamlPresenterStreamError, "")
|
|
e.cause = getCurrentException()
|
|
raise e
|
|
nextState = if length <= 60: dFlowSequenceStart else:
|
|
dBlockSequenceItem
|
|
of ypsJson:
|
|
if levels.len > 0 and levels[levels.high] in
|
|
[dFlowMapStart, dFlowMapValue]:
|
|
raise newException(YamlPresenterJsonError,
|
|
"Cannot have sequence as map key in JSON output!")
|
|
nextState = dFlowSequenceStart
|
|
of ypsMinimal, ypsCanonical:
|
|
nextState = dFlowSequenceStart
|
|
of ypsBlockOnly:
|
|
nextState = dBlockSequenceItem
|
|
|
|
if levels.len == 0:
|
|
if nextState == dBlockSequenceItem:
|
|
if style != ypsJson:
|
|
writeTagAndAnchor(target,
|
|
item.seqTag, tagLib, item.seqAnchor)
|
|
else:
|
|
if style != ypsJson:
|
|
writeTagAndAnchor(target,
|
|
item.seqTag, tagLib, item.seqAnchor)
|
|
safeWrite('\x0A')
|
|
indentation += indentationStep
|
|
else:
|
|
startItem(target, style, indentation, levels[levels.high], true)
|
|
if style != ypsJson:
|
|
writeTagAndAnchor(target,
|
|
item.seqTag, tagLib, item.seqAnchor)
|
|
indentation += indentationStep
|
|
|
|
if nextState == dFlowSequenceStart:
|
|
safeWrite('[')
|
|
if levels.len > 0 and style in [ypsJson, ypsCanonical] and
|
|
levels[levels.high] in
|
|
[dBlockExplicitMapKey, dBlockMapValue,
|
|
dBlockImplicitMapKey, dBlockSequenceItem]:
|
|
indentation += indentationStep
|
|
levels.add(nextState)
|
|
of yamlStartMap:
|
|
var nextState: DumperState
|
|
case style
|
|
of ypsDefault:
|
|
type mapParseState = enum
|
|
mpInitial, mpKey, mpValue, mpNeedBlock
|
|
var mps = mpInitial
|
|
while mps != mpNeedBlock:
|
|
try:
|
|
let next = s()
|
|
assert (not finished(s))
|
|
cached.enqueue(next)
|
|
case next.kind
|
|
of yamlScalar, yamlAlias:
|
|
case mps
|
|
of mpInitial: mps = mpKey
|
|
of mpKey: mps = mpValue
|
|
else: mps = mpNeedBlock
|
|
of yamlEndMap:
|
|
break
|
|
else:
|
|
mps = mpNeedBlock
|
|
except:
|
|
var e = newException(YamlPresenterStreamError, "")
|
|
e.cause = getCurrentException()
|
|
raise e
|
|
nextState = if mps == mpNeedBlock: dBlockMapValue else:
|
|
dBlockInlineMap
|
|
of ypsMinimal:
|
|
nextState = dFlowMapStart
|
|
of ypsCanonical:
|
|
nextState = dFlowMapStart
|
|
of ypsJson:
|
|
if levels.len > 0 and levels[levels.high] in
|
|
[dFlowMapStart, dFlowMapValue]:
|
|
raise newException(YamlPresenterJsonError,
|
|
"Cannot have map as map key in JSON output!")
|
|
nextState = dFlowMapStart
|
|
of ypsBlockOnly:
|
|
nextState = dBlockMapValue
|
|
if levels.len == 0:
|
|
if nextState == dBlockMapValue:
|
|
if style != ypsJson:
|
|
writeTagAndAnchor(target,
|
|
item.mapTag, tagLib, item.mapAnchor)
|
|
else:
|
|
if style != ypsJson:
|
|
safeWrite('\x0A')
|
|
writeTagAndAnchor(target,
|
|
item.mapTag, tagLib, item.mapAnchor)
|
|
indentation += indentationStep
|
|
else:
|
|
if nextState in [dBlockMapValue, dBlockImplicitMapKey]:
|
|
if style != ypsJson:
|
|
writeTagAndAnchor(target,
|
|
item.mapTag, tagLib, item.mapAnchor)
|
|
startItem(target, style, indentation, levels[levels.high],
|
|
true)
|
|
else:
|
|
startItem(target, style, indentation, levels[levels.high],
|
|
true)
|
|
if style != ypsJson:
|
|
writeTagAndAnchor(target,
|
|
item.mapTag, tagLib, item.mapAnchor)
|
|
indentation += indentationStep
|
|
|
|
if nextState == dFlowMapStart:
|
|
safeWrite('{')
|
|
if levels.len > 0 and style in [ypsJson, ypsCanonical] and
|
|
levels[levels.high] in
|
|
[dBlockExplicitMapKey, dBlockMapValue,
|
|
dBlockImplicitMapKey, dBlockSequenceItem]:
|
|
indentation += indentationStep
|
|
levels.add(nextState)
|
|
|
|
of yamlEndSequence:
|
|
assert levels.len > 0
|
|
case levels.pop()
|
|
of dFlowSequenceItem:
|
|
case style
|
|
of ypsDefault, ypsMinimal, ypsBlockOnly:
|
|
safeWrite(']')
|
|
of ypsJson, ypsCanonical:
|
|
indentation -= indentationStep
|
|
try:
|
|
target.write('\x0A')
|
|
target.write(repeat(' ', indentation))
|
|
target.write(']')
|
|
except:
|
|
var e = newException(YamlPresenterOutputError, "")
|
|
e.cause = getCurrentException()
|
|
raise e
|
|
if levels.len == 0 or levels[levels.high] notin
|
|
[dBlockExplicitMapKey, dBlockMapValue,
|
|
dBlockImplicitMapKey, dBlockSequenceItem]:
|
|
continue
|
|
of dFlowSequenceStart:
|
|
if levels.len > 0 and style in [ypsJson, ypsCanonical] and
|
|
levels[levels.high] in
|
|
[dBlockExplicitMapKey, dBlockMapValue,
|
|
dBlockImplicitMapKey, dBlockSequenceItem]:
|
|
indentation -= indentationStep
|
|
safeWrite(']')
|
|
of dBlockSequenceItem:
|
|
discard
|
|
else:
|
|
assert false
|
|
indentation -= indentationStep
|
|
of yamlEndMap:
|
|
assert levels.len > 0
|
|
let level = levels.pop()
|
|
case level
|
|
of dFlowMapValue:
|
|
case style
|
|
of ypsDefault, ypsMinimal, ypsBlockOnly:
|
|
safeWrite('}')
|
|
of ypsJson, ypsCanonical:
|
|
indentation -= indentationStep
|
|
try:
|
|
target.write('\x0A')
|
|
target.write(repeat(' ', indentation))
|
|
target.write('}')
|
|
except:
|
|
var e = newException(YamlPresenterOutputError, "")
|
|
e.cause = getCurrentException()
|
|
raise e
|
|
if levels.len == 0 or levels[levels.high] notin
|
|
[dBlockExplicitMapKey, dBlockMapValue,
|
|
dBlockImplicitMapKey, dBlockSequenceItem]:
|
|
continue
|
|
of dFlowMapStart:
|
|
if levels.len > 0 and style in [ypsJson, ypsCanonical] and
|
|
levels[levels.high] in
|
|
[dBlockExplicitMapKey, dBlockMapValue,
|
|
dBlockImplicitMapKey, dBlockSequenceItem]:
|
|
indentation -= indentationStep
|
|
safeWrite('}')
|
|
of dBlockMapValue, dBlockInlineMap:
|
|
discard
|
|
else:
|
|
assert false
|
|
indentation -= indentationStep
|
|
of yamlEndDocument:
|
|
try:
|
|
let next = s()
|
|
if finished(s):
|
|
break
|
|
cached.enqueue(next)
|
|
except:
|
|
var e = newException(YamlPresenterStreamError, "")
|
|
e.cause = getCurrentException()
|
|
raise e
|
|
safeWrite("...\x0A")
|
|
|
|
proc transform*(input: Stream, output: Stream, style: YamlPresentationStyle,
|
|
indentationStep: int = 2) =
|
|
var
|
|
tagLib = extendedTagLibrary()
|
|
parser = newParser(tagLib)
|
|
events = parser.parse(input)
|
|
if style == ypsCanonical:
|
|
var specificTagEvents = iterator(): YamlStreamEvent =
|
|
for e in events():
|
|
var event = e
|
|
case event.kind
|
|
of yamlStartDocument, yamlEndDocument, yamlEndMap, yamlAlias,
|
|
yamlEndSequence:
|
|
discard
|
|
of yamlStartMap:
|
|
if event.mapTag in [yTagQuestionMark, yTagExclamationMark]:
|
|
event.mapTag = yTagMap
|
|
of yamlStartSequence:
|
|
if event.seqTag in [yTagQuestionMark, yTagExclamationMark]:
|
|
event.seqTag = yTagSequence
|
|
of yamlScalar:
|
|
if event.scalarTag == yTagQuestionMark:
|
|
case event.scalarType
|
|
of yTypeInteger:
|
|
event.scalarTag = yTagInteger
|
|
of yTypeFloat, yTypeFloatInf, yTypeFloatNaN:
|
|
event.scalarTag = yTagFloat
|
|
of yTypeBoolTrue, yTypeBoolFalse:
|
|
event.scalarTag = yTagBoolean
|
|
of yTypeNull:
|
|
event.scalarTag = yTagNull
|
|
of yTypeString, yTypeUnknown:
|
|
event.scalarTag = yTagString
|
|
elif event.scalarTag == yTagExclamationMark:
|
|
event.scalarTag = yTagString
|
|
yield event
|
|
present(specificTagEvents, output, tagLib, style,
|
|
indentationStep)
|
|
else:
|
|
present(parser.parse(input), output, tagLib, style,
|
|
indentationStep)
|