OOP PDF: Change the save toolbar button to match the save menu behavior.

extensions/browser/guest_view/mime_handler_view/mime_handler_view_guest.cc

 // Copyright 2014 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.
 
 #include "extensions/browser/guest_view/mime_handler_view/mime_handler_view_guest.h"
 
 #include "base/strings/stringprintf.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/render_process_host.h"
 #include "content/public/common/url_constants.h"
@@ skipping 60 matching lines @@
     const GURL& embedder_site_url,
     const base::DictionaryValue& create_params,
     const WebContentsCreatedCallback& callback) {
   std::string orig_mime_type;
   create_params.GetString(mime_handler_view::kMimeType, &orig_mime_type);
   DCHECK(!orig_mime_type.empty());
 
   std::string extension_src;
   create_params.GetString(mime_handler_view::kSrc, &extension_src);
   DCHECK(!extension_src.empty());
+  std::string content_url_str;
+  create_params.GetString(mime_handler_view::kContentUrl, &content_url_str);
+  content_url_ = GURL(content_url_str);
+  if (!content_url_.is_valid()) {
+    content_url_ = GURL();
+    callback.Run(nullptr);
+    return;
+  }
 
   GURL mime_handler_extension_url(extension_src);
   if (!mime_handler_extension_url.is_valid()) {
     callback.Run(NULL);
     return;
   }
 
   const Extension* mime_handler_extension =
       // TODO(lazyboy): Do we need handle the case where the extension is
       // terminated (ExtensionRegistry::TERMINATED)?
@@ skipping 86 matching lines @@
     return;
 
   embedder_web_contents()->GetDelegate()->FindReply(embedder_web_contents(),
                                                     request_id,
                                                     number_of_matches,
                                                     selection_rect,
                                                     active_match_ordinal,
                                                     final_update);
 }
 
+bool MimeHandlerViewGuest::SaveFrame(const GURL& url,
+                                     const content::Referrer& referrer) {
+  if (!attached())
+    return false;
+
+  embedder_web_contents()->SaveFrame(content_url_, referrer);
+  return true;
+}
+
 bool MimeHandlerViewGuest::OnMessageReceived(const IPC::Message& message) {
   bool handled = true;
   IPC_BEGIN_MESSAGE_MAP(MimeHandlerViewGuest, message)
     IPC_MESSAGE_HANDLER(ExtensionHostMsg_Request, OnRequest)
     IPC_MESSAGE_UNHANDLED(handled = false)
   IPC_END_MESSAGE_MAP()
   return handled;
 }
 
 void MimeHandlerViewGuest::OnRequest(
     const ExtensionHostMsg_Request_Params& params) {
   if (extension_function_dispatcher_) {
     extension_function_dispatcher_->Dispatch(
         params, web_contents()->GetRenderViewHost());
   }
 }
 
 }  // namespace extensions