| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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_EXTENSIONS_API_PAGE_CAPTURE_PAGE_CAPTURE_API_H_ | 5 #ifndef CHROME_BROWSER_EXTENSIONS_API_PAGE_CAPTURE_PAGE_CAPTURE_API_H_ |
| 6 #define CHROME_BROWSER_EXTENSIONS_API_PAGE_CAPTURE_PAGE_CAPTURE_API_H_ | 6 #define CHROME_BROWSER_EXTENSIONS_API_PAGE_CAPTURE_PAGE_CAPTURE_API_H_ |
| 7 | 7 |
| 8 #include <string> | 8 #include <string> |
| 9 | 9 |
| 10 #include "base/memory/ref_counted.h" | 10 #include "base/memory/ref_counted.h" |
| 11 #include "chrome/browser/extensions/extension_function.h" | 11 #include "chrome/browser/extensions/extension_function.h" |
| 12 #include "chrome/common/extensions/api/page_capture.h" | 12 #include "chrome/common/extensions/api/page_capture.h" |
| 13 #include "webkit/blob/shareable_file_reference.h" | 13 #include "webkit/blob/shareable_file_reference.h" |
| 14 | 14 |
| 15 namespace base { |
| 15 class FilePath; | 16 class FilePath; |
| 17 } |
| 16 | 18 |
| 17 namespace content { | 19 namespace content { |
| 18 class WebContents; | 20 class WebContents; |
| 19 } | 21 } |
| 20 | 22 |
| 21 namespace extensions { | 23 namespace extensions { |
| 22 | 24 |
| 23 class PageCaptureSaveAsMHTMLFunction : public AsyncExtensionFunction { | 25 class PageCaptureSaveAsMHTMLFunction : public AsyncExtensionFunction { |
| 24 public: | 26 public: |
| 25 PageCaptureSaveAsMHTMLFunction(); | 27 PageCaptureSaveAsMHTMLFunction(); |
| 26 | 28 |
| 27 // Test specific delegate used to test that the temporary file gets deleted. | 29 // Test specific delegate used to test that the temporary file gets deleted. |
| 28 class TestDelegate { | 30 class TestDelegate { |
| 29 public: | 31 public: |
| 30 // Called on the UI thread when the temporary file that contains the | 32 // Called on the UI thread when the temporary file that contains the |
| 31 // generated data has been created. | 33 // generated data has been created. |
| 32 virtual void OnTemporaryFileCreated(const FilePath& temp_file) = 0; | 34 virtual void OnTemporaryFileCreated(const base::FilePath& temp_file) = 0; |
| 33 }; | 35 }; |
| 34 static void SetTestDelegate(TestDelegate* delegate); | 36 static void SetTestDelegate(TestDelegate* delegate); |
| 35 | 37 |
| 36 private: | 38 private: |
| 37 virtual ~PageCaptureSaveAsMHTMLFunction(); | 39 virtual ~PageCaptureSaveAsMHTMLFunction(); |
| 38 virtual bool RunImpl() OVERRIDE; | 40 virtual bool RunImpl() OVERRIDE; |
| 39 virtual bool OnMessageReceivedFromRenderView( | 41 virtual bool OnMessageReceivedFromRenderView( |
| 40 const IPC::Message& message) OVERRIDE; | 42 const IPC::Message& message) OVERRIDE; |
| 41 | 43 |
| 42 // Called on the file thread. | 44 // Called on the file thread. |
| 43 void CreateTemporaryFile(); | 45 void CreateTemporaryFile(); |
| 44 | 46 |
| 45 // Called on the UI thread. | 47 // Called on the UI thread. |
| 46 void TemporaryFileCreated(bool success); | 48 void TemporaryFileCreated(bool success); |
| 47 void ReturnFailure(const std::string& error); | 49 void ReturnFailure(const std::string& error); |
| 48 void ReturnSuccess(int64 file_size); | 50 void ReturnSuccess(int64 file_size); |
| 49 | 51 |
| 50 // Callback called once the MHTML generation is done. | 52 // Callback called once the MHTML generation is done. |
| 51 void MHTMLGenerated(const FilePath& file_path, int64 mhtml_file_size); | 53 void MHTMLGenerated(const base::FilePath& file_path, int64 mhtml_file_size); |
| 52 | 54 |
| 53 // Returns the WebContents we are associated with, NULL if it's been closed. | 55 // Returns the WebContents we are associated with, NULL if it's been closed. |
| 54 content::WebContents* GetWebContents(); | 56 content::WebContents* GetWebContents(); |
| 55 | 57 |
| 56 scoped_ptr<extensions::api::page_capture::SaveAsMHTML::Params> params_; | 58 scoped_ptr<extensions::api::page_capture::SaveAsMHTML::Params> params_; |
| 57 | 59 |
| 58 // The path to the temporary file containing the MHTML data. | 60 // The path to the temporary file containing the MHTML data. |
| 59 FilePath mhtml_path_; | 61 base::FilePath mhtml_path_; |
| 60 | 62 |
| 61 // The file containing the MHTML. | 63 // The file containing the MHTML. |
| 62 scoped_refptr<webkit_blob::ShareableFileReference> mhtml_file_; | 64 scoped_refptr<webkit_blob::ShareableFileReference> mhtml_file_; |
| 63 | 65 |
| 64 DECLARE_EXTENSION_FUNCTION("pageCapture.saveAsMHTML", PAGECAPTURE_SAVEASMHTML) | 66 DECLARE_EXTENSION_FUNCTION("pageCapture.saveAsMHTML", PAGECAPTURE_SAVEASMHTML) |
| 65 }; | 67 }; |
| 66 | 68 |
| 67 } // namespace extensions | 69 } // namespace extensions |
| 68 | 70 |
| 69 #endif // CHROME_BROWSER_EXTENSIONS_API_PAGE_CAPTURE_PAGE_CAPTURE_API_H_ | 71 #endif // CHROME_BROWSER_EXTENSIONS_API_PAGE_CAPTURE_PAGE_CAPTURE_API_H_ |
| OLD | NEW |