Add PDF extension support (issue #1565)

libcef/browser/extensions/api/streams_private/streams_private_api.h

@@ -0,0 +1,104 @@
+// 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.
+
+#ifndef CEF_LIBCEF_BROWSER_EXTENSIONS_API_STREAMS_PRIVATE_STREAMS_PRIVATE_API_H_
+#define CEF_LIBCEF_BROWSER_EXTENSIONS_API_STREAMS_PRIVATE_STREAMS_PRIVATE_API_H_
+
+#include <map>
+#include <string>
+
+#include "base/scoped_observer.h"
+#include "extensions/browser/browser_context_keyed_api_factory.h"
+#include "extensions/browser/extension_function.h"
+#include "extensions/browser/extension_registry_observer.h"
+
+namespace content {
+class BrowserContext;
+class StreamHandle;
+struct StreamInfo;
+}
+
+namespace extensions {
+class ExtensionRegistry;
+
+class StreamsPrivateAPI : public BrowserContextKeyedAPI,
+                          public ExtensionRegistryObserver {
+ public:
+  // Convenience method to get the StreamsPrivateAPI for a BrowserContext.
+  static StreamsPrivateAPI* Get(content::BrowserContext* context);
+
+  explicit StreamsPrivateAPI(content::BrowserContext* context);
+  ~StreamsPrivateAPI() override;
+
+  // Send the onExecuteMimeTypeHandler event to |extension_id|.
+  // |tab_id| is used to determine the tabId where the document is being
+  // opened. The data for the document will be readable from |stream|, and
+  // should be |expected_content_size| bytes long. If the viewer is being opened
+  // in a BrowserPlugin, specify a non-empty |view_id| of the plugin. |embedded|
+  // should be set to whether the document is embedded within another document.
+  void ExecuteMimeTypeHandler(const std::string& extension_id,
+                              int tab_id,
+                              scoped_ptr<content::StreamInfo> stream,
+                              const std::string& view_id,
+                              int64 expected_content_size,
+                              bool embedded,
+                              int render_process_id,
+                              int render_frame_id);
+
+  void AbortStream(const std::string& extension_id,
+                   const GURL& stream_url,
+                   const base::Closure& callback);
+
+  // BrowserContextKeyedAPI implementation.
+  static BrowserContextKeyedAPIFactory<StreamsPrivateAPI>* GetFactoryInstance();
+
+ private:
+  friend class BrowserContextKeyedAPIFactory<StreamsPrivateAPI>;
+  typedef std::map<std::string,
+                   std::map<GURL,
+                            linked_ptr<content::StreamHandle> > > StreamMap;
+
+  // ExtensionRegistryObserver implementation.
+  void OnExtensionUnloaded(content::BrowserContext* browser_context,
+                           const Extension* extension,
+                           UnloadedExtensionInfo::Reason reason) override;
+
+  // BrowserContextKeyedAPI implementation.
+  static const char* service_name() {
+    return "StreamsPrivateAPI";
+  }
+  static const bool kServiceIsNULLWhileTesting = true;
+  static const bool kServiceRedirectedInIncognito = true;
+
+  content::BrowserContext* const browser_context_;
+  StreamMap streams_;
+
+  // Listen to extension unloaded notifications.
+  ScopedObserver<ExtensionRegistry, ExtensionRegistryObserver>
+      extension_registry_observer_;
+
+  base::WeakPtrFactory<StreamsPrivateAPI> weak_ptr_factory_;
+
+};
+
+class StreamsPrivateAbortFunction : public UIThreadExtensionFunction {
+ public:
+  StreamsPrivateAbortFunction();
+  DECLARE_EXTENSION_FUNCTION("streamsPrivate.abort", STREAMSPRIVATE_ABORT)
+
+ protected:
+  ~StreamsPrivateAbortFunction() override {}
+
+  // ExtensionFunction:
+  ExtensionFunction::ResponseAction Run() override;
+
+ private:
+  void OnClose();
+
+  std::string stream_url_;
+};
+
+}  // namespace extensions
+
+#endif  // CEF_LIBCEF_BROWSER_EXTENSIONS_API_STREAMS_PRIVATE_STREAMS_PRIVATE_API_H_