MD Downloads: check for removed files on page load

chrome/browser/ui/webui/md_downloads/md_downloads_dom_handler.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/webui/md_downloads/md_downloads_dom_handler.h"
 
 #include <algorithm>
 #include <functional>
 
 #include "base/bind.h"
@@ skipping 64 matching lines @@
 }  // namespace
 
 MdDownloadsDOMHandler::MdDownloadsDOMHandler(
     content::DownloadManager* download_manager, content::WebUI* web_ui)
     : list_tracker_(download_manager, web_ui),
       weak_ptr_factory_(this) {
   // Create our fileicon data source.
   Profile* profile = Profile::FromBrowserContext(
       download_manager->GetBrowserContext());
   content::URLDataSource::Add(profile, new FileIconSource());
+  CheckForRemovedFiles();
 }
 
 MdDownloadsDOMHandler::~MdDownloadsDOMHandler() {
   FinalizeRemovals();
 }
 
 // MdDownloadsDOMHandler, public: ---------------------------------------------
 
 void MdDownloadsDOMHandler::RegisterMessages() {
   web_ui()->RegisterMessageCallback("getDownloads",
@@ skipping 34 matching lines @@
                  weak_ptr_factory_.GetWeakPtr()));
   web_ui()->RegisterMessageCallback("openDownloadsFolder",
       base::Bind(&MdDownloadsDOMHandler::HandleOpenDownloadsFolder,
                  weak_ptr_factory_.GetWeakPtr()));
 }
 
 void MdDownloadsDOMHandler::RenderViewReused(
     content::RenderViewHost* render_view_host) {
   list_tracker_.Stop();
   list_tracker_.Reset();
+  CheckForRemovedFiles();
 }
 
 void MdDownloadsDOMHandler::HandleGetDownloads(const base::ListValue* args) {
   CountDownloadsDOMEvents(DOWNLOADS_DOM_EVENT_GET_DOWNLOADS);
 
   bool terms_changed = list_tracker_.SetSearchTerms(*args);
   if (terms_changed)
     list_tracker_.Reset();
 
   list_tracker_.StartAndSendChunk();
@@ skipping 253 matching lines @@
   if (GetMainNotifierManager())
     item = GetMainNotifierManager()->GetDownload(id);
   if (!item && GetOriginalNotifierManager())
     item = GetOriginalNotifierManager()->GetDownload(id);
   return item;
 }
 
 content::WebContents* MdDownloadsDOMHandler::GetWebUIWebContents() {
   return web_ui()->GetWebContents();
 }
+
+void MdDownloadsDOMHandler::CheckForRemovedFiles() {
+  if (GetMainNotifierManager())
+    GetMainNotifierManager()->CheckForHistoryFilesRemoval();
+  if (GetOriginalNotifierManager())
+    GetOriginalNotifierManager()->CheckForHistoryFilesRemoval();
+}