From af2efbc990aa6d8cd6d8732b43d3cd85ebdca3c8 Mon Sep 17 00:00:00 2001
From: Yuichi Nishiwaki <yuichi.nishiwaki@gmail.com>
Date: Thu, 5 Dec 2013 17:10:28 +0900
Subject: [PATCH] add -e option (for one liner script)

---
 tools/main.c | 113 ++++++++++++++++++++++++++++++++++++++-------------
 1 file changed, 84 insertions(+), 29 deletions(-)

diff --git a/tools/main.c b/tools/main.c
index ff6a5545..96d557fd 100644
--- a/tools/main.c
+++ b/tools/main.c
@@ -14,8 +14,6 @@
 #define CODE_MAX_LENGTH 1024
 #define LINE_MAX_LENGTH 256
 
-static char *fname;
-
 void
 print_help(void)
 {
@@ -30,30 +28,6 @@ print_help(void)
   puts(help);
 }
 
-bool
-parse_opt(int argc, char *argv[])
-{
-  int r;
-
-  while (~(r = getopt(argc, argv, "h"))) {
-    switch (r) {
-    case 'h':
-      print_help();
-      exit(0);
-    }
-  }
-  argc -= optind;
-  argv += optind;
-
-  if (argc == 0) {
-    return 1;
-  }
-  else {
-    fname = argv[0];
-    return 0;
-  }
-}
-
 int
 repl(pic_state *pic)
 {
@@ -200,6 +174,77 @@ exec_file(pic_state *pic, const char *fname)
   return 0;
 }
 
+static int
+exec_string(pic_state *pic, const char *str)
+{
+  int n, i;
+  pic_value vs, v;
+  struct pic_proc *proc;
+  int ai = pic_gc_arena_preserve(pic);
+
+  n = pic_parse_cstr(pic, str, &vs);
+  if (n < 0) {
+    return 1;
+  }
+
+  for (i = 0; i < n; ++i) {
+    v = pic_car(pic, vs);
+
+    proc = pic_codegen(pic, v);
+    if (proc == NULL) {
+      return 1;
+    }
+    v = pic_apply(pic, proc, pic_nil_value());
+    if (pic_undef_p(v)) {
+      return 1;
+    }
+
+    vs = pic_cdr(pic, vs);
+
+    pic_gc_arena_restore(pic, ai);
+  }
+
+  return 0;
+}
+
+static char *fname;
+static char *one_liner;
+
+enum {
+  NO_MODE = 0,
+  INTERACTIVE_MODE,
+  FILE_EXEC_MODE,
+  ONE_LINER_MODE,
+} mode;
+
+void
+parse_opt(int argc, char *argv[])
+{
+  int r;
+
+  while (~(r = getopt(argc, argv, "he:"))) {
+    switch (r) {
+    case 'h':
+      print_help();
+      exit(0);
+    case 'e':
+      one_liner = optarg;
+      mode = ONE_LINER_MODE;
+    }
+  }
+  argc -= optind;
+  argv += optind;
+
+  if (argc == 0) {
+    if (mode == NO_MODE)
+      mode = INTERACTIVE_MODE;
+  }
+  else {
+    fname = argv[0];
+    mode = FILE_EXEC_MODE;
+  }
+}
+
 int
 main(int argc, char *argv[], char **envp)
 {
@@ -208,11 +253,21 @@ main(int argc, char *argv[], char **envp)
 
   pic = pic_open(argc, argv, envp);
 
-  if (parse_opt(argc, argv)) {
+  parse_opt(argc, argv);
+
+  switch (mode) {
+  case NO_MODE:
+    puts("logic flaw");
+    abort();
+  case INTERACTIVE_MODE:
     res = repl(pic);
-  }
-  else {
+    break;
+  case FILE_EXEC_MODE:
     res = exec_file(pic, fname);
+    break;
+  case ONE_LINER_MODE:
+    res = exec_string(pic, one_liner);
+    break;
   }
 
   pic_close(pic);