Merge pull request #66 from which-ecosystem/fix-googledrive

Patch 1.0.2

1 files changed, 6 insertions, 2 deletions

diff --git a/src/components/UploadImage/UploadImage.tsx b/src/components/UploadImage/UploadImage.tsx
index 5b2f38e..e9f2168 100644
--- a/src/components/UploadImage/UploadImage.tsx
+++ b/src/components/UploadImage/UploadImage.tsx
@@ -25,9 +25,13 @@ const UploadImage: React.FC<PropTypes> = ({ setIsOpen, isOpen, callback }) => {
   const handleSubmit = () => {
     let result = url;
     if (url.startsWith('https://www.instagram.com/')) {
-      const lastSlashIndex = url.lastIndexOf('/');
-      const baseUrl = url.slice(0, lastSlashIndex);
+      const appendixIndex = url.indexOf('/?utm_source=');
+      const baseUrl = url.slice(0, appendixIndex);
       result = `${baseUrl}/media/?size=l`;
+    } else if (url.startsWith('https://drive.google.com/')) {
+      const match = url.match('/d/(.*)/');
+      const fileId = match && match[1];
+      result = `https://drive.google.com/uc?export=view&id=${fileId}`;
     }
     callback(result || '');
     handleClose();