forked from lix-project/lix
Support adding variables to the scope
The command ":a <expr>" evaluates <expr> and adds the attributes in the resulting attribute set to the interpreter scope. For instance: nix-repl> :a import <nixpkgs> {} nix-repl> lib.range 0 10 [ 0 1 2 3 4 5 6 7 8 9 10 ]
This commit is contained in:
parent
e90569905e
commit
287c88ca59
95
nix-repl.cc
95
nix-repl.cc
|
@ -6,6 +6,8 @@
|
||||||
|
|
||||||
#include "shared.hh"
|
#include "shared.hh"
|
||||||
#include "eval.hh"
|
#include "eval.hh"
|
||||||
|
#include "eval-inline.hh"
|
||||||
|
#include "store-api.hh"
|
||||||
|
|
||||||
using namespace std;
|
using namespace std;
|
||||||
using namespace nix;
|
using namespace nix;
|
||||||
|
@ -14,6 +16,23 @@ using namespace nix;
|
||||||
string programId = "nix-repl";
|
string programId = "nix-repl";
|
||||||
|
|
||||||
|
|
||||||
|
struct NixRepl
|
||||||
|
{
|
||||||
|
string curDir;
|
||||||
|
EvalState state;
|
||||||
|
|
||||||
|
StaticEnv staticEnv;
|
||||||
|
Env * env;
|
||||||
|
int displ;
|
||||||
|
|
||||||
|
NixRepl();
|
||||||
|
void mainLoop();
|
||||||
|
void processLine(string line);
|
||||||
|
void addVar(const Symbol & name, Value * v);
|
||||||
|
Expr * parseString(string s);
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
void printHelp()
|
void printHelp()
|
||||||
{
|
{
|
||||||
std::cout << "Usage: nix-repl\n";
|
std::cout << "Usage: nix-repl\n";
|
||||||
|
@ -22,7 +41,7 @@ void printHelp()
|
||||||
|
|
||||||
bool getLine(string & line)
|
bool getLine(string & line)
|
||||||
{
|
{
|
||||||
char * s = readline ("nix-repl> ");
|
char * s = readline("nix-repl> ");
|
||||||
if (!s) return false;
|
if (!s) return false;
|
||||||
line = chomp(string(s));
|
line = chomp(string(s));
|
||||||
free(s);
|
free(s);
|
||||||
|
@ -31,21 +50,33 @@ bool getLine(string & line)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
void run(nix::Strings args)
|
NixRepl::NixRepl()
|
||||||
|
: staticEnv(false, &state.staticBaseEnv)
|
||||||
{
|
{
|
||||||
EvalState state;
|
curDir = absPath(".");
|
||||||
Path curDir = absPath(".");
|
|
||||||
|
env = &state.allocEnv(32768);
|
||||||
|
env->up = &state.baseEnv;
|
||||||
|
displ = 0;
|
||||||
|
|
||||||
|
store = openStore();
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
void NixRepl::mainLoop()
|
||||||
|
{
|
||||||
|
std::cerr << "Welcome to Nix version " << NIX_VERSION << ". Type :? for help." << std::endl << std::endl;
|
||||||
|
|
||||||
while (true) {
|
while (true) {
|
||||||
string line;
|
string line;
|
||||||
if (!getLine(line)) break;
|
if (!getLine(line)) break;
|
||||||
|
|
||||||
|
/* Remove preceeding whitespace. */
|
||||||
|
size_t n = line.find_first_not_of(" \n\r\t");
|
||||||
|
if (n != string::npos) line = string(line, n);
|
||||||
|
|
||||||
try {
|
try {
|
||||||
Expr * e = state.parseExprFromString(line, curDir);
|
processLine(line);
|
||||||
Value v;
|
|
||||||
state.eval(e, v);
|
|
||||||
state.strictForceValue(v);
|
|
||||||
std::cout << v << std::endl;
|
|
||||||
} catch (Error & e) {
|
} catch (Error & e) {
|
||||||
printMsg(lvlError, e.msg());
|
printMsg(lvlError, e.msg());
|
||||||
}
|
}
|
||||||
|
@ -55,3 +86,49 @@ void run(nix::Strings args)
|
||||||
|
|
||||||
std::cout << std::endl;
|
std::cout << std::endl;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
void NixRepl::processLine(string line)
|
||||||
|
{
|
||||||
|
if (string(line, 0, 2) == ":a") {
|
||||||
|
Expr * e = parseString(string(line, 2));
|
||||||
|
Value v;
|
||||||
|
e->eval(state, *env, v);
|
||||||
|
state.forceAttrs(v);
|
||||||
|
foreach (Bindings::iterator, i, *v.attrs)
|
||||||
|
addVar(i->name, i->value);
|
||||||
|
}
|
||||||
|
|
||||||
|
else if (string(line, 0, 1) == ":") {
|
||||||
|
throw Error(format("unknown command ‘%1%’") % string(line, 0, 2));
|
||||||
|
}
|
||||||
|
|
||||||
|
else {
|
||||||
|
Expr * e = parseString(line);
|
||||||
|
Value v;
|
||||||
|
e->eval(state, *env, v);
|
||||||
|
state.strictForceValue(v);
|
||||||
|
std::cout << v << std::endl;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
void NixRepl::addVar(const Symbol & name, Value * v)
|
||||||
|
{
|
||||||
|
staticEnv.vars[name] = displ;
|
||||||
|
env->values[displ++] = v;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
Expr * NixRepl::parseString(string s)
|
||||||
|
{
|
||||||
|
Expr * e = state.parseExprFromString(s, curDir, staticEnv);
|
||||||
|
return e;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
void run(nix::Strings args)
|
||||||
|
{
|
||||||
|
NixRepl repl;
|
||||||
|
repl.mainLoop();
|
||||||
|
}
|
||||||
|
|
Loading…
Reference in a new issue