Fix up a few minor plugin inaccuracies

Author: stephencookdev

WrappedPlugin/index.js

@@ -2,16 +2,24 @@ let idInc = 0;
 
 const genPluginMethod = (orig, pluginName, smp, type) =>
   function(method, func) {
-    const id = idInc++;
-    const timeEventName = type + "/" + method;
-
     const wrappedFunc = (...args) => {
+      const id = idInc++;
+      const timeEventName = pluginName + "/" + type + "/" + method;
+      // we don't know if there's going to be a callback applied to a particular
+      // call, so we just set it multiple times, letting each one override the last
+      const addEndEvent = () =>
+        smp.addTimeEvent("plugins", timeEventName, "end", { id });
+
       smp.addTimeEvent("plugins", timeEventName, "start", {
         id,
         name: pluginName,
       });
-      const ret = func.apply(this, args.map(a => wrap(a, pluginName, smp)));
-      smp.addTimeEvent("plugins", timeEventName, "end", { id });
+      const ret = func.apply(
+        this,
+        args.map(a => wrap(a, pluginName, smp, addEndEvent))
+      );
+      addEndEvent();
+
       return ret;
     };
 
@@ -27,17 +35,42 @@ const construcNamesToWrap = [
   "ContextModuleFactory",
 ];
 
-const wrap = (orig, pluginName, smp) => {
-  const origConstrucName = orig && orig.constructor && orig.constructor.name;
-  const shouldWrap = construcNamesToWrap.includes(origConstrucName);
-  if (!shouldWrap) return orig;
+const wrap = (orig, pluginName, smp, addEndEvent) => {
+  if (!orig) return orig;
+
+  const getOrigConstrucName = target =>
+    target && target.constructor && target.constructor.name;
+  const getShouldWrap = target => {
+    const origConstrucName = getOrigConstrucName(target);
+    return construcNamesToWrap.includes(origConstrucName);
+  };
+  const shouldWrap = getShouldWrap(orig);
+  const shouldSoftWrap = Object.values(orig).some(getShouldWrap);
+
+  if (!shouldWrap && !shouldSoftWrap) {
+    const vanillaFunc =
+      typeof orig === "function" &&
+      orig &&
+      orig.prototype &&
+      orig.prototype.constructor !== orig;
+    return vanillaFunc
+      ? function() {
+          const ret = orig();
+          addEndEvent();
+          return ret;
+        }
+      : orig;
+  }
 
   const proxy = new Proxy(orig, {
     get: (target, property) => {
-      if (property === "plugin")
-        return genPluginMethod(orig, pluginName, smp, origConstrucName).bind(
-          proxy
-        );
+      if (shouldWrap && property === "plugin")
+        return genPluginMethod(
+          orig,
+          pluginName,
+          smp,
+          getOrigConstrucName(target)
+        ).bind(proxy);
 
       if (typeof orig[property] === "function")
         return orig[property].bind(proxy);
@@ -47,7 +80,7 @@ const wrap = (orig, pluginName, smp) => {
       return Reflect.set(target, property, value);
     },
     deleteProperty: (target, property) => {
-      delete target[property];
+      return Reflect.deleteProperty(target, property);
     },
   });
 

index.js

@@ -75,10 +75,15 @@ module.exports = class SpeedMeasurePlugin {
       data.start = curTime;
       eventList.push(data);
     } else if (eventType === "end") {
-      const matchingEvent = eventList.find(
-        e => !e.end && (e.id === data.id || e.name === data.name)
-      );
-      const eventToModify = matchingEvent || eventList.find(e => !e.end);
+      const matchingEvent = eventList.find(e => {
+        const allowOverwrite = !e.end || !data.fillLast;
+        const idMatch = e.id !== undefined && e.id === data.id;
+        const nameMatch =
+          !data.id && e.name !== undefined && e.name === data.name;
+        return allowOverwrite && (idMatch || nameMatch);
+      });
+      const eventToModify =
+        matchingEvent || (data.fillLast && eventList.find(e => !e.end));
       if (!eventToModify) {
         console.log(
           "Could not find a matching event to end",
@@ -100,7 +105,7 @@ module.exports = class SpeedMeasurePlugin {
       this.addTimeEvent("misc", "compile", "start", { watch: false });
     });
     compiler.plugin("done", () => {
-      this.addTimeEvent("misc", "compile", "end");
+      this.addTimeEvent("misc", "compile", "end", { fillLast: true });
 
       const output = this.getOutput();
       if (typeof this.options.outputTarget === "string") {
@@ -126,6 +131,7 @@ module.exports = class SpeedMeasurePlugin {
         if (name) {
           this.addTimeEvent("loaders", "build", "start", {
             name,
+            fillLast: true,
             loaders: getLoaderNames(module.loaders),
           });
         }
@@ -134,7 +140,10 @@ module.exports = class SpeedMeasurePlugin {
       compilation.plugin("succeed-module", module => {
         const name = getModuleName(module);
         if (name) {
-          this.addTimeEvent("loaders", "build", "end", { name });
+          this.addTimeEvent("loaders", "build", "end", {
+            name,
+            fillLast: true,
+          });
         }
       });
     });

package.json

@@ -1,6 +1,6 @@
 {
   "name": "speed-measure-webpack-plugin",
-  "version": "0.3.1",
+  "version": "0.3.2",
   "description": "Measure + analyse the speed of your webpack loaders and plugins",
   "main": "index.js",
   "repository": {