Implementing custom notification icons for the DisplayBalloon method on Windows.

chrome/browser/ui/views/status_icons/status_icon_win.cc

 // Copyright (c) 2011 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/views/status_icons/status_icon_win.h"
 
 #include "base/sys_string_conversions.h"
+#include "base/win/windows_version.h"
 #include "third_party/skia/include/core/SkBitmap.h"
 #include "ui/gfx/icon_util.h"
 #include "ui/gfx/point.h"
 #if !defined(USE_AURA)
 #include "views/controls/menu/menu_2.h"
 #endif
 
 StatusIconWin::StatusIconWin(UINT id, HWND window, UINT message)
     : icon_id_(id),
       window_(window),
       message_id_(message) {
   NOTIFYICONDATA icon_data;
   InitIconData(&icon_data);
   icon_data.uFlags = NIF_MESSAGE;
   icon_data.uCallbackMessage = message_id_;
   BOOL result = Shell_NotifyIcon(NIM_ADD, &icon_data);
   // This can happen if the explorer process isn't running when we try to
   // create the icon for some reason (for example, at startup).
   if (!result)
     LOG(WARNING) << "Unable to create status tray icon.";
 }
 
 StatusIconWin::~StatusIconWin() {
+  // Force to hide the balloon if any. Otherwise it might keep the icon in
+  // the tray until the balloon goes away.
+  HideBalloon();
+
   // Remove our icon.
   NOTIFYICONDATA icon_data;
   InitIconData(&icon_data);
   Shell_NotifyIcon(NIM_DELETE, &icon_data);
 }
 
 void StatusIconWin::SetImage(const SkBitmap& image) {
   // Create the icon.
   NOTIFYICONDATA icon_data;
   InitIconData(&icon_data);
@@ skipping 36 matching lines @@
   icon_data.uFlags = NIF_TIP;
   wcscpy_s(icon_data.szTip, tool_tip.c_str());
   BOOL result = Shell_NotifyIcon(NIM_MODIFY, &icon_data);
   if (!result)
     LOG(WARNING) << "Unable to set tooltip for status tray icon";
 }
 
 void StatusIconWin::DisplayBalloon(const SkBitmap& icon,
                                    const string16& title,
                                    const string16& contents) {
-  // TODO(leandrogracia): implement custom icons for notification balloons.
-
   NOTIFYICONDATA icon_data;
   InitIconData(&icon_data);
   icon_data.uFlags = NIF_INFO;
   icon_data.dwInfoFlags = NIIF_INFO;
   wcscpy_s(icon_data.szInfoTitle, title.c_str());
   wcscpy_s(icon_data.szInfo, contents.c_str());
   icon_data.uTimeout = 0;
+
+  base::win::Version win_version = base::win::OSInfo::GetInstance()->version();
+  if (!icon.empty() && win_version != base::win::VERSION_PRE_XP) {
+    balloon_icon_.Set(IconUtil::CreateHICONFromSkBitmap(icon));
+    if (win_version >= base::win::VERSION_VISTA) {
+      icon_data.hBalloonIcon = balloon_icon_.Get();
+      icon_data.dwInfoFlags = NIIF_USER | NIIF_LARGE_ICON;
+    } else {
+      icon_data.hIcon = balloon_icon_.Get();
+      icon_data.uFlags |= NIF_ICON;
+      icon_data.dwInfoFlags = NIIF_USER;
+    }
+  }
+
   BOOL result = Shell_NotifyIcon(NIM_MODIFY, &icon_data);
   if (!result)
     LOG(WARNING) << "Unable to create status tray balloon.";
 }
 
+void StatusIconWin::HideBalloon() {
+  NOTIFYICONDATA icon_data;
+  InitIconData(&icon_data);
+  icon_data.szInfo[0] = '\0';
+  BOOL result = Shell_NotifyIcon(NIM_MODIFY, &icon_data);
+  if (!result)
+    LOG(WARNING) << "Unable to hide status tray balloon.";
+}
+
 void StatusIconWin::UpdatePlatformContextMenu(ui::MenuModel* menu) {
 #if defined(USE_AURA)
   // crbug.com/99489.
   NOTIMPLEMENTED();
 #else
   // If no items are passed, blow away our context menu.
   if (!menu) {
     context_menu_.reset();
     return;
   }
@@ skipping 16 matching lines @@
   if (context_menu_.get()) {
     // Set our window as the foreground window, so the context menu closes when
     // we click away from it.
     SetForegroundWindow(window_);
     context_menu_->RunContextMenuAt(gfx::Point(x, y));
   }
 #endif
 }
 
 void StatusIconWin::InitIconData(NOTIFYICONDATA* icon_data) {
-  icon_data->cbSize = sizeof(NOTIFYICONDATA);
+  if (base::win::OSInfo::GetInstance()->version() >= base::win::VERSION_VISTA)
+    icon_data->cbSize = sizeof(NOTIFYICONDATA);
+  else
+    icon_data->cbSize = NOTIFYICONDATA_V3_SIZE;
+
   icon_data->hWnd = window_;
   icon_data->uID = icon_id_;
 }