From 55191f143a1d1f7aa4e6724d415b674c0863bf4c Mon Sep 17 00:00:00 2001
From: Cyberarm <matthewlikesrobots@gmail.com>
Date: Sun, 26 Dec 2021 21:32:54 -0600
Subject: [PATCH] Added option to force fetch to Cache.fetch, force user
 profile picture to be fetched on start up or on login

---
 lib/cache.rb       | 8 ++++----
 lib/pages/login.rb | 2 +-
 lib/states/boot.rb | 1 +
 3 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/lib/cache.rb b/lib/cache.rb
index df72cf0..13dcdd1 100644
--- a/lib/cache.rb
+++ b/lib/cache.rb
@@ -7,18 +7,18 @@ class W3DHub
     end
 
     # Fetch a generic uri
-    def self.fetch(internet, uri)
+    def self.fetch(internet, uri, force_fetch = false)
       path = path(uri)
 
-      if File.exist?(path)
+      if !force_fetch && File.exist?(path)
         path
       else
-        response = internet.get(uri, [["user-agent", W3DHub::Api::USER_AGENT]])
+        response = internet.get(uri, W3DHub::Api::DEFAULT_HEADERS)
 
         if response.success?
           response.save(path, "wb")
 
-          path
+          return path
         end
 
         false
diff --git a/lib/pages/login.rb b/lib/pages/login.rb
index 4c325a0..f07a261 100644
--- a/lib/pages/login.rb
+++ b/lib/pages/login.rb
@@ -46,7 +46,7 @@ class W3DHub
                       Store.settings[:account][:refresh_token] = account.refresh_token
                       Store.settings.save_settings
 
-                      Cache.fetch(internet, account.avatar_uri)
+                      Cache.fetch(internet, account.avatar_uri, true)
 
                       populate_account_info
                       page(W3DHub::Pages::Games)
diff --git a/lib/states/boot.rb b/lib/states/boot.rb
index d9c80b9..d8f0607 100644
--- a/lib/states/boot.rb
+++ b/lib/states/boot.rb
@@ -71,6 +71,7 @@ class W3DHub
 
           if @account
             Store.settings[:account][:refresh_token] = @account.refresh_token
+            Cache.fetch(internet, @account.avatar_uri, true)
           else
             Store.settings[:account][:refresh_token] = nil
           end