Hooking MHTML generation to the browser.

content/browser/renderer_host/render_message_filter.cc

Index: content/browser/renderer_host/render_message_filter.cc
diff --git a/content/browser/renderer_host/render_message_filter.cc b/content/browser/renderer_host/render_message_filter.cc
index a20698aa32433f949aa17b4b532c80b001a648b4..d5b5c5217c9ea4c4adcfbda672ab9b51fd9e8482 100644
--- a/content/browser/renderer_host/render_message_filter.cc
+++ b/content/browser/renderer_host/render_message_filter.cc
@@ -22,6 +22,7 @@
 #include "content/browser/browser_thread.h"
 #include "content/browser/child_process_security_policy.h"
 #include "content/browser/content_browser_client.h"
+#include "chrome/browser/download/mhtml_generation_manager.h"

[jam, 2011/06/09 22:32:03]

you'll get a DEPS failure for this, since you can't include chrome code from
content (we're removing all existing includes, and are down to < 30.  adding
ones isn't allowed at this point).

the good news is that this is trivial to fix, just dispatch this message in
chrome/browser/renderer_host/chrome_render_message_filter.cc instead.  that's
where we filter chrome specific messages.

similarly, the IPC message, since it's chrome-specific, need to be in
chrome\common\render_messages.h instead of in content.

[Jay Civelli, 2011/06/10 22:38:22]

I knew I was doing something wrong!
Moved things around as you suggested.

[jam, 2011/06/10 23:41:31]

I don't see that?

 #include "content/browser/plugin_process_host.h"
 #include "content/browser/plugin_service.h"
 #include "content/browser/ppapi_plugin_process_host.h"
@@ -372,6 +373,7 @@ bool RenderMessageFilter::OnMessageReceived(const IPC::Message& message,
     IPC_MESSAGE_HANDLER(ViewHostMsg_EnableSpdy, OnEnableSpdy)
     IPC_MESSAGE_HANDLER_DELAY_REPLY(ViewHostMsg_Keygen, OnKeygen)
     IPC_MESSAGE_HANDLER(ViewHostMsg_AsyncOpenFile, OnAsyncOpenFile)
+    IPC_MESSAGE_HANDLER(ViewHostMsg_SavedPageAsMHTML, OnSavedPageAsMHTML)
     IPC_MESSAGE_UNHANDLED(handled = false)
   IPC_END_MESSAGE_MAP_EX()
 
@@ -870,3 +872,18 @@ void RenderMessageFilter::AsyncOpenFileOnFileThread(const FilePath& path,
       BrowserThread::IO, FROM_HERE, NewRunnableMethod(
           this, &RenderMessageFilter::Send, reply));
 }
+
+void RenderMessageFilter::OnSavedPageAsMHTML(int job_id, bool success) {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
+  BrowserThread::PostTask(
+      BrowserThread::UI, FROM_HERE, NewRunnableMethod(
+          this, &RenderMessageFilter::SavedPageAsMHTMLOnUIThread,
+          job_id, success));
+}
+
+void RenderMessageFilter::SavedPageAsMHTMLOnUIThread(int job_id, bool success) {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
+  g_browser_process->mhtml_generation_manager()->
+      MHTMLGenerated(job_id, success);
+
+}