stefangomez · July 1, 2019 04:39 · Jul 1, 2019 · Jul 1, 2019
diff --git a/appsync-client.js b/appsync-client.js
@@ -0,0 +1,49 @@
+import Amplify from "@aws-amplify/core";
+import Auth from "@aws-amplify/auth";
+import { setContext } from "apollo-link-context";
+import { ApolloLink } from "apollo-link";
+import { createHttpLink } from "apollo-link-http";
+import AWSAppSyncClient, { AUTH_TYPE, createAppSyncLink } from "aws-appsync";
+import awsConfig from "../aws-exports";
+
+Amplify.configure({
+  ...awsConfig,
+  Analytics: { disabled: true }
+});
+
+const appsyncClient = new AWSAppSyncClient(awsConfig, {
+  link: createAppSyncLink({
+    url: awsConfig.aws_appsync_graphqlEndpoint,
+    region: awsConfig.aws_appsync_region,
+    auth: {
+      type: AUTH_TYPE.AWS_IAM,
+      credentials: async () => await Auth.currentCredentials()
+    },
+    resultsFetcherLink: ApolloLink.from([
+      setContext(async (request, previousContext) => {
+        try {
+          const session = await Auth.currentSession();
+          if (session) {
+            delete previousContext.headers["X-Amz-Security-Token"];
+            return {
+              headers: {
+                ...previousContext.headers,
+                Authorization: session.getIdToken().getJwtToken()
+              }
+            };
+          }
+        } catch (e) {
+          // log or catch error
+        }
+        return {
+          headers: previousContext.headers
+        };
+      }),
+      createHttpLink({
+        uri: awsConfig.aws_appsync_graphqlEndpoint
+      })
+    ])
+  })
+});
+
+export default appsyncClient;