[header-link] adding header-link

nutshell-lab · Nov 29, 2018 · bc28b7b · bc28b7b
1 parent a02c409
commit bc28b7b
Show file tree

Hide file tree

Showing 5 changed files with 31 additions and 5 deletions.
diff --git a/package.json b/package.json
@@ -33,6 +33,7 @@
     "apollo-cache-inmemory": "^1.3.11",
     "apollo-client": "^2.4.7",
     "apollo-link": "^1.2.4",
+    "apollo-link-context": "^1.0.10",
     "apollo-link-http": "^1.5.7",
     "aws-sdk": "^2.363.0",
     "aws4": "^1.8.0",

diff --git a/src/header-link.js b/src/header-link.js
@@ -0,0 +1,14 @@
+import { setContext } from 'apollo-link-context'
+
+export default headers =>
+  setContext((_request, context) => {
+    console.log('context', context)
+    if (!context.graphqlContext) return {}
+    if (!context.graphqlContext.headers) return {}
+    return {
+      headers: {
+        ...context.graphqlContext.headers,
+        ...headers
+      }
+    }
+  })
diff --git a/src/index.js b/src/index.js
@@ -1,5 +1,6 @@
 export { default as graphqlHttpRequest } from './http-request'
 export { default as graphqlLambdaRequest } from './lambda-request'
 export { default as createLambdaLink } from './lambda-link'
+export { default as createHeaderLink } from './header-link'
 export { default as hmacfetch } from './aws4-signer'
 export { default as formatError } from './format-error'
diff --git a/src/lambda-request.js b/src/lambda-request.js
@@ -1,11 +1,15 @@
+import { ApolloLink } from 'apollo-link'
 import { graphqlRequest } from './core/graphql-request'
+import createHeaderLink from './header-link'
 import createLambdaLink from './lambda-link'
 
 export default ({ lambdaName, region, request, variables, headers = {} }) => {
-  const link = createLambdaLink({
-    lambdaName,
-    region,
-    headers
-  })
+  const link = ApolloLink.from([
+    createHeaderLink(),
+    createLambdaLink({
+      lambdaName,
+      region
+    })
+  ])
   return graphqlRequest({ name: lambdaName, link, request, variables })
 }
diff --git a/yarn.lock b/yarn.lock
@@ -1070,6 +1070,12 @@ apollo-client@^2.4.7:
   optionalDependencies:
     "@types/async" "2.0.50"
 
+apollo-link-context@^1.0.10:
+  version "1.0.10"
+  resolved "https://registry.yarnpkg.com/apollo-link-context/-/apollo-link-context-1.0.10.tgz#c20b0b32c6e4e08c0d3967174f9c2a897c8431fe"
+  dependencies:
+    apollo-link "^1.2.4"
+
 apollo-link-dedup@^1.0.0:
   version "1.0.11"
   resolved "https://registry.yarnpkg.com/apollo-link-dedup/-/apollo-link-dedup-1.0.11.tgz#6f34ea748d2834850329ad03111ef18445232b05"