stream-chat-react-native-core 4.4.0-beta.2 → 4.4.0-beta.3

package/lib/commonjs/contexts/messageInputContext/MessageInputContext.js

@@ -71,6 +71,8 @@ var MessageInputContext = _react["default"].createContext({});
 exports.MessageInputContext = MessageInputContext;
 
 var MessageInputProvider = function MessageInputProvider(_ref) {
+  var _getFileUploadConfig, _getFileUploadConfig2, _getImageUploadConfig, _getImageUploadConfig2;
+
   var children = _ref.children,
       value = _ref.value;
 
@@ -81,8 +83,38 @@ var MessageInputProvider = function MessageInputProvider(_ref) {
       setSelectedPicker = _useAttachmentPickerC.setSelectedPicker;
 
   var _useChatContext = (0, _ChatContext.useChatContext)(),
+      appSettings = _useChatContext.appSettings,
       client = _useChatContext.client;
 
+  var getFileUploadConfig = function getFileUploadConfig() {
+    var _appSettings$app;
+
+    var fileConfig = appSettings == null ? void 0 : (_appSettings$app = appSettings.app) == null ? void 0 : _appSettings$app.file_upload_config;
+
+    if (fileConfig !== null || fileConfig !== undefined) {
+      return fileConfig;
+    } else {
+      return {};
+    }
+  };
+
+  var blockedFileExtensionTypes = (_getFileUploadConfig = getFileUploadConfig()) == null ? void 0 : _getFileUploadConfig.blocked_file_extensions;
+  var blockedFileMimeTypes = (_getFileUploadConfig2 = getFileUploadConfig()) == null ? void 0 : _getFileUploadConfig2.blocked_mime_types;
+
+  var getImageUploadConfig = function getImageUploadConfig() {
+    var _appSettings$app2;
+
+    var imageConfig = appSettings == null ? void 0 : (_appSettings$app2 = appSettings.app) == null ? void 0 : _appSettings$app2.image_upload_config;
+
+    if (imageConfig !== null || imageConfig !== undefined) {
+      return imageConfig;
+    }
+
+    return {};
+  };
+
+  var blockedImageExtensionTypes = (_getImageUploadConfig = getImageUploadConfig()) == null ? void 0 : _getImageUploadConfig.blocked_file_extensions;
+  var blockedImageMimeTypes = (_getImageUploadConfig2 = getImageUploadConfig()) == null ? void 0 : _getImageUploadConfig2.blocked_mime_types;
   var channelCapabities = (0, _OwnCapabilitiesContext.useOwnCapabilitiesContext)();
 
   var _useChannelContext = (0, _ChannelContext.useChannelContext)(),
@@ -649,9 +681,57 @@ var MessageInputProvider = function MessageInputProvider(_ref) {
     };
   }();
 
+  var regExcondition = /File (extension \.\w{2,4}|type \S+) is not supported/;
+
+  var getUploadSetStateAction = function getUploadSetStateAction(id, fileState) {
+    var extraData = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : {};
+
+    var uploads = function uploads(prevUploads) {
+      return prevUploads.map(function (prevUpload) {
+        if (prevUpload.id === id) {
+          return (0, _extends2["default"])({}, prevUpload, extraData, {
+            state: fileState
+          });
+        }
+
+        return prevUpload;
+      });
+    };
+
+    return uploads;
+  };
+
+  var handleFileOrImageUploadError = function handleFileOrImageUploadError(error, isImageError, id) {
+    if (isImageError) {
+      setNumberOfUploads(function (prevNumberOfUploads) {
+        return prevNumberOfUploads - 1;
+      });
+
+      if (error instanceof Error) {
+        if (regExcondition.test(error.message)) {
+          return setImageUploads(getUploadSetStateAction(id, _utils.FileState.NOT_SUPPORTED));
+        }
+
+        return setImageUploads(getUploadSetStateAction(id, _utils.FileState.UPLOAD_FAILED));
+      }
+    } else {
+      setNumberOfUploads(function (prevNumberOfUploads) {
+        return prevNumberOfUploads - 1;
+      });
+
+      if (error instanceof Error) {
+        if (regExcondition.test(error.message)) {
+          return setFileUploads(getUploadSetStateAction(id, _utils.FileState.NOT_SUPPORTED));
+        }
+
+        return setFileUploads(getUploadSetStateAction(id, _utils.FileState.UPLOAD_FAILED));
+      }
+    }
+  };
+
   var uploadFile = function () {
     var _ref6 = (0, _asyncToGenerator2["default"])(_regenerator["default"].mark(function _callee4(_ref5) {
-      var newFile, file, id, response;
+      var newFile, file, id, response, extraData;
       return _regenerator["default"].wrap(function _callee4$(_context4) {
         while (1) {
           switch (_context4.prev = _context4.next) {
@@ -667,17 +747,7 @@ var MessageInputProvider = function MessageInputProvider(_ref) {
 
             case 3:
               file = newFile.file, id = newFile.id;
-              setFileUploads(function (prevFileUploads) {
-                return prevFileUploads.map(function (fileUpload) {
-                  if (fileUpload.id === id) {
-                    return (0, _extends2["default"])({}, fileUpload, {
-                      state: _utils.FileState.UPLOADING
-                    });
-                  }
-
-                  return fileUpload;
-                });
-              });
+              setFileUploads(getUploadSetStateAction(id, _utils.FileState.UPLOADING));
               response = {};
               _context4.prev = 6;
 
@@ -707,57 +777,24 @@ var MessageInputProvider = function MessageInputProvider(_ref) {
               response = _context4.sent;
 
             case 17:
+              extraData = {
+                url: response.file
+              };
+              setFileUploads(getUploadSetStateAction(id, _utils.FileState.UPLOADED, extraData));
               _context4.next = 24;
               break;
 
-            case 19:
-              _context4.prev = 19;
+            case 21:
+              _context4.prev = 21;
               _context4.t0 = _context4["catch"](6);
-              console.warn(_context4.t0);
-
-              if (!newFile) {
-                setNumberOfUploads(function (prevNumberOfUploads) {
-                  return prevNumberOfUploads - 1;
-                });
-              } else {
-                setFileUploads(function (prevFileUploads) {
-                  return prevFileUploads.map(function (fileUpload) {
-                    if (fileUpload.id === id) {
-                      return (0, _extends2["default"])({}, fileUpload, {
-                        state: _utils.FileState.UPLOAD_FAILED
-                      });
-                    }
-
-                    return fileUpload;
-                  });
-                });
-                setNumberOfUploads(function (prevNumberOfUploads) {
-                  return prevNumberOfUploads - 1;
-                });
-              }
-
-              return _context4.abrupt("return");
+              handleFileOrImageUploadError(_context4.t0, false, id);
 
             case 24:
-              setFileUploads(function (prevFileUploads) {
-                return prevFileUploads.map(function (fileUpload) {
-                  if (fileUpload.id === id) {
-                    return (0, _extends2["default"])({}, fileUpload, {
-                      state: _utils.FileState.UPLOADED,
-                      url: response.file
-                    });
-                  }
-
-                  return fileUpload;
-                });
-              });
-
-            case 25:
             case "end":
               return _context4.stop();
           }
         }
-      }, _callee4, null, [[6, 19]]);
+      }, _callee4, null, [[6, 21]]);
     }));
 
     return function uploadFile(_x) {
@@ -870,18 +907,9 @@ var MessageInputProvider = function MessageInputProvider(_ref) {
                     return prevAsyncUploads;
                   });
                 } else {
-                  setImageUploads(function (prevImageUploads) {
-                    return prevImageUploads.map(function (imageUpload) {
-                      if (imageUpload.id === id) {
-                        return (0, _extends2["default"])({}, imageUpload, {
-                          state: _utils.FileState.UPLOADED,
-                          url: res.file
-                        });
-                      }
-
-                      return imageUpload;
-                    });
-                  });
+                  setImageUploads(getUploadSetStateAction(id, _utils.FileState.UPLOADED, {
+                    url: res.file
+                  }));
                 }
               });
               _context5.next = 41;
@@ -896,50 +924,22 @@ var MessageInputProvider = function MessageInputProvider(_ref) {
 
             case 41:
               if (Object.keys(response).length) {
-                setImageUploads(function (prevImageUploads) {
-                  return prevImageUploads.map(function (imageUpload) {
-                    if (imageUpload.id === id) {
-                      return (0, _extends2["default"])({}, imageUpload, {
-                        height: file.height,
-                        state: _utils.FileState.UPLOADED,
-                        url: response.file,
-                        width: file.width
-                      });
-                    }
-
-                    return imageUpload;
-                  });
-                });
+                setImageUploads(getUploadSetStateAction(id, _utils.FileState.UPLOADED, {
+                  height: file.height,
+                  url: response.file,
+                  width: file.width
+                }));
               }
 
-              _context5.next = 50;
+              _context5.next = 47;
               break;
 
             case 44:
               _context5.prev = 44;
               _context5.t2 = _context5["catch"](5);
-              console.warn(_context5.t2);
-
-              if (newImage) {
-                setImageUploads(function (prevImageUploads) {
-                  return prevImageUploads.map(function (imageUpload) {
-                    if (imageUpload.id === id) {
-                      return (0, _extends2["default"])({}, imageUpload, {
-                        state: _utils.FileState.UPLOAD_FAILED
-                      });
-                    }
-
-                    return imageUpload;
-                  });
-                });
-              }
+              handleFileOrImageUploadError(_context5.t2, true, id);
 
-              setNumberOfUploads(function (prevNumberOfUploads) {
-                return prevNumberOfUploads - 1;
-              });
-              return _context5.abrupt("return");
-
-            case 50:
+            case 47:
             case "end":
               return _context5.stop();
           }
@@ -954,33 +954,46 @@ var MessageInputProvider = function MessageInputProvider(_ref) {
 
   var uploadNewFile = function () {
     var _ref10 = (0, _asyncToGenerator2["default"])(_regenerator["default"].mark(function _callee6(file) {
-      var id, mimeType, newFile;
+      var id, mimeType, isBlockedFileExtension, isBlockedFileMimeType, fileState, newFile;
       return _regenerator["default"].wrap(function _callee6$(_context6) {
         while (1) {
           switch (_context6.prev = _context6.next) {
             case 0:
               id = (0, _utils.generateRandomId)();
               mimeType = (0, _mimeTypes.lookup)(file.name);
+              isBlockedFileExtension = blockedFileExtensionTypes == null ? void 0 : blockedFileExtensionTypes.some(function (fileExtensionType) {
+                var _file$name;
+
+                return (_file$name = file.name) == null ? void 0 : _file$name.includes(fileExtensionType);
+              });
+              isBlockedFileMimeType = blockedFileMimeTypes == null ? void 0 : blockedFileMimeTypes.some(function (mimeType) {
+                var _file$name2;
+
+                return (_file$name2 = file.name) == null ? void 0 : _file$name2.includes(mimeType);
+              });
+              fileState = isBlockedFileExtension || isBlockedFileMimeType ? _utils.FileState.NOT_SUPPORTED : _utils.FileState.UPLOADING;
               newFile = {
                 file: (0, _extends2["default"])({}, file, {
                   type: mimeType || (file == null ? void 0 : file.type)
                 }),
                 id: id,
-                state: _utils.FileState.UPLOADING
+                state: fileState
               };
-              _context6.next = 5;
+              _context6.next = 8;
               return Promise.all([setFileUploads(function (prevFileUploads) {
                 return prevFileUploads.concat([newFile]);
               }), setNumberOfUploads(function (prevNumberOfUploads) {
                 return prevNumberOfUploads + 1;
               })]);
 
-            case 5:
-              uploadFile({
-                newFile: newFile
-              });
+            case 8:
+              if (!isBlockedFileExtension) {
+                uploadFile({
+                  newFile: newFile
+                });
+              }
 
-            case 6:
+            case 9:
             case "end":
               return _context6.stop();
           }
@@ -995,30 +1008,43 @@ var MessageInputProvider = function MessageInputProvider(_ref) {
 
   var uploadNewImage = function () {
     var _ref11 = (0, _asyncToGenerator2["default"])(_regenerator["default"].mark(function _callee7(image) {
-      var id, newImage;
+      var id, isBlockedImageMimeType, isBlockedImageExtension, imageState, newImage;
       return _regenerator["default"].wrap(function _callee7$(_context7) {
         while (1) {
           switch (_context7.prev = _context7.next) {
             case 0:
               id = (0, _utils.generateRandomId)();
+              isBlockedImageMimeType = blockedImageMimeTypes == null ? void 0 : blockedImageMimeTypes.some(function (mimeType) {
+                var _image$uri;
+
+                return (_image$uri = image.uri) == null ? void 0 : _image$uri.includes(mimeType);
+              });
+              isBlockedImageExtension = blockedImageExtensionTypes == null ? void 0 : blockedImageExtensionTypes.some(function (imageExtensionType) {
+                var _image$uri2;
+
+                return (_image$uri2 = image.uri) == null ? void 0 : _image$uri2.includes(imageExtensionType);
+              });
+              imageState = isBlockedImageExtension || isBlockedImageMimeType ? _utils.FileState.NOT_SUPPORTED : _utils.FileState.UPLOADING;
               newImage = {
                 file: image,
                 id: id,
-                state: _utils.FileState.UPLOADING
+                state: imageState
               };
-              _context7.next = 4;
+              _context7.next = 7;
               return Promise.all([setImageUploads(function (prevImageUploads) {
                 return prevImageUploads.concat([newImage]);
               }), setNumberOfUploads(function (prevNumberOfUploads) {
                 return prevNumberOfUploads + 1;
               })]);
 
-            case 4:
-              uploadImage({
-                newImage: newImage
-              });
+            case 7:
+              if (!isBlockedImageExtension) {
+                uploadImage({
+                  newImage: newImage
+                });
+              }
 
-            case 5:
+            case 8:
             case "end":
               return _context7.stop();
           }
@@ -1087,7 +1113,7 @@ var MessageInputProvider = function MessageInputProvider(_ref) {
     __self: _this,
     __source: {
       fileName: _jsxFileName,
-      lineNumber: 1086,
+      lineNumber: 1117,
       columnNumber: 5
     }
   }, children);
@@ -1108,7 +1134,7 @@ var withMessageInputContext = function withMessageInputContext(Component) {
       __self: _this,
       __source: {
         fileName: _jsxFileName,
-        lineNumber: 1114,
+        lineNumber: 1145,
         columnNumber: 12
       }
     }));