conservatory
/
picrin
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add -e option (for one liner script)

This commit is contained in:
Yuichi Nishiwaki 2013-12-05 17:10:28 +09:00
parent 220c7856cd
commit af2efbc990
1 changed files with 84 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

113
tools/main.c
View File

@ -14,8 +14,6 @@
#define CODE_MAX_LENGTH 1024 #define CODE_MAX_LENGTH 1024
#define LINE_MAX_LENGTH 256 #define LINE_MAX_LENGTH 256
static char *fname;
void void
print_help(void) print_help(void)
{ {
@ -30,30 +28,6 @@ print_help(void)
puts(help); 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 int
repl(pic_state *pic) repl(pic_state *pic)
{ {
@ -200,6 +174,77 @@ exec_file(pic_state *pic, const char *fname)
return 0; 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 int
main(int argc, char *argv[], char **envp) main(int argc, char *argv[], char **envp)
{ {
@ -208,11 +253,21 @@ main(int argc, char *argv[], char **envp)
pic = pic_open(argc, argv, 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); res = repl(pic);
} break;
else { case FILE_EXEC_MODE:
res = exec_file(pic, fname); res = exec_file(pic, fname);
break;
case ONE_LINER_MODE:
res = exec_string(pic, one_liner);
break;
} }
pic_close(pic); pic_close(pic);