stream-chat-react-native 9.0.2-beta.2 → 9.1.0-beta.1

package/src/optionalDependencies/__tests__/pickDocument.test.ts

@@ -0,0 +1,86 @@
+describe('native pickDocument', () => {
+  afterEach(() => {
+    jest.resetModules();
+    jest.clearAllMocks();
+  });
+
+  it('adds a thumbnail for picked video files', async () => {
+    const generateThumbnails = jest.fn().mockResolvedValue({
+      'file:///video.mp4': { uri: 'file:///video-thumb.jpg' },
+    });
+
+    jest.doMock(
+      '@react-native-documents/picker',
+      () => ({
+        pick: jest.fn().mockResolvedValue([
+          {
+            name: 'video.mp4',
+            size: 42,
+            type: 'video/mp4',
+            uri: 'file:///video.mp4',
+          },
+        ]),
+        types: { allFiles: '*/*' },
+      }),
+      { virtual: true },
+    );
+    jest.doMock('../generateThumbnail', () => ({
+      generateThumbnails,
+    }));
+
+    const { pickDocument } = require('../pickDocument');
+
+    await expect(pickDocument({ maxNumberOfFiles: 2 })).resolves.toEqual({
+      assets: [
+        {
+          name: 'video.mp4',
+          size: 42,
+          thumb_url: 'file:///video-thumb.jpg',
+          type: 'video/mp4',
+          uri: 'file:///video.mp4',
+        },
+      ],
+      cancelled: false,
+    });
+    expect(generateThumbnails).toHaveBeenCalledWith(['file:///video.mp4']);
+  });
+
+  it('does not generate thumbnails for non-video files', async () => {
+    const generateThumbnails = jest.fn().mockResolvedValue({});
+
+    jest.doMock(
+      '@react-native-documents/picker',
+      () => ({
+        pick: jest.fn().mockResolvedValue([
+          {
+            name: 'doc.pdf',
+            size: 42,
+            type: 'application/pdf',
+            uri: 'file:///doc.pdf',
+          },
+        ]),
+        types: { allFiles: '*/*' },
+      }),
+      { virtual: true },
+    );
+    jest.doMock('../generateThumbnail', () => ({
+      generateThumbnails,
+    }));
+
+    const { pickDocument } = require('../pickDocument');
+
+    await expect(pickDocument({ maxNumberOfFiles: 2 })).resolves.toEqual({
+      assets: [
+        {
+          name: 'doc.pdf',
+          size: 42,
+          thumb_url: undefined,
+          type: 'application/pdf',
+          uri: 'file:///doc.pdf',
+        },
+      ],
+      cancelled: false,
+    });
+    expect(generateThumbnails).toHaveBeenCalledWith([]);
+  });
+});

package/src/optionalDependencies/pickDocument.ts

@@ -3,6 +3,8 @@
  *
  * For its full API, see https://github.com/react-native-documents/document-picker/blob/main/packages/document-picker/src/index.ts
  * */
+import { generateThumbnails } from './generateThumbnail';
+
 type ResponseValue = {
   name: string;
   size: number;
@@ -31,6 +33,20 @@ try {
 export const pickDocument = DocumentPicker
   ? async ({ maxNumberOfFiles }: { maxNumberOfFiles: number }) => {
       try {
+        const addVideoThumbnails = async <T extends { type?: string; uri?: string }>(
+          assets: T[],
+        ) => {
+          const videoUris = assets
+            .filter(({ type, uri }) => type?.startsWith('video/') && !!uri)
+            .map(({ uri }) => uri as string);
+          const thumbnailResults = await generateThumbnails(videoUris);
+
+          return assets.map((asset) => ({
+            ...asset,
+            thumb_url: asset.uri ? thumbnailResults[asset.uri]?.uri || undefined : undefined,
+          }));
+        };
+
         if (!DocumentPicker) return { cancelled: true };
         let res: ResponseValue[] = await DocumentPicker.pick({
           allowMultiSelection: true,
@@ -42,12 +58,14 @@ export const pickDocument = DocumentPicker
         }
 
         return {
-          assets: res.map(({ name, size, type, uri }) => ({
-            name,
-            size,
-            type,
-            uri,
-          })),
+          assets: await addVideoThumbnails(
+            res.map(({ name, size, type, uri }) => ({
+              name,
+              size,
+              type,
+              uri,
+            })),
+          ),
           cancelled: false,
         };
       } catch (err) {