withApollo.js

import Amplify, { Auth } from 'aws-amplify';
import { withAuthenticator } from 'aws-amplify-react/dist/Auth';
import AWSAppSyncClient from 'aws-appsync';
import { ApolloProvider } from 'react-apollo';
const client = new AWSAppSyncClient({
  url: 'https://xxxx.appsync-api.us-east-1.amazonaws.com/graphql',
  region: 'us-east-1',
  auth: {
    // AWS Cognito User Pool
    type: AUTH_TYPE.AMAZON_COGNITO_USER_POOLS,
    jwtToken: async () => (await Auth.currentSession()).getIdToken().getJwtToken(),
    // API KEY
    type: AUTH_TYPE.API_KEY,
    apiKey: 'xxxxxxxxxxxxx',
    // AWS IAM
    type: AUTH_TYPE.AWS_IAM
    credentials: () => Auth.currentCredentials(),
  },
});

const WithProvider = () => (
  <Router>
    <ApolloProvider client={client}>
  </Router>
);

export default withAuthenticator(WithProvider);