Detect removed files and reflect the state in chrome://downloads and the download shelf

chrome/browser/ui/webui/downloads_dom_handler.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/webui/downloads_dom_handler.h"
 
 #include <algorithm>
 #include <functional>
 
 #include "base/basictypes.h"
@@ skipping 101 matching lines @@
 }
 
 // A download has started or been deleted. Query our DownloadManager for the
 // current set of downloads.
 void DownloadsDOMHandler::ModelChanged() {
   ClearDownloadItems();
   download_manager_->SearchDownloads(WideToUTF16(search_text_),
                                      &download_items_);
   sort(download_items_.begin(), download_items_.end(), DownloadItemSorter());
 
-  // Scan for any in progress downloads and add ourself to them as an observer.
+  // Add oneself to all download items as an observer.
   for (OrderedDownloads::iterator it = download_items_.begin();
        it != download_items_.end(); ++it) {
     if (static_cast<int>(it - download_items_.begin()) > kMaxDownloads)
       break;
-
-    DownloadItem* download = *it;
-    if (download->IsInProgress()) {
-      // We want to know what happens as the download progresses.
-      download->AddObserver(this);
-    } else if (download->safety_state() == DownloadItem::DANGEROUS) {
-      // We need to be notified when the user validates the dangerous download.
-      download->AddObserver(this);
-    }
+    (*it)->AddObserver(this);
   }
 
   SendCurrentDownloads();
 }
 
 void DownloadsDOMHandler::HandleGetDownloads(const ListValue* args) {
   std::wstring new_search = UTF16ToWideHack(ExtractStringValue(args));
   if (search_text_.compare(new_search) != 0) {
     search_text_ = new_search;
     ModelChanged();
   } else {
     SendCurrentDownloads();
   }
+
+  download_manager_->CheckForFilesRemoval();

[Randy Smith (Not in Mondays), 2011/05/13 20:45:32]

And I'd also nuke these, as well.  I think doing the check for file removal
occasionally (even as infrequently as when you startup the browser) is fine; the
"real" fix is handling the case when we try to do an operation on a file that's
been removed.

[haraken1, 2011/05/16 11:42:27]

I removed it.

 }
 
 void DownloadsDOMHandler::HandleOpenFile(const ListValue* args) {
   DownloadItem* file = GetDownloadByValue(args);
   if (file)
     file->OpenDownload();
+
+  download_manager_->CheckForFilesRemoval();
 }
 
 void DownloadsDOMHandler::HandleDrag(const ListValue* args) {
   DownloadItem* file = GetDownloadByValue(args);
   if (file) {
     IconManager* im = g_browser_process->icon_manager();
     gfx::Image* icon = im->LookupIcon(file->GetUserVerifiedFilePath(),
                                       IconLoader::NORMAL);
     gfx::NativeView view = web_ui_->tab_contents()->GetNativeView();
     download_util::DragDownload(file, icon, view);
   }
+
+  download_manager_->CheckForFilesRemoval();
 }
 
 void DownloadsDOMHandler::HandleSaveDangerous(const ListValue* args) {
   DownloadItem* file = GetDownloadByValue(args);
   if (file)
     download_manager_->DangerousDownloadValidated(file);
+
+  download_manager_->CheckForFilesRemoval();
 }
 
 void DownloadsDOMHandler::HandleDiscardDangerous(const ListValue* args) {
   DownloadItem* file = GetDownloadByValue(args);
   if (file)
     file->Delete(DownloadItem::DELETE_DUE_TO_USER_DISCARD);
+
+  download_manager_->CheckForFilesRemoval();
 }
 
 void DownloadsDOMHandler::HandleShow(const ListValue* args) {
   DownloadItem* file = GetDownloadByValue(args);
   if (file)
     file->ShowDownloadInShell();
+
+  download_manager_->CheckForFilesRemoval();
 }
 
 void DownloadsDOMHandler::HandlePause(const ListValue* args) {
   DownloadItem* file = GetDownloadByValue(args);
   if (file)
     file->TogglePause();
+
+  download_manager_->CheckForFilesRemoval();
 }
 
 void DownloadsDOMHandler::HandleRemove(const ListValue* args) {
   DownloadItem* file = GetDownloadByValue(args);
   if (file)
     file->Remove();
+
+  download_manager_->CheckForFilesRemoval();
 }
 
 void DownloadsDOMHandler::HandleCancel(const ListValue* args) {
   DownloadItem* file = GetDownloadByValue(args);
   if (file)
     file->Cancel(true);
+
+  download_manager_->CheckForFilesRemoval();
 }
 
 void DownloadsDOMHandler::HandleClearAll(const ListValue* args) {
   download_manager_->RemoveAllDownloads();
+
+  download_manager_->CheckForFilesRemoval();
 }
 
 // DownloadsDOMHandler, private: ----------------------------------------------
 
 void DownloadsDOMHandler::SendCurrentDownloads() {
   ListValue results_value;
   for (OrderedDownloads::iterator it = download_items_.begin();
       it != download_items_.end(); ++it) {
     int index = static_cast<int>(it - download_items_.begin());
     if (index > kMaxDownloads)
@@ skipping 24 matching lines @@
   return NULL;
 }
 
 DownloadItem* DownloadsDOMHandler::GetDownloadByValue(const ListValue* args) {
   int id;
   if (ExtractIntegerValue(args, &id)) {
     return GetDownloadById(id);
   }
   return NULL;
 }