NimYAML/private/json.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
    Level = tuple[node: JsonNode, key: string]

proc initLevel(node: JsonNode): Level {.raises: [].} =
    (node: node, key: cast[string](nil))

proc jsonFromScalar(content: string, tag: TagId): JsonNode
       {.raises: [YamlConstructionError].}=
    new(result)
    var mappedType: TypeHint
    
    case tag
    of yTagQuestionMark: mappedType = guessType(content)
    of yTagExclamationMark, yTagString: mappedType = yTypeUnknown
    of yTagBoolean:
        case guessType(content)
        of yTypeBoolTrue: mappedType = yTypeBoolTrue
        of yTypeBoolFalse: mappedType = yTypeBoolFalse
        else:
            raise newException(YamlConstructionError,
                               "Invalid boolean value: " & content)
    of yTagInteger: mappedType = yTypeInteger
    of yTagNull: mappedType = yTypeNull
    of yTagFloat:
        case guessType(content)
        of yTypeFloat: mappedType = yTypeFloat
        of yTypeFloatInf: mappedType = yTypeFloatInf
        of yTypeFloatNaN: mappedType = yTypeFloatNaN
        else:
            raise newException(YamlConstructionError,
                               "Invalid float value: " & content)
    else: mappedType = yTypeUnknown
    
    try:
        case mappedType
        of yTypeInteger:
            result.kind = JInt
            result.num = parseBiggestInt(content)
        of yTypeFloat:
            result.kind = JFloat
            result.fnum = parseFloat(content)
        of yTypeFloatInf:
            result.kind = JFloat
            result.fnum = if content[0] == '-': NegInf else: Inf
        of yTypeFloatNaN:
            result.kind = JFloat
            result.fnum = NaN
        of yTypeBoolTrue:
            result.kind = JBool
            result.bval = true
        of yTypeBoolFalse:
            result.kind = JBool
            result.bval = false
        of yTypeNull:
            result.kind = JNull
        else:
            result.kind = JString
            shallowCopy(result.str, content)
    except ValueError:
        var e = newException(YamlConstructionError,
                             "Cannot parse numeric value")
        e.parent = getCurrentException()
        raise e

proc constructJson*(s: var YamlStream): seq[JsonNode] =
    newSeq(result, 0)
    
    var
        levels  = newSeq[Level]()
        anchors = initTable[AnchorId, JsonNode]()
    for event in s:
        case event.kind
        of yamlStartDoc:
            # we don't need to do anything here; root node will be created
            # by first scalar, sequence or map event
            discard
        of yamlEndDoc:
            # we can savely assume that levels has e length of exactly 1.
            result.add(levels.pop().node)
        of yamlStartSeq:
            levels.add(initLevel(newJArray()))
            if event.seqAnchor != yAnchorNone:
                anchors[event.seqAnchor] = levels[levels.high].node
        of yamlStartMap:
            levels.add(initLevel(newJObject()))
            if event.mapAnchor != yAnchorNone:
                anchors[event.mapAnchor] = levels[levels.high].node
        of yamlScalar:
            if levels.len == 0:
                # parser ensures that next event will be yamlEndDocument
                levels.add((node: jsonFromScalar(event.scalarContent,
                                                 event.scalarTag), key: nil))
                continue
            
            case levels[levels.high].node.kind
            of JArray:
                let jsonScalar = jsonFromScalar(event.scalarContent,
                                                event.scalarTag)
                levels[levels.high].node.elems.add(jsonScalar)
                if event.scalarAnchor != yAnchorNone:
                    anchors[event.scalarAnchor] = jsonScalar
            of JObject:
                if isNil(levels[levels.high].key):
                    # JSON only allows strings as keys
                    levels[levels.high].key = event.scalarContent
                    if event.scalarAnchor != yAnchorNone:
                        raise newException(YamlConstructionError,
                                "scalar keys may not have anchors in JSON")
                else:
                    let jsonScalar = jsonFromScalar(event.scalarContent,
                                                    event.scalarTag)
                    levels[levels.high].node[levels[levels.high].key] =
                            jsonScalar
                    levels[levels.high].key = nil
                    if event.scalarAnchor != yAnchorNone:
                        anchors[event.scalarAnchor] = jsonScalar
            else: discard # will never happen
        of yamlEndSeq, yamlEndMap:
            if levels.len > 1:
                let level = levels.pop()
                case levels[levels.high].node.kind
                of JArray: levels[levels.high].node.elems.add(level.node)
                of JObject:
                    if isNil(levels[levels.high].key):
                        raise newException(YamlConstructionError,
                                "non-scalar as key not allowed in JSON")
                    else:
                        levels[levels.high].node[
                            levels[levels.high].key] = level.node
                        levels[levels.high].key = nil
                else: discard # will never happen
            else: discard # wait for yamlEndDocument
        of yamlAlias:
            # we can savely assume that the alias exists in anchors
            # (else the parser would have already thrown an exception)
            case levels[levels.high].node.kind
            of JArray:
                try:
                    levels[levels.high].node.elems.add(
                            anchors[event.aliasTarget])
                except KeyError:
                    # we can safely assume that this doesn't happen. It would
                    # have resulted in a parser error earlier.
                    assert(false)
            of JObject:
                if isNil(levels[levels.high].key):
                    raise newException(YamlConstructionError,
                            "cannot use alias node as key in JSON")
                else:
                    try:
                        levels[levels.high].node.fields.add(
                                levels[levels.high].key,
                                anchors[event.aliasTarget])
                    except KeyError:
                        # we can safely assume that this doesn't happen. It would
                        # have resulted in a parser error earlier.
                        assert(false)
                    levels[levels.high].key = nil
            else: discard # will never happen

proc loadToJson*(s: Stream): seq[JsonNode] =
    var
        parser = newYamlParser(initCoreTagLibrary())
        events = parser.parse(s)
    try:
        return constructJson(events)
    except YamlConstructionError:
        var e = cast[ref YamlConstructionError](getCurrentException())
        e.line = parser.getLineNumber()
        e.column = parser.getColNumber()
        e.lineContent = parser.getLineContent()
        raise e
    except YamlStreamError:
        let e = getCurrentException()
        if e.parent of IOError:
            raise cast[ref IOError](e.parent)
        elif e.parent of YamlParserError:
            raise cast[ref YamlParserError](e.parent)
        else:
            # can never happen
            assert(false)
    except AssertionError: raise
    except Exception:
        # compiler bug: https://github.com/nim-lang/Nim/issues/3772
        assert(false)