From 6ac685c4b846343ae181634c1b1dd730d712df80 Mon Sep 17 00:00:00 2001
From: John Maguire <john.maguire@gmail.com>
Date: Wed, 4 Jan 2012 12:39:22 +0000
Subject: [PATCH] Use correct icon for DI/SkyFM.

---
 src/internet/digitallyimportedurlhandler.cpp | 9 +++++++++
 src/internet/digitallyimportedurlhandler.h   | 2 +-
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/src/internet/digitallyimportedurlhandler.cpp b/src/internet/digitallyimportedurlhandler.cpp
index ff16e60e2..ce48d9780 100644
--- a/src/internet/digitallyimportedurlhandler.cpp
+++ b/src/internet/digitallyimportedurlhandler.cpp
@@ -33,6 +33,15 @@ QString DigitallyImportedUrlHandler::scheme() const {
   return service_->api_service_name();
 }
 
+QIcon DigitallyImportedUrlHandler::icon() const {
+  if (scheme() == "di") {
+    return QIcon(":providers/digitallyimported.png");
+  } else if (scheme() == "sky") {
+    return QIcon(":providers/skyfm.png");
+  }
+  return QIcon();
+}
+
 UrlHandler::LoadResult DigitallyImportedUrlHandler::StartLoading(const QUrl& url) {
   LoadResult ret(url);
   if (task_id_ != -1) {
diff --git a/src/internet/digitallyimportedurlhandler.h b/src/internet/digitallyimportedurlhandler.h
index 1183284a0..dcaa133ff 100644
--- a/src/internet/digitallyimportedurlhandler.h
+++ b/src/internet/digitallyimportedurlhandler.h
@@ -28,7 +28,7 @@ public:
   DigitallyImportedUrlHandler(DigitallyImportedServiceBase* service);
 
   QString scheme() const;
-  QIcon icon() const { return QIcon(":providers/digitallyimported.png"); }
+  QIcon icon() const;
   LoadResult StartLoading(const QUrl& url);
 
   void CancelTask();