Fix URL

menthorlabs · Nov 12, 2023 · aae1646 · aae1646
1 parent 85e376d
commit aae1646
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 6 deletions.
diff --git a/src/content.js b/src/content.js
@@ -62,7 +62,10 @@ const getSignedUrlPromise = async (ContentType) => {
     const signedUrl = await getSignedUrl(s3Client, command, {
       expiresIn: 3600,
     });
-    return signedUrl;
+    return {
+      url: signedUrl,
+      fileName,
+    };
   } catch (error) {
     console.error('Error uploading file:', error);
     throw new Error('Failed to upload file');

diff --git a/src/submission.js b/src/submission.js
@@ -56,7 +56,10 @@ const getSignedUrlPromise = async (ContentType) => {
     const signedUrl = await getSignedUrl(s3Client, command, {
       expiresIn: 3600,
     });
-    return signedUrl;
+    return {
+      url: signedUrl,
+      fileName,
+    };
   } catch (error) {
     console.error('Error uploading file:', error);
     throw new Error('Failed to upload file');
@@ -402,10 +405,10 @@ module.exports.requestSubmissionUrl = async (event) => {
       }
     }
 
-    const url = await getSignedUrlPromise(ContentType);
+    const signedUrl = await getSignedUrlPromise(ContentType);
 
     connection.query('UPDATE Submission SET Filename = ? WHERE Id = ?', [
-      url.fileName,
+      signedUrl.fileName,
       submissionId,
       userEmail,
     ]);
@@ -418,8 +421,8 @@ module.exports.requestSubmissionUrl = async (event) => {
       statusCode: 200,
       body: JSON.stringify({
         message: 'Submission url generated successfully',
-        url,
-        fileName: url.fileName,
+        url: signedUrl.url,
+        fileName: signedUrl.fileName,
       }),
     };
   } catch (error) {