handle uncaught exceptions in nodejs

Author: RealyUniqueName

haxelib.json

@@ -4,8 +4,8 @@
   "license"      : "MIT",
   "tags"         : ["js", "php7", "stack", "callstack", "stacktrace"],
   "description"  : "Friendly stack traces for JS and PHP7 targets. Makes them point to haxe sources.",
-  "version"      : "2.1.0",
-  "releasenote"  : "Support for php7 target",
+  "version"      : "2.1.1",
+  "releasenote"  : "Handle uncaught exceptions in nodejs",
   "classPath"    : "src",
   "contributors" : ["RealyUniqueName"],
   "dependencies" : {

src/jstack/js/JStack.hx

@@ -2,22 +2,20 @@ package jstack.js;
 
 import haxe.CallStack;
 import sourcemap.SourcePos;
-import js.Lib;
+import js.Error;
 import haxe.io.Path;
 import js.Browser;
 import haxe.Http;
 
-using StringTools;
-
 /**
  * Handles source map
  */
 class JStack {
 
     /** Create instance just to invoke `inject()` */
-    static private var instance = new JStack();
+    static var instance = new JStack();
     /** User-defined callback which will be invoked when sourceMap is loaded */
-    static private var onReadyCallback : Void->Void;
+    static var onReadyCallback : Void->Void;
 
     /** Indicates if source map is loaded */
     public var ready (default,null) : Bool = false;
@@ -28,25 +26,35 @@ class JStack {
      * A call to this method is automatically injected in `static main()` function of your app.
      * You don't need to use this method manually.
      */
-    static public function onReady (callback:Void->Void) : Void
-    {
+    static public function onReady (callback:Void->Void) {
         onReadyCallback = callback;
         if (instance.ready) callback();
     }
 
+    /**
+     * Overwrite this method to handle uncaught exceptions manually.
+     * Any returned value will be written to stderr.
+     * @param e - Uncaught exception.
+     */
+    @:access(haxe.CallStack)
+    static public dynamic function uncaughtExceptionHandler (e:Error) : String {
+        var stack = CallStack.getStack(e);
+        var error = e.message + CallStack.toString(stack) + '\n';
+        return error;
+    }
 
-    private function new ()
-    {
+    function new () {
+        if(isNode()) {
+            untyped __js__('process.on("uncaughtException", {0})', _uncaughtExceptionHandler);
+        }
         inject();
     }
 
-
     /**
      * Loads source map and injects hooks into `haxe.CallStack`.
      * It's asynchronous process so haxe-related positions in call stack might be not available right away.
      */
-    public function inject () : Void
-    {
+    public function inject () {
         loadSourceMap(function (sourceMapData:String) {
             var mapper = new SourceMap(sourceMapData);
 
@@ -66,9 +74,8 @@ class JStack {
     /**
      * Load source map and pass it to `callback`
      */
-    private function loadSourceMap (callback : String->Void) : Void
-    {
-        if (untyped __js__("typeof window != 'undefined'")) {
+    function loadSourceMap (callback : String->Void) {
+        if (!isNode()) {
             loadInBrowser(callback);
         } else {
             loadInNode(callback);
@@ -79,8 +86,7 @@ class JStack {
     /**
      * Do the job in browser environment
      */
-    private function loadInBrowser (callback:String->Void) : Void
-    {
+    function loadInBrowser (callback:String->Void) {
         var file = getCurrentDirInBrowser() + '/' + Tools.getSourceMapFileName();
         var http = new Http(file);
 
@@ -97,8 +103,7 @@ class JStack {
     /**
      * Do the job in nodejs environment
      */
-    public function loadInNode (callback:String->Void) : Void
-    {
+    public function loadInNode (callback:String->Void) {
         var dir : String = untyped __js__("__dirname");
         var fs = untyped __js__("require('fs')");
 
@@ -115,8 +120,7 @@ class JStack {
     /**
      * Scans DOM for <script> tags to find current script directory
      */
-    public function getCurrentDirInBrowser () : String
-    {
+    public function getCurrentDirInBrowser () : String {
         var file = Tools.getOutputFileName();
         var scripts = Browser.document.getElementsByTagName('script');
 
@@ -132,22 +136,37 @@ class JStack {
         return path.dir;
     }
 
+    /**
+     * Actual handler used for uncaught exceptions
+     * @param e -
+     */
+    static function _uncaughtExceptionHandler (e:Error) {
+        var error = uncaughtExceptionHandler(e);
+        if(error != null && error.length > 0) {
+            untyped __js__('console.log({0})', error);
+        }
+    }
+
+    /**
+        Check if currently run on nodejs.
+    **/
+    static function isNode () : Bool {
+        return untyped __js__("typeof window == 'undefined'");
+    }
 }
 
 
 /**
  * Represents call stack position
  */
 @:keep
-private class StackPos
-{
+class StackPos {
     /** JS side */
-    private var js : Dynamic;
+    var js : Dynamic;
     /** HX side */
-    private var hx : Dynamic;
+    var hx : Dynamic;
 
-    public function new (js:Dynamic, hx:SourcePos)
-    {
+    public function new (js:Dynamic, hx:SourcePos) {
         this.js = js;
         this.hx = hx;
     }