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.
This commit is contained in:
BlackDex 2019-02-04 17:27:40 +01:00
parent 574b040142
commit 9026cc8d42
1 changed files with 4 additions and 5 deletions

View File

@ -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() {