Add support for devtools with new query and mutations store location

apollographql · Jul 10, 2017 · c823fed · c823fed
1 parent 0cfc9a1
commit c823fed
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 2 deletions.
diff --git a/src/ApolloClient.ts b/src/ApolloClient.ts
@@ -445,7 +445,10 @@ export default class ApolloClient implements DataProxy {
         if (this.devToolsHookCb) {
           this.devToolsHookCb({
             action,
-            state: this.queryManager.getApolloState(),
+            state: {
+              queries: this.queryManager.queryStore.__getStore(),
+              mutations: this.queryManager.mutationStore.__getStore(),
+            },
             dataWithOptimisticResults: this.queryManager.getDataWithOptimisticResults(),
           });
         }
@@ -484,7 +487,10 @@ export default class ApolloClient implements DataProxy {
         if (this.devToolsHookCb) {
           this.devToolsHookCb({
             action,
-            state: this.queryManager.getApolloState(),
+            state: {
+              queries: this.queryManager.queryStore.__getStore(),
+              mutations: this.queryManager.mutationStore.__getStore(),
+            },
             dataWithOptimisticResults: this.queryManager.getDataWithOptimisticResults(),
           });
         }

diff --git a/src/mutations/store.ts b/src/mutations/store.ts
@@ -1,6 +1,10 @@
 export class MutationStore {
   private store: {[mutationId: string]: MutationStoreValue} = {};
 
+  public __getStore(): {[mutationId: string]: MutationStoreValue} {
+    return this.store;
+  }
+
   public get(mutationId: string): MutationStoreValue {
     return this.store[mutationId];
   }

diff --git a/src/queries/store.ts b/src/queries/store.ts
@@ -16,6 +16,10 @@ import { NetworkStatus } from './networkStatus';
 export class QueryStore {
   private store: {[queryId: string]: QueryStoreValue} = {};
 
+  public __getStore(): {[queryId: string]: QueryStoreValue} {
+    return this.store;
+  }
+
   public get(queryId: string): QueryStoreValue {
     return this.store[queryId];
   }