922 lines
29 KiB
Nim
922 lines
29 KiB
Nim
# SPDX-FileCopyrightText: ☭ Emery Hemingway
|
|
# SPDX-License-Identifier: Unlicense
|
|
|
|
import std/[assertions, options, sets, sequtils, strutils, tables, typetraits]
|
|
from std/algorithm import sort
|
|
from std/json import escapeJson, escapeJsonUnquoted
|
|
import bigints
|
|
import ./preserves/private/[buffering, encoding, decoding, dot, macros, parsing, texts, values]
|
|
|
|
export buffering, encoding, decoding, parsing, texts, values
|
|
|
|
when defined(tracePreserves):
|
|
when defined(posix):
|
|
template trace(args: varargs[untyped]) = {.cast(noSideEffect).}: stderr.writeLine(args)
|
|
else:
|
|
template trace(args: varargs[untyped]) = {.cast(noSideEffect).}: echo(args)
|
|
else:
|
|
template trace(args: varargs[untyped]) = discard
|
|
|
|
type Preserve*[E] {.deprecated: "preserves.Preserve[E] is now preserves.Value".} = Value
|
|
|
|
proc sortDict(pr: var Value) =
|
|
sort(pr.dict) do (x, y: DictEntry) -> int:
|
|
cmp(x.key, y.key)
|
|
|
|
proc cannonicalize*(pr: var Value) =
|
|
## Cannonicalize a compound Preserves value by total ordering.
|
|
case pr.kind
|
|
of pkSequence:
|
|
apply(pr.sequence, cannonicalize)
|
|
of pkSet:
|
|
apply(pr.set, cannonicalize)
|
|
sort(pr.set)
|
|
of pkDictionary:
|
|
apply(pr.dict) do (e: var DictEntry):
|
|
cannonicalize(e.val)
|
|
sortDict(pr)
|
|
else:
|
|
discard
|
|
|
|
proc toInt*(pr: Value): Option[int] =
|
|
case pr.kind
|
|
of pkRegister: result = some pr.register
|
|
of pkBigInt: result = toInt[int](pr.bigint)
|
|
else: discard
|
|
|
|
func isAtomic*(pr: Value): bool = pr.kind in atomKinds
|
|
|
|
func isBoolean*(pr: Value): bool {.inline.} = pr.kind == pkBoolean
|
|
## Check if ``pr`` is a Preserve boolean.
|
|
|
|
func isFalse*(pr: Value): bool {.inline.} =
|
|
## Check if ``pr`` is equivalent to the zero-initialized ``Preserve``.
|
|
pr.kind == pkBoolean and pr.bool == false
|
|
|
|
func isFloat*(pr: Value): bool {.inline.} = pr.kind == pkFloat
|
|
## Check if ``pr`` is a Preserve float.
|
|
|
|
func isDouble*(pr: Value): bool {.deprecated: "use isFloat".} = pr.kind == pkFloat
|
|
|
|
func isInteger*(pr: Value): bool {.inline.} =
|
|
## Check if ``pr`` is a Preserve integer.
|
|
pr.kind == pkRegister or pr.kind == pkBigInt
|
|
|
|
func isInteger*(pr: Value; i: SomeInteger): bool {.inline.} =
|
|
## Check if ``pr`` is a Preserve integer equivalent to `i`.
|
|
case pr.kind
|
|
of pkRegister: pr.register == i.int
|
|
of pkBigInt: pr.int == i.initBigInt
|
|
else: false
|
|
|
|
func isString*(pr: Value): bool {.inline.} = pr.kind == pkString
|
|
## Check if ``pr`` is a Preserve text string.
|
|
|
|
func isString*(pr: Value; s: string): bool {.inline.} =
|
|
## Check if ``pr`` is a Preserve text string equivalent to `s`.
|
|
pr.kind == pkString and pr.string == s
|
|
|
|
func isByteString*(pr: Value): bool {.inline.} = pr.kind == pkByteString
|
|
## Check if ``pr`` is a Preserves byte string.
|
|
|
|
func isSymbol*(pr: Value): bool {.inline.} = pr.kind == pkSymbol
|
|
## Check if `pr` is a Preserves symbol.
|
|
|
|
func isSymbol*(pr: Value; sym: string|Symbol): bool {.inline.} =
|
|
## Check if ``pr`` is a Preserves symbol of ``sym``.
|
|
(pr.kind == pkSymbol) and (pr.symbol == Symbol(sym))
|
|
|
|
proc label*(pr: Value): Value {.inline.} =
|
|
## Return the label of record value.
|
|
pr.record[pr.record.high]
|
|
|
|
proc arity*(pr: Value): int {.inline.} =
|
|
## Return the number of fields in record `pr`.
|
|
pred(pr.record.len)
|
|
|
|
func isRecord*(pr: Value): bool {.inline.} = (pr.kind == pkRecord) and (pr.record.len > 0)
|
|
## Check if ``pr`` is a Preserves record.
|
|
|
|
func isRecord*(pr: Value; label: string): bool {.inline.} =
|
|
## Check if ``pr`` is a Preserves record with the given label symbol.
|
|
pr.kind == pkRecord and
|
|
pr.record.len > 0 and
|
|
pr.label.isSymbol(label)
|
|
|
|
func isRecord*(pr: Value; label: string; arity: Natural): bool {.inline.} =
|
|
## Check if ``pr`` is a Preserves record with the given label symbol and field arity.
|
|
pr.kind == pkRecord and
|
|
pr.record.len == succ(arity) and
|
|
pr.label.isSymbol(label)
|
|
|
|
proc isSequence*(pr: Value): bool {.inline.} = pr.kind == pkSequence
|
|
## Check if ``pr`` is a Preserves sequence.
|
|
|
|
proc isSet*(pr: Value): bool {.inline.} = pr.kind == pkSet
|
|
## Check if ``pr`` is a Preserves set.
|
|
|
|
proc isDictionary*(pr: Value): bool {.inline.} = pr.kind == pkDictionary
|
|
## Check if ``pr`` is a Preserves dictionary.
|
|
|
|
func isEmbedded*(pr: Value): bool {.inline.} =
|
|
## Check if ``pr`` is an embedded value.
|
|
pr.embedded or pr.kind == pkEmbedded
|
|
|
|
proc `&`*(x, y: Value): Value =
|
|
## Concatenate operator.
|
|
if x.kind != y.kind:
|
|
raise newException(ValueError, "cannot concatenate heterogenous values")
|
|
case x.kind
|
|
of pkString:
|
|
result = Value(kind: pkString, string: x.string & y.string)
|
|
of pkByteString:
|
|
result = Value(kind: pkByteString, bytes: x.bytes & y.bytes)
|
|
of pkSequence:
|
|
result = Value(kind: pkSequence, sequence: x.sequence & y.sequence)
|
|
else:
|
|
raise newException(ValueError, "cannot concatenate this value type")
|
|
|
|
proc `&`*(x: Value; y: seq[Value]): Value =
|
|
if x.kind != pkSequence:
|
|
raise newException(ValueError, "cannot concatenate to non-sequence value")
|
|
result = Value(kind: pkSequence, sequence: x.sequence & y)
|
|
|
|
proc pop*(pr: var Value; key: Value; val: var Value): bool =
|
|
## Deletes the `key` from a Preserves dictionary.
|
|
## Returns true, if the key existed, and sets `val` to the mapping
|
|
## of the key. Otherwise, returns false, and the `val` is unchanged.
|
|
if pr.kind == pkDictionary:
|
|
var i = 0
|
|
while i < pr.dict.len:
|
|
if pr.dict[i].key == key:
|
|
val = move pr.dict[i].val
|
|
delete(pr.dict, i .. i)
|
|
return true
|
|
|
|
proc `[]`*(pr, key: Value): Value {.deprecated: "use step instead".} =
|
|
## Select a value by `key` from `pr`.
|
|
## Works for sequences, records, and dictionaries.
|
|
case pr.kind
|
|
of pkDictionary:
|
|
for (k, v) in pr.dict.items:
|
|
if k == key: return v
|
|
raise newException(KeyError, "value not in Preserves dictionary")
|
|
of pkRecord, pkSequence:
|
|
let idx = key.toInt
|
|
if idx.isSome:
|
|
result = pr[get idx]
|
|
else:
|
|
raise newException(ValueError, "invalid Preserves index")
|
|
else:
|
|
raise newException(ValueError, "invalid Preserves indexing")
|
|
|
|
proc toSymbol*(s: sink string): Value {.inline.} =
|
|
## Create a Preserves symbol value.
|
|
Value(kind: pkSymbol, symbol: Symbol s)
|
|
|
|
proc toSymbol*(s: sink string; E: typedesc): Value {.deprecated.} =
|
|
s.toSymbol
|
|
|
|
proc initRecord*(label: Value; arity: Natural = 0): Value =
|
|
## Create a Preserves record value.
|
|
result = Value(kind: pkRecord, record: newSeq[Value](arity.succ))
|
|
result.record[arity] = label
|
|
|
|
proc initRecord*(label: Value; E: typedesc): Value {.deprecated.} =
|
|
initRecord(label)
|
|
|
|
proc initRecord*(label: Value; args: varargs[Value]): Value =
|
|
## Create a Preserves record value.
|
|
result = Value(kind: pkRecord,
|
|
record: newSeqOfCap[Value](1+args.len))
|
|
for arg in args:
|
|
result.record.add(arg)
|
|
result.record.add(label)
|
|
|
|
proc initRecord*(label: string; args: varargs[Value]): Value {.inline.} =
|
|
## Create a Preserves record value.
|
|
initRecord(toSymbol(label), args)
|
|
|
|
proc toRecord*(items: varargs[Value, toPreserves]): Value =
|
|
assert items.len > 0
|
|
result = initRecord(items[0], items.len.pred)
|
|
for i in 0..<items.high: result.record[i] = items[succ i]
|
|
|
|
proc initSequence*(len: Natural = 0): Value =
|
|
## Create a Preserves sequence value.
|
|
Value(kind: pkSequence, sequence: newSeq[Value](len))
|
|
|
|
proc initSequence*(len: Natural; E: typedesc): Value {.deprecated.} =
|
|
initSequence(len)
|
|
|
|
proc initSequenceOfCap*(cap: Natural): Value =
|
|
## Create a Preserves sequence value.
|
|
Value(kind: pkSequence, sequence: newSeqOfCap[Value](cap))
|
|
|
|
proc initSet*(): Value = Value(kind: pkSet)
|
|
## Create a Preserves set value.
|
|
|
|
proc initDictionary*(): Value = Value(kind: pkDictionary)
|
|
## Create a Preserves dictionary value.
|
|
|
|
proc initDictionary*(E: typedesc): Value =
|
|
initDictionary()
|
|
|
|
proc toDictionary*(pairs: openArray[(string, Value)]): Value =
|
|
## Create a Preserves dictionary value.
|
|
result = Value(kind: pkDictionary)
|
|
for (key, val) in pairs: result[toSymbol(key)] = val
|
|
|
|
proc embed*(pr: sink Value): Value =
|
|
## Mark a Preserves value as embedded.
|
|
result = pr
|
|
result.embedded = true
|
|
|
|
proc embed*(e: sink EmbeddedRef): Value =
|
|
## Create a Preserves value that embeds ``e``.
|
|
Value(kind: pkEmbedded, embeddedRef: e, embedded: true)
|
|
|
|
proc len*(pr: Value): int =
|
|
## Return the shallow count of values in ``pr``, that is the number of
|
|
## fields in a record, items in a sequence, items in a set, or key-value pairs
|
|
## in a dictionary.
|
|
case pr.kind
|
|
of pkRecord: pr.record.len.pred
|
|
of pkSequence: pr.sequence.len
|
|
of pkSet: pr.set.len
|
|
of pkDictionary: pr.dict.len
|
|
else: 0
|
|
|
|
iterator items*(pr: Value): Value =
|
|
## Shallow iterator over `pr`, yield the fields in a record,
|
|
## the items of a sequence, the items of a set, or the pairs
|
|
## of a dictionary.
|
|
case pr.kind
|
|
of pkRecord:
|
|
for i in 0..pr.record.high.pred:
|
|
yield pr.record[i]
|
|
of pkSequence:
|
|
for e in pr.sequence.items: yield e
|
|
of pkSet:
|
|
for e in pr.set.items: yield e
|
|
of pkDictionary:
|
|
for (k, v) in pr.dict.items:
|
|
yield k # key can be an arbitrary Preserve
|
|
yield v
|
|
else: discard
|
|
|
|
iterator pairs*(pr: Value): DictEntry =
|
|
assert(pr.kind == pkDictionary, "not a dictionary")
|
|
for i in 0..pr.dict.high: yield pr.dict[i]
|
|
|
|
proc fields*(pr: Value): seq[Value] {.inline.} =
|
|
## Return the fields of a record value.
|
|
pr.record[0..pr.record.high.pred]
|
|
|
|
iterator fields*(pr: Value): Value =
|
|
## Iterate the fields of a record value.
|
|
for i in 0..<pr.record.high: yield pr.record[i]
|
|
|
|
proc unembed*(pr: Value; E: typedesc): Option[E] =
|
|
## Unembed an `E` value from a `Value` value.
|
|
if pr.kind == pkEmbedded and pr.embeddedRef of E:
|
|
result = some(E pr.embeddedRef)
|
|
|
|
template preservesRecord*(label: string) {.pragma.}
|
|
## Serialize this object or tuple as a record.
|
|
## See ``toPreserve``.
|
|
|
|
proc hasPreservesRecordPragma*(T: typedesc): bool =
|
|
## Test if a type has a `{.preservesRecord: "…".}` pragma attached.
|
|
hasCustomPragma(T, preservesRecord)
|
|
|
|
proc recordLabel*(T: typedesc): string =
|
|
## Get the record label set by a pragma on a type.
|
|
runnableExamples:
|
|
type Foo {.preservesRecord: "bar".} = object
|
|
n: int
|
|
assert recordLabel(Foo) == "bar"
|
|
T.getCustomPragmaVal(preservesRecord)
|
|
|
|
template preservesTuple*() {.pragma.}
|
|
## Serialize this object or tuple as a tuple.
|
|
## See ``toPreserves``.
|
|
|
|
template preservesTupleTail*() {.pragma.}
|
|
## Serialize this object field to the end of its containing tuple.
|
|
## See ``toPreserves``.
|
|
|
|
proc hasPreservesTuplePragma*(T: typedesc): bool =
|
|
## Test if a type has a `preservesTuple` pragma attached.
|
|
hasCustomPragma(T, preservesTuple)
|
|
|
|
template preservesDictionary*() {.pragma.}
|
|
## Serialize this object or tuple as a dictionary.
|
|
## See ``toPreserves``.
|
|
|
|
proc hasPreservesDictionaryPragma*(T: typedesc): bool =
|
|
## Test if a type has a `preservesDictionary` pragma attached.
|
|
hasCustomPragma(T, preservesDictionary)
|
|
|
|
template preservesOr*() {.pragma.}
|
|
## Serialize this object as an ``or`` alternative.
|
|
## See ``toPreserves``.
|
|
|
|
template preservesLiteral*(value: typed) {.pragma.}
|
|
## Serialize a Preserves literal within this object.
|
|
## See ``toPreserves``.
|
|
|
|
template preservesEmbedded*() {.pragma.}
|
|
## Pragma to mark a type or value as embedded.
|
|
|
|
template unpreservable*() {.pragma.}
|
|
## Pragma to forbid a type from being converted by ``toPreserves``.
|
|
## Useful for preventing an embeded type from being encoded
|
|
## as its native type.
|
|
## Unpreservability is asserted at runtime.
|
|
|
|
converter preserve*(i: SomeInteger): Value =
|
|
Value(kind: pkRegister, register: BiggestInt i)
|
|
|
|
proc toPreserves*[T](x: T): Value {.gcsafe.} =
|
|
## Serializes ``x`` to Preserves. Can be customized by defining
|
|
## ``toPreservesHook(x: T; E: typedesc)`` in the calling scope.
|
|
## Any ``toPreservesHook`` that does not compile will be discarded;
|
|
## *Write tests for your hooks!*
|
|
##
|
|
## When `tracePreserves` is defined (`-d:tracePreserves`) a diagnostic
|
|
## trace is printing during `toPreserve`.
|
|
when (T is Value): result = x
|
|
elif T is Value:
|
|
result = cast[Value](x)
|
|
elif compiles(toPreservesHook(x)):
|
|
result = toPreservesHook(x)
|
|
cannonicalize(result)
|
|
elif T is enum:
|
|
result = toSymbol($x)
|
|
elif T is seq[byte]:
|
|
result = Value(kind: pkByteString, bytes: x)
|
|
elif T is array | seq:
|
|
result = Value(kind: pkSequence, sequence: newSeqOfCap[Value](x.len))
|
|
for v in x.items: result.sequence.add(toPreserves(v))
|
|
elif T is set:
|
|
result = Value(kind: pkSet, set: newSeqOfCap[Value](x.len))
|
|
for v in x.items: result.incl(toPreserves(v))
|
|
cannonicalize(result)
|
|
elif T is bool:
|
|
result = Value(kind: pkBoolean, bool: x)
|
|
elif T is SomeFloat:
|
|
result = Value(kind: pkFloat, float: float(x))
|
|
elif T is tuple:
|
|
result = Value(kind: pkSequence,
|
|
sequence: newSeqOfCap[Value](tupleLen(T)))
|
|
for xf in fields(x):
|
|
result.sequence.add(toPreserves(xf))
|
|
elif T is Ordinal:
|
|
result = Value(kind: pkRegister, register: x.ord)
|
|
assert result.register.T == x
|
|
elif T is ptr | ref:
|
|
when T.hasCustomPragma(preservesEmbedded):
|
|
result = embed(x)
|
|
else:
|
|
if system.`==`(x, nil): result = initRecord("null")
|
|
else: result = toPreserves(x[])
|
|
elif T is string:
|
|
result = Value(kind: pkString, string: x)
|
|
elif T is SomeInteger:
|
|
result = Value(kind: pkRegister, register: x.int)
|
|
assert result.register.T == x
|
|
elif T is Symbol:
|
|
result = Value(kind: pkSymbol, symbol: x)
|
|
elif T is distinct:
|
|
result = toPreserves(x.distinctBase)
|
|
elif T is object:
|
|
template fieldToPreserve[F](key: string; field: F): Value {.used.} =
|
|
when x.dot(key).hasCustomPragma(preservesLiteral):
|
|
const atom = x.dot(key).getCustomPragmaVal(preservesLiteral).parsePreservesAtom
|
|
when x.dot(key).hasCustomPragma(preservesEmbedded):
|
|
atom.toPreservesHook().embed
|
|
else:
|
|
atom.toPreservesHook()
|
|
elif x.dot(key).hasCustomPragma(preservesEmbedded) and F is EmbeddedRef:
|
|
embed(field)
|
|
else:
|
|
field.toPreserves
|
|
# checking for the embedded pragma here yields false positives
|
|
when T.hasCustomPragma(unpreservable):
|
|
raiseAssert($T & " is unpreservable")
|
|
elif T.hasCustomPragma(preservesOr):
|
|
var hasKind, hasVariant: bool
|
|
for k, v in x.fieldPairs:
|
|
if k == "orKind":
|
|
assert(not hasKind)
|
|
hasKind = true
|
|
else:
|
|
assert(hasKind and not hasVariant)
|
|
result = fieldToPreserve(k, v)
|
|
hasVariant = true
|
|
elif T.hasCustomPragma(preservesRecord):
|
|
result = Value(kind: pkRecord)
|
|
for label, field in x.fieldPairs:
|
|
when x.dot(label).hasCustomPragma(preservesTupleTail):
|
|
for e in field.items:
|
|
result.record.add(toPreserves(e))
|
|
else:
|
|
var pr = fieldToPreserve(label, field)
|
|
result.record.add(pr)
|
|
result.record.add(toSymbol(T.getCustomPragmaVal(preservesRecord)))
|
|
elif T.hasCustomPragma(preservesTuple):
|
|
result = initSequence(0)
|
|
for label, field in x.fieldPairs:
|
|
when x.dot(label).hasCustomPragma(preservesTupleTail):
|
|
for y in field.items:
|
|
result.sequence.add(toPreserves(y))
|
|
# TODO: what if there are fields after the tail?
|
|
else:
|
|
var pr = fieldToPreserve(label, field)
|
|
result.sequence.add(pr)
|
|
elif T.hasCustomPragma(preservesDictionary):
|
|
result = initDictionary()
|
|
for key, val in x.fieldPairs:
|
|
when val is Option:
|
|
if val.isSome:
|
|
var pr = fieldToPreserve(key, val.get)
|
|
if x.dot(key).hasCustomPragma(preservesEmbedded):
|
|
pr.embedded = true
|
|
result[key.toSymbol] = pr
|
|
else:
|
|
var pr = fieldToPreserve(key, val)
|
|
if x.dot(key).hasCustomPragma(preservesEmbedded):
|
|
pr.embedded = true
|
|
result[key.toSymbol] = pr
|
|
sortDict(result)
|
|
else:
|
|
{.warning: "failed to preserve object " & $T .}
|
|
result = toPreservesHook(x)
|
|
else:
|
|
{.warning: "failed to preserve " & $T .}
|
|
result = toPreservesHook(x)
|
|
# the hook doesn't compile but produces a useful error
|
|
trace T, " -> ", result
|
|
|
|
proc toPreserve*[T](x: T; E: typedesc = void): Value {.deprecated.} =
|
|
x.topreserves()
|
|
|
|
proc toPreservesHook*(a: Atom): Value =
|
|
result = Value(kind: a.kind)
|
|
case a.kind
|
|
of pkBoolean:
|
|
result.bool = a.bool
|
|
of pkFloat:
|
|
result.float = a.float
|
|
of pkRegister:
|
|
result.register = a.register
|
|
of pkBigInt:
|
|
result.bigint = a.bigint
|
|
of pkString:
|
|
result.string = a.string
|
|
of pkByteString:
|
|
result.bytes = a.bytes
|
|
of pkSymbol:
|
|
result.symbol = a.symbol
|
|
else:
|
|
discard
|
|
|
|
proc toPreservesHook*[T](set: HashSet[T]): Value =
|
|
## Hook for preserving ``HashSet``.
|
|
result = Value(kind: pkSet, set: newSeqOfCap[Value](set.len))
|
|
for e in set: result.incl toPreserves(e)
|
|
cannonicalize(result)
|
|
|
|
proc toPreservesHook*[A, B](table: Table[A, B]|TableRef[A, B]): Value =
|
|
## Hook for preserving ``Table``.
|
|
result = initDictionary()
|
|
for k, v in table.pairs:
|
|
result[toPreserves(k)] = toPreserves(v)
|
|
sortDict(result)
|
|
|
|
proc toPreservesHook*(o: Option): Value =
|
|
o.get.toPreserves
|
|
|
|
proc fromAtom*[T](v: var T; a: ATom): bool =
|
|
if T is Atom:
|
|
v = a
|
|
result = true
|
|
if T is Value:
|
|
v = a.toPreservesHook
|
|
result = true
|
|
elif T is enum:
|
|
if a.kind == pkSymbol:
|
|
try:
|
|
v = parseEnum[T](string a.symbol)
|
|
result = true
|
|
except ValueError: discard
|
|
elif T is bool:
|
|
if a.kind == pkBoolean:
|
|
v = a.bool
|
|
result = true
|
|
elif T is SomeInteger:
|
|
if a.kind == pkRegister:
|
|
result = a.register.T < high(T)
|
|
if result:
|
|
v = T a.register
|
|
elif a.kind == pkBigInt:
|
|
var o = toInt[T](a.bigint)
|
|
result = o.isSome
|
|
if result: v = o.get
|
|
elif T is seq[byte]:
|
|
if a.kind == pkByteString:
|
|
v = a.bytes
|
|
result = true
|
|
elif T is SomeFloat:
|
|
case a.kind
|
|
of pkFloat:
|
|
v = T a.float
|
|
result = true
|
|
else: discard
|
|
elif T is Ordinal | SomeInteger:
|
|
if a.kind == pkRegister:
|
|
v = T(a.register)
|
|
result = int(v) == a.register
|
|
elif a.kind == pkBigInt:
|
|
var o = toInt[T](a.bigint)
|
|
if o.isSome:
|
|
v = get o
|
|
result = true
|
|
elif T is string:
|
|
if a.kind == pkString:
|
|
v = a.string
|
|
result = true
|
|
elif T is Symbol:
|
|
if a.kind == pkSymbol:
|
|
v = a.symbol
|
|
result = true
|
|
elif T is distinct:
|
|
result = fromAtom(v.distinctBase, a)
|
|
|
|
proc fromPreserves*[T](v: var T; pr: Value): bool =
|
|
## Inplace version of `preservesTo`. Returns ``true`` on
|
|
## a complete match, otherwise returns ``false``.
|
|
## Can be customized with `fromPreservesHook(x: T; var pr: Value): bool`.
|
|
## Any ``fromPreservesHook`` that does not compile will be discarded;
|
|
## *Write tests for your hooks!*
|
|
##
|
|
## When `tracePreserves` is defined (`-d:tracePreserves`) a diagnostic
|
|
## trace is printing during `fromPreserve`.
|
|
# TODO: {.raises: [].}
|
|
runnableExamples:
|
|
type Foo {.preservesRecord: "foo".} = object
|
|
x, y: int
|
|
var foo: Foo
|
|
assert(fromPreserves(foo, parsePreserves("""<foo 1 2>""")))
|
|
assert(foo.x == 1)
|
|
assert(foo.y == 2)
|
|
when T is Value:
|
|
v = pr
|
|
result = true
|
|
elif T is Value:
|
|
v = pr
|
|
result = true
|
|
elif compiles(fromPreservesHook(v, pr)):
|
|
result = fromPreservesHook(v, pr)
|
|
elif T is enum:
|
|
if pr.isSymbol:
|
|
try:
|
|
v = parseEnum[T](string pr.symbol)
|
|
result = true
|
|
except ValueError: discard
|
|
elif T is bool:
|
|
if pr.kind == pkBoolean:
|
|
v = pr.bool
|
|
result = true
|
|
elif T is SomeInteger:
|
|
if pr.kind == pkRegister:
|
|
v = T(pr.register)
|
|
result = true
|
|
elif T is seq[byte]:
|
|
if pr.kind == pkByteString:
|
|
v = pr.bytes
|
|
result = true
|
|
elif T is seq:
|
|
if pr.kind == pkSequence:
|
|
v.setLen(pr.len)
|
|
result = true
|
|
for i, e in pr.sequence:
|
|
result = result and fromPreserves(v[i], pr.sequence[i])
|
|
if not result:
|
|
v.setLen 0
|
|
break
|
|
elif T is SomeFloat:
|
|
if pr.kind == pkFloat:
|
|
v = (T)pr.float
|
|
result = true
|
|
elif T is Ordinal | SomeInteger:
|
|
case pr.kind
|
|
of pkRegister:
|
|
v = (T)pr.register
|
|
result = true
|
|
of pkBigInt:
|
|
var o = toInt[T](pr.bigint)
|
|
result = o.isSome
|
|
if result: v = get o
|
|
else: discard
|
|
elif T is string:
|
|
if pr.kind == pkString:
|
|
v = pr.string
|
|
result = true
|
|
elif T is Symbol:
|
|
if pr.kind == pkSymbol:
|
|
v = pr.symbol
|
|
result = true
|
|
elif T is distinct:
|
|
result = fromPreserve(v.distinctBase, pr)
|
|
elif T is tuple:
|
|
case pr.kind
|
|
of pkRecord, pkSequence:
|
|
if pr.len <= tupleLen(T):
|
|
result = true
|
|
var i {.used.}: int
|
|
for f in fields(v):
|
|
if result and i < pr.len:
|
|
result = result and fromPreserves(f, pr[i])
|
|
inc i
|
|
of pkDictionary:
|
|
if tupleLen(T) <= pr.len:
|
|
result = true
|
|
for key, val in fieldPairs(v):
|
|
let pv = step(pr, toSymbol(key))
|
|
result =
|
|
if pv.isSome: fromPreserves(val, get pv)
|
|
else: false
|
|
if not result: break
|
|
else: discard
|
|
elif T is ref:
|
|
when T.hasCustomPragma(preservesEmbedded):
|
|
if (pr.kind == pkEmbedded) and (pr.embeddedRef of T):
|
|
v = T(pr.embeddedRef)
|
|
result = true
|
|
else:
|
|
if isNil(v): new(v)
|
|
result = fromPreserves(v[], pr)
|
|
elif T is object:
|
|
template fieldFromPreserve[T](key: string; val: T; pr: Value): bool {.used.} =
|
|
when v.dot(key).hasCustomPragma(preservesLiteral):
|
|
const atom = v.dot(key).getCustomPragmaVal(preservesLiteral).parsePreservesAtom
|
|
pr == atom.toPreservesHook()
|
|
elif v.dot(key).hasCustomPragma(preservesEmbedded):
|
|
when T is EmbeddedRef:
|
|
if pr.kind == pkEmbedded and pr.embeddedRef of T:
|
|
val = T(pr.embeddedRef)
|
|
true
|
|
else: false
|
|
else: fromPreserves(val, pr)
|
|
else: fromPreserves(val, pr)
|
|
when T.hasCustomPragma(unpreservable):
|
|
raiseAssert($T & " is unpreservable")
|
|
elif T.hasCustomPragma(preservesRecord):
|
|
if pr.isRecord and pr.label.isSymbol(T.getCustomPragmaVal(preservesRecord)):
|
|
result = true
|
|
var i: int
|
|
for name, field in fieldPairs(v):
|
|
when v.dot(name).hasCustomPragma(preservesTupleTail):
|
|
v.dot(name).setLen(pr.record.len.pred - i)
|
|
var j: int
|
|
while result and i < pr.record.high:
|
|
result = result and fromPreserves(v.dot(name)[j], pr.record[i])
|
|
inc i
|
|
inc j
|
|
break
|
|
else:
|
|
if result and i <= pr.len:
|
|
result = result and fieldFromPreserve(name, field, pr.record[i])
|
|
inc i
|
|
result = result and (i <= pr.len)
|
|
elif T.hasCustomPragma(preservesTuple):
|
|
if pr.isSequence:
|
|
result = true
|
|
var i: int
|
|
for name, field in fieldPairs(v):
|
|
when v.dot(name).hasCustomPragma(preservesTupleTail):
|
|
if pr.len >= i:
|
|
setLen(v.dot(name), pr.len - i)
|
|
var j: int
|
|
while result and i < pr.len:
|
|
result = result and fieldFromPreserve(name, v.dot(name)[j], pr.sequence[i])
|
|
inc i
|
|
inc j
|
|
else:
|
|
if result and i < pr.len:
|
|
result = result and fieldFromPreserve(name, field, pr.sequence[i])
|
|
inc i
|
|
result = result and (i == pr.len)
|
|
elif T.hasCustomPragma(preservesDictionary):
|
|
if pr.isDictionary:
|
|
result = true
|
|
var i: int
|
|
for key, field in fieldPairs(v):
|
|
if not result: break
|
|
let val = step(pr, key.toSymbol)
|
|
when field is Option:
|
|
if val.isSome:
|
|
discard fieldFromPreserve(key, v.dot(key), val.get)
|
|
else:
|
|
inc i
|
|
result = result and val.isSome
|
|
if result:
|
|
var pr = val.get
|
|
result = result and fieldFromPreserve(key, v.dot(key), pr)
|
|
result = result and (i <= pr.len)
|
|
elif T.hasCustomPragma(preservesOr):
|
|
for kind in typeof(T.orKind):
|
|
v = T(orKind: kind)
|
|
var fieldWasFound = false
|
|
for key, val in fieldPairs(v):
|
|
when key != "orKind": # fieldPairs unwraps early
|
|
result = fieldFromPreserve(key, v.dot(key), pr)
|
|
fieldWasFound = true
|
|
break
|
|
if not fieldWasFound:
|
|
# hopefully a `discard` of-branch, so discard `pr`
|
|
result = true
|
|
if result: break
|
|
else:
|
|
if pr.isDictionary:
|
|
result = true
|
|
var i: int
|
|
for key, _ in fieldPairs(v):
|
|
if not result: break
|
|
let val = step(pr, key.toSymbol)
|
|
result = result and val.isSome
|
|
if result:
|
|
result = result and fieldFromPreserve(key, v.dot(key), val.get)
|
|
inc i
|
|
result = result and (i <= pr.len)
|
|
else:
|
|
result = fromPreservesHook(v, pr)
|
|
# a previous branch determined that the hook does not compile but
|
|
# calling it here explicitly produces a reasonable compiler error
|
|
when defined(tracePreserves):
|
|
if not result:
|
|
trace T, " !- ", pr
|
|
else:
|
|
trace T, " <- ", pr
|
|
|
|
proc fromPreserve*[T](v: var T; pr: Value): bool {.deprecated.} = v.frompreserves(pr)
|
|
|
|
proc preservesTo*(pr: Value; T: typedesc): Option[T] =
|
|
## Reverse of `toPreserve`.
|
|
# TODO: {.raises: [].}
|
|
runnableExamples:
|
|
import std/options
|
|
type Foo {.preservesRecord: "foo".} = object
|
|
x, y: int
|
|
|
|
assert(parsePreserves("""<foo "abc">""").preservesTo(Foo).isNone)
|
|
assert(parsePreserves("""<bar 1 2>""").preservesTo(Foo).isNone)
|
|
assert(parsePreserves("""<foo 1 2>""").preservesTo(Foo).isSome)
|
|
var v: T
|
|
if fromPreserves(v, pr):
|
|
result = some(move v)
|
|
|
|
proc fromPreservesHook*[T, E](v: var set[T]; pr: Value): bool =
|
|
## Hook for unpreserving a `set`.
|
|
if pr.kind == pkSet:
|
|
reset v
|
|
result = true
|
|
var vv: T
|
|
for e in pr.set:
|
|
result = fromPreserves(vv, e)
|
|
if result:
|
|
v.incl vv
|
|
else:
|
|
reset v
|
|
break
|
|
|
|
proc fromPreservesHook*[T, E](set: var HashSet[T]; pr: Value): bool =
|
|
## Hook for preserving ``HashSet``.
|
|
if pr.kind == pkSet:
|
|
result = true
|
|
set.init(pr.set.len)
|
|
var e: T
|
|
for pe in pr.set:
|
|
result = fromPreserves(e, pe)
|
|
if not result: break
|
|
set.incl(move e)
|
|
|
|
proc fromPreservesHook*[A,B](t: var (Table[A,B]|TableRef[A,B]); pr: Value): bool =
|
|
if pr.isDictionary:
|
|
when t is TableRef[A,B]:
|
|
if t.isNil: new t
|
|
result = true
|
|
var a: A
|
|
var b: B
|
|
for (k, v) in pr.dict.items:
|
|
result = fromPreserves(a, k) and fromPreserves(b, v)
|
|
if not result:
|
|
clear t
|
|
break
|
|
t[move a] = move b
|
|
|
|
proc fromPreservesHook*[T](opt: var Option[T]; pr: Value): bool =
|
|
opt = some(default T)
|
|
result = opt.get.fromPreserves(pr)
|
|
if not result:
|
|
opt = none(T)
|
|
|
|
when isMainModule:
|
|
var t: Table[int, string]
|
|
var pr = t.toPreservesHook()
|
|
assert fromPreservesHook(t, pr)
|
|
|
|
func step(pr, idx: Value): Option[Value] =
|
|
if pr.isDictionary:
|
|
for (k, v) in pr.dict.items:
|
|
if k == idx:
|
|
result = some(v)
|
|
break
|
|
elif (pr.isRecord or pr.isSequence):
|
|
var o = idx.toInt
|
|
if o.isSome:
|
|
var i = get o
|
|
if i < pr.len:
|
|
result = some(pr[i])
|
|
|
|
func step*(pr: Value; path: varargs[Value, toPreserves]): Option[Value] =
|
|
## Step into `pr` by index `idx`.
|
|
## Works for sequences, records, and dictionaries.
|
|
runnableExamples:
|
|
import std/options
|
|
assert step(parsePreserves("""<foo 1 2>"""), 1.toPreserves) == some(2.toPreserves)
|
|
assert step(parsePreserves("""{ foo: 1 bar: 2}"""), "foo".toSymbol) == some(1.toPreserves)
|
|
assert step(parsePreserves("""[ ]"""), 1.toPreserves) == none(Value)
|
|
result = some(pr)
|
|
for index in path:
|
|
if result.isSome:
|
|
result = step(result.get, index)
|
|
|
|
func step*(pr: Value; key: Symbol): Option[Value] =
|
|
## Step into dictionary by a `Symbol` key.
|
|
if pr.isDictionary:
|
|
for (k, v) in pr.dict.items:
|
|
if k.isSymbol and k.symbol == key:
|
|
result = some(v)
|
|
break
|
|
|
|
proc apply*(result: var Value; op: proc(_: var Value) {.gcsafe.}) {.gcsafe.} =
|
|
proc recurse(result: var Value) = apply(result, op)
|
|
op(result)
|
|
case result.kind
|
|
of pkBoolean, pkFloat, pkRegister, pkBigInt,
|
|
pkString, pkByteString, pkSymbol, pkEmbedded:
|
|
discard
|
|
of pkRecord:
|
|
apply(result.record, recurse)
|
|
of pkSequence:
|
|
apply(result.sequence, recurse)
|
|
of pkSet:
|
|
apply(result.set, recurse)
|
|
of pkDictionary:
|
|
apply(result.dict) do (e: var DictEntry):
|
|
recurse(e.key)
|
|
recurse(e.val)
|
|
cannonicalize(result)
|
|
|
|
proc mapEmbeds*(pr: sink Value; op: proc (x: Value): Value): Value =
|
|
## Process all embeds in a `Value`.
|
|
case pr.kind
|
|
of pkBoolean, pkFloat, pkRegister, pkBigInt,
|
|
pkString, pkByteString, pkSymbol, pkEmbedded:
|
|
result = pr
|
|
of pkRecord:
|
|
result = Value(kind: pr.kind)
|
|
result.record = map(pr.record) do (x: Value) -> Value:
|
|
mapEmbeds(x, op)
|
|
of pkSequence:
|
|
result = Value(kind: pr.kind)
|
|
result.sequence = map(pr.sequence) do (x: Value) -> Value:
|
|
mapEmbeds(x, op)
|
|
of pkSet:
|
|
result = Value(kind: pr.kind)
|
|
result.set = map(pr.set) do (x: Value) -> Value:
|
|
mapEmbeds(x, op)
|
|
of pkDictionary:
|
|
result = Value(kind: pr.kind)
|
|
result.dict = map(pr.dict) do (e: DictEntry) -> DictEntry:
|
|
(mapEmbeds(e.key, op), mapEmbeds(e.val, op))
|
|
if pr.embedded or pr.kind == pkEmbedded:
|
|
result = op(result)
|
|
result.embedded = true
|
|
cannonicalize(result)
|
|
# TODO: is this necessary?
|
|
|
|
proc getOrDefault*[T, V](pr: Value; key: string; default: V): V =
|
|
## Retrieves the value of `pr[key]` if `pr` is a dictionary containing `key`
|
|
## or returns the `default` value.
|
|
var sym = toSymbol(key, T)
|
|
if pr.kind == pkDictionary:
|
|
for (k, v) in pr.dict:
|
|
if sym == k:
|
|
if fromPreserves(result, v): return
|
|
else: break
|
|
default
|