Optimize wrappers

Author: tshemsedinov

JavaScript/8-cancelable.js

@@ -6,6 +6,7 @@ const cancelable = (fn) => {
   };
   wrapper.cancel = () => {
     fn = null;
+    return wrapper;
   };
   return wrapper;
 };

JavaScript/9-wrap.js

@@ -5,8 +5,12 @@ const wrap = (fn) => {
   let counter = 0;
 
   const wrapper = (...args) => {
-    if (limit && counter++ === limit) wrapper.cancel();
-    if (fn) return fn(...args);
+    if (limit && counter === limit) wrapper.cancel();
+    if (fn) {
+      const res = fn(...args);
+      counter++;
+      return res;
+    }
   };
 
   wrapper.cancel = () => {

JavaScript/a-optimized.js

@@ -8,13 +8,15 @@ const wrap = (func) => {
 
   const wrapper = (...args) => {
     if (!fn) return;
-    if (limit && counter++ === limit) {
+    if (limit && counter === limit) {
       limit = 0;
       counter = 0;
       this.cancel();
       return;
     }
-    return fn(...args);
+    const res = fn(...args);
+    counter++
+    return res;
   };
 
   const methods = {