Merge pull request #9 from hjvvoorthuijsen/Callback-function-in-setState

Callback function in set state
antoinejaussoin · Jun 17, 2020 · e71fa50 · e71fa50
2 parents 9f03f37 + b48883b
commit e71fa50
Show file tree

Hide file tree

Showing 6 changed files with 14,280 additions and 6 deletions.
diff --git a/Jooks.js b/Jooks.js
@@ -261,7 +261,16 @@ var Jooks = /** @class */ (function () {
     Jooks.prototype.mockUseState = function (defaultValue) {
         var _this = this;
         var localPointer = this.stateStore.pointer;
+        var isFunction = function (arg) { return typeof arg === 'function'; };
         var setState = function (v) {
+            if (isFunction(v)) {
+                var newState = v(_this.stateStore.store[localPointer]);
+                if (_this.verbose) {
+                    console.log('Set state to ', newState, ' pointer ', localPointer);
+                }
+                _this.stateStore.store[localPointer] = newState;
+                return;
+            }
             if (_this.verbose) {
                 console.log('Set state to ', v, ' pointer ', localPointer);
             }