Add support for devtools with new query and mutations store location

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

Hide file tree

Showing 4 changed files with 17 additions and 3 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/core/QueryManager.ts b/src/core/QueryManager.ts
@@ -456,7 +456,7 @@ export class QueryManager {
       variables,
       isPoll: fetchType === FetchType.poll,
       isRefetch: fetchType === FetchType.refetch,
-      metadata, fetchMoreForQueryId
+      metadata, fetchMoreForQueryId,
     });
 
     this.broadcastQueries();

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];
   }