OLD | NEW |
1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #ifndef CHROME_BROWSER_DOWNLOAD_SAVE_PACKAGE_H_ | 5 #ifndef CHROME_BROWSER_DOWNLOAD_SAVE_PACKAGE_H_ |
6 #define CHROME_BROWSER_DOWNLOAD_SAVE_PACKAGE_H_ | 6 #define CHROME_BROWSER_DOWNLOAD_SAVE_PACKAGE_H_ |
7 #pragma once | 7 #pragma once |
8 | 8 |
9 #include <queue> | 9 #include <queue> |
10 #include <string> | 10 #include <string> |
11 #include <vector> | 11 #include <vector> |
12 | 12 |
13 #include "base/basictypes.h" | 13 #include "base/basictypes.h" |
14 #include "base/file_path.h" | 14 #include "base/file_path.h" |
15 #include "base/gtest_prod_util.h" | 15 #include "base/gtest_prod_util.h" |
16 #include "base/hash_tables.h" | 16 #include "base/hash_tables.h" |
17 #include "base/ref_counted.h" | 17 #include "base/ref_counted.h" |
18 #include "base/task.h" | 18 #include "base/task.h" |
19 #include "chrome/browser/renderer_host/render_view_host_delegate.h" | |
20 #include "chrome/browser/shell_dialogs.h" | 19 #include "chrome/browser/shell_dialogs.h" |
| 20 #include "chrome/browser/tab_contents/web_navigation_observer.h" |
21 #include "googleurl/src/gurl.h" | 21 #include "googleurl/src/gurl.h" |
22 | 22 |
23 class SaveFileManager; | 23 class SaveFileManager; |
24 class SaveItem; | 24 class SaveItem; |
25 class SavePackage; | 25 class SavePackage; |
26 class DownloadItem; | 26 class DownloadItem; |
27 class DownloadManager; | 27 class DownloadManager; |
28 class GURL; | 28 class GURL; |
29 class MessageLoop; | 29 class MessageLoop; |
30 class PrefService; | 30 class PrefService; |
(...skipping 16 matching lines...) Expand all Loading... |
47 // file regardless internal sub resources and sub frames. | 47 // file regardless internal sub resources and sub frames. |
48 // Saving page as complete-html page means we save not only the main html file | 48 // Saving page as complete-html page means we save not only the main html file |
49 // the user told it to save but also a directory for the auxiliary files such | 49 // the user told it to save but also a directory for the auxiliary files such |
50 // as all sub-frame html files, image files, css files and js files. | 50 // as all sub-frame html files, image files, css files and js files. |
51 // | 51 // |
52 // Each page saving job may include one or multiple files which need to be | 52 // Each page saving job may include one or multiple files which need to be |
53 // saved. Each file is represented by a SaveItem, and all SaveItems are owned | 53 // saved. Each file is represented by a SaveItem, and all SaveItems are owned |
54 // by the SavePackage. SaveItems are created when a user initiates a page | 54 // by the SavePackage. SaveItems are created when a user initiates a page |
55 // saving job, and exist for the duration of one tab's life time. | 55 // saving job, and exist for the duration of one tab's life time. |
56 class SavePackage : public base::RefCountedThreadSafe<SavePackage>, | 56 class SavePackage : public base::RefCountedThreadSafe<SavePackage>, |
57 public RenderViewHostDelegate::Save, | 57 public WebNavigationObserver, |
58 public SelectFileDialog::Listener { | 58 public SelectFileDialog::Listener { |
59 public: | 59 public: |
60 enum SavePackageType { | 60 enum SavePackageType { |
61 // The value of the save type before its set by the user. | 61 // The value of the save type before its set by the user. |
62 SAVE_TYPE_UNKNOWN = -1, | 62 SAVE_TYPE_UNKNOWN = -1, |
63 // User chose to save only the HTML of the page. | 63 // User chose to save only the HTML of the page. |
64 SAVE_AS_ONLY_HTML = 0, | 64 SAVE_AS_ONLY_HTML = 0, |
65 // User chose to save complete-html page. | 65 // User chose to save complete-html page. |
66 SAVE_AS_COMPLETE_HTML = 1 | 66 SAVE_AS_COMPLETE_HTML = 1 |
67 }; | 67 }; |
(...skipping 53 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
121 void ShowDownloadInShell(); | 121 void ShowDownloadInShell(); |
122 | 122 |
123 bool canceled() const { return user_canceled_ || disk_error_occurred_; } | 123 bool canceled() const { return user_canceled_ || disk_error_occurred_; } |
124 bool finished() const { return finished_; } | 124 bool finished() const { return finished_; } |
125 SavePackageType save_type() const { return save_type_; } | 125 SavePackageType save_type() const { return save_type_; } |
126 int tab_id() const { return tab_id_; } | 126 int tab_id() const { return tab_id_; } |
127 int id() const { return unique_id_; } | 127 int id() const { return unique_id_; } |
128 | 128 |
129 void GetSaveInfo(); | 129 void GetSaveInfo(); |
130 | 130 |
131 // RenderViewHostDelegate::Save ---------------------------------------------- | |
132 | |
133 // Process all of the current page's savable links of subresources, resources | |
134 // referrers and frames (including the main frame and subframes) from the | |
135 // render view host. | |
136 virtual void OnReceivedSavableResourceLinksForCurrentPage( | |
137 const std::vector<GURL>& resources_list, | |
138 const std::vector<GURL>& referrers_list, | |
139 const std::vector<GURL>& frames_list); | |
140 | |
141 // Process the serialized html content data of a specified web page | |
142 // gotten from render process. | |
143 virtual void OnReceivedSerializedHtmlData(const GURL& frame_url, | |
144 const std::string& data, | |
145 int32 status); | |
146 | |
147 // Statics ------------------------------------------------------------------- | 131 // Statics ------------------------------------------------------------------- |
148 | 132 |
149 // Used to disable prompting the user for a directory/filename of the saved | 133 // Used to disable prompting the user for a directory/filename of the saved |
150 // web page. This is available for testing. | 134 // web page. This is available for testing. |
151 static void SetShouldPromptUser(bool should_prompt); | 135 static void SetShouldPromptUser(bool should_prompt); |
152 | 136 |
153 // Check whether we can do the saving page operation for the specified URL. | 137 // Check whether we can do the saving page operation for the specified URL. |
154 static bool IsSavableURL(const GURL& url); | 138 static bool IsSavableURL(const GURL& url); |
155 | 139 |
156 // Check whether we can do the saving page operation for the contents which | 140 // Check whether we can do the saving page operation for the contents which |
(...skipping 15 matching lines...) Expand all Loading... |
172 ~SavePackage(); | 156 ~SavePackage(); |
173 | 157 |
174 // Notes from Init() above applies here as well. | 158 // Notes from Init() above applies here as well. |
175 void InternalInit(); | 159 void InternalInit(); |
176 | 160 |
177 void Stop(); | 161 void Stop(); |
178 void CheckFinish(); | 162 void CheckFinish(); |
179 void SaveNextFile(bool process_all_remainder_items); | 163 void SaveNextFile(bool process_all_remainder_items); |
180 void DoSavingProcess(); | 164 void DoSavingProcess(); |
181 | 165 |
| 166 // WebNavigationObserver implementation. |
| 167 virtual bool OnMessageReceived(const IPC::Message& message); |
| 168 |
182 // Create a file name based on the response from the server. | 169 // Create a file name based on the response from the server. |
183 bool GenerateFileName(const std::string& disposition, | 170 bool GenerateFileName(const std::string& disposition, |
184 const GURL& url, | 171 const GURL& url, |
185 bool need_html_ext, | 172 bool need_html_ext, |
186 FilePath::StringType* generated_name); | 173 FilePath::StringType* generated_name); |
187 | 174 |
188 // Get all savable resource links from current web page, include main | 175 // Get all savable resource links from current web page, include main |
189 // frame and sub-frame. | 176 // frame and sub-frame. |
190 void GetAllSavableResourceLinksForCurrentPage(); | 177 void GetAllSavableResourceLinksForCurrentPage(); |
191 // Get html data by serializing all frames of current page with lists | 178 // Get html data by serializing all frames of current page with lists |
192 // which contain all resource links that have local copy. | 179 // which contain all resource links that have local copy. |
193 void GetSerializedHtmlDataForCurrentPageWithLocalLinks(); | 180 void GetSerializedHtmlDataForCurrentPageWithLocalLinks(); |
194 | 181 |
195 SaveItem* LookupItemInProcessBySaveId(int32 save_id); | 182 SaveItem* LookupItemInProcessBySaveId(int32 save_id); |
196 void PutInProgressItemToSavedMap(SaveItem* save_item); | 183 void PutInProgressItemToSavedMap(SaveItem* save_item); |
197 | 184 |
198 // Retrieves the URL to be saved from tab_contents_ variable. | 185 // Retrieves the URL to be saved from tab_contents_ variable. |
199 GURL GetUrlToBeSaved(); | 186 GURL GetUrlToBeSaved(); |
200 | 187 |
201 void CreateDirectoryOnFileThread(const FilePath& website_save_dir, | 188 void CreateDirectoryOnFileThread(const FilePath& website_save_dir, |
202 const FilePath& download_save_dir, | 189 const FilePath& download_save_dir, |
203 const std::string& mime_type); | 190 const std::string& mime_type); |
204 void ContinueGetSaveInfo(const FilePath& suggested_path, | 191 void ContinueGetSaveInfo(const FilePath& suggested_path, |
205 bool can_save_as_complete); | 192 bool can_save_as_complete); |
206 void ContinueSave(const FilePath& final_name, int index); | 193 void ContinueSave(const FilePath& final_name, int index); |
207 | 194 |
| 195 void OnReceivedSavableResourceLinksForCurrentPage( |
| 196 const std::vector<GURL>& resources_list, |
| 197 const std::vector<GURL>& referrers_list, |
| 198 const std::vector<GURL>& frames_list); |
| 199 |
| 200 void OnReceivedSerializedHtmlData(const GURL& frame_url, |
| 201 const std::string& data, |
| 202 int32 status); |
| 203 |
208 | 204 |
209 typedef base::hash_map<std::string, SaveItem*> SaveUrlItemMap; | 205 typedef base::hash_map<std::string, SaveItem*> SaveUrlItemMap; |
210 // in_progress_items_ is map of all saving job in in-progress state. | 206 // in_progress_items_ is map of all saving job in in-progress state. |
211 SaveUrlItemMap in_progress_items_; | 207 SaveUrlItemMap in_progress_items_; |
212 // saved_failed_items_ is map of all saving job which are failed. | 208 // saved_failed_items_ is map of all saving job which are failed. |
213 SaveUrlItemMap saved_failed_items_; | 209 SaveUrlItemMap saved_failed_items_; |
214 | 210 |
215 // The number of in process SaveItems. | 211 // The number of in process SaveItems. |
216 int in_process_count() const { | 212 int in_process_count() const { |
217 return static_cast<int>(in_progress_items_.size()); | 213 return static_cast<int>(in_progress_items_.size()); |
(...skipping 97 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
315 | 311 |
316 friend class SavePackageTest; | 312 friend class SavePackageTest; |
317 FRIEND_TEST_ALL_PREFIXES(SavePackageTest, TestSuggestedSaveNames); | 313 FRIEND_TEST_ALL_PREFIXES(SavePackageTest, TestSuggestedSaveNames); |
318 | 314 |
319 ScopedRunnableMethodFactory<SavePackage> method_factory_; | 315 ScopedRunnableMethodFactory<SavePackage> method_factory_; |
320 | 316 |
321 DISALLOW_COPY_AND_ASSIGN(SavePackage); | 317 DISALLOW_COPY_AND_ASSIGN(SavePackage); |
322 }; | 318 }; |
323 | 319 |
324 #endif // CHROME_BROWSER_DOWNLOAD_SAVE_PACKAGE_H_ | 320 #endif // CHROME_BROWSER_DOWNLOAD_SAVE_PACKAGE_H_ |
OLD | NEW |