coderoad
diff --git a/‎web-app/src/services/apollo/index.ts
Lines changed: 41 additions & 1 deletion b/‎web-app/src/services/apollo/index.ts
Lines changed: 41 additions & 1 deletion
diff --git a/‎web-app/stories/utils/ApolloDecorator.tsx
Lines changed: 2 additions & 9 deletions b/‎web-app/stories/utils/ApolloDecorator.tsx
Lines changed: 2 additions & 9 deletions
@@ -1,7 +1,47 @@
-import ApolloClient from 'apollo-boost'
+import ApolloClient, { InMemoryCache } from 'apollo-boost'
 
 const client = new ApolloClient({
     uri: process.env.REACT_APP_GQL_URI,
+    headers: {
+        Authorization: process.env.GQL_AUTH_TOKEN,
+    },
+    cache: new InMemoryCache(),
+    resolvers: {
+        Mutation: {
+            setStatus: (_root, variables, { cache, getCacheKey }) => {
+                // TODO: optimize status setting to act on diffs
+
+                // set local cache
+                function set(typename: string, id: string, status: 'ACTIVE' | 'COMPLETE') {
+                    const writeId = getCacheKey({ __typename: typename, id })
+                    const data = { status }
+                    cache.writeData({ id: writeId, data })
+                }
+
+                const { progress, position } = variables
+
+                // set level progress & active
+                for (const levelId of Object.keys(progress.levels)) {
+                    set('Level', levelId, 'COMPLETE')
+                }
+                set('Level', position.levelId, 'ACTIVE')
+
+                // set stage progress & active
+                for (const stageId of Object.keys(progress.stages)) {
+                    set('Stage', stageId, 'COMPLETE')
+                }
+                set('Stage', position.stageId, 'ACTIVE')
+
+                // set step progress & active
+                for (const stepId of Object.keys(progress.steps)) {
+                    set('Step', stepId, 'COMPLETE')
+                }
+                set('Step', position.stepId, 'ACTIVE')
+
+                return null
+            },
+        },
+    },
 })
 
 export default client
@@ -1,17 +1,10 @@
 import React, { Fragment } from 'react'
-import ApolloClient from 'apollo-boost'
+import client from '../../src/services/apollo'
 import { ApolloProvider } from '@apollo/react-hooks'
 
-const graphqlClient = new ApolloClient({
-  uri: 'http://localhost:4000/graphql',
-  headers: {
-    Authorization: process.env.GQL_AUTH_TOKEN,
-  },
-})
-
 function StorybookProvider({ children }) {
   return (
-    <ApolloProvider client={graphqlClient}>
+    <ApolloProvider client={client}>
       <Fragment>{children}</Fragment>
     </ApolloProvider>
   )