Added a filter for specifying the variable to proof (#18)

* Added a filter for specifying the variable to proof

* removed console.log

* Update version library

package.json

@@ -1,7 +1,7 @@
 {
   "name": "zk-states",
   "packageManager": "[email protected]",
-  "version": "0.0.14",
+  "version": "0.0.15",
   "description": "Verifiable state manager for React based on SnarkyJS & Mina protocol",
   "typesVersions": {
     "<4.0": {

src/hooks.ts

@@ -6,10 +6,10 @@ import { ZkAppWorkerClient } from "./zkAppWorkerClient";
 
 let oldState = INITIAL_STATE;
 
-const stringifyState = (state: object) => {
+const stringifyState = <T extends object>(state: T, toProof: (keyof T)[]) => {
   const stateVariables: Record<string, unknown> = {};
   Object.entries(state).forEach(([key, value]) => {
-    if (typeof value !== "function") {
+    if (typeof value !== "function" && toProof.includes(key as keyof T)) {
       stateVariables[key] = value;
     }
   });
@@ -22,6 +22,7 @@ const stringifyState = (state: object) => {
 export const createZKState = <T extends object>(
   worker: Worker,
   createState: StateCreator<T, [], []>,
+  toProof: (keyof T)[],
 ) => {
   const useZKStore = create<T>(createState);
   const zkAppWorkerClient = new ZkAppWorkerClient(worker);
@@ -55,7 +56,7 @@ export const createZKState = <T extends object>(
     useEffect(() => {
       if (!isInitialized) return;
 
-      const newState = stringifyState(state);
+      const newState = stringifyState<T>(state, toProof);
 
       if (newState === oldState) return;