From 574b040142ab29d80baac134084c4a29ec0dab83 Mon Sep 17 00:00:00 2001
From: BlackDex <black.dex@gmail.com>
Date: Mon, 4 Feb 2019 16:59:52 +0100
Subject: [PATCH 1/2] Loop through the iconlist until an icon is found

Loop for a maximum of 5 times through the iconlist or until a
successful download of an icon.
---
 src/api/icons.rs | 35 +++++++++++++++++++++++------------
 1 file changed, 23 insertions(+), 12 deletions(-)

diff --git a/src/api/icons.rs b/src/api/icons.rs
index c7b3829..fcfbe7d 100644
--- a/src/api/icons.rs
+++ b/src/api/icons.rs
@@ -137,19 +137,19 @@ struct IconList {
     href: String,
 }
 
-/// Returns a Result with a String which holds the preferend favicon location.
-/// There will always be a result with a string which will contain https://example.com/favicon.ico
-/// This does not mean that that location does exists, but it is the default location.
+/// Returns a Result/Tuple which holds a Vector IconList and a string which holds the cookies from the last response.
+/// There will always be a result with a string which will contain https://example.com/favicon.ico and an empty string for the cookies.
+/// This does not mean that that location does exists, but it is the default location browser use.
 ///
 /// # Argument
 /// * `domain` - A string which holds the domain with extension.
 ///
 /// # Example
 /// ```
-/// favicon_location1 = get_icon_url("github.com");
-/// favicon_location2 = get_icon_url("gitlab.com");
+/// let (mut iconlist, cookie_str) = get_icon_url("github.com")?;
+/// let (mut iconlist, cookie_str) = get_icon_url("gitlab.com")?;
 /// ```
-fn get_icon_url(domain: &str) -> Result<(String, String), Error> {
+fn get_icon_url(domain: &str) -> Result<(Vec<IconList>, String), Error> {
     // Default URL with secure and insecure schemes
     let ssldomain = format!("https://{}", domain);
     let httpdomain = format!("http://{}", domain);
@@ -200,7 +200,7 @@ fn get_icon_url(domain: &str) -> Result<(String, String), Error> {
     iconlist.sort_by_key(|x| x.priority);
 
     // There always is an icon in the list, so no need to check if it exists, and just return the first one
-    Ok((iconlist.remove(0).href, cookie_str))
+    Ok((iconlist, cookie_str))
 }
 
 fn get_page(url: &str) -> Result<Response, Error> {
@@ -292,13 +292,24 @@ fn parse_sizes(sizes: &str) -> (u16, u16) {
 }
 
 fn download_icon(domain: &str) -> Result<Vec<u8>, Error> {
-    let (url, cookie_str) = get_icon_url(&domain)?;
-
-    info!("Downloading icon for {} via {}...", domain, url);
-    let mut res = get_page_with_cookies(&url, &cookie_str)?;
+    let (mut iconlist, cookie_str) = get_icon_url(&domain)?;
 
     let mut buffer = Vec::new();
-    res.copy_to(&mut buffer)?;
+
+    let mut attempts = 0;
+    while attempts < 5 {
+        let url = &iconlist.remove(0).href;
+        info!("Downloading icon for {} via {}...", domain, url);
+        match get_page_with_cookies(&url, &cookie_str) {
+            Ok(mut res) => {
+                info!("Download finished for {}", url);
+                res.copy_to(&mut buffer)?;
+                break;
+            },
+            Err(_) => info!("Download failed for {}", url),
+        };
+        attempts += 1;
+    }
 
     if buffer.is_empty() {
         err!("Empty response")

From 9026cc8d426107f140f029a1a9a73da53beb2e1a Mon Sep 17 00:00:00 2001
From: BlackDex <black.dex@gmail.com>
Date: Mon, 4 Feb 2019 17:27:40 +0100
Subject: [PATCH 2/2] Fixed issue when the iconlist is smaller then 5

When the iconlist was smaller then 5 items, it would cause a panic.
Solved by using .truncate() on the iconlist.
---
 src/api/icons.rs | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/src/api/icons.rs b/src/api/icons.rs
index fcfbe7d..cb3669c 100644
--- a/src/api/icons.rs
+++ b/src/api/icons.rs
@@ -265,7 +265,7 @@ fn get_icon_priority(href: &str, sizes: &str) -> u8 {
 ///
 /// # Arguments
 /// * `sizes` - The size of the icon if available as a <width>x<height> value like 32x32.
-/// 
+///
 /// # Example
 /// ```
 /// let (width, height) = parse_sizes("64x64"); // (64, 64)
@@ -296,9 +296,9 @@ fn download_icon(domain: &str) -> Result<Vec<u8>, Error> {
 
     let mut buffer = Vec::new();
 
-    let mut attempts = 0;
-    while attempts < 5 {
-        let url = &iconlist.remove(0).href;
+    iconlist.truncate(5);
+    for icon in iconlist {
+        let url = icon.href;
         info!("Downloading icon for {} via {}...", domain, url);
         match get_page_with_cookies(&url, &cookie_str) {
             Ok(mut res) => {
@@ -308,7 +308,6 @@ fn download_icon(domain: &str) -> Result<Vec<u8>, Error> {
             },
             Err(_) => info!("Download failed for {}", url),
         };
-        attempts += 1;
     }
 
     if buffer.is_empty() {