Fix issue where images wouldn't load if just posted

Macludde · Dec 16, 2021 · fcb879c · fcb879c
1 parent 1d89994
commit fcb879c
Showing 1 changed file with 15 additions and 3 deletions.
diff --git a/src/pages/Feed/Post.tsx b/src/pages/Feed/Post.tsx
@@ -54,9 +54,21 @@ const Post: React.FC<PostProps> = ({ post, hideComments, children }) => {
             post.type === 'file' ||
             post.type === undefined
         ) {
-            getDownloadURL(
-                ref(storage, `posts/${post.author}/${post.id}`)
-            ).then((url) => setContentURL(url))
+            const getURL = async (level: number) => {
+                if (level > 5) return
+                try {
+                    const url = await getDownloadURL(
+                        ref(storage, `posts/${post.author}/${post.id}`)
+                    )
+                    setContentURL(url)
+                } catch (error) {
+                    setContentURL(null)
+                    setTimeout(() => {
+                        getURL(level + 1)
+                    }, 200)
+                }
+            }
+            getURL(0)
         }
     }, [post.id, post.author, post.type])