Retry if gallery image query for i.imgur.com fails.

koffeinsource · Mar 31, 2016 · 10584eb · 10584eb
1 parent b811d7e
commit 10584eb
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/fromURL.go b/fromURL.go
@@ -30,8 +30,14 @@ func (client *Client) GetInfoFromURL(url string) (*GenericInfo, int, error) {
 		}
 		id := url[start:end]
 		client.Log.Debugf("Detected imgur image ID %v. Was going down the i.imgur.com/ path.", id)
-		ii, status, err := client.GetGalleryImageInfo(id)
-		ret.GImage = ii
+		gii, status, err := client.GetGalleryImageInfo(id)
+		if status < 400 {
+			ret.GImage = gii
+		} else {
+			var ii *ImageInfo
+			ii, status, err = client.GetImageInfo(id)
+			ret.Image = ii
+		}
 		return &ret, status, err
 	}