Support for adding save page download items into downloads history.

content/browser/download/save_package.h

 // 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.
 
 #ifndef CONTENT_BROWSER_DOWNLOAD_SAVE_PACKAGE_H_
 #define CONTENT_BROWSER_DOWNLOAD_SAVE_PACKAGE_H_
 #pragma once
 
 #include <queue>
 #include <string>
 #include <vector>
 
 #include "base/basictypes.h"
 #include "base/file_path.h"
 #include "base/gtest_prod_util.h"
 #include "base/hash_tables.h"
 #include "base/memory/ref_counted.h"
 #include "base/memory/weak_ptr.h"
 #include "base/task.h"
+#include "chrome/browser/download/download_manager.h"
 #include "content/browser/tab_contents/tab_contents_observer.h"
 #include "googleurl/src/gurl.h"
 
-class DownloadItem;
-class DownloadManager;
 class GURL;
 class MessageLoop;
 class PrefService;
 class Profile;
 class SaveFileManager;
 class SaveItem;
 class SavePackage;
 class TabContents;
 struct SaveFileCreateInfo;
 struct SavePackageParam;
@@ skipping 11 matching lines @@
 // Saving page as complete-html page means we save not only the main html file
 // the user told it to save but also a directory for the auxiliary files such
 // as all sub-frame html files, image files, css files and js files.
 //
 // Each page saving job may include one or multiple files which need to be
 // saved. Each file is represented by a SaveItem, and all SaveItems are owned
 // by the SavePackage. SaveItems are created when a user initiates a page
 // saving job, and exist for the duration of one tab's life time.
 class SavePackage : public base::RefCountedThreadSafe<SavePackage>,
                     public TabContentsObserver,
+                    public DownloadItem::Observer,
                     public base::SupportsWeakPtr<SavePackage> {
  public:
   enum SavePackageType {
     // The value of the save type before its set by the user.
     SAVE_TYPE_UNKNOWN = -1,
     // User chose to save only the HTML of the page.
     SAVE_AS_ONLY_HTML = 0,
     // User chose to save complete-html page.
     SAVE_AS_COMPLETE_HTML = 1
   };
@@ skipping 87 matching lines @@
 
   // Notes from Init() above applies here as well.
   void InternalInit();
 
   void Stop();
   void CheckFinish();
   void SaveNextFile(bool process_all_remainder_items);
   void DoSavingProcess();
 
   // TabContentsObserver implementation.
-  virtual bool OnMessageReceived(const IPC::Message& message);
+  virtual bool OnMessageReceived(const IPC::Message& message) OVERRIDE;
+
+  // DownloadItem::Observer implementation.
+  virtual void OnDownloadUpdated(DownloadItem* download) OVERRIDE;
+  virtual void OnDownloadOpened(DownloadItem* download) OVERRIDE {}
+
+  // Update the download history of this item upon completion.
+  void FinalizeDownloadEntry();
+
+  // Detach from DownloadManager.
+  void StopObservation();
 
   // Return max length of a path for a specific base directory.
   // This is needed on POSIX, which restrict the length of file names in
   // addition to the restriction on the length of path names.
   // |base_dir| is assumed to be a directory name with no trailing slash.
   static uint32 GetMaxPathLengthForDirectory(const FilePath& base_dir);
 
   static bool GetSafePureFileName(const FilePath& dir_path,
                                   const FilePath::StringType& file_name_ext,
                                   uint32 max_file_path_len,
@@ skipping 30 matching lines @@
 
   void OnReceivedSavableResourceLinksForCurrentPage(
       const std::vector<GURL>& resources_list,
       const std::vector<GURL>& referrers_list,
       const std::vector<GURL>& frames_list);
 
   void OnReceivedSerializedHtmlData(const GURL& frame_url,
                                     const std::string& data,
                                     int32 status);
 
-  // Convenience function.
-  DownloadManager* GetDownloadManager();
-
-
   typedef base::hash_map<std::string, SaveItem*> SaveUrlItemMap;
   // in_progress_items_ is map of all saving job in in-progress state.
   SaveUrlItemMap in_progress_items_;
   // saved_failed_items_ is map of all saving job which are failed.
   SaveUrlItemMap saved_failed_items_;
 
   // The number of in process SaveItems.
   int in_process_count() const {
     return static_cast<int>(in_progress_items_.size());
   }
@@ skipping 30 matching lines @@
   // A queue for items we are about to start saving.
   SaveItemQueue waiting_item_queue_;
 
   typedef base::hash_map<int32, SaveItem*> SavedItemMap;
   // saved_success_items_ is map of all saving job which are successfully saved.
   SavedItemMap saved_success_items_;
 
   // Non-owning pointer for handling file writing on the file thread.
   SaveFileManager* file_manager_;
 
-  // We use a fake DownloadItem here in order to reuse the DownloadItemView.
-  // This class owns the pointer.
+  // DownloadManager owns the DownloadItem and handles history and UI.
+  DownloadManager* download_manager_;
   DownloadItem* download_;
 
   // The URL of the page the user wants to save.
   GURL page_url_;
   FilePath saved_main_file_path_;
   FilePath saved_main_directory_path_;
 
   // The title of the page the user wants to save.
   string16 title_;
 
@@ skipping 34 matching lines @@
   friend class SavePackageTest;
   FRIEND_TEST_ALL_PREFIXES(SavePackageTest, TestSuggestedSaveNames);
   FRIEND_TEST_ALL_PREFIXES(SavePackageTest, TestLongSafePureFilename);
 
   ScopedRunnableMethodFactory<SavePackage> method_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(SavePackage);
 };
 
 #endif  // CONTENT_BROWSER_DOWNLOAD_SAVE_PACKAGE_H_