@superblocksteam/shared 0.9611.0 → 2.0.0-SNAPSHOT.12b3e4ccc970c4bd

package/src/utils/attachment-upload.test.ts

@@ -5,6 +5,7 @@ import {
   buildAttachmentUploadMetadata,
   buildAttachmentUploadFormData,
   createUploadedAttachmentFromServer,
+  getAttachmentMimeTypeFromFileName,
   getTextAttachmentType,
   parseUserFacingUploadError,
   isOfficeXml,
@@ -12,6 +13,27 @@ import {
   toAttachmentUploadDescriptor
 } from './attachment-upload.js';
 
+describe('getAttachmentMimeTypeFromFileName', () => {
+  test.each([
+    ['logo.png', 'image/png'],
+    ['logo.PNG', 'image/png'],
+    ['photo.jpg', 'image/jpeg'],
+    ['photo.jpeg', 'image/jpeg'],
+    ['animation.gif', 'image/gif'],
+    ['shot.webp', 'image/webp'],
+    ['report.pdf', 'application/pdf'],
+    ['server.log', 'text/plain'],
+    ['rows.csv', 'text/csv'],
+    ['config.json', 'application/json']
+  ])('reads %s as %s', (fileName, expected) => {
+    expect(getAttachmentMimeTypeFromFileName(fileName)).toBe(expected);
+  });
+
+  test.each(['app.exe', 'archive.dmg', 'noextension', ''])('has no type for %s', (fileName) => {
+    expect(getAttachmentMimeTypeFromFileName(fileName)).toBeUndefined();
+  });
+});
+
 describe('toAttachmentUploadDescriptor', () => {
   test('preserves explicit names for non-File uploads', () => {
     const descriptor = toAttachmentUploadDescriptor({

package/src/utils/attachment-upload.ts

@@ -291,6 +291,34 @@ function getTextLikeMimeTypeFromFileName(fileName: string | undefined): string |
   return textAttachmentType ? TEXT_ATTACHMENT_MIME_TYPES[textAttachmentType] : undefined;
 }
 
+/**
+ * Names the media type for a file that arrives with nothing but a name, as it
+ * does when a caller points at a path instead of handing over the bytes.
+ *
+ * Derived from {@link MIME_TYPE_FILE_EXTENSIONS} rather than a general-purpose
+ * MIME database so the answer stays inside what attachments actually accept:
+ * a wider lookup would name a type for `.exe` that upload then refuses.
+ */
+export function getAttachmentMimeTypeFromFileName(fileName?: string): string | undefined {
+  const normalizedFileName = normalizeNonEmptyString(fileName)?.toLowerCase();
+  if (!normalizedFileName) {
+    return undefined;
+  }
+
+  for (const [mimeType, extension] of Object.entries(MIME_TYPE_FILE_EXTENSIONS)) {
+    // '.jpeg' resolves through the text-like pass below only if listed there,
+    // so both spellings are matched here against the canonical extension.
+    if (normalizedFileName.endsWith(extension)) {
+      return mimeType === 'image/jpg' ? 'image/jpeg' : mimeType;
+    }
+  }
+  if (normalizedFileName.endsWith('.jpeg')) {
+    return 'image/jpeg';
+  }
+
+  return getTextLikeMimeTypeFromFileName(normalizedFileName);
+}
+
 function isProviderSafeTextMimeType(mimeType: string): boolean {
   return mimeType.startsWith('text/') || mimeType === 'application/json';
 }