Optionally Return Full Response from Direct Uploads Endpoint

src/DirectUploadProvider.js

@@ -33,12 +33,13 @@ export type DelegatedProps = {|
     xhr: XMLHttpRequest,
   }) => mixed,
   render: RenderProps => React.Node,
+  fullAttributes?: boolean,
 |}
 
 type Props = {
   ...DelegatedProps,
   origin: Origin,
-  onSuccess: (string[]) => mixed,
+  onSuccess: (any[]) => mixed,
   headers?: CustomHeaders,
 }
 
@@ -87,11 +88,11 @@ class DirectUploadProvider extends React.Component<Props, State> {
 
     this.setState({ uploading: true })
 
-    const signedIds = await Promise.all(
+    const resultArr = await Promise.all(
       this.uploads.map(upload => upload.start())
     )
 
-    this.props.onSuccess(signedIds)
+    this.props.onSuccess(resultArr)
     this.uploads = []
     this.setState({ fileUploads: {}, uploading: false })
   }
@@ -110,6 +111,7 @@ class DirectUploadProvider extends React.Component<Props, State> {
       onBeforeBlobRequest,
       onBeforeStorageRequest,
       origin,
+      fullAttributes,
     } = this.props
 
     return new Upload(file, {
@@ -119,6 +121,7 @@ class DirectUploadProvider extends React.Component<Props, State> {
       onBeforeStorageRequest,
       onChangeFile: this.handleChangeFileUpload,
       origin,
+      fullAttributes,
     })
   }
 }

src/Upload.js

@@ -25,6 +25,7 @@ type Options = {|
   }) => mixed,
   onChangeFile: ({ [string]: ActiveStorageFileUpload }) => mixed,
   headers?: CustomHeaders,
+  fullAttributes?: boolean,
 |}
 
 class Upload {
@@ -85,13 +86,13 @@ class Upload {
     })
   }
 
-  handleSuccess = (signedId: string) => {
+  handleSuccess = (result: string | Object) => {
     this.handleChangeFile({
       state: 'finished',
       id: this.id,
       file: this.directUpload.file,
     })
-    return signedId
+    return result
   }
 
   handleError = (error: string) => {
@@ -104,11 +105,14 @@ class Upload {
     throw error
   }
 
-  start(): Promise<string> {
+  start(): Promise<string> | Promise<Object> {
     const promise = new Promise((resolve, reject) => {
       this.directUpload.create((error, attributes) => {
         if (error) reject(error)
-        else resolve(attributes.signed_id)
+        else {
+          let result = this.options.fullAttributes ? attributes : attributes.signed_id
+          resolve(result)
+        }
       })
     })
 

src/Upload.test.js

@@ -10,7 +10,7 @@ jest.mock('activestorage', () => {
       id: 'id',
       file,
       create(cb) {
-        cb(null, { signed_id: 'signedId' })
+        cb(null, { signed_id: 'signedId', key: 'key' })
       },
     })),
   }
@@ -79,6 +79,11 @@ describe('Upload', () => {
       expect(await upload.start()).toEqual('signedId')
     })
 
+    it('resolves with fullAttributes from the direct upload when option "fullAttributes" is true', async () => {
+      const upload = new Upload(file, { ...options, fullAttributes: true })
+      expect(await upload.start()).toEqual({ signed_id: 'signedId', key: 'key' })
+    })
+
     it('reports that the upload is finished if it does so', async () => {
       const upload = new Upload(file, options)
       await upload.start()