Use statvoo API to fetch favicons. (trustwallet#445)

* Use `statvoo` API to fetch favicons. * Use `guard` instead of coalescing operator for `host` * Add new `linkURL` to `Bookmark` * Return .none when host is empty
Baleen-Y · Mar 11, 2018 · 1738dfb · 1738dfb
1 parent c0d940b
commit 1738dfb
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 2 deletions.
diff --git a/Trust/Browser/Coordinators/BrowserCoordinator.swift b/Trust/Browser/Coordinators/BrowserCoordinator.swift
@@ -100,7 +100,7 @@ class BrowserCoordinator: Coordinator {
     }
 
     func openBookmark(bookmark: Bookmark) {
-        guard let url = URL(string: bookmark.url) else { return }
+        guard let url = bookmark.linkURL else { return }
         rootViewController.goTo(url: url)
     }
 

diff --git a/Trust/Browser/Types/Bookmark.swift b/Trust/Browser/Types/Bookmark.swift
@@ -18,6 +18,10 @@ class Bookmark: Object {
         self.id = incrementID()
     }
 
+    var linkURL: URL? {
+        return URL(string: url)
+    }
+
     override class func primaryKey() -> String? {
         return "id"
     }

diff --git a/Trust/Browser/ViewModel/BookmarkViewModel.swift b/Trust/Browser/ViewModel/BookmarkViewModel.swift
@@ -16,6 +16,9 @@ struct BookmarkViewModel {
         return bookmark.title
     }
     var imageURL: URL? {
-        return URL(string: "\(bookmark.url)favicon.ico")
+        guard let host = bookmark.linkURL?.host else {
+            return .none
+        }
+        return URL(string: "https://api.statvoo.com/favicon/?url=\(host)")
     }
 }