Add DO-SOURCE macro for iterating over source files
[jscl.git] / src / compiler-codegen.lisp
index f3b05cc..9bb915d 100644 (file)
 ;;; It is intended to be used with the new compiler. However, it is
 ;;; quite independent so it has been integrated early in JSCL.
 
+(/debug "loading compiler-codegen.lisp!")
+
+(defvar *js-macros* nil)
+(defmacro define-js-macro (name lambda-list &body body)
+  (let ((form (gensym)))
+    `(push (cons ',name
+                 (lambda (,form)
+                   (block ,name
+                     (destructuring-bind ,lambda-list ,form
+                       ,@body))))
+           *js-macros*)))
+
+(defun js-macroexpand (js)
+  (if (and (consp js) (assoc (car js) *js-macros*))
+      (let ((expander (cdr (assoc (car js) *js-macros*))))
+        (multiple-value-bind (expansion stop-expand-p)
+            (funcall expander (cdr js))
+          (if stop-expand-p
+              expansion
+              (js-macroexpand expansion))))
+      js))
+
+
+(defconstant no-comma 12)
+
 (defvar *js-output* t)
 
 ;;; Two seperate functions are needed for escaping strings:
 
 (defun valid-js-identifier (string-designator)
   (let ((string (typecase string-designator
-                  (symbol (string-downcase (symbol-name string-designator)))
+                  (symbol (symbol-name string-designator))
                   (string string-designator)
                   (t
                    (return-from valid-js-identifier (values nil nil))))))
      (js-format "~a" (js-escape-string form)))
     ((symbolp form)
      (case form
-       (true  (js-format "true"))
-       (false (js-format "false"))
-       (null  (js-format "null"))
-       (this  (js-format "this"))
+       (true      (js-format "true"))
+       (false     (js-format "false"))
+       (null      (js-format "null"))
+       (this      (js-format "this"))
+       (undefined (js-format "undefined"))
        (otherwise
         (js-identifier form))))
     (t
     (dotimes (i (1- size))
       (let ((elt (aref vector i)))
         (unless (eq elt 'null)
-          (js-expr elt))
+          (js-expr elt no-comma))
         (js-format ",")))
     (when (plusp size)
-      (js-expr (aref vector (1- size))))
+      (js-expr (aref vector (1- size)) no-comma))
     (js-format "]")))
 
 (defun js-object-initializer (plist)
         (declare (ignore identifier))
         (if identifier-p
             (js-identifier key)
-            (js-expr (string key))))
+            (js-expr (string key) no-comma)))
       (js-format ": ")
-      (js-expr value)
+      (js-expr value no-comma)
       (unless (null (cddr tail))
         (js-format ","))))
   (js-format "}"))
       (js-format ",")
       (js-identifier arg)))
   (js-format ")")
-  (js-stmt `(group ,@body)))
+  (js-stmt `(group ,@body) t))
 
 (defun check-lvalue (x)
   (unless (or (symbolp x)
               (nth-value 1 (valid-js-identifier x))
               (and (consp x)
-                   (member (car x) '(get =))))
+                   (member (car x) '(get = property))))
     (error "Bad Javascript lvalue ~S" x)))
 
 ;;; Process the Javascript AST to reduce some syntax sugar.
          (case (length (cdr form))
            (1 `(unary- ,(cadr form)))
            (t (reduce (lambda (x y) `(- ,x ,y)) (cdr form)))))
+        ((and or)
+         (reduce (lambda (x y) `(,(car form) ,x ,y)) (cdr form)))
         ((progn comma)
          (reduce (lambda (x y) `(comma ,x ,y)) (cdr form) :from-end t))
-        (t form))
+        (t
+         (js-macroexpand form)))
       form))
 
-;; Initialized to any value larger than any operator precedence
-(defvar *js-operator-precedence* 1000)
-(defvar *js-operator-associativity* 'left)
-(defvar *js-operand-order* 'left)
-
-;; Format an expression optionally wrapped with parenthesis if the
-;; precedence rules require it.
-(defmacro with-operator ((precedence associativity) &body body)
-  (let ((g!parens (gensym))
-        (g!precedence (gensym)))
-    `(let* ((,g!precedence ,precedence)
-            (,g!parens
-             (cond
-               ((> ,g!precedence *js-operator-precedence*))
-               ((< ,g!precedence *js-operator-precedence*) nil)
-               ;; Same precedence. Let us consider associativity.
-               (t
-                (not (eq *js-operand-order* *js-operator-associativity*)))))
-            (*js-operator-precedence* ,g!precedence)
-            (*js-operator-associativity* ,associativity)
-            (*js-operand-order* 'left))
-       (when ,g!parens (js-format "("))
-       (progn ,@body)
-       (when ,g!parens (js-format ")")))))
-
-(defun js-operator (string)
-  (js-format "~a" string)
-  (setq *js-operand-order* 'right))
-
-(defun js-operator-expression (op args)
+(defun js-operator-expression (op args precedence associativity operand-order)
   (let ((op1 (car args))
         (op2 (cadr args)))
     (case op
+      ;; Accessors
+      (property
+       (js-expr (car args) 0)
+       (js-format "[")
+       (js-expr (cadr args) no-comma)
+       (js-format "]"))
+      (get
+       (multiple-value-bind (accessor accessorp)
+           (valid-js-identifier (cadr args))
+         (unless accessorp
+           (error "Invalid accessor ~S" (cadr args)))
+         (js-expr (car args) 0)
+         (js-format ".")
+         (js-identifier accessor)))      
       ;; Function call
       (call
-       (js-expr (car args))
+       (js-expr (car args) 1)
        (js-format "(")
        (when (cdr args)
-         (with-operator (13 'left)
-           (js-expr (cadr args))
-           (dolist (operand (cddr args))
-             (let ((*js-output* t))
-               (js-format ",")
-               (js-expr operand)))))
+         (js-expr (cadr args) no-comma)
+         (dolist (operand (cddr args))
+           (js-format ",")
+           (js-expr operand no-comma)))
        (js-format ")"))
-      ;; Accessors
-      (get
-       (multiple-value-bind (identifier identifierp)
-           (valid-js-identifier (car args))
-         (multiple-value-bind (accessor accessorp)
-             (valid-js-identifier (cadr args))
-           (cond
-             ((and identifierp accessorp)
-              (js-identifier identifier)
-              (js-format ".")
-              (js-identifier accessor))
-             (t
-              (js-expr (car args))
-              (js-format "[")
-              (js-expr (cadr args))
-              (js-format "]"))))))
       ;; Object syntax
       (object
        (js-object-initializer args))
        (apply #'js-function args)
        (js-format ")"))
       (t
-       (flet ((%unary-op (operator string precedence associativity post lvalue)
-                (when (eq op operator)
-                  (with-operator (precedence associativity)
+       (labels ((low-precedence-p (op-precedence)
+                  (cond
+                    ((> op-precedence precedence))
+                    ((< op-precedence precedence) nil)
+                    (t (not (eq operand-order associativity)))))
+                
+                (%unary-op (operator string operator-precedence operator-associativity post lvalue)
+                  (when (eq op operator)
                     (when lvalue (check-lvalue op1))
+                    (when (low-precedence-p operator-precedence) (js-format "("))
                     (cond
                       (post
-                       (js-expr op1)
-                       (js-operator string))
+                       (js-expr op1 operator-precedence operator-associativity 'left)
+                       (js-format "~a" string))
                       (t
-                       (js-operator string)
-                       (js-expr op1))))
-                  (return-from js-operator-expression)))
-              (%binary-op (operator string precedence associativity lvalue)
-                (when (eq op operator)
-                  (when lvalue (check-lvalue op1))
-                  (with-operator (precedence associativity)
-                    (js-expr op1)
-                    (js-operator string)
-                    (js-expr op2))
-                  (return-from js-operator-expression))))
+                       (js-format "~a" string)
+                       (js-expr op1 operator-precedence operator-associativity 'right)))
+                    (when (low-precedence-p operator-precedence) (js-format ")"))
+                    (return-from js-operator-expression)))
+                
+                (%binary-op (operator string operator-precedence operator-associativity lvalue)
+                  (when (eq op operator)
+                    (when lvalue (check-lvalue op1))
+                    (when (low-precedence-p operator-precedence) (js-format "("))
+                    (js-expr op1 operator-precedence operator-associativity 'left)
+                    (js-format "~a" string)
+                    (js-expr op2 operator-precedence operator-associativity 'right)
+                    (when (low-precedence-p operator-precedence) (js-format ")"))
+                    (return-from js-operator-expression))))
 
          (macrolet ((unary-op (operator string precedence associativity &key post lvalue)
                       `(%unary-op ',operator ',string ',precedence ',associativity ',post ',lvalue))
                     (binary-op (operator string precedence associativity &key lvalue)
                       `(%binary-op ',operator ',string ',precedence ',associativity ',lvalue)))
 
-           (unary-op pre++       "++"            1    right :lvalue t)
-           (unary-op pre--       "--"            1    right :lvalue t)
-           (unary-op post++      "++"            1    right :lvalue t :post t)
-           (unary-op post--      "--"            1    right :lvalue t :post t)
-           (unary-op not         "!"             1    right)
-           (unary-op bit-not     "~"             1    right)
+           (unary-op pre++       "++"            2    right :lvalue t)
+           (unary-op pre--       "--"            2    right :lvalue t)
+           (unary-op post++      "++"            2    right :lvalue t :post t)
+           (unary-op post--      "--"            2    right :lvalue t :post t)
+           (unary-op not         "!"             2    right)
+           (unary-op bit-not     "~"             2    right)
            ;; Note that the leading space is necessary because it
            ;; could break with post++, for example. TODO: Avoid
            ;; leading space when it's possible.
-           (unary-op unary+      " +"            1    right)
-           (unary-op unary-      " -"            1    right)
-           (unary-op delete      "delete "       1    right)
-           (unary-op void        "void "         1    right)
-           (unary-op typeof      "typeof "       1    right)
-           (unary-op new         "new "          1    right)
+           (unary-op unary+      " +"            2    right)
+           (unary-op unary-      " -"            2    right)
+           (unary-op delete      "delete "       2    right)
+           (unary-op void        "void "         2    right)
+           (unary-op typeof      "typeof "       2    right)
+           (unary-op new         "new "          2    right)
 
-           (binary-op *          "*"             2    left)
-           (binary-op /          "/"             2    left)
-           (binary-op mod        "%"             2    left)
-           (binary-op %          "%"             2    left)
-           (binary-op +          "+"             3    left)
-           (binary-op -          "-"             3    left)
-           (binary-op <<         "<<"            4    left)
-           (binary-op >>         "<<"            4    left)
-           (binary-op >>>        ">>>"           4    left)
-           (binary-op <=         "<="            5    left)
-           (binary-op <          "<"             5    left)
-           (binary-op >          ">"             5    left)
-           (binary-op >=         ">="            5    left)
-           (binary-op instanceof " instanceof "  5    left)
-           (binary-op in         " in "          5    left)
-           (binary-op ==         "=="            6    left)
-           (binary-op !=         "!="            6    left)
-           (binary-op ===        "==="           6    left)
-           (binary-op !==        "!=="           6    left)
-           (binary-op bit-and    "&"             7    left)
-           (binary-op bit-xor    "^"             8    left)
-           (binary-op bit-or     "|"             9    left)
-           (binary-op and        "&&"           10    left)
-           (binary-op or         "||"           11    left)
+           (binary-op *          "*"             3    left)
+           (binary-op /          "/"             3    left)
+           (binary-op mod        "%"             3    left)
+           (binary-op %          "%"             3    left)
+           (binary-op +          "+"             4    left)
+           (binary-op -          "-"             5    left)
+           (binary-op <<         "<<"            5    left)
+           (binary-op >>         "<<"            5    left)
+           (binary-op >>>        ">>>"           5    left)
+           (binary-op <=         "<="            6    left)
+           (binary-op <          "<"             6    left)
+           (binary-op >          ">"             6    left)
+           (binary-op >=         ">="            6    left)
+           (binary-op instanceof " instanceof "  6    left)
+           (binary-op in         " in "          6    left)
+           (binary-op ==         "=="            7    left)
+           (binary-op !=         "!="            7    left)
+           (binary-op ===        "==="           7    left)
+           (binary-op !==        "!=="           7    left)
+           (binary-op bit-and    "&"             8    left)
+           (binary-op bit-xor    "^"             9    left)
+           (binary-op bit-or     "|"            10    left)
+           (binary-op and        "&&"           11    left)
+           (binary-op or         "||"           12    left)
            (binary-op =          "="            13    right :lvalue t)
            (binary-op +=         "+="           13    right :lvalue t)
            (binary-op incf       "+="           13    right :lvalue t)
            (binary-op progn      ","            13    right)
 
            (when (member op '(? if))
-             (with-operator (12 'right)
-               (js-expr (first args))
-               (js-operator "?")
-               (js-expr (second args))
-               (js-format ":")
-               (js-expr (third args)))
+             (when (low-precedence-p 12) (js-format "("))
+             (js-expr (first args) 12 'right 'left)
+             (js-format "?")
+             (js-expr (second args) 12 'right 'right)
+             (js-format ":")
+             (js-expr (third args) 12 'right 'right)
+             (when (low-precedence-p 12) (js-format ")"))
              (return-from js-operator-expression))
 
            (error "Unknown operator `~S'" op)))))))
 
-(defun js-expr (form)
+(defun js-expr (form &optional (precedence 1000) associativity operand-order)
   (let ((form (js-expand-expr form)))
     (cond
       ((or (symbolp form) (numberp form) (stringp form))
       ((vectorp form)
        (js-vector-initializer form))
       (t
-       (js-operator-expression (car form) (cdr form))))))
+       (js-operator-expression (car form) (cdr form) precedence associativity operand-order)))))
 
 (defun js-expand-stmt (form)
   (cond
     ((and (consp form) (eq (car form) 'progn))
      (destructuring-bind (&body body) (cdr form)
        (cond
-         ((null body)           '(empty))
-         ((null (cdr body))     (js-expand-stmt (car body)))
-         (t                     `(group ,@(cdr form))))))
+         ((null body)
+          nil)
+         ((null (cdr body))
+          (js-expand-stmt (car body)))
+         (t
+          `(group ,@(cdr form))))))
     (t
-     form)))
+     (js-macroexpand form))))
 
 (defun js-stmt (form &optional parent)
   (let ((form (js-expand-stmt form)))
     (flet ((js-stmt (x) (js-stmt x form)))
-      (if (atom form)
-          (progn
-            (js-expr form)
-            (js-format ";"))
-          (case (car form)
-            (empty
-             (unless (and (consp parent) (eq (car parent) 'group))
-               (js-format ";")))
-            (label
-             (destructuring-bind (label &body body) (cdr form)
-               (js-identifier label)
-               (js-format ":")
-               (js-stmt `(progn ,@body))))
-            (break
-             (destructuring-bind (label) (cdr form)
-               (js-format "break ")
-               (js-identifier label)
+      (cond
+        ((null form)
+         (unless (or (and (consp parent) (eq (car parent) 'group))
+                     (null parent))
+           (js-format ";")))
+        ((atom form)
+         (progn
+           (js-expr form)
+           (js-format ";")))
+        (t
+         (case (car form)
+           (label
+            (destructuring-bind (label &body body) (cdr form)
+              (js-identifier label)
+              (js-format ":")
+              (js-stmt `(progn ,@body))))
+           (break
+            (destructuring-bind (&optional label) (cdr form)
+              (js-format "break")
+              (when label
+                (js-format " ")
+                (js-identifier label))
+              (js-format ";")))
+           (return
+             (destructuring-bind (value) (cdr form)
+               (js-format "return ")
+               (js-expr value)
                (js-format ";")))
-            (return
-              (destructuring-bind (value) (cdr form)
-                (js-format "return ")
-                (js-expr value)
-                (js-format ";")))
-            (var
-             (flet ((js-var (spec)
-                      (destructuring-bind (variable &optional initial)
-                          (ensure-list spec)
-                        (js-identifier variable)
-                        (when initial
-                          (js-format "=")
-                          (js-expr initial)))))
-               (destructuring-bind (var &rest vars) (cdr form)
-                 (let ((*js-operator-precedence* 12))
-                   (js-format "var ")
-                   (js-var var)
-                   (dolist (var vars)
-                     (js-format ",")
-                     (js-var var))
-                   (js-format ";")))))
-            (if
-             (destructuring-bind (condition true &optional false) (cdr form)
-               (js-format "if (")
-               (js-expr condition)
-               (js-format ") ")
-               (js-stmt true)
-               (when false
-                 (js-format " else ")
-                 (js-stmt false))))
-            (group
-             (let ((in-group-p
-                    (or (null parent)
-                        (and (consp parent) (eq (car parent) 'group)))))
-               (unless  in-group-p (js-format "{"))
-               (mapc #'js-stmt (cdr form))
-               (unless in-group-p (js-format "}"))))
-            (while
-                (destructuring-bind (condition &body body) (cdr form)
-                  (js-format "while (")
-                  (js-expr condition)
-                  (js-format ")")
-                  (js-stmt `(progn ,@body))))
-            (throw
-                (destructuring-bind (object) (cdr form)
-                  (js-format "throw ")
-                  (js-expr object)
-                  (js-format ";")))
-            (t
-             (js-expr form)
-             (js-format ";")))))))
+           (var
+            (flet ((js-var (spec)
+                     (destructuring-bind (variable &optional initial)
+                         (ensure-list spec)
+                       (js-identifier variable)
+                       (when initial
+                         (js-format "=")
+                         (js-expr initial no-comma)))))
+              (destructuring-bind (var &rest vars) (cdr form)
+                (js-format "var ")
+                (js-var var)
+                (dolist (var vars)
+                  (js-format ",")
+                  (js-var var))
+                (js-format ";"))))
+           (if
+            (destructuring-bind (condition true &optional false) (cdr form)
+              (js-format "if (")
+              (js-expr condition)
+              (js-format ") ")
+              (js-stmt true)
+              (when false
+                (js-format " else ")
+                (js-stmt false))))
+           (group
+            (let ((in-group-p
+                   (or (null parent)
+                       (and (consp parent) (eq (car parent) 'group)))))
+              (unless  in-group-p (js-format "{"))
+              (mapc #'js-stmt (cdr form))
+              (unless in-group-p (js-format "}"))))
+           (while
+               (destructuring-bind (condition &body body) (cdr form)
+                 (js-format "while (")
+                 (js-expr condition)
+                 (js-format ")")
+                 (js-stmt `(progn ,@body))))
+           (switch
+            (destructuring-bind (value &rest cases) (cdr form)
+              (js-format "switch(")
+              (js-expr value)
+              (js-format "){")
+              (dolist (case cases)
+                (cond
+                  ((and (consp case) (eq (car case) 'case))
+                   (js-format "case ")
+                   (let ((value (cadr case)))
+                     (unless (or (stringp value) (integerp value))
+                       (error "Non-constant switch case `~S'." value))
+                     (js-expr value))
+                   (js-format ":"))
+                  ((eq case 'default)
+                   (js-format "default:"))
+                  (t
+                   (js-stmt case))))
+              (js-format "}")))
+           (for
+            (destructuring-bind ((start condition step) &body body) (cdr form)
+              (js-format "for (")
+              (js-expr start)
+              (js-format ";")
+              (js-expr condition)
+              (js-format ";")
+              (js-expr step)
+              (js-format ")")
+              (js-stmt `(progn ,@body))))
+           (for-in
+            (destructuring-bind ((x object) &body body) (cdr form)
+              (js-format "for (")
+              (js-identifier x)
+              (js-format " in ")
+              (js-expr object)
+              (js-format ")")
+              (js-stmt `(progn ,@body))))
+           (try
+            (destructuring-bind (&rest body) (cdr form)
+              (js-format "try")
+              (js-stmt `(group ,@body))))
+           (catch
+               (destructuring-bind ((var) &rest body) (cdr form)
+                 (js-format "catch (")
+                 (js-identifier var)
+                 (js-format ")")
+                 (js-stmt `(group ,@body))))
+           (finally
+            (destructuring-bind (&rest body) (cdr form)
+              (js-format "finally")
+              (js-stmt `(group ,@body))))
+           (throw
+               (destructuring-bind (object) (cdr form)
+                 (js-format "throw ")
+                 (js-expr object)
+                 (js-format ";")))
+           (t
+            (js-expr form)
+            (js-format ";"))))))))
 
 (defun js (&rest stmts)
   (mapc #'js-stmt stmts)