-use strict;
-use warnings FATAL => qw(all);
-no if $] >= 5.018, warnings => "experimental::smartmatch";
-use File::Basename;
-use lib dirname (__FILE__);
-use readline qw(mal_readline set_rl_mode);
-use feature qw(switch);
-use Data::Dumper;
-
-use types qw($nil $true $false _sequential_Q _symbol_Q _list_Q);
-use reader;
-use printer;
-use env;
-use core qw($core_ns);
-use interop qw(pl_to_mal);
-
-# read
-sub READ {
- my $str = shift;
- return reader::read_str($str);
-}
-
-# eval
-sub is_pair {
- my ($x) = @_;
- return _sequential_Q($x) && scalar(@{$x->{val}}) > 0;
-}
-
-sub quasiquote {
- my ($ast) = @_;
- if (!is_pair($ast)) {
- return List->new([Symbol->new("quote"), $ast]);
- } elsif (_symbol_Q($ast->nth(0)) && ${$ast->nth(0)} eq 'unquote') {
- return $ast->nth(1);
- } elsif (is_pair($ast->nth(0)) && _symbol_Q($ast->nth(0)->nth(0)) &&
- ${$ast->nth(0)->nth(0)} eq 'splice-unquote') {
- return List->new([Symbol->new("concat"),
- $ast->nth(0)->nth(1),
- quasiquote($ast->rest())]);
- } else {
- return List->new([Symbol->new("cons"),
- quasiquote($ast->nth(0)),
- quasiquote($ast->rest())]);
- }
-}
-
-sub is_macro_call {
- my ($ast, $env) = @_;
- if (_list_Q($ast) &&
- _symbol_Q($ast->nth(0)) &&
- $env->find($ast->nth(0))) {
- my ($f) = $env->get($ast->nth(0));
- if ((ref $f) =~ /^Function/) {
- return $f->{ismacro};
- }
- }
- return 0;
-}
-
-sub macroexpand {
- my ($ast, $env) = @_;
- while (is_macro_call($ast, $env)) {
- my $mac = $env->get($ast->nth(0));
- $ast = $mac->apply($ast->rest());
- }
- return $ast;
-}
-
-
-sub eval_ast {
- my($ast, $env) = @_;
- given (ref $ast) {
- when (/^Symbol/) {
- $env->get($ast);
- }
- when (/^List/) {
- my @lst = map {EVAL($_, $env)} @{$ast->{val}};
- return List->new(\@lst);
- }
- when (/^Vector/) {
- my @lst = map {EVAL($_, $env)} @{$ast->{val}};
- return Vector->new(\@lst);
- }
- when (/^HashMap/) {
- my $new_hm = {};
- foreach my $k (keys( %{ $ast->{val} })) {
- $new_hm->{$k} = EVAL($ast->get($k), $env);
- }
- return HashMap->new($new_hm);
- }
- default {
- return $ast;
- }
- }
-}
-
-sub EVAL {
- my($ast, $env) = @_;
-
- while (1) {
-
- #print "EVAL: " . printer::_pr_str($ast) . "\n";
- if (! _list_Q($ast)) {
- return eval_ast($ast, $env);
- }
-
- # apply list
- $ast = macroexpand($ast, $env);
- if (! _list_Q($ast)) {
- return eval_ast($ast, $env);
- }
-
- my ($a0, $a1, $a2, $a3) = @{$ast->{val}};
- given ((ref $a0) =~ /^Symbol/ ? $$a0 : $a0) {
- when (/^def!$/) {
- my $res = EVAL($a2, $env);
- return $env->set($a1, $res);
- }
- when (/^let\*$/) {
- my $let_env = Env->new($env);
- for(my $i=0; $i < scalar(@{$a1->{val}}); $i+=2) {
- $let_env->set($a1->nth($i), EVAL($a1->nth($i+1), $let_env));
- }
- $ast = $a2;
- $env = $let_env;
- # Continue loop (TCO)
- }
- when (/^quote$/) {
- return $a1;
- }
- when (/^quasiquote$/) {
- $ast = quasiquote($a1);
- # Continue loop (TCO)
- }
- when (/^defmacro!$/) {
- my $func = EVAL($a2, $env);
- $func->{ismacro} = 1;
- return $env->set($a1, $func);
- }
- when (/^macroexpand$/) {
- return macroexpand($a1, $env);
- }
- when (/^try\*$/) {
- do {
- local $@;
- my $ret;
- eval {
- use autodie; # always "throw" errors
- $ret = EVAL($a1, $env);
- 1;
- } or do {
- my $err = $@;
- if ($a2 && ${$a2->nth(0)} eq "catch\*") {
- my $exc;
- if (ref $err) {
- $exc = $err;
- } else {
- $exc = String->new(substr $err, 0, -1);
- }
- return EVAL($a2->nth(2), Env->new($env,
- List->new([$a2->nth(1)]),
- List->new([$exc])));
- } else {
- die $err;
- }
- };
- return $ret;
- };
- }
- when (/^do$/) {
- eval_ast($ast->slice(1, $#{$ast->{val}}-1), $env);
- $ast = $ast->nth($#{$ast->{val}});
- # Continue loop (TCO)
- }
- when (/^if$/) {
- my $cond = EVAL($a1, $env);
- if ($cond eq $nil || $cond eq $false) {
- $ast = $a3 ? $a3 : $nil;
- } else {
- $ast = $a2;
- }
- # Continue loop (TCO)
- }
- when (/^fn\*$/) {
- return Function->new(\&EVAL, $a2, $env, $a1);
- }
- default {
- my $el = eval_ast($ast, $env);
- my $f = $el->nth(0);
- if ((ref $f) =~ /^Function/) {
- $ast = $f->{ast};
- $env = $f->gen_env($el->rest());
- # Continue loop (TCO)
- } else {
- return &{ $f }($el->rest());
- }
- }
- }
-
- } # TCO while loop
-}
-
-# print
-sub PRINT {
- my $exp = shift;
- return printer::_pr_str($exp);
-}
-
-# repl
-my $repl_env = Env->new();
-sub REP {
- my $str = shift;
- return PRINT(EVAL(READ($str), $repl_env));
-}
-
-# core.pl: defined using perl
-foreach my $n (%$core_ns) {
- $repl_env->set(Symbol->new($n), $core_ns->{$n});
-}
-$repl_env->set(Symbol->new('eval'), sub { EVAL($_[0]->nth(0), $repl_env); });
-my @_argv = map {String->new($_)} @ARGV[1..$#ARGV];
-$repl_env->set(Symbol->new('*ARGV*'), List->new(\@_argv));
-
-# core.mal: defined using the language itself
-REP("(def! not (fn* (a) (if a false true)))");
-REP("(def! load-file (fn* (f) (eval (read-string (str \"(do \" (slurp f) \")\")))))");
-REP("(defmacro! cond (fn* (& xs) (if (> (count xs) 0) (list 'if (first xs) (if (> (count xs) 1) (nth xs 1) (throw \"odd number of forms to cond\")) (cons 'cond (rest (rest xs)))))))");
-REP("(defmacro! or (fn* (& xs) (if (empty? xs) nil (if (= 1 (count xs)) (first xs) `(let* (or_FIXME ~(first xs)) (if or_FIXME or_FIXME (or ~@(rest xs))))))))");
-
-
-if (scalar(@ARGV) > 0 && $ARGV[0] eq "--raw") {
- set_rl_mode("raw");
- shift @ARGV;
-}
-if (scalar(@ARGV) > 0) {
- REP("(load-file \"" . $ARGV[0] . "\")");
- exit 0;
-}
-while (1) {
- my $line = mal_readline("user> ");
- if (! defined $line) { last; }
- do {
- local $@;
- my $ret;
- eval {
- use autodie; # always "throw" errors
- print(REP($line), "\n");
- 1;
- } or do {
- my $err = $@;
- given (ref $err) {
- when (/^BlankException/) {
- # ignore and continue
- }
- default {
- chomp $err;
- print "Error: $err\n";
- }
- }
- };
- };
-}
+use strict;
+use warnings FATAL => "recursion";
+no if $] >= 5.018, warnings => "experimental::smartmatch";
+use feature qw(switch);
+use File::Basename;
+use lib dirname (__FILE__);
+
+use Data::Dumper;
+use List::Util qw(pairs pairmap);
+use Scalar::Util qw(blessed);
+
+use readline qw(mal_readline set_rl_mode);
+use types qw($nil $true $false _symbol_Q _list_Q);
+use reader;
+use printer;
+use env;
+use core;
+use interop qw(pl_to_mal);
+
+# read
+sub READ {
+ my $str = shift;
+ return reader::read_str($str);
+}
+
+# eval
+sub is_pair {
+ my ($x) = @_;
+ return $x->isa('Mal::Sequence') && @$x;
+}
+
+sub quasiquote {
+ my ($ast) = @_;
+ if (!is_pair($ast)) {
+ return Mal::List->new([Mal::Symbol->new("quote"), $ast]);
+ } elsif (_symbol_Q($ast->[0]) && ${$ast->[0]} eq 'unquote') {
+ return $ast->[1];
+ } elsif (is_pair($ast->[0]) && _symbol_Q($ast->[0]->[0]) &&
+ ${$ast->[0]->[0]} eq 'splice-unquote') {
+ return Mal::List->new([Mal::Symbol->new("concat"),
+ $ast->[0]->[1],
+ quasiquote($ast->rest())]);
+ } else {
+ return Mal::List->new([Mal::Symbol->new("cons"),
+ quasiquote($ast->[0]),
+ quasiquote($ast->rest())]);
+ }
+}
+
+sub is_macro_call {
+ my ($ast, $env) = @_;
+ if (_list_Q($ast) &&
+ _symbol_Q($ast->[0]) &&
+ $env->find($ast->[0])) {
+ my ($f) = $env->get($ast->[0]);
+ if ($f->isa('Mal::Function')) {
+ return $f->{ismacro};
+ }
+ }
+ return 0;
+}
+
+sub macroexpand {
+ my ($ast, $env) = @_;
+ while (is_macro_call($ast, $env)) {
+ my @args = @$ast;
+ my $mac = $env->get(shift @args);
+ $ast = &$mac(@args);
+ }
+ return $ast;
+}
+
+
+sub eval_ast {
+ my($ast, $env) = @_;
+ if ($ast->isa('Mal::Symbol')) {
+ return $env->get($ast);
+ } elsif ($ast->isa('Mal::Sequence')) {
+ return ref($ast)->new([ map { EVAL($_, $env) } @$ast ]);
+ } elsif ($ast->isa('Mal::HashMap')) {
+ return Mal::HashMap->new({ pairmap { $a => EVAL($b, $env) } %$ast });
+ } else {
+ return $ast;
+ }
+}
+
+sub EVAL {
+ my($ast, $env) = @_;
+
+ while (1) {
+
+ #print "EVAL: " . printer::_pr_str($ast) . "\n";
+ if (! _list_Q($ast)) {
+ return eval_ast($ast, $env);
+ }
+ @$ast or return $ast;
+
+ # apply list
+ $ast = macroexpand($ast, $env);
+ if (! _list_Q($ast)) {
+ return eval_ast($ast, $env);
+ }
+
+ my ($a0, $a1, $a2, $a3) = @$ast;
+ if (!$a0) { return $ast; }
+ given ($a0->isa('Mal::Symbol') ? $$a0 : $a0) {
+ when ('def!') {
+ my $res = EVAL($a2, $env);
+ return $env->set($a1, $res);
+ }
+ when ('let*') {
+ my $let_env = Mal::Env->new($env);
+ foreach my $pair (pairs @$a1) {
+ my ($k, $v) = @$pair;
+ $let_env->set($k, EVAL($v, $let_env));
+ }
+ $ast = $a2;
+ $env = $let_env;
+ # Continue loop (TCO)
+ }
+ when ('quote') {
+ return $a1;
+ }
+ when ('quasiquote') {
+ $ast = quasiquote($a1);
+ # Continue loop (TCO)
+ }
+ when ('defmacro!') {
+ my $func = EVAL($a2, $env)->clone;
+ $func->{ismacro} = 1;
+ return $env->set($a1, $func);
+ }
+ when ('macroexpand') {
+ return macroexpand($a1, $env);
+ }
+ when ('try*') {
+ local $@;
+ my $ret = eval { EVAL($a1, $env) };
+ return $ret unless $@;
+ if ($a2 && ${$a2->[0]} eq 'catch*') {
+ my $exc;
+ if (defined(blessed $@) && $@->isa('Mal::Type')) {
+ $exc = $@;
+ } else {
+ chomp(my $msg = $@);
+ $exc = Mal::String->new($msg);
+ }
+ my $catch_env = Mal::Env->new($env, [$a2->[1]], [$exc]);
+ return EVAL($a2->[2], $catch_env)
+ } else {
+ die $@;
+ }
+ }
+ when ('do') {
+ eval_ast($ast->slice(1, $#$ast-1), $env);
+ $ast = $ast->[$#$ast];
+ # Continue loop (TCO)
+ }
+ when ('if') {
+ my $cond = EVAL($a1, $env);
+ if ($cond eq $nil || $cond eq $false) {
+ $ast = $a3 ? $a3 : $nil;
+ } else {
+ $ast = $a2;
+ }
+ # Continue loop (TCO)
+ }
+ when ('fn*') {
+ return Mal::Function->new(\&EVAL, $a2, $env, $a1);
+ }
+ default {
+ my @el = @{eval_ast($ast, $env)};
+ my $f = shift @el;
+ if ($f->isa('Mal::Function')) {
+ $ast = $f->{ast};
+ $env = $f->gen_env(\@el);
+ # Continue loop (TCO)
+ } else {
+ return &$f(@el);
+ }
+ }
+ }
+
+ } # TCO while loop
+}
+
+# print
+sub PRINT {
+ my $exp = shift;
+ return printer::_pr_str($exp);
+}
+
+# repl
+my $repl_env = Mal::Env->new();
+sub REP {
+ my $str = shift;
+ return PRINT(EVAL(READ($str), $repl_env));
+}
+
+# core.pl: defined using perl
+foreach my $n (keys %core::ns) {
+ $repl_env->set(Mal::Symbol->new($n), $core::ns{$n});
+}
+$repl_env->set(Mal::Symbol->new('eval'),
+ bless sub { EVAL($_[0], $repl_env); }, 'Mal::CoreFunction');
+my @_argv = map {Mal::String->new($_)} @ARGV[1..$#ARGV];
+$repl_env->set(Mal::Symbol->new('*ARGV*'), Mal::List->new(\@_argv));
+
+# core.mal: defined using the language itself
+REP(q[(def! not (fn* (a) (if a false true)))]);
+REP(q[(def! load-file (fn* (f) (eval (read-string (str "(do " (slurp f) "\nnil)")))))]);
+REP(q[(defmacro! cond (fn* (& xs) (if (> (count xs) 0) (list 'if (first xs) (if (> (count xs) 1) (nth xs 1) (throw "odd number of forms to cond")) (cons 'cond (rest (rest xs)))))))]);
+
+if (@ARGV && $ARGV[0] eq "--raw") {
+ set_rl_mode("raw");
+ shift @ARGV;
+}
+if (@ARGV) {
+ REP(qq[(load-file "$ARGV[0]")]);
+ exit 0;
+}
+while (1) {
+ my $line = mal_readline("user> ");
+ if (! defined $line) { last; }
+ do {
+ local $@;
+ my $ret;
+ eval {
+ print(REP($line), "\n");
+ 1;
+ } or do {
+ my $err = $@;
+ if (defined(blessed $err) && $err->isa('Mal::BlankException')) {
+ # ignore and continue
+ } elsif (defined(blessed $err) && $err->isa('Mal::Type')) {
+ print "Error: ".printer::_pr_str($err)."\n";
+ } else {
+ chomp $err;
+ print "Error: $err\n";
+ }
+ };
+ };
+}
HCoop / jackhill / mal.git / blobdiff