X-Git-Url: http://repo.macrolet.net/gitweb/?a=blobdiff_plain;f=test.lisp;h=adf8c1230c9ce1a3e19e1ad3eaf72f7ef386556a;hb=4f17b56fa136f97d11975d081a861351eb64db76;hp=e5fdc77afdcffc786116f96c82aa1690028a2c71;hpb=5c045e769548bec80d6270a21f47d3d411e69080;p=jscl.git diff --git a/test.lisp b/test.lisp index e5fdc77..adf8c12 100644 --- a/test.lisp +++ b/test.lisp @@ -2,9 +2,9 @@ (eval-when-compile (%compile-defmacro 'defmacro - (lambda (name args &rest body) + '(lambda (name args &rest body) `(eval-when-compile - (%compile-defmacro ',name (lambda ,args ,@body)))))) + (%compile-defmacro ',name '(lambda ,args ,@body)))))) (defmacro defvar (name value) `(progn @@ -12,15 +12,47 @@ (%compile-defvar ',name)) (setq ,name ,value))) -(defvar t 't) -(defvar nil 'nil) - (defmacro defun (name args &rest body) `(progn (eval-when-compile (%compile-defun ',name)) (fsetq ,name (lambda ,args ,@body)))) +(defvar *package* (new)) + +(defvar nil (make-symbol "NIL")) +(set *package* "NIL" nil) + +(defvar t (make-symbol "T")) +(set *package* "T" t) + +(defun internp (name) + (in name *package*)) + +(defun intern (name) + (if (internp name) + (get *package* name) + (set *package* name (make-symbol name)))) + +(defun find-symbol (name) + (get *package* name)) + +(defmacro when (condition &rest body) + `(if ,condition (progn ,@body) nil)) + +(defmacro unless (condition &rest body) + `(if ,condition nil (progn ,@body))) + +(defmacro dolist (iter &rest body) + (let ((var (first iter)) + (g!list (make-symbol "LIST"))) + `(let ((,g!list ,(second iter)) + (,var nil)) + (while ,g!list + (setq ,var (car ,g!list)) + ,@body + (setq ,g!list (cdr ,g!list)))))) + (defun = (x y) (= x y)) (defun + (x y) (+ x y)) (defun - (x y) (- x y)) @@ -34,19 +66,46 @@ (defun truncate (x y) (floor (/ x y))) (defun cons (x y ) (cons x y)) +(defun consp (x) (consp x)) + (defun car (x) (car x)) +(defun cdr (x) (cdr x)) + (defun caar (x) (car (car x))) (defun cadr (x) (car (cdr x))) -(defun cdr (x) (cdr x)) (defun cdar (x) (cdr (car x))) (defun cddr (x) (cdr (cdr x))) -(defun append (list1 list2) +(defun caddr (x) (car (cdr (cdr x)))) +(defun cdddr (x) (cdr (cdr (cdr x)))) + +(defun cadddr (x) (car (cdr (cdr (cdr x))))) + +(defun first (x) (car x)) +(defun second (x) (cadr x)) +(defun third (x) (caddr x)) +(defun fourth (x) (cadddr x)) + +(defun list (&rest args) + args) + +(defun atom (x) + (not (consp x))) + +(defun ensure-list (x) + (if (listp x) + x + (list x))) + +(defun append-two (list1 list2) (if (null list1) list2 (cons (car list1) (append (cdr list1) list2)))) +(defun append (&rest lists) + (!reduce #'append-two lists '())) + (defun reverse-aux (list acc) (if (null list) acc @@ -55,6 +114,24 @@ (defun reverse (list) (reverse-aux list '())) +(defmacro incf (x) + `(setq ,x (1+ ,x))) + +(defmacro decf (x) + `(setq ,x (1- ,x))) + +(defun list-length (list) + (let ((l 0)) + (while (not (null list)) + (incf l) + (setq list (cdr list))) + l)) + +(defun length (seq) + (if (stringp seq) + (string-length seq) + (list-length seq))) + (defun mapcar (func list) (if (null list) '() @@ -64,18 +141,6 @@ (defmacro push (x place) `(setq ,place (cons ,x ,place))) -(defvar *package* (new)) - -(defun intern (name) - (let ((s (get *package* name))) - (if s - s - (set *package* name (make-symbol name))))) - -(defun find-symbol (name) - (get *package* name)) - - (defmacro cond (&rest clausules) (if (null clausules) nil @@ -85,6 +150,25 @@ (progn ,@(cdar clausules)) (cond ,@(cdr clausules)))))) + +(defmacro case (form &rest clausules) + (let ((!form (make-symbol "FORM"))) + `(let ((,!form ,form)) + (cond + ,@(mapcar (lambda (clausule) + (if (eq (car clausule) t) + clausule + `((eql ,!form ,(car clausule)) + ,@(cdr clausule)))) + clausules))))) + +(defmacro ecase (form &rest clausules) + `(case ,form + ,@(append + clausules + `((t + (error "ECASE expression failed.")))))) + (defun !reduce (func list initial) (if (null list) initial @@ -123,9 +207,886 @@ (if (zerop x) "0" (let ((digits nil)) + (while (not (zerop x 0)) + (push (mod x 10) digits) + (setq x (truncate x 10))) + (join (mapcar (lambda (d) (string (char "0123456789" d))) + digits) + "")))) + +(defmacro and (&rest forms) + (cond + ((null forms) + t) + ((null (cdr forms)) + (car forms)) + (t + `(if ,(car forms) + (and ,@(cdr forms)) + nil)))) + + +(defmacro or (&rest forms) + (cond + ((null forms) + nil) + ((null (cdr forms)) + (car forms)) + (t + (let ((g (make-symbol "VAR"))) + `(let ((,g ,(car forms))) + (if ,g ,g (or ,@(cdr forms)))))))) + + +(defmacro prog1 (form &rest body) + (let ((value (make-symbol "VALUE"))) + `(let ((,value ,form)) + ,@body + ,value))) + + +(defun char= (x y) (= x y)) + +(defun <= (x y) (or (< x y) (= x y))) +(defun >= (x y) (not (< x y))) + +(defun listp (x) + (or (consp x) (null x))) + +(defun integerp (x) + (and (numberp x) (= (floor x) x))) + +(defun last (x) + (if (null (cdr x)) + x + (last (cdr x)))) + +(defun butlast (x) + (if (null (cdr x)) + nil + (cons (car x) (butlast (cdr x))))) + +(defun member (x list) + (cond + ((null list) + nil) + ((eql x (car list)) + list) + (t + (member x (cdr list))))) + +(defun remove (x list) + (cond + ((null list) + nil) + ((eql x (car list)) + (remove x (cdr list))) + (t + (cons (car list) (remove x (cdr list)))))) + +(defun remove-if (func list) + (cond + ((null list) + nil) + ((funcall func (car list)) + (remove-if func (cdr list))) + (t + (cons (car list) (remove-if func (cdr list)))))) + +(defun remove-if-not (func list) + (cond + ((null list) + nil) + ((funcall func (car list)) + (cons (car list) (remove-if-not func (cdr list)))) + (t + (remove-if-not func (cdr list))))) + +(defun digit-char-p (x) + (if (and (<= #\0 x) (<= x #\9)) + (- x #\0) + nil)) + +(defun parse-integer (string) + (let ((value 0) + (index 0) + (size (length string))) + (while (< index size) + (setq value (+ (* value 10) (digit-char-p (char string index)))) + (incf index)) + value)) + +(defun every (function seq) + ;; string + (let ((ret t) + (index 0) + (size (length seq))) + (while (and ret (< index size)) + (unless (funcall function (char seq index)) + (setq ret nil)) + (incf index)) + ret)) + +(defun eql (x y) + (eq x y)) + +(defun assoc (x alist) + (cond + ((null alist) + nil) + ((eql x (caar alist)) + (car alist)) + (t + (assoc x (cdr alist))))) + +(defun string= (s1 s2) + (equal s1 s2)) + +;; ---------------------------------------------------------- + +;;; Utils + +#+common-lisp +(progn + (defmacro while (condition &body body) + `(do () + ((not ,condition)) + ,@body)) + + (defun concat-two (s1 s2) + (concatenate 'string s1 s2)) + + (defun setcar (cons new) + (setf (car cons) new)) + (defun setcdr (cons new) + (setf (cdr cons) new))) + +(defvar *newline* (string (code-char 10))) + +(defun concat (&rest strs) + (!reduce (lambda (s1 s2) (concat-two s1 s2)) + strs + "")) + +;;; Concatenate a list of strings, with a separator +(defun join (list separator) + (cond + ((null list) + "") + ((null (cdr list)) + (car list)) + (t + (concat (car list) + separator + (join (cdr list) separator))))) + +(defun join-trailing (list separator) + (cond + ((null list) + "") + ((null (car list)) + (join-trailing (cdr list) separator)) + (t + (concat (car list) separator (join-trailing (cdr list) separator))))) + +(defun integer-to-string (x) + (if (zerop x) + "0" + (let ((digits nil)) (while (not (= x 0)) (push (mod x 10) digits) (setq x (truncate x 10))) (join (mapcar (lambda (d) (string (char "0123456789" d))) digits) "")))) + +;;;; Reader + +;;; It is a basic Lisp reader. It does not use advanced stuff +;;; intentionally, because we want to use it to bootstrap a simple +;;; Lisp. The main entry point is the function `ls-read', which +;;; accepts a strings as argument and return the Lisp expression. +(defun make-string-stream (string) + (cons string 0)) + +(defun %peek-char (stream) + (and (< (cdr stream) (length (car stream))) + (char (car stream) (cdr stream)))) + +(defun %read-char (stream) + (and (< (cdr stream) (length (car stream))) + (prog1 (char (car stream) (cdr stream)) + (setcdr stream (1+ (cdr stream)))))) + +(defun whitespacep (ch) + (or (char= ch #\space) (char= ch #\newline) (char= ch #\tab))) + +(defun skip-whitespaces (stream) + (let (ch) + (setq ch (%peek-char stream)) + (while (and ch (whitespacep ch)) + (%read-char stream) + (setq ch (%peek-char stream))))) + +(defun terminalp (ch) + (or (null ch) (whitespacep ch) (char= #\) ch) (char= #\( ch))) + +(defun read-until (stream func) + (let ((string "") + (ch)) + (setq ch (%peek-char stream)) + (while (not (funcall func ch)) + (setq string (concat string (string ch))) + (%read-char stream) + (setq ch (%peek-char stream))) + string)) + +(defun skip-whitespaces-and-comments (stream) + (let (ch) + (skip-whitespaces stream) + (setq ch (%peek-char stream)) + (while (and ch (char= ch #\;)) + (read-until stream (lambda (x) (char= x #\newline))) + (skip-whitespaces stream) + (setq ch (%peek-char stream))))) + +(defun %read-list (stream) + (skip-whitespaces-and-comments stream) + (let ((ch (%peek-char stream))) + (cond + ((null ch) + (error "Unspected EOF")) + ((char= ch #\)) + (%read-char stream) + nil) + ((char= ch #\.) + (%read-char stream) + (prog1 (ls-read stream) + (skip-whitespaces-and-comments stream) + (unless (char= (%read-char stream) #\)) + (error "')' was expected.")))) + (t + (cons (ls-read stream) (%read-list stream)))))) + +(defun read-string (stream) + (let ((string "") + (ch nil)) + (setq ch (%read-char stream)) + (while (not (eql ch #\")) + (when (null ch) + (error "Unexpected EOF")) + (when (eql ch #\\) + (setq ch (%read-char stream))) + (setq string (concat string (string ch))) + (setq ch (%read-char stream))) + string)) + +(defvar *eof* (make-symbol "EOF")) +(defun ls-read (stream) + (skip-whitespaces-and-comments stream) + (let ((ch (%peek-char stream))) + (cond + ((null ch) + *eof*) + ((char= ch #\() + (%read-char stream) + (%read-list stream)) + ((char= ch #\') + (%read-char stream) + (list 'quote (ls-read stream))) + ((char= ch #\`) + (%read-char stream) + (list 'backquote (ls-read stream))) + ((char= ch #\") + (%read-char stream) + (read-string stream)) + ((char= ch #\,) + (%read-char stream) + (if (eql (%peek-char stream) #\@) + (progn (%read-char stream) (list 'unquote-splicing (ls-read stream))) + (list 'unquote (ls-read stream)))) + ((char= ch #\#) + (%read-char stream) + (ecase (%read-char stream) + (#\' + (list 'function (ls-read stream))) + (#\\ + (let ((cname + (concat (string (%read-char stream)) + (read-until stream #'terminalp)))) + (cond + ((string= cname "space") (char-code #\space)) + ((string= cname "tab") (char-code #\tab)) + ((string= cname "newline") (char-code #\newline)) + (t (char-code (char cname 0)))))) + (#\+ + (let ((feature (read-until stream #'terminalp))) + (cond + ((string= feature "common-lisp") + (ls-read stream) ;ignore + (ls-read stream)) + ((string= feature "lispstrack") + (ls-read stream)) + (t + (error "Unknown reader form."))))))) + (t + (let ((string (read-until stream #'terminalp))) + (if (every #'digit-char-p string) + (parse-integer string) + (intern (string-upcase string)))))))) + +(defun ls-read-from-string (string) + (ls-read (make-string-stream string))) + + +;;;; Compiler + +(defvar *compilation-unit-checks* '()) + +(defvar *env* '()) +(defvar *fenv* '()) + +(defun make-binding (name type js declared) + (list name type js declared)) + +(defun binding-name (b) (first b)) +(defun binding-type (b) (second b)) +(defun binding-translation (b) (third b)) +(defun binding-declared (b) + (and b (fourth b))) +(defun mark-binding-as-declared (b) + (setcar (cdddr b) t)) + + +(defvar *variable-counter* 0) +(defun gvarname (symbol) + (concat "v" (integer-to-string (incf *variable-counter*)))) + +(defun lookup-variable (symbol env) + (or (assoc symbol env) + (assoc symbol *env*) + (let ((name (symbol-name symbol)) + (binding (make-binding symbol 'variable (gvarname symbol) nil))) + (push binding *env*) + (push (lambda () + (unless (binding-declared (assoc symbol *env*)) + (error (concat "Undefined variable `" name "'")))) + *compilation-unit-checks*) + binding))) + +(defun lookup-variable-translation (symbol env) + (binding-translation (lookup-variable symbol env))) + +(defun extend-local-env (args env) + (append (mapcar (lambda (symbol) + (make-binding symbol 'variable (gvarname symbol) t)) + args) + env)) + +(defvar *function-counter* 0) +(defun lookup-function (symbol env) + (or (assoc symbol env) + (assoc symbol *fenv*) + (let ((name (symbol-name symbol)) + (binding + (make-binding symbol + 'function + (concat "f" (integer-to-string (incf *function-counter*))) + nil))) + (push binding *fenv*) + (push (lambda () + (unless (binding-declared (assoc symbol *fenv*)) + (error (concat "Undefined function `" name "'")))) + *compilation-unit-checks*) + binding))) + +(defun lookup-function-translation (symbol env) + (binding-translation (lookup-function symbol env))) + + +(defvar *toplevel-compilations* nil) + +(defun %compile-defvar (name) + (let ((b (lookup-variable name *env*))) + (mark-binding-as-declared b) + (push (concat "var " (binding-translation b)) *toplevel-compilations*))) + +(defun %compile-defun (name) + (let ((b (lookup-function name *env*))) + (mark-binding-as-declared b) + (push (concat "var " (binding-translation b)) *toplevel-compilations*))) + +(defun %compile-defmacro (name lambda) + (push (make-binding name 'macro lambda t) *fenv*)) + + +(defvar *compilations* nil) + +(defun ls-compile-block (sexps env fenv) + (join-trailing + (remove nil (mapcar (lambda (x) + (ls-compile x env fenv)) + sexps)) + "; +")) +(defmacro define-compilation (name args &rest body) + ;; Creates a new primitive `name' with parameters args and + ;; @body. The body can access to the local environment through the + ;; variable ENV. + `(push (list ',name (lambda (env fenv ,@args) ,@body)) + *compilations*)) + +(define-compilation if (condition true false) + (concat "(" + (ls-compile condition env fenv) " !== " (ls-compile nil nil nil) + " ? " + (ls-compile true env fenv) + " : " + (ls-compile false env fenv) + ")")) + +;;; Return the required args of a lambda list +(defun lambda-list-required-argument (lambda-list) + (if (or (null lambda-list) (eq (car lambda-list) '&rest)) + nil + (cons (car lambda-list) (lambda-list-required-argument (cdr lambda-list))))) + +(defun lambda-list-rest-argument (lambda-list) + (second (member '&rest lambda-list))) + +(define-compilation lambda (lambda-list &rest body) + (let ((required-arguments (lambda-list-required-argument lambda-list)) + (rest-argument (lambda-list-rest-argument lambda-list))) + (let ((new-env (extend-local-env + (append (and rest-argument (list rest-argument)) + required-arguments) + env))) + (concat "(function (" + (join (mapcar (lambda (x) + (lookup-variable-translation x new-env)) + required-arguments) + ",") + "){" + *newline* + (if rest-argument + (let ((js!rest (lookup-variable-translation rest-argument new-env))) + (concat "var " js!rest "= " (ls-compile nil env fenv) ";" *newline* + "for (var i = arguments.length-1; i>=" + (integer-to-string (length required-arguments)) + "; i--)" *newline* + js!rest " = " + "{car: arguments[i], cdr: " js!rest "};" + *newline*)) + "") + (concat (ls-compile-block (butlast body) new-env fenv) + "return " (ls-compile (car (last body)) new-env fenv) ";") + *newline* + "})")))) + +(define-compilation fsetq (var val) + (concat (lookup-function-translation var fenv) + " = " + (ls-compile val env fenv))) + +(define-compilation setq (var val) + (concat (lookup-variable-translation var env) + " = " + (ls-compile val env fenv))) + +;;; Literals + +(defun escape-string (string) + (let ((output "") + (index 0) + (size (length string))) + (while (< index size) + (let ((ch (char string index))) + (when (or (char= ch #\") (char= ch #\\)) + (setq output (concat output "\\"))) + (when (or (char= ch #\newline)) + (setq output (concat output "\\")) + (setq ch #\n)) + (setq output (concat output (string ch)))) + (incf index)) + output)) + +(defun literal->js (sexp) + (cond + ((integerp sexp) (integer-to-string sexp)) + ((stringp sexp) (concat "\"" (escape-string sexp) "\"")) + ((symbolp sexp) (ls-compile `(intern ,(escape-string (symbol-name sexp))) *env* *fenv*)) + ((consp sexp) (concat "{car: " + (literal->js (car sexp)) + ", cdr: " + (literal->js (cdr sexp)) "}")))) + +(defvar *literal-counter* 0) +(defun literal (form) + (let ((var (concat "l" (integer-to-string (incf *literal-counter*))))) + (push (concat "var " var " = " (literal->js form)) *toplevel-compilations*) + var)) + +(define-compilation quote (sexp) + (literal sexp)) + +(define-compilation debug (form) + (concat "console.log(" (ls-compile form env fenv) ")")) + +(define-compilation while (pred &rest body) + (concat "(function(){ while(" + (ls-compile pred env fenv) " !== " (ls-compile nil nil nil) + "){" + (ls-compile-block body env fenv) + "}})()")) + +(define-compilation function (x) + (cond + ((and (listp x) (eq (car x) 'lambda)) + (ls-compile x env fenv)) + ((symbolp x) + (lookup-function-translation x fenv)))) + +#+common-lisp +(defmacro eval-when-compile (&body body) + `(eval-when (:compile-toplevel :load-toplevel :execute) + ,@body)) + +(define-compilation eval-when-compile (&rest body) + (eval (cons 'progn body)) + nil) + +(defmacro define-transformation (name args form) + `(define-compilation ,name ,args + (ls-compile ,form env fenv))) + +(define-transformation progn (&rest body) + `((lambda () ,@body))) + +(define-transformation let (bindings &rest body) + (let ((bindings (mapcar #'ensure-list bindings))) + `((lambda ,(mapcar #'car bindings) ,@body) + ,@(mapcar #'cadr bindings)))) + +;;; A little backquote implementation without optimizations of any +;;; kind for lispstrack. +(defun backquote-expand-1 (form) + (cond + ((symbolp form) + (list 'quote form)) + ((atom form) + form) + ((eq (car form) 'unquote) + (car form)) + ((eq (car form) 'backquote) + (backquote-expand-1 (backquote-expand-1 (cadr form)))) + (t + (cons 'append + (mapcar (lambda (s) + (cond + ((and (listp s) (eq (car s) 'unquote)) + (list 'list (cadr s))) + ((and (listp s) (eq (car s) 'unquote-splicing)) + (cadr s)) + (t + (list 'list (backquote-expand-1 s))))) + form))))) + +(defun backquote-expand (form) + (if (and (listp form) (eq (car form) 'backquote)) + (backquote-expand-1 (cadr form)) + form)) + +(defmacro backquote (form) + (backquote-expand-1 form)) + +(define-transformation backquote (form) + (backquote-expand-1 form)) + +;;; Primitives + +(defun compile-bool (x) + (concat "(" x "?" (ls-compile t nil nil) ": " (ls-compile nil nil nil) ")")) + +(define-compilation + (x y) + (concat "((" (ls-compile x env fenv) ") + (" (ls-compile y env fenv) "))")) + +(define-compilation - (x y) + (concat "((" (ls-compile x env fenv) ") - (" (ls-compile y env fenv) "))")) + +(define-compilation * (x y) + (concat "((" (ls-compile x env fenv) ") * (" (ls-compile y env fenv) "))")) + +(define-compilation / (x y) + (concat "((" (ls-compile x env fenv) ") / (" (ls-compile y env fenv) "))")) + +(define-compilation < (x y) + (compile-bool (concat "((" (ls-compile x env fenv) ") < (" (ls-compile y env fenv) "))"))) + +(define-compilation = (x y) + (compile-bool (concat "((" (ls-compile x env fenv) ") == (" (ls-compile y env fenv) "))"))) + +(define-compilation numberp (x) + (compile-bool (concat "(typeof (" (ls-compile x env fenv) ") == \"number\")"))) + + +(define-compilation mod (x y) + (concat "((" (ls-compile x env fenv) ") % (" (ls-compile y env fenv) "))")) + +(define-compilation floor (x) + (concat "(Math.floor(" (ls-compile x env fenv) "))")) + +(define-compilation null (x) + (compile-bool (concat "(" (ls-compile x env fenv) "===" (ls-compile nil env fenv) ")"))) + +(define-compilation cons (x y) + (concat "({car: " (ls-compile x env fenv) ", cdr: " (ls-compile y env fenv) "})")) + +(define-compilation consp (x) + (compile-bool + (concat "(function(){ var tmp = " + (ls-compile x env fenv) + "; return (typeof tmp == 'object' && 'car' in tmp);})()"))) + +(define-compilation car (x) + (concat "(function () { var tmp = " (ls-compile x env fenv) + "; return tmp === " (ls-compile nil nil nil) "? " + (ls-compile nil nil nil) + ": tmp.car; })()")) + +(define-compilation cdr (x) + (concat "(function () { var tmp = " (ls-compile x env fenv) + "; return tmp === " (ls-compile nil nil nil) "? " + (ls-compile nil nil nil) + ": tmp.cdr; })()")) + +(define-compilation setcar (x new) + (concat "((" (ls-compile x env fenv) ").car = " (ls-compile new env fenv) ")")) + +(define-compilation setcdr (x new) + (concat "((" (ls-compile x env fenv) ").cdr = " (ls-compile new env fenv) ")")) + +(define-compilation symbolp (x) + (compile-bool + (concat "(function(){ var tmp = " + (ls-compile x env fenv) + "; return (typeof tmp == 'object' && 'name' in tmp); })()"))) + +(define-compilation make-symbol (name) + (concat "{name: " (ls-compile name env fenv) "}")) + +(define-compilation symbol-name (x) + (concat "(" (ls-compile x env fenv) ").name")) + +(define-compilation eq (x y) + (compile-bool + (concat "(" (ls-compile x env fenv) " === " (ls-compile y env fenv) ")"))) + +(define-compilation equal (x y) + (compile-bool + (concat "(" (ls-compile x env fenv) " == " (ls-compile y env fenv) ")"))) + +(define-compilation string (x) + (concat "String.fromCharCode(" (ls-compile x env fenv) ")")) + +(define-compilation stringp (x) + (compile-bool + (concat "(typeof(" (ls-compile x env fenv) ") == \"string\")"))) + +(define-compilation string-upcase (x) + (concat "(" (ls-compile x env fenv) ").toUpperCase()")) + +(define-compilation string-length (x) + (concat "(" (ls-compile x env fenv) ").length")) + +(define-compilation char (string index) + (concat "(" + (ls-compile string env fenv) + ").charCodeAt(" + (ls-compile index env fenv) + ")")) + +(define-compilation concat-two (string1 string2) + (concat "(" + (ls-compile string1 env fenv) + ").concat(" + (ls-compile string2 env fenv) + ")")) + +(define-compilation funcall (func &rest args) + (concat "(" + (ls-compile func env fenv) + ")(" + (join (mapcar (lambda (x) + (ls-compile x env fenv)) + args) + ", ") + ")")) + +(define-compilation apply (func &rest args) + (if (null args) + (concat "(" (ls-compile func env fenv) ")()") + (let ((args (butlast args)) + (last (car (last args)))) + (concat "(function(){" *newline* + "var f = " (ls-compile func env fenv) ";" *newline* + "var args = [" (join (mapcar (lambda (x) + (ls-compile x env fenv)) + args) + ", ") + "];" *newline* + "var tail = (" (ls-compile last env fenv) ");" *newline* + "while (tail != " (ls-compile nil env fenv) "){" *newline* + " args.push(tail.car);" *newline* + " tail = tail.cdr;" *newline* + "}" *newline* + "return f.apply(this, args);" *newline* + "})()" *newline*)))) + +(define-compilation js-eval (string) + (concat "eval.apply(window, [" (ls-compile string env fenv) "])")) + + +(define-compilation error (string) + (concat "(function (){ throw " (ls-compile string env fenv) ";" "return 0;})()")) + +(define-compilation new () + "{}") + +(define-compilation get (object key) + (concat "(function(){ var tmp = " + "(" (ls-compile object env fenv) ")[" (ls-compile key env fenv) "]" + ";" + "return tmp == undefined? " (ls-compile nil nil nil) ": tmp ;" + "})()")) + +(define-compilation set (object key value) + (concat "((" + (ls-compile object env fenv) + ")[" + (ls-compile key env fenv) "]" + " = " (ls-compile value env fenv) ")")) + +(define-compilation in (key object) + (compile-bool + (concat "(" (ls-compile key env fenv) " in " (ls-compile object env fenv) ")"))) + + +(defun macrop (x) + (and (symbolp x) (eq (binding-type (lookup-function x *fenv*)) 'macro))) + +(defun ls-macroexpand-1 (form env fenv) + (if (macrop (car form)) + (let ((binding (lookup-function (car form) *env*))) + (if (eq (binding-type binding) 'macro) + (apply (eval (binding-translation binding)) (cdr form)) + form)) + form)) + +(defun compile-funcall (function args env fenv) + (cond + ((symbolp function) + (concat (lookup-function-translation function fenv) + "(" + (join (mapcar (lambda (x) (ls-compile x env fenv)) args) + ", ") + ")")) + ((and (listp function) (eq (car function) 'lambda)) + (concat "(" (ls-compile function env fenv) ")(" + (join (mapcar (lambda (x) (ls-compile x env fenv)) args) + ", ") + ")")) + (t + (error (concat "Invalid function designator " (symbol-name function)))))) + +(defun ls-compile (sexp env fenv) + (cond + ((symbolp sexp) (lookup-variable-translation sexp env)) + ((integerp sexp) (integer-to-string sexp)) + ((stringp sexp) (concat "\"" (escape-string sexp) "\"")) + ((listp sexp) + (if (assoc (car sexp) *compilations*) + (let ((comp (second (assoc (car sexp) *compilations*)))) + (apply comp env fenv (cdr sexp))) + (if (macrop (car sexp)) + (ls-compile (ls-macroexpand-1 sexp env fenv) env fenv) + (compile-funcall (car sexp) (cdr sexp) env fenv)))))) + +(defun ls-compile-toplevel (sexp) + (setq *toplevel-compilations* nil) + (let ((code (ls-compile sexp nil nil))) + (prog1 + (concat #+common-lisp (concat "/* " (princ-to-string sexp) " */") + (join (mapcar (lambda (x) (concat x ";" *newline*)) + *toplevel-compilations*) + "") + code) + (setq *toplevel-compilations* nil)))) + +#+common-lisp +(progn + (defun read-whole-file (filename) + (with-open-file (in filename) + (let ((seq (make-array (file-length in) :element-type 'character))) + (read-sequence seq in) + seq))) + + (defun ls-compile-file (filename output) + (setq *env* nil *fenv* nil) + (setq *compilation-unit-checks* nil) + (with-open-file (out output :direction :output :if-exists :supersede) + (let* ((source (read-whole-file filename)) + (in (make-string-stream source))) + (loop + for x = (ls-read in) + until (eq x *eof*) + for compilation = (ls-compile-toplevel x) + when (plusp (length compilation)) + do (write-line (concat compilation "; ") out)) + (dolist (check *compilation-unit-checks*) + (funcall check)) + (setq *compilation-unit-checks* nil)))) + + (defun bootstrap () + (ls-compile-file "lispstrack.lisp" "lispstrack.js"))) + +;;; ---------------------------------------------------------- + +(defmacro with-compilation-unit (&rest body) + `(prog1 + (progn + (setq *compilation-unit-checks* nil) + (setq *env* (remove-if-not #'binding-declared *env*)) + (setq *fenv* (remove-if-not #'binding-declared *fenv*)) + ,@body) + (dolist (check *compilation-unit-checks*) + (funcall check)))) + +(defun eval (x) + (let ((code + (with-compilation-unit + (ls-compile-toplevel x nil nil)))) + (js-eval code))) + + +;; Set the initial global environment to be equal to the host global +;; environment at this point of the compilation. +(eval-when-compile + (let ((c1 (ls-compile `(setq *fenv* ',*fenv*) nil nil)) + (c2 (ls-compile `(setq *env* ',*env*) nil nil)) + (c3 (ls-compile `(setq *variable-counter* ',*variable-counter*) nil nil)) + (c4 (ls-compile `(setq *function-counter* ',*function-counter*) nil nil)) + (c5 (ls-compile `(setq *literal-counter* ',*literal-counter*) nil nil))) + (setq *toplevel-compilations* + (append *toplevel-compilations* (list c1 c2 c3 c4 c5))))) + +(js-eval + (concat "var lisp = {};" + "lisp.read = " (lookup-function-translation 'ls-read-from-string nil) ";" *newline* + "lisp.eval = " (lookup-function-translation 'eval nil) ";" *newline* + "lisp.compile = " (lookup-function-translation 'ls-compile-toplevel nil) ";" *newline* + "lisp.evalString = function(str){" *newline* + " return lisp.eval(lisp.read(str));" *newline* + "}" *newline* + "lisp.compileString = function(str){" *newline* + " return lisp.compile(lisp.read(str));" *newline* + "}" *newline*))