# NimYAML - YAML implementation in Nim # (c) Copyright 2016 Felix Krause # # See the file "copying.txt", included in this # distribution, for details about the copyright. ## ========================= ## Module yaml.serialization ## ========================= ## ## This is the most high-level API of NimYAML. It enables you to parse YAML ## character streams directly into native YAML types and vice versa. It builds ## on top of the low-level parser and presenter APIs. ## ## It is possible to define custom construction and serialization procs for any ## type. Please consult the serialization guide on the NimYAML website for more ## information. import tables, typetraits, strutils, macros, streams, times import parser, taglib, presenter, stream, ../private/internal, hints export stream # *something* in here needs externally visible `==`(x,y: AnchorId), # but I cannot figure out what. binding it would be the better option. type SerializationContext* = ref object ## Context information for the process of serializing YAML from Nim values. refs*: Table[pointer, AnchorId] style: AnchorStyle nextAnchorId*: AnchorId put*: proc(e: YamlStreamEvent) {.raises: [], closure.} ConstructionContext* = ref object ## Context information for the process of constructing Nim values from YAML. refs*: Table[AnchorId, pointer] YamlConstructionError* = object of YamlLoadingError ## Exception that may be raised when constructing data objects from a ## `YamlStream <#YamlStream>`_. The fields ``line``, ``column`` and ## ``lineContent`` are only available if the costructing proc also does ## parsing, because otherwise this information is not available to the ## costruction proc. # forward declares proc constructChild*[T](s: var YamlStream, c: ConstructionContext, result: var T) {.raises: [YamlConstructionError, YamlStreamError].} ## Constructs an arbitrary Nim value from a part of a YAML stream. ## The stream will advance until after the finishing token that was used ## for constructing the value. The ``ConstructionContext`` is needed for ## potential child objects which may be refs. proc constructChild*(s: var YamlStream, c: ConstructionContext, result: var string) {.raises: [YamlConstructionError, YamlStreamError].} ## Constructs a Nim value that is a string from a part of a YAML stream. ## This specialization takes care of possible nil strings. proc constructChild*[T](s: var YamlStream, c: ConstructionContext, result: var seq[T]) {.raises: [YamlConstructionError, YamlStreamError].} ## Constructs a Nim value that is a string from a part of a YAML stream. ## This specialization takes care of possible nil seqs. proc constructChild*[O](s: var YamlStream, c: ConstructionContext, result: var ref O) {.raises: [YamlStreamError].} ## Constructs an arbitrary Nim value from a part of a YAML stream. ## The stream will advance until after the finishing token that was used ## for constructing the value. The object may be constructed from an alias ## node which will be resolved using the ``ConstructionContext``. proc representChild*[O](value: ref O, ts: TagStyle, c: SerializationContext) {.raises: [].} ## Represents an arbitrary Nim reference value as YAML object. The object ## may be represented as alias node if it is already present in the ## ``SerializationContext``. proc representChild*(value: string, ts: TagStyle, c: SerializationContext) {.inline, raises: [].} ## Represents a Nim string. Supports nil strings. proc representChild*[O](value: O, ts: TagStyle, c: SerializationContext) ## Represents an arbitrary Nim object as YAML object. proc newConstructionContext*(): ConstructionContext = new(result) result.refs = initTable[AnchorId, pointer]() proc newSerializationContext*(s: AnchorStyle, putImpl: proc(e: YamlStreamEvent) {.raises: [], closure.}): SerializationContext = SerializationContext(refs: initTable[pointer, AnchorId](), style: s, nextAnchorId: 0.AnchorId, put: putImpl) template presentTag*(t: typedesc, ts: TagStyle): TagId = ## Get the TagId that represents the given type in the given style if ts == tsNone: yTagQuestionMark else: yamlTag(t) proc lazyLoadTag(uri: string): TagId {.inline, raises: [].} = try: result = serializationTagLibrary.tags[uri] except KeyError: result = serializationTagLibrary.registerUri(uri) proc safeTagUri(id: TagId): string {.raises: [].} = try: var uri = serializationTagLibrary.uri(id) i = 0 # '!' is not allowed inside a tag handle if uri.len > 0 and uri[0] == '!': uri = uri[1..^1] # ',' is not allowed after a tag handle in the suffix because it's a flow # indicator for c in uri.mitems(): if c == ',': c = ';' inc(i) return uri except KeyError: internalError("Unexpected KeyError for TagId " & $id) proc constructionError(s: YamlStream, msg: string): ref YamlConstructionError = result = newException(YamlConstructionError, msg) if not s.getLastTokenContext(result.line, result.column, result.lineContent): (result.line, result.column) = (-1, -1) result.lineContent = "" template constructScalarItem*(s: var YamlStream, i: untyped, t: typedesc, content: untyped) = ## Helper template for implementing ``constructObject`` for types that ## are constructed from a scalar. ``i`` is the identifier that holds ## the scalar as ``YamlStreamEvent`` in the content. Exceptions raised in ## the content will be automatically catched and wrapped in ## ``YamlConstructionError``, which will then be raised. bind constructionError let i = s.next() if i.kind != yamlScalar: raise constructionError(s, "Expected scalar") try: content except YamlConstructionError: raise except Exception: var e = constructionError(s, "Cannot construct to " & name(t) & ": " & item.scalarContent) e.parent = getCurrentException() raise e proc yamlTag*(T: typedesc[string]): TagId {.inline, noSideEffect, raises: [].} = yTagString proc constructObject*(s: var YamlStream, c: ConstructionContext, result: var string) {.raises: [YamlConstructionError, YamlStreamError].} = ## costructs a string from a YAML scalar constructScalarItem(s, item, string): result = item.scalarContent proc representObject*(value: string, ts: TagStyle, c: SerializationContext, tag: TagId) {.raises: [].} = ## represents a string as YAML scalar c.put(scalarEvent(value, tag, yAnchorNone)) proc parseHex[T: int8|int16|int32|int64|uint8|uint16|uint32|uint64]( s: YamlStream, val: string): T = result = 0 for i in 2.. recListIndex: inc(result, child[bIndex][recListIndex].len) macro matchMatrix(t: typedesc): untyped = result = newNimNode(nnkBracket) let numFields = fieldCount(t) for i in 0.. 0: case s.next().kind of yamlStartMap, yamlStartSeq: inc(depth) of yamlEndMap, yamlEndSeq: dec(depth) of yamlScalar: discard else: internalError("Unexpected event kind.") when isVariantObject(O): e = s.next() if e.kind != yamlEndMap: raise s.constructionError("Expected end of single-pair map, got " & $e.kind) discard s.next() when result is tuple: var i = 0 for fname, value in fieldPairs(result): if not matched[i]: raise s.constructionError("While constructing " & typetraits.name(O) & ": Missing field: " & escape(fname)) inc(i) else: ensureAllFieldsPresent(s, O, tIndex, result, matched) macro genRepresentObject(t: typedesc, value, childTagStyle: typed): typed = result = newStmtList() let tSym = genSym(nskConst, ":tSym") result.add(quote do: when compiles(`transientBitvectorProc`(`t`)): const `tSym` = `transientBitvectorProc`(`t`) else: const `tSym` = -1 ) let tDecl = getType(t) tDesc = getType(tDecl[1]) isVO = isVariantObject(t) var fieldIndex = 0'i16 for child in tDesc[2].children: if child.kind == nnkRecCase: let fieldName = $child[0] fieldAccessor = newDotExpr(value, newIdentNode(fieldName)) result.add(quote do: c.put(startMapEvent(yTagQuestionMark, yAnchorNone)) c.put(scalarEvent(`fieldName`, if `childTagStyle` == tsNone: yTagQuestionMark else: yTagNimField, yAnchorNone)) representChild(`fieldAccessor`, `childTagStyle`, c) c.put(endMapEvent()) ) let enumName = $getTypeInst(child[0]) var caseStmt = newNimNode(nnkCaseStmt).add(fieldAccessor) for bIndex in 1 .. len(child) - 1: var curBranch: NimNode var recListIndex = 0 case child[bIndex].kind of nnkOfBranch: curBranch = newNimNode(nnkOfBranch) while child[bIndex][recListIndex].kind == nnkIntLit: curBranch.add(newCall(enumName, newLit(child[bIndex][recListIndex].intVal))) inc(recListIndex) of nnkElse: curBranch = newNimNode(nnkElse) else: internalError("Unexpected child kind: " & $child[bIndex].kind) doAssert child[bIndex][recListIndex].kind == nnkRecList var curStmtList = newStmtList() if child[bIndex][recListIndex].len > 0: for item in child[bIndex][recListIndex].children: inc(fieldIndex) let name = $item itemAccessor = newDotExpr(value, newIdentNode(name)) curStmtList.add(quote do: when `tSym` == -1 or `fieldIndex` notin transientVectors[`tSym`]: c.put(startMapEvent(yTagQuestionMark, yAnchorNone)) c.put(scalarEvent(`name`, if `childTagStyle` == tsNone: yTagQuestionMark else: yTagNimField, yAnchorNone)) representChild(`itemAccessor`, `childTagStyle`, c) c.put(endMapEvent()) ) else: curStmtList.add(newNimNode(nnkDiscardStmt).add(newEmptyNode())) curBranch.add(curStmtList) caseStmt.add(curBranch) result.add(caseStmt) else: let name = $child childAccessor = newDotExpr(value, newIdentNode(name)) result.add(quote do: when `tSym` == -1 or `fieldIndex` notin transientVectors[`tSym`]: when `isVO`: c.put(startMapEvent(yTagQuestionMark, yAnchorNone)) c.put(scalarEvent(`name`, if `childTagStyle` == tsNone: yTagQuestionMark else: yTagNimField, yAnchorNone)) representChild(`childAccessor`, `childTagStyle`, c) when `isVO`: c.put(endMapEvent()) ) inc(fieldIndex) proc representObject*[O: object](value: O, ts: TagStyle, c: SerializationContext, tag: TagId) = ## represents a Nim object or tuple as YAML mapping let childTagStyle = if ts == tsRootOnly: tsNone else: ts when isVariantObject(O): c.put(startSeqEvent(tag, yAnchorNone)) else: c.put(startMapEvent(tag, yAnchorNone)) genRepresentObject(O, value, childTagStyle) when isVariantObject(O): c.put(endSeqEvent()) else: c.put(endMapEvent()) proc representObject*[O: tuple](value: O, ts: TagStyle, c: SerializationContext, tag: TagId) = let childTagStyle = if ts == tsRootOnly: tsNone else: ts var fieldIndex = 0'i16 c.put(startMapEvent(tag, yAnchorNone)) for name, fvalue in fieldPairs(value): c.put(scalarEvent(name, if childTagStyle == tsNone: yTagQuestionMark else: yTagNimField, yAnchorNone)) representChild(fvalue, childTagStyle, c) inc(fieldIndex) c.put(endMapEvent()) proc constructObject*[O: enum](s: var YamlStream, c: ConstructionContext, result: var O) {.raises: [YamlConstructionError, YamlStreamError].} = ## constructs a Nim enum from a YAML scalar let e = s.next() if e.kind != yamlScalar: raise s.constructionError("Expected scalar, got " & $e.kind) try: result = parseEnum[O](e.scalarContent) except ValueError: var ex = s.constructionError("Cannot parse '" & escape(e.scalarContent) & "' as " & type(O).name) ex.parent = getCurrentException() raise ex proc representObject*[O: enum](value: O, ts: TagStyle, c: SerializationContext, tag: TagId) {.raises: [].} = ## represents a Nim enum as YAML scalar c.put(scalarEvent($value, tag, yAnchorNone)) proc yamlTag*[O](T: typedesc[ref O]): TagId {.inline, raises: [].} = yamlTag(O) macro constructImplicitVariantObject(s, c, r, possibleTagIds: untyped, t: typedesc): typed = let tDesc = getType(getType(t)[1]) yAssert tDesc.kind == nnkObjectTy let recCase = tDesc[2][0] yAssert recCase.kind == nnkRecCase let discriminant = newDotExpr(r, newIdentNode($recCase[0])) var ifStmt = newNimNode(nnkIfStmt) for i in 1 .. recCase.len - 1: yAssert recCase[i].kind == nnkOfBranch var branch = newNimNode(nnkElifBranch) var branchContent = newStmtList(newAssignment(discriminant, recCase[i][0])) case recCase[i][1].len of 0: branch.add(infix(newIdentNode("yTagNull"), "in", possibleTagIds)) branchContent.add(newNimNode(nnkDiscardStmt).add(newCall("next", s))) of 1: let field = newDotExpr(r, newIdentNode($recCase[i][1][0])) branch.add(infix( newCall("yamlTag", newCall("type", field)), "in", possibleTagIds)) branchContent.add(newCall("constructChild", s, c, field)) else: internalError("Too many children: " & $recCase[i][1].len) branch.add(branchContent) ifStmt.add(branch) let raiseStmt = newNimNode(nnkRaiseStmt).add( newCall(bindSym("constructionError"), s, infix(newStrLitNode("This value type does not map to any field in " & typetraits.name(t) & ": "), "&", newCall("uri", newIdentNode("serializationTagLibrary"), newNimNode(nnkBracketExpr).add(possibleTagIds, newIntLitNode(0))) ) )) ifStmt.add(newNimNode(nnkElse).add(newNimNode(nnkTryStmt).add( newStmtList(raiseStmt), newNimNode(nnkExceptBranch).add( newIdentNode("KeyError"), newNimNode(nnkDiscardStmt).add(newEmptyNode()) )))) result = newStmtList(newCall("reset", r), ifStmt) macro isImplicitVariantObject(o: typed): untyped = result = newCall("compiles", newCall(implicitVariantObjectMarker, o)) proc constructChild*[T](s: var YamlStream, c: ConstructionContext, result: var T) = let item = s.peek() when isImplicitVariantObject(result): var possibleTagIds = newSeq[TagId]() case item.kind of yamlScalar: case item.scalarTag of yTagQuestionMark: case guessType(item.scalarContent) of yTypeInteger: possibleTagIds.add([yamlTag(int), yamlTag(int8), yamlTag(int16), yamlTag(int32), yamlTag(int64)]) if item.scalarContent[0] != '-': possibleTagIds.add([yamlTag(uint), yamlTag(uint8), yamlTag(uint16), yamlTag(uint32), yamlTag(uint64)]) of yTypeFloat, yTypeFloatInf, yTypeFloatNaN: possibleTagIds.add([yamlTag(float), yamlTag(float32), yamlTag(float64)]) of yTypeBoolTrue, yTypeBoolFalse: possibleTagIds.add(yamlTag(bool)) of yTypeNull: raise s.constructionError("not implemented!") of yTypeUnknown: possibleTagIds.add(yamlTag(string)) of yTypeTimestamp: possibleTagIds.add(yamlTag(Time)) of yTagExclamationMark: possibleTagIds.add(yamlTag(string)) else: possibleTagIds.add(item.scalarTag) of yamlStartMap: if item.mapTag in [yTagQuestionMark, yTagExclamationMark]: raise s.constructionError( "Complex value of implicit variant object type must have a tag.") possibleTagIds.add(item.mapTag) of yamlStartSeq: if item.seqTag in [yTagQuestionMark, yTagExclamationMark]: raise s.constructionError( "Complex value of implicit variant object type must have a tag.") possibleTagIds.add(item.seqTag) else: internalError("Unexpected item kind: " & $item.kind) constructImplicitVariantObject(s, c, result, possibleTagIds, T) else: case item.kind of yamlScalar: if item.scalarTag notin [yTagQuestionMark, yTagExclamationMark, yamlTag(T)]: raise s.constructionError("Wrong tag for " & typetraits.name(T)) elif item.scalarAnchor != yAnchorNone: raise s.constructionError("Anchor on non-ref type") of yamlStartMap: if item.mapTag notin [yTagQuestionMark, yamlTag(T)]: raise s.constructionError("Wrong tag for " & typetraits.name(T)) elif item.mapAnchor != yAnchorNone: raise s.constructionError("Anchor on non-ref type") of yamlStartSeq: if item.seqTag notin [yTagQuestionMark, yamlTag(T)]: raise s.constructionError("Wrong tag for " & typetraits.name(T)) elif item.seqAnchor != yAnchorNone: raise s.constructionError("Anchor on non-ref type") else: internalError("Unexpected item kind: " & $item.kind) constructObject(s, c, result) proc constructChild*(s: var YamlStream, c: ConstructionContext, result: var string) = let item = s.peek() if item.kind == yamlScalar: if item.scalarTag == yTagNimNilString: discard s.next() result = nil return elif item.scalarTag notin [yTagQuestionMark, yTagExclamationMark, yamlTag(string)]: raise s.constructionError("Wrong tag for string") elif item.scalarAnchor != yAnchorNone: raise s.constructionError("Anchor on non-ref type") constructObject(s, c, result) proc constructChild*[T](s: var YamlStream, c: ConstructionContext, result: var seq[T]) = let item = s.peek() if item.kind == yamlScalar: if item.scalarTag == yTagNimNilSeq: discard s.next() result = nil return elif item.kind == yamlStartSeq: if item.seqTag notin [yTagQuestionMark, yamlTag(seq[T])]: raise s.constructionError("Wrong tag for " & typetraits.name(seq[T])) elif item.seqAnchor != yAnchorNone: raise s.constructionError("Anchor on non-ref type") constructObject(s, c, result) proc constructChild*[O](s: var YamlStream, c: ConstructionContext, result: var ref O) = var e = s.peek() if e.kind == yamlScalar: if e.scalarTag == yTagNull or (e.scalarTag == yTagQuestionMark and guessType(e.scalarContent) == yTypeNull): result = nil discard s.next() return elif e.kind == yamlAlias: result = cast[ref O](c.refs.getOrDefault(e.aliasTarget)) discard s.next() return new(result) template removeAnchor(anchor: var AnchorId) {.dirty.} = if anchor != yAnchorNone: yAssert(not c.refs.hasKey(anchor)) c.refs[anchor] = cast[pointer](result) anchor = yAnchorNone case e.kind of yamlScalar: removeAnchor(e.scalarAnchor) of yamlStartMap: removeAnchor(e.mapAnchor) of yamlStartSeq: removeAnchor(e.seqAnchor) else: internalError("Unexpected event kind: " & $e.kind) s.peek = e try: constructChild(s, c, result[]) except YamlConstructionError, YamlStreamError: raise except Exception: var e = newException(YamlStreamError, getCurrentExceptionMsg()) e.parent = getCurrentException() raise e proc representChild*(value: string, ts: TagStyle, c: SerializationContext) = if isNil(value): c.put(scalarEvent("", yTagNimNilString)) else: representObject(value, ts, c, presentTag(string, ts)) proc representChild*[T](value: seq[T], ts: TagStyle, c: SerializationContext) = if isNil(value): c.put(scalarEvent("", yTagNimNilSeq)) else: representObject(value, ts, c, presentTag(seq[T], ts)) proc representChild*[O](value: ref O, ts: TagStyle, c: SerializationContext) = if isNil(value): c.put(scalarEvent("~", yTagNull)) elif c.style == asNone: representChild(value[], ts, c) else: let p = cast[pointer](value) if c.refs.hasKey(p): if c.refs.getOrDefault(p) == yAnchorNone: c.refs[p] = c.nextAnchorId c.nextAnchorId = AnchorId(int(c.nextAnchorId) + 1) c.put(aliasEvent(c.refs.getOrDefault(p))) return if c.style == asAlways: c.refs[p] = c.nextAnchorId c.nextAnchorId = AnchorId(int(c.nextAnchorId) + 1) else: c.refs[p] = yAnchorNone let a = if c.style == asAlways: c.refs.getOrDefault(p) else: cast[AnchorId](p) childTagStyle = if ts == tsAll: tsAll else: tsRootOnly let origPut = c.put c.put = proc(e: YamlStreamEvent) = var ex = e case ex.kind of yamlStartMap: ex.mapAnchor = a if ts == tsNone: ex.mapTag = yTagQuestionMark of yamlStartSeq: ex.seqAnchor = a if ts == tsNone: ex.seqTag = yTagQuestionMark of yamlScalar: ex.scalarAnchor = a if ts == tsNone and guessType(ex.scalarContent) != yTypeNull: ex.scalarTag = yTagQuestionMark else: discard c.put = origPut c.put(ex) representChild(value[], childTagStyle, c) proc representChild*[O](value: O, ts: TagStyle, c: SerializationContext) = when isImplicitVariantObject(value): # todo: this would probably be nicer if constructed with a macro var count = 0 for name, field in fieldPairs(value): if count > 0: representChild(field, if ts == tsAll: tsAll else: tsRootOnly, c) inc(count) if count == 1: c.put(scalarEvent("~", yTagNull)) else: representObject(value, ts, c, if ts == tsNone: yTagQuestionMark else: yamlTag(O)) proc construct*[T](s: var YamlStream, target: var T) {.raises: [YamlStreamError].} = ## Constructs a Nim value from a YAML stream. var context = newConstructionContext() try: var e = s.next() yAssert(e.kind == yamlStartDoc) constructChild(s, context, target) e = s.next() yAssert(e.kind == yamlEndDoc) except YamlConstructionError: raise (ref YamlConstructionError)(getCurrentException()) except YamlStreamError: let cur = getCurrentException() var e = newException(YamlStreamError, cur.msg) e.parent = cur.parent raise e except Exception: # may occur while calling s() var ex = newException(YamlStreamError, "") ex.parent = getCurrentException() raise ex proc load*[K](input: Stream | string, target: var K) {.raises: [YamlConstructionError, IOError, YamlParserError].} = ## Loads a Nim value from a YAML character stream. var parser = newYamlParser(serializationTagLibrary) events = parser.parse(input) try: construct(events, target) except YamlStreamError: let e = (ref YamlStreamError)(getCurrentException()) if e.parent of IOError: raise (ref IOError)(e.parent) elif e.parent of YamlParserError: raise (ref YamlParserError)(e.parent) else: internalError("Unexpected exception: " & e.parent.repr) proc loadMultiDoc*[K](input: Stream | string, target: var seq[K]) = if target.isNil: target = newSeq[K]() var parser = newYamlParser(serializationTagLibrary) events = parser.parse(input) try: while not events.finished(): var item: K construct(events, item) target.add(item) except YamlConstructionError: var e = (ref YamlConstructionError)(getCurrentException()) discard events.getLastTokenContext(e.line, e.column, e.lineContent) raise e except YamlStreamError: let e = (ref YamlStreamError)(getCurrentException()) if e.parent of IOError: raise (ref IOError)(e.parent) elif e.parent of YamlParserError: raise (ref YamlParserError)(e.parent) else: internalError("Unexpected exception: " & e.parent.repr) proc setAnchor(a: var AnchorId, q: var Table[pointer, AnchorId]) {.inline.} = if a != yAnchorNone: a = q.getOrDefault(cast[pointer](a)) proc represent*[T](value: T, ts: TagStyle = tsRootOnly, a: AnchorStyle = asTidy): YamlStream = ## Represents a Nim value as ``YamlStream`` var bys = newBufferYamlStream() var context = newSerializationContext(a, proc(e: YamlStreamEvent) = bys.put(e) ) bys.put(startDocEvent()) representChild(value, ts, context) bys.put(endDocEvent()) if a == asTidy: for item in bys.mitems(): case item.kind of yamlStartMap: item.mapAnchor.setAnchor(context.refs) of yamlStartSeq: item.seqAnchor.setAnchor(context.refs) of yamlScalar: item.scalarAnchor.setAnchor(context.refs) else: discard result = bys proc dump*[K](value: K, target: Stream, tagStyle: TagStyle = tsRootOnly, anchorStyle: AnchorStyle = asTidy, options: PresentationOptions = defaultPresentationOptions) {.raises: [YamlPresenterJsonError, YamlPresenterOutputError, YamlStreamError].} = ## Dump a Nim value as YAML character stream. var events = represent(value, if options.style == psCanonical: tsAll else: tagStyle, if options.style == psJson: asNone else: anchorStyle) try: present(events, target, serializationTagLibrary, options) except YamlStreamError: internalError("Unexpected exception: " & getCurrentException().repr) proc dump*[K](value: K, tagStyle: TagStyle = tsRootOnly, anchorStyle: AnchorStyle = asTidy, options: PresentationOptions = defaultPresentationOptions): string = ## Dump a Nim value as YAML into a string var events = represent(value, if options.style == psCanonical: tsAll else: tagStyle, if options.style == psJson: asNone else: anchorStyle) try: result = present(events, serializationTagLibrary, options) except YamlStreamError: internalError("Unexpected exception: " & getCurrentException().repr) proc canBeImplicit(t: typedesc): bool {.compileTime.} = let tDesc = getType(t) if tDesc.kind != nnkObjectTy: return false if tDesc[2].len != 1: return false if tDesc[2][0].kind != nnkRecCase: return false var foundEmptyBranch = false for i in 1.. tDesc[2][0].len - 1: case tDesc[2][0][i][1].len # branch contents of 0: if foundEmptyBranch: return false else: foundEmptyBranch = true of 1: discard else: return false return true macro setImplicitVariantObjectMarker(t: typedesc): untyped = result = quote do: when compiles(`transientBitvectorProc`(`t`)): {.fatal: "Cannot mark object with transient fields as implicit".} proc `implicitVariantObjectMarker`*(unused: `t`) = discard template markAsImplicit*(t: typedesc): typed = ## Mark a variant object type as implicit. This requires the type to consist ## of nothing but a case expression and each branch of the case expression ## containing exactly one field - with the exception that one branch may ## contain zero fields. when canBeImplicit(t): # this will be checked by means of compiles(implicitVariantObject(...)) setImplicitVariantObjectMarker(t) else: {. fatal: "This type cannot be marked as implicit" .} macro getFieldIndex(t: typedesc, field: untyped): untyped = let tDecl = getType(t) tName = $tDecl[1] tDesc = getType(tDecl[1]) fieldName = $field var fieldIndex = 0 found = false block outer: for child in tDesc[2].children: if child.kind == nnkRecCase: for bIndex in 1 .. len(child) - 1: var bChildIndex = 0 case child[bIndex].kind of nnkOfBranch: while child[bIndex][bChildIndex].kind == nnkIntLit: inc(bChildIndex) of nnkElse: discard else: internalError("Unexpected child kind: " & $child[bIndex][bChildIndex].kind) yAssert child[bIndex][bChildIndex].kind == nnkRecList for item in child[bIndex][bChildIndex].children: inc(fieldIndex) yAssert item.kind == nnkSym if $item == fieldName: found = true break outer else: yAssert child.kind == nnkSym if $child == fieldName: found = true break inc(fieldIndex) if not found: let msg = "Type " & tName & " has no field " & fieldName & '!' result = quote do: {.fatal: `msg`.} else: result = newNimNode(nnkInt16Lit) result.intVal = fieldIndex macro markAsTransient*(t: typedesc, field: untyped): typed = let nextBitvectorIndex = transientVectors.len result = quote do: when compiles(`implicitVariantObjectMarker`(`t`)): {.fatal: "Cannot mark fields of implicit variant objects as transient." .} when not compiles(`transientBitvectorProc`(`t`)): proc `transientBitvectorProc`*(myType: typedesc[`t`]): int {.compileTime.} = `nextBitvectorIndex` static: transientVectors.add({}) static: transientVectors[`transientBitvectorProc`(`t`)].incl(getFieldIndex(`t`, `field`)) macro setDefaultValue*(t: typedesc, field: untyped, value: typed): typed = let dSym = genSym(nskVar, ":default") nextBitvectorIndex = defaultVectors.len result = quote do: when compiles(`transientBitvectorProc`(`t`)): {.fatal: "Cannot set default value of transient field".} elif compiles(`implicitVariantObjectMarker`(`t`)): {.fatal: "Cannot set default value of implicit variant objects.".} when not compiles(`defaultValueGetter`(`t`)): var `dSym` {.compileTime.}: `t` template `defaultValueGetter`(t: typedesc[`t`]): auto = `dSym` proc `defaultBitvectorProc`*(myType: typedesc[`t`]): int {.compileTime.} = `nextBitvectorIndex` static: defaultVectors.add({}) static: `defaultValueGetter`(`t`).`field` = `value` defaultVectors[`defaultBitvectorProc`(`t`)].incl(getFieldIndex(`t`, `field`)) macro ignoreInputKey*(t: typedesc, name: string{lit}): typed = let nextIgnoredKeyList = ignoredKeyLists.len result = quote do: when not compiles(`ignoredKeyListProc`(`t`)): proc `ignoredKeyListProc`*(t: typedesc[`t`]): int {.compileTime.} = `nextIgnoredKeyList` static: ignoredKeyLists.add(@[]) when `name` in ignoredKeyLists[`ignoredKeyListProc`(`t`)]: {.fatal: "Input key " & `name` & " is already ignored!".} static: ignoredKeyLists[`ignoredKeyListProc`(`t`)].add(`name`)