trace forms

gistfile1.clj

(ns trace
  (:use clojure.pprint))

(def *ignore*
  '#{def quote var try fn* monitor-enter monitor-exit})

(defmulti trace-special-form (fn [form] (first form)))

(defn- trace-bindings [bindings]
  (vec (apply concat
              (map (fn [[sym value]]
                     `[~sym (trace-forms ~value)]) (partition 2 bindings)))))

(defmethod trace-special-form 'let* [[_ bindings & body]]
  `(let ~(trace-bindings bindings)
     (trace-forms ~@body)))

(defmethod trace-special-form 'loop* [[_ bindings & body]]
  `(loop ~(trace-bindings bindings)
     (trace-forms ~@body)))

(defmethod trace-special-form :default [form]
  :default)

(defn trace-form [form]
  `(try
     ~(if (and (or (list? form)
                   (seq? form))
               (> (count form) 0))
        (if (*ignore* (first form))
          form
          (let [sform (trace-special-form form)]
            (if (= sform :default)
              (let [mform (macroexpand-1 form)]
                (if (= form mform)
                  (map trace-form mform)
                  (trace-form mform)))
              sform)))
        form)
       (catch Exception e#
         (print "Form failed: ")
         (pprint '~form)
         (throw e#))))

(defmacro trace-forms [& body]
  `(do
     ~@(map trace-form body)))