Merge pull request #192 from gigasquid/exception_printing_querying

Exception printing and querying

Author: heyLu

pixie/stdlib.pxi

@@ -2104,7 +2104,6 @@ Expands to calls to `extend-type`."
   ([sb] (str sb))
   ([sb item] (conj! sb item)))
 
-
 (defmacro using [bindings & body]
   (let [pairs (partition 2 bindings)
         names (map first pairs)]
@@ -2114,3 +2113,16 @@ Expands to calls to `extend-type`."
                 `(-dispose! ~nm))
               names)
        result#)))
+(defn pst
+  {:doc "Prints the trace of a Runtime Exception if given, or the last Runtime Exception in *e"
+   :signatures [[] [e]]
+   :added "0.1"}
+  ([] (pst *e))
+  ([e] (when e (print (str e)))))
+
+(defn trace
+  {:doc "Returns a seq of the trace of a Runtime Exception or the last Runtime Exception in *e"
+   :signatures [[] [e]]
+   :added "0.1"}
+  ([] (trace *e))
+  ([e] (seq e)))

pixie/vm/object.py

@@ -122,7 +122,6 @@ def __repr__(self):
 
         s.extend([u"RuntimeException: " + rt.name(rt.str(self._data)) + u"\n"])
 
-
         return u"".join(s)
 
 class WrappedException(Exception):
@@ -176,13 +175,34 @@ def __repr__(self):
     def interpreter_code_info_state(self):
         return self._line, self._line_number, self._column_number, self._file
 
+    def trace_map(self):
+        from pixie.vm.string import String
+        from pixie.vm.numbers import Integer
+        from pixie.vm.keyword import keyword
+
+        tm = {keyword(u"type") : keyword(u"interpreter")}
+        tm[keyword(u"line")] = String(self._line.__repr__())
+        tm[keyword(u"line-number")] = Integer(self._line_number)
+        tm[keyword(u"column-number")] = Integer(self._column_number)
+        tm[keyword(u"file")] = String(self._file)
+        return tm
+
 class NativeCodeInfo(ErrorInfo):
     def __init__(self, name):
         self._name = name
 
     def __repr__(self):
         return u"in internal function " + self._name + u"\n"
 
+    def trace_map(self):
+        from pixie.vm.string import String
+        from pixie.vm.numbers import Integer
+        from pixie.vm.keyword import keyword
+
+        tm = {keyword(u"type") : keyword(u"native")}
+        tm[keyword(u"name")] = String(self._name)
+        return tm
+
 class PolymorphicCodeInfo(ErrorInfo):
     def __init__(self, name, tp):
         self._name = name
@@ -193,11 +213,30 @@ def __repr__(self):
         assert isinstance(tp, Type)
         return u"in polymorphic function " + self._name + u" dispatching on " + tp._name + u"\n"
 
+    def trace_map(self):
+        from pixie.vm.string import String
+        from pixie.vm.numbers import Integer
+        from pixie.vm.keyword import keyword
 
+        tp = self._tp
+        assert isinstance(tp, Type)
+        tm = {keyword(u"type") : keyword(u"polymorphic")}
+        tm[keyword(u"name")] = String(self._name)
+        tm[keyword(u"tp")] = String(tp._name)
+        return tm
 
 class PixieCodeInfo(ErrorInfo):
     def __init__(self, name):
         self._name = name
 
     def __repr__(self):
         return u"in pixie function " + self._name + u"\n"
+
+    def trace_map(self):
+        from pixie.vm.string import String
+        from pixie.vm.numbers import Integer
+        from pixie.vm.keyword import keyword
+
+        tm = {keyword(u"type") : keyword(u"pixie")}
+        tm[keyword(u"name")] = String(self._name)
+        return tm

pixie/vm/rt.py

@@ -143,7 +143,7 @@ def reinit():
     # stacklet.with_stacklets(run_load_stdlib)
 
     init_fns = [u"reduce", u"get", u"reset!", u"assoc", u"key", u"val", u"keys", u"vals", u"vec", u"load-file", u"compile-file",
-                u"load-ns"]
+                u"load-ns", u"hashmap"]
     for x in init_fns:
         globals()[py_str(code.munge(x))] = unwrap(code.intern_var(u"pixie.stdlib", x))
 

pixie/vm/stdlib.py

@@ -774,8 +774,28 @@ def _str(self):
     assert isinstance(self, RuntimeException)
     return rt.wrap(self.__repr__())
 
+@extend(_seq, RuntimeException)
+def _seq(self):
+    import pixie.vm.persistent_vector as vector
+    import pixie.vm.persistent_hash_map as hmap
+    from pixie.vm.keyword import keyword
+    assert isinstance(self, RuntimeException)
+    trace = vector.EMPTY
+    trace_element = rt.hashmap(keyword(u"type"), keyword(u"runtime"))
+    trace_element = rt.assoc(trace_element, keyword(u"data"), rt.wrap(self._data))
+    trace = rt.conj(trace, trace_element)
+    for x in self._trace:
+        tmap = x.trace_map()
+        trace_element = hmap.EMPTY
+        for key in tmap:
+            val = tmap[key]
+            trace_element = rt.assoc(trace_element, key, val)
+
+        trace = rt.conj(trace, trace_element)
+
+    return rt._seq(trace)
+
 @as_var("ex-msg")
 def ex_msg(e):
     assert isinstance(e, RuntimeException)
     return e._data
-

tests/pixie/tests/test-stdlib.pxi

@@ -377,3 +377,11 @@
   (t/assert= (transduce (drop-while even?) conj [2 4 6 7 8]) [7 8])
   (t/assert= (transduce (drop-while even?) conj [0 2] [1 4 6]) [0 2 1 4 6])
   (t/assert= (transduce (drop-while even?) conj [0 2] [2 4 6 7 8]) [0 2 7 8]))
+
+
+(t/deftest test-trace
+  (try
+    (/ 0 0)
+    (catch e
+        (t/assert= (first (trace e)) {:type :runtime :data "Divide by zero"})
+        (t/assert= (second (trace e)) {:type :native :name "_div"} ))))