angular-cn/packages/zone.js/lib/zone-spec/wtf.ts

/**
 * @license
 * Copyright Google LLC 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
 */
/**
 * @fileoverview
 * @suppress {missingRequire}
 */

(function(global: any) {
interface Wtf {
  trace: WtfTrace;
}
interface WtfScope {}
interface WtfRange {}
interface WtfTrace {
  events: WtfEvents;
  leaveScope(scope: WtfScope, returnValue?: any): void;
  beginTimeRange(rangeType: string, action: string): WtfRange;
  endTimeRange(range: WtfRange): void;
}
interface WtfEvents {
  createScope(signature: string, flags?: any): WtfScopeFn;
  createInstance(signature: string, flags?: any): WtfEventFn;
}

type WtfScopeFn = (...args: any[]) => WtfScope;
type WtfEventFn = (...args: any[]) => any;

// Detect and setup WTF.
let wtfTrace: WtfTrace|null = null;
let wtfEvents: WtfEvents|null = null;
const wtfEnabled: boolean = (function(): boolean {
  const wtf: Wtf = global['wtf'];
  if (wtf) {
    wtfTrace = wtf.trace;
    if (wtfTrace) {
      wtfEvents = wtfTrace.events;
      return true;
    }
  }
  return false;
})();

class WtfZoneSpec implements ZoneSpec {
  name: string = 'WTF';

  static forkInstance =
      wtfEnabled ? wtfEvents!.createInstance('Zone:fork(ascii zone, ascii newZone)') : null;
  static scheduleInstance: {[key: string]: WtfEventFn} = {};
  static cancelInstance: {[key: string]: WtfEventFn} = {};
  static invokeScope: {[key: string]: WtfEventFn} = {};
  static invokeTaskScope: {[key: string]: WtfEventFn} = {};

  onFork(parentZoneDelegate: ZoneDelegate, currentZone: Zone, targetZone: Zone, zoneSpec: ZoneSpec):
      Zone {
    const retValue = parentZoneDelegate.fork(targetZone, zoneSpec);
    WtfZoneSpec.forkInstance!(zonePathName(targetZone), retValue.name);
    return retValue;
  }

  onInvoke(
      parentZoneDelegate: ZoneDelegate, currentZone: Zone, targetZone: Zone, delegate: Function,
      applyThis: any, applyArgs?: any[], source?: string): any {
    const src = source || 'unknown';
    let scope = WtfZoneSpec.invokeScope[src];
    if (!scope) {
      scope = WtfZoneSpec.invokeScope[src] =
          wtfEvents!.createScope(`Zone:invoke:${source}(ascii zone)`);
    }
    return wtfTrace!.leaveScope(
        scope(zonePathName(targetZone)),
        parentZoneDelegate.invoke(targetZone, delegate, applyThis, applyArgs, source));
  }


  onHandleError(parentZoneDelegate: ZoneDelegate, currentZone: Zone, targetZone: Zone, error: any):
      boolean {
    return parentZoneDelegate.handleError(targetZone, error);
  }

  onScheduleTask(parentZoneDelegate: ZoneDelegate, currentZone: Zone, targetZone: Zone, task: Task):
      any {
    const key = task.type + ':' + task.source;
    let instance = WtfZoneSpec.scheduleInstance[key];
    if (!instance) {
      instance = WtfZoneSpec.scheduleInstance[key] =
          wtfEvents!.createInstance(`Zone:schedule:${key}(ascii zone, any data)`);
    }
    const retValue = parentZoneDelegate.scheduleTask(targetZone, task);
    instance(zonePathName(targetZone), shallowObj(task.data, 2));
    return retValue;
  }


  onInvokeTask(
      parentZoneDelegate: ZoneDelegate, currentZone: Zone, targetZone: Zone, task: Task,
      applyThis?: any, applyArgs?: any[]): any {
    const source = task.source;
    let scope = WtfZoneSpec.invokeTaskScope[source];
    if (!scope) {
      scope = WtfZoneSpec.invokeTaskScope[source] =
          wtfEvents!.createScope(`Zone:invokeTask:${source}(ascii zone)`);
    }
    return wtfTrace!.leaveScope(
        scope(zonePathName(targetZone)),
        parentZoneDelegate.invokeTask(targetZone, task, applyThis, applyArgs));
  }

  onCancelTask(parentZoneDelegate: ZoneDelegate, currentZone: Zone, targetZone: Zone, task: Task):
      any {
    const key = task.source;
    let instance = WtfZoneSpec.cancelInstance[key];
    if (!instance) {
      instance = WtfZoneSpec.cancelInstance[key] =
          wtfEvents!.createInstance(`Zone:cancel:${key}(ascii zone, any options)`);
    }
    const retValue = parentZoneDelegate.cancelTask(targetZone, task);
    instance(zonePathName(targetZone), shallowObj(task.data, 2));
    return retValue;
  }
}

function shallowObj(obj: {[k: string]: any}|undefined, depth: number): any {
  if (!obj || !depth) return null;
  const out: {[k: string]: any} = {};
  for (const key in obj) {
    if (obj.hasOwnProperty(key)) {
      // explicit : any due to https://github.com/microsoft/TypeScript/issues/33191
      let value: any = obj[key];
      switch (typeof value) {
        case 'object':
          const name = value && value.constructor && (<any>value.constructor).name;
          value = name == (<any>Object).name ? shallowObj(value, depth - 1) : name;
          break;
        case 'function':
          value = value.name || undefined;
          break;
      }
      out[key] = value;
    }
  }
  return out;
}

function zonePathName(zone: Zone) {
  let name: string = zone.name;
  let localZone = zone.parent;
  while (localZone != null) {
    name = localZone.name + '::' + name;
    localZone = localZone.parent;
  }
  return name;
}

(Zone as any)['wtfZoneSpec'] = !wtfEnabled ? null : new WtfZoneSpec();
})(typeof window === 'object' && window || typeof self === 'object' && self || global);