1 files changed, 81 insertions, 0 deletions
diff --git a/chrome/browser/download/mhtml_generation_manager.h b/chrome/browser/download/mhtml_generation_manager.h
new file mode 100644
index 0000000..8e8557f
--- /dev/null
+++ b/chrome/browser/download/mhtml_generation_manager.h
@@ -0,0 +1,81 @@
+// 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_MHTML_GENERATION_MANAGER_H_
+#define CHROME_BROWSER_DOWNLOAD_MHTML_GENERATION_MANAGER_H_
+
+#include <map>
+
+#include "base/memory/ref_counted.h"
+#include "base/platform_file.h"
+#include "base/process.h"
+#include "content/browser/browser_thread.h"
+#include "ipc/ipc_platform_file.h"
+
+class FilePath;
+class TabContents;
+
+class MHTMLGenerationManager
+    : public base::RefCountedThreadSafe<MHTMLGenerationManager,
+                                        BrowserThread::DeleteOnUIThread> {
+ public:
+  MHTMLGenerationManager();
+  ~MHTMLGenerationManager();
+
+  // Instructs the render view to generate a MHTML representation of the current
+  // page for |tab_contents|.
+  void GenerateMHTML(TabContents* tab_contents, const FilePath& file);
+
+  // Notification from the renderer that the MHTML generation succeeded/failed.
+  void MHTMLGenerated(int job_id, bool success);
+
+  // The details sent along with the MHTML_GENERATED notification.
+  struct NotificationDetails {
+    FilePath file_path;
+    bool success;
+  };
+
+ private:
+  struct Job{
+    Job();
+
+    FilePath file_path;
+
+    // The handles to file the MHTML is saved to, for the browser and renderer
+    // processes.
+    base::PlatformFile browser_file;
+    IPC::PlatformFileForTransit renderer_file;
+
+    // The IDs mapping to a specific tab.
+    int process_id;
+    int routing_id;
+  };
+
+  // Called on the file thread to create |file|.
+  void CreateFile(int job_id,
+                  const FilePath& file,
+                  base::ProcessHandle renderer_process);
+
+  // Called on the UI thread when the file that should hold the MHTML data has
+  // been created.  This returns a handle to that file for the browser process
+  // and one for the renderer process. These handles are
+  // kInvalidPlatformFileValue if the file could not be opened.
+  void FileCreated(int job_id,
+                   base::PlatformFile browser_file,
+                   IPC::PlatformFileForTransit renderer_file);
+
+  // Called on the file thread to close the file the MHTML was saved to.
+  void CloseFile(base::PlatformFile file);
+
+  // Called on the UI thread when a job has been processed (successfully or
+  // not).  Closes the file and removes the job from the job map.
+  void JobFinished(int job_id, bool success);
+
+  typedef std::map<int, Job> IDToJobMap;
+  IDToJobMap id_to_job_;
+
+  DISALLOW_COPY_AND_ASSIGN(MHTMLGenerationManager);
+};
+
+#endif  // CHROME_BROWSER_DOWNLOAD_MHTML_GENERATION_MANAGER_H_