#829 | Set mime type when uploading files to s3

avniproject · Dec 18, 2024 · 91d0ea5 · 91d0ea5
1 parent baaae62
commit 91d0ea5
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/avni-server-api/src/main/java/org/avni/server/service/StorageService.java b/avni-server-api/src/main/java/org/avni/server/service/StorageService.java
@@ -9,9 +9,9 @@
 import com.amazonaws.services.s3.transfer.TransferManagerBuilder;
 import org.apache.commons.io.FileUtils;
 import org.apache.commons.io.FilenameUtils;
-import org.avni.server.domain.S3ExtensionFile;
 import org.avni.server.domain.Organisation;
 import org.avni.server.domain.OrganisationConfig;
+import org.avni.server.domain.S3ExtensionFile;
 import org.avni.server.domain.UserContext;
 import org.avni.server.framework.security.UserContextHolder;
 import org.avni.server.service.media.MediaFolder;
@@ -348,7 +348,11 @@ public String putObject(String objectKey, File tempFile) {
             logger.info(format("[dev] Save file locally. '%s'", objectKey));
             return tempFile.getAbsolutePath();
         }
-        s3Client.putObject(new PutObjectRequest(bucketName, objectKey, tempFile));
+        String mimeType = URLConnection.guessContentTypeFromName(objectKey);
+        ObjectMetadata objectMetadata = new ObjectMetadata();
+        objectMetadata.setContentType(mimeType);
+        PutObjectRequest putObjectRequest = new PutObjectRequest(bucketName, objectKey, tempFile).withMetadata(objectMetadata);
+        s3Client.putObject(putObjectRequest);
         tempFile.delete();
         return objectKey;
     }