Added S3 PutObjectRequest config option

Author: shukob

packages/aws-appsync/src/client.ts

@@ -35,6 +35,7 @@ import { createRetryLink } from './link/retry-link';
 import { boundEnqueueDeltaSync, buildSync, DELTASYNC_KEY, hashForOptions } from "./deltaSync";
 import { Subscription } from 'apollo-client/util/Observable';
 import { CONTROL_EVENTS_KEY } from './link/subscription-handshake-link';
+import { S3Config } from "./link/complex-object-link";
 
 export { defaultDataIdFromObject };
 
@@ -77,18 +78,20 @@ export const createAppSyncLink = ({
     complexObjectsCredentials,
     resultsFetcherLink = createHttpLink({ uri: url }),
     conflictResolver,
+    s3Config
 }: {
         url: string,
         region: string,
         auth: AuthOptions,
         complexObjectsCredentials: CredentialsGetter,
         resultsFetcherLink?: ApolloLink,
         conflictResolver?: ConflictResolver,
+        s3Config?: S3Config,
     }) => {
     const link = ApolloLink.from([
         createLinkWithStore((store) => new OfflineLink(store)),
         new ConflictResolutionLink(conflictResolver),
-        new ComplexObjectLink(complexObjectsCredentials),
+        new ComplexObjectLink(complexObjectsCredentials, s3Config),
         createRetryLink(ApolloLink.from([
             createAuthLink({ url, region, auth }),
             createSubscriptionHandshakeLink(url, resultsFetcherLink)
@@ -135,6 +138,7 @@ export interface AWSAppSyncClientOptions {
     cacheOptions?: ApolloReducerConfig,
     disableOffline?: boolean,
     offlineConfig?: OfflineConfig,
+    s3Config?: S3Config,
 }
 
 export interface OfflineConfig {
@@ -181,6 +185,7 @@ class AWSAppSyncClient<TCacheShape extends NormalizedCacheObject> extends Apollo
             callback = () => { },
             storeCacheRootMutation = false,
         } = {},
+        s3Config = {}
     }: AWSAppSyncClientOptions, options?: Partial<ApolloClientOptions<TCacheShape>>) {
         const { cache: customCache = undefined, link: customLink = undefined } = options || {};
 
@@ -218,7 +223,7 @@ class AWSAppSyncClient<TCacheShape extends NormalizedCacheObject> extends Apollo
                 };
             });
         });
-        const link = waitForRehydrationLink.concat(customLink || createAppSyncLink({ url, region, auth, complexObjectsCredentials, conflictResolver }));
+        const link = waitForRehydrationLink.concat(customLink || createAppSyncLink({ url, region, auth, complexObjectsCredentials, conflictResolver , s3Config}));
 
         const newOptions = {
             ...options,

packages/aws-appsync/src/link/complex-object-link-uploader.native.ts

@@ -8,7 +8,7 @@
  */
 import * as S3 from 'aws-sdk/clients/s3';
 
-export default (fileField, { credentials }) => {
+export default (fileField, { credentials, s3Config }) => {
     const {
         bucket: Bucket,
         key: Key,
@@ -22,10 +22,16 @@ export default (fileField, { credentials }) => {
         region,
     });
 
-    return s3.upload({
+    let params: S3.PutObjectRequest = {
         Bucket,
         Key,
         Body,
         ContentType,
-    }).promise();
+    };
+
+    if (s3Config.modifyPutObjectRequest != null) {
+        params = s3Config.modifyPutObjectRequest(params);
+    }
+
+    return s3.upload(params).promise();
 };

packages/aws-appsync/src/link/complex-object-link-uploader.ts

@@ -8,7 +8,7 @@
  */
 import * as S3 from 'aws-sdk/clients/s3';
 
-export default (fileField, { credentials }) => {
+export default (fileField, { credentials, s3Config }) => {
     const {
         bucket: Bucket,
         key: Key,
@@ -22,10 +22,16 @@ export default (fileField, { credentials }) => {
         region,
     });
 
-    return s3.upload({
+    let params: S3.PutObjectRequest = {
         Bucket,
         Key,
         Body,
         ContentType,
-    }).promise();
+    };
+
+    if (s3Config.modifyPutObjectRequest != null) {
+        params = s3Config.modifyPutObjectRequest(params);
+    }
+
+    return s3.upload(params).promise();
 };

packages/aws-appsync/src/link/complex-object-link.ts

@@ -14,23 +14,30 @@ import { ExecutionResult, GraphQLError } from 'graphql';
 
 import upload from "./complex-object-link-uploader";
 import { AWSAppsyncGraphQLError } from '../types';
+import { S3 } from "aws-sdk";
+
+export interface S3Config{
+    modifyPutObjectRequest?: (request: S3.PutObjectRequest) => S3.PutObjectRequest
+}
 
 export class ComplexObjectLink extends ApolloLink {
 
     private link: ApolloLink;
 
-    constructor(credentials) {
+    constructor(credentials, s3Config = null) {
         super();
 
-        this.link = complexObjectLink(credentials);
+        this.link = complexObjectLink(credentials, s3Config);
     }
 
     request(operation, forward) {
         return this.link.request(operation, forward);
     }
 }
 
-export const complexObjectLink = (credentials) => {
+
+
+export const complexObjectLink = (credentials, s3Config = null) => {
     return new ApolloLink((operation, forward) => {
         return new Observable(observer => {
             let handle;
@@ -46,7 +53,7 @@ export const complexObjectLink = (credentials) => {
 
                 uploadPromise = Promise.resolve(uploadCredentials)
                     .then(credentials => {
-                        const uploadPromises = Object.entries(objectsToUpload).map(([_, fileField]) => upload(fileField, { credentials }));
+                        const uploadPromises = Object.entries(objectsToUpload).map(([_, fileField]) => upload(fileField, { credentials, s3Config }));
 
                         return Promise.all([operation, ...uploadPromises] as Promise<any>[]);
                     })