Add a mimeHandler extension API.

chrome/browser/extensions/api/streams_private/streams_private_api.cc

 // Copyright (c) 2012 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 "chrome/browser/extensions/api/streams_private/streams_private_api.h"
 
 #include "base/lazy_instance.h"
 #include "base/values.h"
 #include "chrome/browser/extensions/extension_tab_util.h"
 #include "chrome/common/extensions/api/streams_private.h"
 #include "chrome/common/extensions/manifest_handlers/mime_types_handler.h"
 #include "content/public/browser/stream_handle.h"
 #include "content/public/browser/stream_info.h"
 #include "extensions/browser/event_router.h"
 #include "extensions/browser/extension_function_registry.h"
 #include "extensions/browser/extension_registry.h"
-#include "extensions/browser/guest_view/guest_view_manager.h"
 #include "extensions/browser/guest_view/mime_handler_view/mime_handler_stream_manager.h"
 #include "extensions/browser/guest_view/mime_handler_view/mime_handler_view_guest.h"
 #include "net/http/http_response_headers.h"
 
 namespace extensions {
 namespace {
 
 void CreateResponseHeadersDictionary(const net::HttpResponseHeaders* headers,
                                      base::DictionaryValue* result) {
   if (!headers)
     return;
 
   void* iter = NULL;
   std::string header_name;
   std::string header_value;
   while (headers->EnumerateHeaderLines(&iter, &header_name, &header_value)) {
     base::Value* existing_value = NULL;
     if (result->Get(header_name, &existing_value)) {
       base::StringValue* existing_string_value =
           static_cast<base::StringValue*>(existing_value);
       existing_string_value->GetString()->append(", ").append(header_value);
     } else {
       result->SetString(header_name, header_value);
     }
   }
 }
 
-// If |guest_web_contents| has a MimeHandlerViewGuest with view id of |view_id|,
-// abort it. Returns true if the MimeHandlerViewGuest has a matching view id.
-bool MaybeAbortStreamInGuest(const std::string& view_id,
-                             content::WebContents* guest_web_contents) {
-  auto guest = MimeHandlerViewGuest::FromWebContents(guest_web_contents);
-  if (!guest)
-    return false;
-  if (guest->view_id() != view_id)
-    return false;
-  auto stream = guest->GetStream();
-  if (stream)
-    stream->Abort();
-  return true;
-}
-
 }  // namespace
 
 namespace streams_private = api::streams_private;
 
 // static
 StreamsPrivateAPI* StreamsPrivateAPI::Get(content::BrowserContext* context) {
   return GetFactoryInstance()->Get(context);
 }
 
 StreamsPrivateAPI::StreamsPrivateAPI(content::BrowserContext* context)
@@ skipping 14 matching lines @@
     int64 expected_content_size,
     bool embedded,
     int render_process_id,
     int render_frame_id) {
   const Extension* extension = ExtensionRegistry::Get(browser_context_)
                                    ->enabled_extensions()
                                    .GetByID(extension_id);
   if (!extension)
     return;
 
+  MimeTypesHandler* handler = MimeTypesHandler::GetHandler(extension);
+  // If the mime handler uses MimeHandlerViewGuest, the MimeHandlerViewGuest
+  // will take ownership of the stream. Otherwise, store the stream handle in
+  // |streams_| and fire an event notifying the extension.
+  if (!handler->handler_url().empty()) {
+    GURL handler_url(Extension::GetBaseURLFromExtensionId(extension_id).spec() +
+                     handler->handler_url());
+    MimeHandlerStreamManager::Get(browser_context_)
+        ->AddStream(
+            view_id,
+            make_scoped_ptr(new StreamContainer(
+                stream.Pass(), ExtensionTabUtil::GetTabId(web_contents),
+                expected_content_size, embedded, handler_url, extension_id)),
+            render_process_id, render_frame_id);

[raymes, 2015/01/12 05:26:04]

Can we split this up into a few statements?? :)

[Sam McNally, 2015/01/12 07:13:35]

Done.

+    return;
+  }
   // Create the event's arguments value.
   streams_private::StreamInfo info;
   info.mime_type = stream->mime_type;
   info.original_url = stream->original_url.spec();
   info.stream_url = stream->handle->GetURL().spec();
   info.tab_id = ExtensionTabUtil::GetTabId(web_contents);
   info.embedded = embedded;
 
   if (!view_id.empty()) {
     info.view_id.reset(new std::string(view_id));
   }
 
   int size = -1;
   if (expected_content_size <= INT_MAX)
     size = expected_content_size;
   info.expected_content_size = size;
 
   CreateResponseHeadersDictionary(stream->response_headers.get(),
                                   &info.response_headers.additional_properties);
 
   scoped_ptr<Event> event(
       new Event(streams_private::OnExecuteMimeTypeHandler::kEventName,
                 streams_private::OnExecuteMimeTypeHandler::Create(info)));
 
   EventRouter::Get(browser_context_)
       ->DispatchEventToExtension(extension_id, event.Pass());
-  MimeTypesHandler* handler = MimeTypesHandler::GetHandler(extension);
-  GURL url = stream->handle->GetURL();
-  // If the mime handler uses MimeHandlerViewGuest, the MimeHandlerViewGuest
-  // will take ownership of the stream. Otherwise, store the stream handle in
-  // |streams_|.
-  if (handler->handler_url().empty()) {
-    streams_[extension_id][url] = make_linked_ptr(stream->handle.release());

[raymes, 2015/01/12 05:26:04]

I think we might need this.

[Sam McNally, 2015/01/12 07:13:35]

Done.

-    return;
-  }
 
-  GURL handler_url(Extension::GetBaseURLFromExtensionId(extension_id).spec() +
-                   handler->handler_url() + "?id=" + view_id);
-  MimeHandlerStreamManager::Get(browser_context_)
-      ->AddStream(view_id, make_scoped_ptr(new StreamContainer(
-                               stream.Pass(), handler_url, extension_id)),
-                  render_process_id, render_frame_id);
-  // If the mime handler uses MimeHandlerViewGuest, we need to be able to look
-  // up the MimeHandlerViewGuest instance that is handling the streamed
-  // resource in order to abort the stream. The embedding WebContents and the
-  // view id are necessary to perform that lookup.
-  mime_handler_streams_[extension_id][url] =
-      std::make_pair(web_contents, view_id);
 }
 
 void StreamsPrivateAPI::AbortStream(const std::string& extension_id,
                                     const GURL& stream_url,
                                     const base::Closure& callback) {
-  auto streams = mime_handler_streams_.find(extension_id);
-  if (streams != mime_handler_streams_.end()) {
-    auto stream_info = streams->second.find(stream_url);
-    if (stream_info != streams->second.end()) {
-      auto stream = MimeHandlerStreamManager::Get(browser_context_)
-                        ->GetStream(stream_info->second.second);
-      // If the mime handler uses MimeHandlerViewGuest, the stream will either
-      // be owned by the particular MimeHandlerViewGuest if it has been created,
-      // or by the MimeHandleStreamManager, otherwise.
-      if (!stream) {
-        GuestViewManager::FromBrowserContext(browser_context_)
-            ->ForEachGuest(stream_info->second.first,
-                           base::Bind(&MaybeAbortStreamInGuest,
-                                      stream_info->second.second));
-      }
-      streams->second.erase(stream_info);
-      callback.Run();
-    }
-    return;
-  }
-
   StreamMap::iterator extension_it = streams_.find(extension_id);
   if (extension_it == streams_.end()) {
     callback.Run();
     return;
   }
 
   StreamMap::mapped_type* url_map = &extension_it->second;
   StreamMap::mapped_type::iterator url_it = url_map->find(stream_url);
   if (url_it == url_map->end()) {
     callback.Run();
     return;
   }
 
   url_it->second->AddCloseListener(callback);
   url_map->erase(url_it);
 }
 
 void StreamsPrivateAPI::OnExtensionUnloaded(
     content::BrowserContext* browser_context,
     const Extension* extension,
     UnloadedExtensionInfo::Reason reason) {
   streams_.erase(extension->id());
-  mime_handler_streams_.erase(extension->id());
 }
 
 StreamsPrivateAbortFunction::StreamsPrivateAbortFunction() {
 }
 
 ExtensionFunction::ResponseAction StreamsPrivateAbortFunction::Run() {
   DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
   EXTENSION_FUNCTION_VALIDATE(args_->GetString(0, &stream_url_));
   StreamsPrivateAPI::Get(browser_context())->AbortStream(
       extension_id(), GURL(stream_url_), base::Bind(
           &StreamsPrivateAbortFunction::OnClose, this));
   return RespondLater();
 }
 
 void StreamsPrivateAbortFunction::OnClose() {
   DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
   Respond(NoArguments());
 }
 
 static base::LazyInstance<BrowserContextKeyedAPIFactory<StreamsPrivateAPI> >
     g_factory = LAZY_INSTANCE_INITIALIZER;
 
 // static
 BrowserContextKeyedAPIFactory<StreamsPrivateAPI>*
 StreamsPrivateAPI::GetFactoryInstance() {
   return g_factory.Pointer();
 }
 
 }  // namespace extensions