Site specific secondary tiles for Windows 8.

chrome/browser/ui/metro_pin_tab_helper.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/ui/metro_pin_tab_helper.h"
 
 #include "base/logging.h"
 #include "base/utf_string_conversions.h"
+#include "chrome/browser/favicon/favicon_tab_helper.h"
+#include "chrome/browser/ui/tab_contents/tab_contents.h"
 #include "content/public/browser/web_contents.h"
+#include "third_party/skia/include/core/SkBitmap.h"
+#include "ui/gfx/image/image.h"
 
 #if defined(OS_WIN)

[sky, 2012/10/19 16:09:56]

Isn't this whole file windows specific?

[benwells, 2012/10/22 06:20:47]

It's only used on windows, yes. I've only put the minimal amount of code behind
#if's as it felt cleaner 

I can move more. E.g. the tab helper creation can go behind an #if in
tab_contents.cc, and this could become _win.

[sky, 2012/10/22 15:51:21]

Yes please. Additionally move this out of TabContents entirely. At this point
you should be using WebContentsUserData.

 #include "base/win/metro.h"
 #endif
 
 DEFINE_WEB_CONTENTS_USER_DATA_KEY(MetroPinTabHelper)
 
 MetroPinTabHelper::MetroPinTabHelper(content::WebContents* web_contents)
     : content::WebContentsObserver(web_contents),
       is_pinned_(false) {}
 
 MetroPinTabHelper::~MetroPinTabHelper() {}
 
 void MetroPinTabHelper::TogglePinnedToStartScreen() {
-#if defined(OS_WIN)
-  HMODULE metro_module = base::win::GetMetroModule();
-  if (metro_module) {
-    typedef void (*MetroTogglePinnedToStartScreen)(const string16&,
-        const string16&);
-    MetroTogglePinnedToStartScreen metro_toggle_pinned_to_start_screen =
-        reinterpret_cast<MetroTogglePinnedToStartScreen>(
-            ::GetProcAddress(metro_module, "MetroTogglePinnedToStartScreen"));
-    if (!metro_toggle_pinned_to_start_screen) {
-      NOTREACHED();
+  UpdatePinnedStateForCurrentURL();
+  bool was_pinned = is_pinned_;
+
+  // TODO(benwells): This will update the state incorrectly if the user
+  // cancels. To fix this some sort of callback needs to be introduced as
+  // the pinning happens on another thread.
+  is_pinned_ = !is_pinned_;
+
+  if (was_pinned) {
+    UnPinPageFromStartScreen();
+    return;
+  }
+
+  // TODO(benwells): Handle downloading a larger favicon if there is one.
+  GURL url = web_contents()->GetURL();
+  string16 url_str = UTF8ToUTF16(url.spec());
+  string16 title = web_contents()->GetTitle();
+  TabContents* tab_contents = TabContents::FromWebContents(web_contents());
+  DCHECK(tab_contents);
+  FaviconTabHelper* favicon_tab_helper = FaviconTabHelper::FromWebContents(
+      tab_contents->web_contents());
+  // TODO(benwells): Handle the case where there is a favicon for the site but
+  // it isn't downloaded yet.
+  if (favicon_tab_helper->FaviconIsValid()) {
+    gfx::Image favicon = favicon_tab_helper->GetFavicon();
+    if (!favicon.IsEmpty()) {
+      const SkBitmap* favicon_skia = favicon.ToSkBitmap();
+      PinPageToStartScreen(title, url_str, favicon_skia);
       return;
     }
+  }
 
-    GURL url = web_contents()->GetURL();
-    string16 title = web_contents()->GetTitle();
-    VLOG(1) << __FUNCTION__ << " calling pin with title: " << title
-            << " and url " << UTF8ToUTF16(url.spec());
-    metro_toggle_pinned_to_start_screen(title, UTF8ToUTF16(url.spec()));
-    // TODO(benwells): This will update the state incorrectly if the user
-    // cancels. To fix this some sort of callback needs to be introduced as
-    // the pinning happens on another thread.
-    is_pinned_ = !is_pinned_;
-    return;
-  }
-#endif
+  PinPageToStartScreen(title, url_str, NULL);
 }
 
 void MetroPinTabHelper::DidNavigateMainFrame(
     const content::LoadCommittedDetails& /*details*/,
     const content::FrameNavigateParams& /*params*/) {
   UpdatePinnedStateForCurrentURL();
 }
 
 void MetroPinTabHelper::UpdatePinnedStateForCurrentURL() {
 #if defined(OS_WIN)
   HMODULE metro_module = base::win::GetMetroModule();
   if (metro_module) {
     typedef BOOL (*MetroIsPinnedToStartScreen)(const string16&);
     MetroIsPinnedToStartScreen metro_is_pinned_to_start_screen =
         reinterpret_cast<MetroIsPinnedToStartScreen>(
             ::GetProcAddress(metro_module, "MetroIsPinnedToStartScreen"));
     if (!metro_is_pinned_to_start_screen) {
       NOTREACHED();
       return;
     }
 
     GURL url = web_contents()->GetURL();
     is_pinned_ = metro_is_pinned_to_start_screen(UTF8ToUTF16(url.spec())) != 0;
     VLOG(1) << __FUNCTION__ << " with url " << UTF8ToUTF16(url.spec())
             << " result: " << is_pinned_;
   }
 #endif
 }
+
+void MetroPinTabHelper::PinPageToStartScreen(
+    const string16& title,
+    const string16& url,
+    const SkBitmap* bitmap) {
+#if defined(OS_WIN)
+  HMODULE metro_module = base::win::GetMetroModule();
+  if (metro_module) {
+    typedef void (*MetroPinToStartScreen)(const string16&,
+        const string16&, const SkBitmap*);
+    MetroPinToStartScreen metro_pin_to_start_screen =
+        reinterpret_cast<MetroPinToStartScreen>(
+            ::GetProcAddress(metro_module, "MetroPinToStartScreen"));
+    if (!metro_pin_to_start_screen) {
+      NOTREACHED();
+      return;
+    }
+
+    VLOG(1) << __FUNCTION__ << " calling pin with title: " << title
+            << " and url: " << url;
+    metro_pin_to_start_screen(title, url, bitmap);
+  }
+#endif
+}
+
+void MetroPinTabHelper::UnPinPageFromStartScreen() {
+#if defined(OS_WIN)
+  HMODULE metro_module = base::win::GetMetroModule();
+  if (metro_module) {
+    typedef void (*MetroUnPinFromStartScreen)(const string16&);
+    MetroUnPinFromStartScreen metro_un_pin_from_start_screen =
+        reinterpret_cast<MetroUnPinFromStartScreen>(
+            ::GetProcAddress(metro_module, "MetroUnPinFromStartScreen"));
+    if (!metro_un_pin_from_start_screen) {
+      NOTREACHED();
+      return;
+    }
+
+    GURL url = web_contents()->GetURL();
+    VLOG(1) << __FUNCTION__ << " calling unpin with url: "
+            << UTF8ToUTF16(url.spec());
+    metro_un_pin_from_start_screen(UTF8ToUTF16(url.spec()));
+  }
+#endif
+}