Support for adding save page download items into downloads history.

chrome/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 CHROME_BROWSER_DOWNLOAD_SAVE_PACKAGE_H_
 #define CHROME_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/task.h"
+#include "chrome/browser/download/download_manager.h"
 #include "chrome/browser/ui/shell_dialogs.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;
 struct SaveFileCreateInfo;
 class SaveFileManager;
 class SaveItem;
 class SavePackage;
 struct SavePackageParam;
 class TabContentsWrapper;
@@ 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 SelectFileDialog::Listener {
+                    public SelectFileDialog::Listener,
+                    public DownloadManager::Observer {
  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 65 matching lines @@
   // web page.  This is available for testing.
   static void SetShouldPromptUser(bool should_prompt);
 
   // Check whether we can do the saving page operation for the specified URL.
   static bool IsSavableURL(const GURL& url);
 
   // Check whether we can do the saving page operation for the contents which
   // have the specified MIME type.
   static bool IsSavableContents(const std::string& contents_mime_type);
 
-  // SelectFileDialog::Listener ------------------------------------------------
-  virtual void FileSelected(const FilePath& path, int index, void* params);
-  virtual void FileSelectionCanceled(void* params);
+  // Interfaces ----------------------------------------------------------------
+
+  // SelectFileDialog::Listener.
+  virtual void FileSelected(const FilePath& path,
+      int index, void* params) OVERRIDE;
+  virtual void FileSelectionCanceled(void* params) OVERRIDE;
+
+  // DownloadManager::Observer.
+  virtual void ModelChanged() OVERRIDE {}
+  virtual void ManagerGoingDown() OVERRIDE;
 
  private:
   friend class base::RefCountedThreadSafe<SavePackage>;
 
   // For testing only.
   SavePackage(TabContentsWrapper* wrapper,
               const FilePath& file_full_path,
               const FilePath& directory_full_path);
 
   virtual ~SavePackage();
 
   // 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;
 
   // 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 69 matching lines @@
   // Ensures that the file name has a proper extension for supported formats
   // if necessary.
   static FilePath EnsureMimeExtension(const FilePath& name,
       const std::string& contents_mime_type);
 
   // Returns extension for supported MIME types (for example, for "text/plain"
   // it returns "txt").
   static const FilePath::CharType* ExtensionForMimeType(
       const std::string& contents_mime_type);
 
+  // Callback when Save Page As entry is commited to the history system.
+  void OnDownloadEntryAdded(int32 download_id, int64 db_handle);
+
+  // Update the download history of this item.
+  void FinalizeDownloadEntry();
+
   // Owning TabContentsWrapper.
   TabContentsWrapper* wrapper_;
 
   typedef std::queue<SaveItem*> SaveItemQueue;
   // 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_;
@@ skipping 43 matching lines @@
   // Since for one tab, it can only have one SavePackage in same time.
   // Now we actually use render_process_id as tab's unique id.
   const int tab_id_;
 
   // Unique ID for this SavePackage.
   const int unique_id_;
 
   // For managing select file dialogs.
   scoped_refptr<SelectFileDialog> select_file_dialog_;
 
+  DownloadManager* download_manager_;
+
   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  // CHROME_BROWSER_DOWNLOAD_SAVE_PACKAGE_H_