repl: put last value in _

1 files changed, 3 insertions, 0 deletions

diff --git a/repl.c b/repl.c
index bf1e41e..7324b2f 100644
--- a/repl.c
+++ b/repl.c
@@ -75,12 +75,15 @@ main(int argc, char **argv) {
         using_history();
         CHECK(pez_pushglobal(cx, "printf"));
 
+        CHECK(pez_pushvoid(cx) && pez_setglobal(cx, "_"));
         while ((src = readline("> "))) {
             add_history(src);
             if (!pez_eval_str(cx, "<repl>", src)) {
                 fprintf(stderr, "error: %s\n", pez_geterr(cx));
             } else {
                 assert(pez_top(cx) == 2);
+                CHECK(pez_push(cx, -1));
+                CHECK(pez_setglobal(cx, "_"));
                 CHECK(printtop(cx));
                 printf("\n");
             }