separate options & results by mediatype

Author: crockalet

src/index.d.ts

@@ -13,21 +13,30 @@ export type Results = {
   width: number;
   height: number;
   mime: string;
-  type: string;
   size: number;
   bucketId?: number;
   realPath?: string;
   parentFolderName?: string;
-  thumbnail?: string;
   creationDate?: string;
 };
 
+export interface VideoResults extends Results {
+  type: MediaType.VIDEO;
+  thumbnail?: string;
+}
+
+export interface ImageResults extends Results {
+  type: MediaType.IMAGE;
+  thumbnail?: undefined;
+}
+
 export type PickerErrorCode =
   | 'PICKER_CANCELLED'
   | 'NO_LIBRARY_PERMISSION'
   | 'NO_CAMERA_PERMISSION';
 
-export type Options = {
+export type Options<T extends MediaType = MediaType.ALL> = {
+  mediaType?: T;
   isPreview?: boolean;
   selectedColor?: string;
   tapHereToChange?: string;
@@ -49,7 +58,6 @@ export type Options = {
   autoPlay?: boolean;
   muteAudio?: boolean;
   preventAutomaticLimitedAccessAlert?: boolean; // newest iOS 14
-  mediaType?: MediaType;
   numberOfColumn?: number;
   maxSelectedAssets?: number;
   fetchOption?: Object;
@@ -63,19 +71,34 @@ export type Options = {
   haveThumbnail?: boolean;
 };
 
-export interface SinglePickerOptions extends Options {
+export interface SinglePickerOptions {
   selectedAssets?: Results;
   singleSelectedMode: true;
 }
 
-export interface MultiPickerOptions extends Options {
+export interface MultiPickerOptions {
   selectedAssets?: Results[];
   singleSelectedMode?: false;
 }
 
+interface MediaTypeOptions {
+  [MediaType.VIDEO]: { isExportThumbnail?: boolean };
+  [MediaType.ALL]: MediaTypeOptions[MediaType.VIDEO];
+}
+
+interface MediaTypeResults {
+  [MediaType.IMAGE]: ImageResults;
+  [MediaType.VIDEO]: VideoResults;
+  [MediaType.ALL]: ImageResults | VideoResults;
+}
+
 type MultipleImagePickerType = {
-  openPicker(options: MultiPickerOptions): Promise<Results[]>;
-  openPicker(options: SinglePickerOptions): Promise<Results>;
+  openPicker<T extends MediaType = MediaType.ALL>(
+    options: MultiPickerOptions & MediaTypeOptions[T] & Options<T>
+  ): Promise<MediaTypeResults[T][]>;
+  openPicker<T extends MediaType = MediaType.ALL>(
+    options: SinglePickerOptions & MediaTypeOptions[T] & Options<T>
+  ): Promise<MediaTypeResults[T]>;
 };
 
 const { MultipleImagePicker } = NativeModules;