chrome/browser/download/download_status_updater_win.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/browser/download/download_status_updater.h"
   6
   7 #include <shobjidl.h>
   8 #include <string>
   9
  10 #include "base/files/file_path.h"
  11 #include "base/logging.h"
  12 #include "base/stl_util.h"
  13 #include "base/strings/string_number_conversions.h"
  14 #include "base/win/metro.h"
  15 #include "base/win/scoped_comptr.h"
  16 #include "base/win/windows_version.h"
  17 #include "chrome/browser/platform_util.h"
  18 #include "chrome/browser/ui/browser.h"
  19 #include "chrome/browser/ui/browser_iterator.h"
  20 #include "chrome/browser/ui/browser_window.h"
  21 #include "content/public/browser/browser_context.h"
  22 #include "content/public/browser/browser_thread.h"
  23 #include "grit/generated_resources.h"
  24 #include "ui/base/l10n/l10n_util.h"
  25 #include "ui/views/win/hwnd_util.h"
  26 #include "url/gurl.h"
  27 #include "win8/util/win8_util.h"
  28
  29 namespace {
  30
  31 const char kDownloadNotificationPrefix[] = "DownloadNotification";
  32 int g_next_notification_id = 0;
  33
  34 void UpdateTaskbarProgressBar(int download_count,
  35                               bool progress_known,
  36                               float progress) {
  37   // Taskbar progress bar is only supported on Win7.
  38   if (base::win::GetVersion() < base::win::VERSION_WIN7)
  39     return;
  40
  41   base::win::ScopedComPtr<ITaskbarList3> taskbar;
  42   HRESULT result = taskbar.CreateInstance(CLSID_TaskbarList, NULL,
  43                                           CLSCTX_INPROC_SERVER);
  44   if (FAILED(result)) {
  45     VLOG(1) << "Failed creating a TaskbarList object: " << result;
  46     return;
  47   }
  48
  49   result = taskbar->HrInit();
  50   if (FAILED(result)) {
  51     LOG(ERROR) << "Failed initializing an ITaskbarList3 interface.";
  52     return;
  53   }
  54
  55   // Iterate through all the browser windows, and draw the progress bar.
  56   for (chrome::BrowserIterator it; !it.done(); it.Next()) {
  57     Browser* browser = *it;
  58     BrowserWindow* window = browser->window();
  59     if (!window)
  60       continue;
  61     HWND frame = views::HWNDForNativeWindow(window->GetNativeWindow());
  62     if (download_count == 0 || progress == 1.0f)
  63       taskbar->SetProgressState(frame, TBPF_NOPROGRESS);
  64     else if (!progress_known)
  65       taskbar->SetProgressState(frame, TBPF_INDETERMINATE);
  66     else
  67       taskbar->SetProgressValue(frame, static_cast<int>(progress * 100), 100);
  68   }
  69 }
  70
  71 void MetroDownloadNotificationClickedHandler(const wchar_t* download_path) {
  72   // Metro chrome will invoke these handlers on the metro thread.
  73   DCHECK(!content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
  74
  75   // Ensure that we invoke the function to display the downloaded item on the
  76   // UI thread.
  77   content::BrowserThread::PostTask(content::BrowserThread::UI, FROM_HERE,
  78                                    base::Bind(platform_util::ShowItemInFolder,
  79                                               static_cast<Profile*>(NULL),
  80                                               base::FilePath(download_path)));
  81 }
  82
  83 }  // namespace
  84
  85 void DownloadStatusUpdater::UpdateAppIconDownloadProgress(
  86     content::DownloadItem* download) {
  87
  88   // Always update overall progress.
  89   float progress = 0;
  90   int download_count = 0;
  91   bool progress_known = GetProgress(&progress, &download_count);
  92   UpdateTaskbarProgressBar(download_count, progress_known, progress);
  93
  94   // Fire notifications when downloads complete.
  95   if (!win8::IsSingleWindowMetroMode())
  96     return;
  97
  98   if (download->GetState() != content::DownloadItem::COMPLETE)
  99     return;
 100
 101   if (download->GetOpenWhenComplete() ||
 102       download->ShouldOpenFileBasedOnExtension() ||
 103       download->IsTemporary() ||
 104       download->GetAutoOpened())
 105     return;
 106
 107   // Don't display the Windows8 metro notifications for an incognito download.
 108   if (download->GetBrowserContext() &&
 109       download->GetBrowserContext()->IsOffTheRecord())
 110     return;
 111
 112   // Don't display the Windows 8 metro notifications if we are in the
 113   // foreground.
 114   HWND foreground_window = ::GetForegroundWindow();
 115   if (::IsWindow(foreground_window)) {
 116     DWORD process_id = 0;
 117     ::GetWindowThreadProcessId(foreground_window, &process_id);
 118     if (process_id == ::GetCurrentProcessId())
 119       return;
 120   }
 121
 122   // In Windows 8 metro mode display a metro style notification which
 123   // informs the user that the download is complete.
 124   HMODULE metro = base::win::GetMetroModule();
 125   base::win::MetroNotification display_notification =
 126       reinterpret_cast<base::win::MetroNotification>(
 127           ::GetProcAddress(metro, "DisplayNotification"));
 128   DCHECK(display_notification);
 129   if (display_notification) {
 130     base::string16 title = l10n_util::GetStringUTF16(
 131         IDS_METRO_DOWNLOAD_COMPLETE_NOTIFICATION_TITLE);
 132     base::string16 body = l10n_util::GetStringUTF16(
 133         IDS_METRO_DOWNLOAD_COMPLETE_NOTIFICATION);
 134
 135     // Dummy notification id. Every metro style notification needs a
 136     // unique notification id.
 137     std::string notification_id = kDownloadNotificationPrefix;
 138     notification_id += base::IntToString(g_next_notification_id++);
 139
 140     display_notification(download->GetURL().spec().c_str(),
 141                          "",
 142                          title.c_str(),
 143                          body.c_str(),
 144                          L"",
 145                          notification_id.c_str(),
 146                          MetroDownloadNotificationClickedHandler,
 147                          download->GetTargetFilePath().value().c_str());
 148   }
 149 }