fix(cli): keep unregister function

Tencent · Sep 3, 2019 · dafd8b2 · dafd8b2
1 parent c8b1b1e
commit dafd8b2
Showing 1 changed file with 15 additions and 15 deletions.
diff --git a/packages/cli/core/hook.js b/packages/cli/core/hook.js
@@ -9,21 +9,9 @@ class Hook {
     }
     this._hooks[key].push(fn);
 
-    const createUnregister = function (key, fn) {
-      return function () {
-        let fns = this._hooks[key];
-        if (fns && typeof fn === 'function') {
-          fns = fns.filter(f => f !== fn);
-          if (fns.length > 0) {
-            this._hooks[key] = fns;
-          } else {
-            delete this._hooks[key];
-          }
-        }
-      };
-    };
-
-    return createUnregister(key, fn).bind(this);
+    return (() => {
+      this.unregister(key, fn);
+    });
   }
 
   hasHook (key) {
@@ -111,6 +99,18 @@ class Hook {
     return (args.length <= 1 ? args[0] : args);
   }
 
+  unregister (key, fn) {
+    let fns = this._hooks[key];
+    if (fns && typeof fn === 'function') {
+      fns = fns.filter(f => f !== fn);
+      if (fns.length > 0) {
+        this._hooks[key] = fns;
+      } else {
+        delete this._hooks[key];
+      }
+    }
+  }
+
   unregisterAll (key) {
     delete this._hooks[key];
   }