From 22ba63df162d3ac980f5c33210929177e0a3dfe0 Mon Sep 17 00:00:00 2001 From: Eelco Dolstra Date: Wed, 16 Aug 2006 21:59:53 +0000 Subject: [PATCH] * More XML output. `--strict' to strictly evaluate attribute sets and so on. * Removed `--print-args', it's subsumed by `--eval-only --xml'. --- src/nix-instantiate/help.txt | 9 +++ src/nix-instantiate/main.cc | 128 ++++++++++++++++++++++------------- 2 files changed, 91 insertions(+), 46 deletions(-) diff --git a/src/nix-instantiate/help.txt b/src/nix-instantiate/help.txt index 18b1d8bd0d..fa0a4590ee 100644 --- a/src/nix-instantiate/help.txt +++ b/src/nix-instantiate/help.txt @@ -18,3 +18,12 @@ Options: --attr / -A PATH: select an attribute from the top-level expression --add-root: add garbage collector roots for the result + +For --eval-only / --parse-only: + + --xml: print an XML representation of the abstract syntax tree + +For --eval-only: + + --strict: compute attributes and list elements, rather than being + lazy diff --git a/src/nix-instantiate/main.cc b/src/nix-instantiate/main.cc index 3e0b10f6a0..3f42de3f78 100644 --- a/src/nix-instantiate/main.cc +++ b/src/nix-instantiate/main.cc @@ -42,13 +42,14 @@ static XMLAttrs singletonAttrs(const string & name, const string & value) } -static void printTermAsXML(EvalState & state, Expr e, XMLWriter & doc) +static void printTermAsXML(Expr e, XMLWriter & doc) { XMLAttrs attrs; ATerm s; int i; - ATermList as; - + ATermList as, formals; + ATerm body, pos; + if (matchStr(e, s)) doc.writeEmptyElement("string", singletonAttrs("value", aterm2String(s))); @@ -62,7 +63,13 @@ static void printTermAsXML(EvalState & state, Expr e, XMLWriter & doc) doc.writeEmptyElement("null"); else if (matchInt(e, i)) - doc.writeEmptyElement("int",singletonAttrs("value", (format("%1%") % i).str())); + doc.writeEmptyElement("int", singletonAttrs("value", (format("%1%") % i).str())); + + else if (e == eTrue) + doc.writeEmptyElement("bool", singletonAttrs("value", "true")); + + else if (e == eFalse) + doc.writeEmptyElement("bool", singletonAttrs("value", "false")); else if (matchAttrs(e, as)) { XMLOpenElement _(doc, "attrs"); @@ -70,57 +77,44 @@ static void printTermAsXML(EvalState & state, Expr e, XMLWriter & doc) queryAllAttrs(e, attrs); for (ATermMap::const_iterator i = attrs.begin(); i != attrs.end(); ++i) { XMLOpenElement _(doc, "attr", singletonAttrs("name", aterm2String(i->key))); - printTermAsXML(state, i->value, doc); + printTermAsXML(i->value, doc); + } + } + + else if (matchFunction(e, formals, body, pos)) { + XMLOpenElement _(doc, "function"); + + for (ATermIterator i(formals); i; ++i) { + Expr name; ValidValues valids; ATerm dummy; + if (!matchFormal(*i, name, valids, dummy)) abort(); + XMLOpenElement _(doc, "arg", singletonAttrs("name", aterm2String(name))); + + ATermList valids2; + if (matchValidValues(valids, valids2)) { + for (ATermIterator j(valids2); j; ++j) { + XMLOpenElement _(doc, "value"); + printTermAsXML(*j, doc); + } + } } } else - doc.writeEmptyElement("unknown"); + doc.writeEmptyElement("unevaluated"); } static void printResult(EvalState & state, Expr e, - bool evalOnly, bool printArgs, bool xmlOutput, - const ATermMap & autoArgs) + bool evalOnly, bool xmlOutput, const ATermMap & autoArgs) { if (evalOnly) if (xmlOutput) { XMLWriter doc(true, cout); XMLOpenElement root(doc, "expr"); - printTermAsXML(state, e, doc); + printTermAsXML(e, doc); } else cout << format("%1%\n") % e; - else if (printArgs) { - XMLWriter doc(true, cout); - XMLOpenElement root(doc, "args"); - - ATermList formals; - ATerm body, pos; - - if (matchFunction(e, formals, body, pos)) { - for (ATermIterator i(formals); i; ++i) { - Expr name; ValidValues valids; ATerm dummy; - if (!matchFormal(*i, name, valids, dummy)) abort(); - - XMLAttrs attrs; - attrs["name"] = aterm2String(name); - XMLOpenElement elem(doc, "arg", attrs); - - ATermList valids2; - if (matchValidValues(valids, valids2)) { - for (ATermIterator j(valids2); j; ++j) { - Expr e = evalExpr(state, *j); - XMLAttrs attrs; - attrs["value"] = showValue(e); - XMLOpenElement elem(doc, "value", attrs); - } - } - } - } else - printMsg(lvlError, "warning: expression does not evaluate to a function"); - } - else { DrvInfos drvs; getDerivations(state, e, "", autoArgs, drvs); @@ -138,6 +132,49 @@ static void printResult(EvalState & state, Expr e, } +Expr strictEval(EvalState & state, Expr e) +{ + e = evalExpr(state, e); + + ATermList as; + + if (matchAttrs(e, as)) { + ATermList as2 = ATempty; + for (ATermIterator i(as); i; ++i) { + ATerm name; Expr e; ATerm pos; + if (!matchBind(*i, name, e, pos)) abort(); /* can't happen */ + as2 = ATinsert(as2, makeBind(name, strictEval(state, e), pos)); + } + return makeAttrs(ATreverse(as2)); + } + + ATermList formals; + ATerm body, pos; + + if (matchFunction(e, formals, body, pos)) { + ATermList formals2 = ATempty; + + for (ATermIterator i(formals); i; ++i) { + Expr name; ValidValues valids; ATerm dummy; + if (!matchFormal(*i, name, valids, dummy)) abort(); + + ATermList valids2; + if (matchValidValues(valids, valids2)) { + ATermList valids3 = ATempty; + for (ATermIterator j(valids2); j; ++j) + valids3 = ATinsert(valids3, strictEval(state, *j)); + valids = makeValidValues(ATreverse(valids3)); + } + + formals2 = ATinsert(formals2, makeFormal(name, valids, dummy)); + } + return makeFunction(ATreverse(formals2), body, pos); + } + + return e; +} + + void run(Strings args) { EvalState state; @@ -145,8 +182,8 @@ void run(Strings args) bool readStdin = false; bool evalOnly = false; bool parseOnly = false; - bool printArgs = false; bool xmlOutput = false; + bool strict = false; string attrPath; ATermMap autoArgs(128); @@ -165,10 +202,6 @@ void run(Strings args) readOnlyMode = true; parseOnly = evalOnly = true; } - else if (arg == "--print-args") { - readOnlyMode = true; - printArgs = true; - } else if (arg == "--attr" || arg == "-A") { if (i == args.end()) throw UsageError("`--attr' requires an argument"); @@ -192,6 +225,8 @@ void run(Strings args) indirectRoot = true; else if (arg == "--xml") xmlOutput = true; + else if (arg == "--strict") + strict = true; else if (arg[0] == '-') throw UsageError(format("unknown flag `%1%'") % arg); else @@ -203,7 +238,7 @@ void run(Strings args) if (readStdin) { Expr e = findAlongAttrPath(state, attrPath, parseStdin(state)); if (!parseOnly) e = evalExpr(state, e); - printResult(state, e, evalOnly, printArgs, xmlOutput, autoArgs); + printResult(state, e, evalOnly, xmlOutput, autoArgs); } for (Strings::iterator i = files.begin(); @@ -213,7 +248,8 @@ void run(Strings args) Expr e = findAlongAttrPath(state, attrPath, parseExprFromFile(state, path)); if (!parseOnly) e = evalExpr(state, e); - printResult(state, e, evalOnly, printArgs, xmlOutput, autoArgs); + if (strict) e = strictEval(state, e); + printResult(state, e, evalOnly, xmlOutput, autoArgs); } printEvalStats(state);