Patch captureStackTrace/prepareStackTrace to ZoneAwareError, patch process.nextTick, fix removeAllListeners bug

lib/common/utils.ts

@@ -155,7 +155,7 @@ function findAllExistingRegisteredTasks(target: any, name: string, capture: bool
   const eventTasks: Task[] = target[EVENT_TASKS];
   if (eventTasks) {
     const result = [];
-    for (var i = eventTasks.length - 1; i >= 0; i --) {
+    for (let i = eventTasks.length - 1; i >= 0; i --) {
       const eventTask = eventTasks[i];
       const data = <ListenerTaskMeta>eventTask.data;
       if (data.eventName === name && data.useCapturing === capture) {
@@ -280,17 +280,24 @@ export function makeZoneAwareRemoveAllListeners(fnName: string, useCapturingPara
   const defaultUseCapturing = useCapturingParam ? false : undefined;
 
   return function zoneAwareRemoveAllListener(self: any, args: any[]) {
-    var eventName = args[0];
-    var useCapturing = args[1] || defaultUseCapturing;
-    var target = self || _global;
-    var eventTasks = findAllExistingRegisteredTasks(target, eventName, useCapturing, true);
-    if (eventTasks) {
-      for (var i = 0; i < eventTasks.length; i ++) {
-        var eventTask = eventTasks[i];
-        eventTask.zone.cancelTask(eventTask);
-      }
+    const target = self || _global;
+    if (args.length === 0) {
+      // remove all listeners without eventName
+      target[EVENT_TASKS] = [];
+      // we don't cancel Task either, because call native eventEmitter.removeAllListeners will
+      // will do remove listener(cancelTask) for us
+      target[symbol]();
+      return;
     }
-    target[symbol](eventName, useCapturing);
+    const eventName = args[0];
+    const useCapturing = args[1] || defaultUseCapturing;
+    // call this function just remove the related eventTask from target[EVENT_TASKS]
+    findAllExistingRegisteredTasks(target, eventName, useCapturing, true);
+    // we don't need useCapturing here because useCapturing is just for DOM, and
+    // removeAllListeners should only be called by node eventEmitter
+    // and we don't cancel Task either, because call native eventEmitter.removeAllListeners will
+    // will do remove listener(cancelTask) for us
+    target[symbol](eventName);
   }
 }
 

lib/node/node.ts

@@ -11,6 +11,7 @@ import './events';
 import './fs';
 
 import {patchTimer} from '../common/timers';
+import {patchMethod} from '../common/utils';
 
 const set = 'set';
 const clear = 'clear';
@@ -30,6 +31,7 @@ if (shouldPatchGlobalTimers) {
   patchTimer(_global, set, clear, 'Immediate');
 }
 
+patchNextTick();
 
 // Crypto
 let crypto;
@@ -83,3 +85,28 @@ if (httpClient && httpClient.ClientRequest) {
     }
   };
 }
+
+function patchNextTick() {
+  let setNative = null;
+
+  function scheduleTask(task: Task) {
+    const args = task.data;
+    args[0] = function() {
+      task.invoke.apply(this, arguments);
+    };
+    setNative.apply(process, args);
+    return task;
+  }
+
+  setNative =
+      patchMethod(process, 'nextTick', (delegate: Function) => function(self: any, args: any[]) {
+        if (typeof args[0] === 'function') {
+          const zone = Zone.current;
+          const task = zone.scheduleMicroTask('nextTick', args[0], args, scheduleTask);
+          return task;
+        } else {
+          // cause an error by calling it directly.
+          return delegate.apply(process, args);
+        }
+      });
+}

lib/zone.ts

@@ -1264,7 +1264,7 @@ const Zone: ZoneType = (function(global: any) {
     /// Skip this frame when printing out stack
     blackList,
     /// This frame marks zone transition
-    trasition
+    transition
   };
   const NativeError = global[__symbol__('Error')] = global.Error;
   // Store the frames which should be removed from the stack frames
@@ -1304,7 +1304,7 @@ const Zone: ZoneType = (function(global: any) {
           if (frameType === FrameType.blackList) {
             frames.splice(i, 1);
             i--;
-          } else if (frameType === FrameType.trasition) {
+          } else if (frameType === FrameType.transition) {
             if (zoneFrame.parent) {
               // This is the special frame where zone changed. Print and process it accordingly
               frames[i] += ` [${zoneFrame.parent.zone.name} => ${zoneFrame.zone.name}]`;
@@ -1337,6 +1337,21 @@ const Zone: ZoneType = (function(global: any) {
     });
   }
 
+  if (NativeError.hasOwnProperty('captureStackTrace')) {
+    Object.defineProperty(ZoneAwareError, 'captureStackTrace', {
+      value: function(targetObject: Object, constructorOpt?: Function) {
+        NativeError.captureStackTrace(targetObject, constructorOpt);
+      }
+    });
+  }
+
+  Object.defineProperty(ZoneAwareError, 'prepareStackTrace', {
+    get: function() { return NativeError.prepareStackTrace; },
+    set: function(value) { return NativeError.prepareStackTrace = value; }
+  });
+
+  // Now we need to populet the `blacklistedStackFrames` as well as find the
+
   // Now we need to populet the `blacklistedStackFrames` as well as find the
   // run/runGuraded/runTask frames. This is done by creating a detect zone and then threading
   // the execution through all of the above methods so that we can look at the stack trace and
@@ -1364,7 +1379,7 @@ const Zone: ZoneType = (function(global: any) {
             // FireFox: Zone.prototype.run@http://localhost:9876/base/build/lib/zone.js:101:24
             // Safari: run@http://localhost:9876/base/build/lib/zone.js:101:24
             let fnName: string = frame.split('(')[0].split('@')[0];
-            let frameType = FrameType.trasition;
+            let frameType = FrameType.transition;
             if (fnName.indexOf('ZoneAwareError') !== -1) {
               zoneAwareFrame = frame;
             }

package.json

@@ -64,6 +64,6 @@
     "ts-loader": "^0.6.0",
     "tslint": "^3.15.1",
     "typescript": "^2.0.2",
-    "whatwg-fetch": "https://github.com/jimmywarting/fetch.git"
+    "whatwg-fetch": "^2.0.1"
   }
 }

test/node/Error.spec.ts

@@ -0,0 +1,29 @@
+/**
+ * @license
+ * Copyright Google Inc. All Rights Reserved.
+ *
+ * Use of this source code is governed by an MIT-style license that can be
+ * found in the LICENSE file at https://angular.io/license
+ */
+
+describe('ZoneAwareError', () => {
+    // If the environment does not supports stack rewrites, then these tests will fail
+    // and there is no point in running them.
+    if (!Error['stackRewrite']) return;
+
+    it ('should have all properties from NativeError', () => {
+        let obj: any = new Object();
+        Error.captureStackTrace(obj);
+        expect(obj.stack).not.toBeUndefined();
+    });
+
+    it ('should support prepareStackTrace', () => {
+        (<any>Error).prepareStackTrace = function(error, stack) {
+            return stack;
+        }
+        let obj: any = new Object();
+        Error.captureStackTrace(obj);
+        expect(obj.stack[0].getFileName()).not.toBeUndefined();
+    });
+});
+

test/node/events.spec.ts

@@ -105,6 +105,15 @@ describe('nodejs EventEmitter', () => {
       expect(emitter.listeners('test').length).toEqual(0);
     });
   });
+  it ('should remove All listeners properly even without a type parameter', () => {
+    zoneA.run(() => {
+      emitter.on('test', shouldNotRun);
+      emitter.on('test1', shouldNotRun);
+      emitter.removeAllListeners();
+      expect(emitter.listeners('test').length).toEqual(0);
+      expect(emitter.listeners('test1').length).toEqual(0);
+    });
+  });
   it ('should remove once listener after emit', () => {
     zoneA.run(() => {
       emitter.once('test', expectZoneA);
@@ -119,4 +128,43 @@ describe('nodejs EventEmitter', () => {
       emitter.emit('test');
     });
   });
+  it ('should trigger removeListener when remove listener', () => {
+    zoneA.run(() => {
+      emitter.on('removeListener', function(type, handler) {
+         zoneResults.push('remove' + type);
+      });
+      emitter.on('newListener', function(type, handler) {
+        zoneResults.push('new' + type);
+      });
+      emitter.on('test', shouldNotRun);
+      emitter.removeListener('test', shouldNotRun);
+      expect(zoneResults).toEqual(['newtest', 'removetest']);
+    });
+  });
+  it ('should trigger removeListener when remove all listeners with eventname ', () => {
+    zoneA.run(() => {
+      emitter.on('removeListener', function(type, handler) {
+        zoneResults.push('remove' + type);
+      });
+      emitter.on('test', shouldNotRun);
+      emitter.on('test1', expectZoneA);
+      emitter.removeAllListeners('test');
+      expect(zoneResults).toEqual(['removetest']);
+      expect(emitter.listeners('removeListener').length).toBe(1);
+    });
+  });
+  it ('should trigger removeListener when remove all listeners without eventname', () => {
+    zoneA.run(() => {
+      emitter.on('removeListener', function(type, handler) {
+        zoneResults.push('remove' + type);
+      });
+      emitter.on('test', shouldNotRun);
+      emitter.on('test1', shouldNotRun);
+      emitter.removeAllListeners();
+      expect(zoneResults).toEqual(['removetest', 'removetest1']);
+      expect(emitter.listeners('test').length).toBe(0);
+      expect(emitter.listeners('test1').length).toBe(0);
+      expect(emitter.listeners('removeListener').length).toBe(0);
+    });
+  });
 });

test/node/process.spec.ts

@@ -0,0 +1,82 @@
+/**
+ * @license
+ * Copyright Google Inc. All Rights Reserved.
+ *
+ * Use of this source code is governed by an MIT-style license that can be
+ * found in the LICENSE file at https://angular.io/license
+ */
+
+describe('process related test', () => {
+    let zoneA, result;
+    beforeEach(() => {
+        zoneA = Zone.current.fork({name: 'zoneA'});
+        result = [];
+    });
+    it('process.nextTick callback should in zone', (done) => {
+      zoneA.run(function() {
+          process.nextTick(() => {
+            expect(Zone.current.name).toEqual('zoneA');
+            done();
+          });
+      });
+    });
+    it('process.nextTick should be excuted before macroTask and promise', (done) => {
+        zoneA.run(function() {
+            setTimeout(() => {
+              result.push('timeout');
+            }, 0);
+            process.nextTick(() => {
+              result.push('tick');
+            });
+            setTimeout(() => {
+              expect(result).toEqual(['tick', 'timeout']);
+              done();
+            });
+        });
+    });
+    it ('process.nextTick should be treated as microTask', (done) => {
+       let zoneTick = Zone.current.fork({
+           name: 'zoneTick',
+           onScheduleTask: (parentZoneDelegate: ZoneDelegate, currentZone: Zone, targetZone: Zone, task: Task): Task => {
+             result.push(
+                 {
+                   callback: 'scheduleTask',
+                   targetZone: targetZone.name,
+                   task: task.source
+                 });
+             return parentZoneDelegate.scheduleTask(targetZone, task);
+           },
+           onInvokeTask: (parentZoneDelegate: ZoneDelegate, currentZone: Zone, targetZone: Zone, task: Task, applyThis?: any, applyArgs?: any): any => {
+             result.push(
+                 {
+                     callback: 'invokeTask',
+                     targetZone: targetZone.name,
+                     task: task.source
+                 });
+             return parentZoneDelegate.invokeTask(targetZone, task, applyThis, applyArgs);
+           }
+       });
+       zoneTick.run(() => {
+         process.nextTick(() => {
+           result.push('tick');
+         });
+       });
+       setTimeout(() => {
+           expect(result.length).toBe(3);
+           expect(result[0]).toEqual(
+               {
+                   callback: 'scheduleTask',
+                   targetZone: 'zoneTick',
+                   task: 'nextTick'
+               });
+           expect(result[1]).toEqual(
+               {
+                   callback: 'invokeTask',
+                   targetZone: 'zoneTick',
+                   task: 'nextTick'
+               }
+           );
+           done();
+       });
+    })
+});

test/node_tests.ts

@@ -7,4 +7,6 @@
  */
 
 import './node/events.spec';
-import './node/fs.spec';
+import './node/fs.spec';
+import './node/process.spec';
+import './node/Error.spec';