OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #ifndef CHROME_BROWSER_EXTENSIONS_API_TAB_CAPTURE_TAB_CAPTURE_REGISTRY_H_ | 5 #ifndef CHROME_BROWSER_EXTENSIONS_API_TAB_CAPTURE_TAB_CAPTURE_REGISTRY_H_ |
6 #define CHROME_BROWSER_EXTENSIONS_API_TAB_CAPTURE_TAB_CAPTURE_REGISTRY_H_ | 6 #define CHROME_BROWSER_EXTENSIONS_API_TAB_CAPTURE_TAB_CAPTURE_REGISTRY_H_ |
7 | 7 |
8 #include <map> | 8 #include <map> |
| 9 #include <string> |
| 10 #include <utility> |
| 11 #include <vector> |
9 | 12 |
10 #include "chrome/browser/media/media_internals.h" | 13 #include "chrome/browser/media/media_internals.h" |
11 #include "chrome/browser/media/media_internals_observer.h" | 14 #include "chrome/browser/media/media_internals_observer.h" |
12 #include "chrome/browser/profiles/profile_keyed_service.h" | 15 #include "chrome/browser/profiles/profile_keyed_service.h" |
13 #include "chrome/common/extensions/api/tab_capture.h" | 16 #include "chrome/common/extensions/api/tab_capture.h" |
14 #include "content/public/browser/browser_thread.h" | 17 #include "content/public/browser/browser_thread.h" |
15 #include "content/public/browser/media_request_state.h" | 18 #include "content/public/browser/media_request_state.h" |
16 #include "content/public/browser/notification_observer.h" | 19 #include "content/public/browser/notification_observer.h" |
17 #include "content/public/browser/notification_registrar.h" | 20 #include "content/public/browser/notification_registrar.h" |
18 #include "content/public/common/media_stream_request.h" | 21 #include "content/public/common/media_stream_request.h" |
19 | 22 |
20 class Profile; | 23 class Profile; |
21 | 24 |
22 namespace extensions { | 25 namespace extensions { |
23 | 26 |
24 namespace tab_capture = extensions::api::tab_capture; | 27 namespace tab_capture = extensions::api::tab_capture; |
25 | 28 |
26 class TabCaptureRegistry : public ProfileKeyedService, | 29 class TabCaptureRegistry : public ProfileKeyedService, |
27 public content::NotificationObserver { | 30 public content::NotificationObserver { |
28 public: | 31 public: |
29 struct TabCaptureRequest { | 32 struct TabCaptureRequest { |
30 std::string extension_id; | 33 std::string extension_id; |
31 int tab_id; | 34 int tab_id; |
32 tab_capture::TabCaptureState status; | 35 tab_capture::TabCaptureState status; |
33 | 36 |
34 TabCaptureRequest() {} | 37 TabCaptureRequest() : tab_id(-1) {} |
35 TabCaptureRequest(std::string extension_id, int tab_id, | 38 TabCaptureRequest(std::string extension_id, int tab_id, |
36 tab_capture::TabCaptureState status) | 39 tab_capture::TabCaptureState status) |
37 : extension_id(extension_id), tab_id(tab_id), status(status) {} | 40 : extension_id(extension_id), tab_id(tab_id), status(status) {} |
38 }; | 41 }; |
39 typedef std::vector<TabCaptureRequest> CaptureRequestList; | 42 typedef std::vector<TabCaptureRequest> CaptureRequestList; |
40 | 43 |
41 explicit TabCaptureRegistry(Profile* profile); | 44 explicit TabCaptureRegistry(Profile* profile); |
42 | 45 |
43 const CaptureRequestList GetCapturedTabs(const std::string& extension_id); | 46 const CaptureRequestList GetCapturedTabs(const std::string& extension_id); |
44 bool AddRequest(const std::string& key, const TabCaptureRequest& request); | 47 bool AddRequest(const std::pair<int, int>, const TabCaptureRequest& request); |
45 bool VerifyRequest(const std::string& key); | 48 bool VerifyRequest(int render_process_id, int render_view_id); |
46 | 49 |
47 private: | 50 private: |
48 // Maps device_id to information about the media stream request. This is | 51 // Maps device_id to information about the media stream request. This is |
49 // expected to be small since maintaining a media stream is expensive. | 52 // expected to be small since maintaining a media stream is expensive. |
50 typedef std::map<std::string, TabCaptureRequest> DeviceCaptureRequestMap; | 53 typedef std::map<const std::pair<int, int>, TabCaptureRequest> |
| 54 DeviceCaptureRequestMap; |
51 | 55 |
52 class MediaObserverProxy : public MediaInternalsObserver, | 56 class MediaObserverProxy : public MediaInternalsObserver, |
53 public base::RefCountedThreadSafe< | 57 public base::RefCountedThreadSafe< |
54 MediaObserverProxy> { | 58 MediaObserverProxy> { |
55 public: | 59 public: |
56 MediaObserverProxy() : handler_(NULL) {} | 60 MediaObserverProxy() : handler_(NULL) {} |
57 void Attach(TabCaptureRegistry* handler); | 61 void Attach(TabCaptureRegistry* handler); |
58 void Detach(); | 62 void Detach(); |
59 | 63 |
60 private: | 64 private: |
61 friend class base::RefCountedThreadSafe<MediaObserverProxy>; | 65 friend class base::RefCountedThreadSafe<MediaObserverProxy>; |
62 virtual ~MediaObserverProxy() {} | 66 virtual ~MediaObserverProxy() {} |
63 | 67 |
64 // MediaInternalsObserver. | 68 // MediaInternalsObserver. |
65 virtual void OnRequestUpdate( | 69 virtual void OnRequestUpdate( |
| 70 int render_process_id, |
| 71 int render_view_id, |
66 const content::MediaStreamDevice& device, | 72 const content::MediaStreamDevice& device, |
67 const content::MediaRequestState state) OVERRIDE; | 73 const content::MediaRequestState state) OVERRIDE; |
68 | 74 |
69 void RegisterAsMediaObserverOnIOThread(bool unregister); | 75 void RegisterAsMediaObserverOnIOThread(bool unregister); |
70 void UpdateOnUIThread( | 76 void UpdateOnUIThread( |
| 77 int render_process_id, |
| 78 int render_view_id, |
71 const content::MediaStreamDevice& device, | 79 const content::MediaStreamDevice& device, |
72 const content::MediaRequestState new_state); | 80 const content::MediaRequestState new_state); |
73 | 81 |
74 TabCaptureRegistry* handler_; | 82 TabCaptureRegistry* handler_; |
75 }; | 83 }; |
76 | 84 |
77 virtual ~TabCaptureRegistry(); | 85 virtual ~TabCaptureRegistry(); |
78 | 86 |
79 void HandleRequestUpdateOnUIThread( | 87 void HandleRequestUpdateOnUIThread( |
| 88 int render_process_id, |
| 89 int render_view_id, |
80 const content::MediaStreamDevice& device, | 90 const content::MediaStreamDevice& device, |
81 const content::MediaRequestState state); | 91 const content::MediaRequestState state); |
82 | 92 |
83 // content::NotificationObserver implementation. | 93 // content::NotificationObserver implementation. |
84 virtual void Observe(int type, | 94 virtual void Observe(int type, |
85 const content::NotificationSource& source, | 95 const content::NotificationSource& source, |
86 const content::NotificationDetails& details) OVERRIDE; | 96 const content::NotificationDetails& details) OVERRIDE; |
87 | 97 |
88 scoped_refptr<MediaObserverProxy> proxy_; | 98 scoped_refptr<MediaObserverProxy> proxy_; |
89 content::NotificationRegistrar registrar_; | 99 content::NotificationRegistrar registrar_; |
90 Profile* const profile_; | 100 Profile* const profile_; |
91 DeviceCaptureRequestMap requests_; | 101 DeviceCaptureRequestMap requests_; |
92 | 102 |
93 DISALLOW_COPY_AND_ASSIGN(TabCaptureRegistry); | 103 DISALLOW_COPY_AND_ASSIGN(TabCaptureRegistry); |
94 }; | 104 }; |
95 | 105 |
96 } // namespace extension | 106 } // namespace extension |
97 | 107 |
98 #endif // CHROME_BROWSER_EXTENSIONS_API_TAB_CAPTURE_TAB_CAPTURE_REGISTRY_H_ | 108 #endif // CHROME_BROWSER_EXTENSIONS_API_TAB_CAPTURE_TAB_CAPTURE_REGISTRY_H_ |
OLD | NEW |