| OLD | NEW |
| (Empty) |
| 1 // Copyright 2017 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #ifndef CHROME_BROWSER_MEDIA_ROUTER_DISCOVERY_DIAL_DIAL_MEDIA_SINK_SERVICE_H_ | |
| 6 #define CHROME_BROWSER_MEDIA_ROUTER_DISCOVERY_DIAL_DIAL_MEDIA_SINK_SERVICE_H_ | |
| 7 | |
| 8 #include <memory> | |
| 9 #include <set> | |
| 10 | |
| 11 #include "chrome/browser/media/router/discovery/dial/device_description_service.
h" | |
| 12 #include "chrome/browser/media/router/discovery/dial/dial_registry.h" | |
| 13 #include "chrome/common/media_router/discovery/media_sink_internal.h" | |
| 14 #include "chrome/common/media_router/discovery/media_sink_service.h" | |
| 15 | |
| 16 namespace media_router { | |
| 17 | |
| 18 class DeviceDescriptionService; | |
| 19 class DialRegistry; | |
| 20 | |
| 21 // A service which can be used to start background discovery and resolution of | |
| 22 // DIAL devices (Smart TVs, Game Consoles, etc.). | |
| 23 // This class is not thread safe. All methods must be called from the IO thread. | |
| 24 class DialMediaSinkService : public MediaSinkService, | |
| 25 public DialRegistry::Observer { | |
| 26 public: | |
| 27 DialMediaSinkService(const OnSinksDiscoveredCallback& callback, | |
| 28 net::URLRequestContextGetter* request_context); | |
| 29 ~DialMediaSinkService() override; | |
| 30 | |
| 31 // Stops listening for DIAL device events. | |
| 32 virtual void Stop(); | |
| 33 | |
| 34 // MediaSinkService implementation | |
| 35 void Start() override; | |
| 36 | |
| 37 protected: | |
| 38 | |
| 39 // Returns instance of device description service. Create a new one if none | |
| 40 // exists. | |
| 41 DeviceDescriptionService* GetDescriptionService(); | |
| 42 | |
| 43 // Does not take ownership of |dial_registry|. | |
| 44 void SetDialRegistryForTest(DialRegistry* dial_registry); | |
| 45 void SetDescriptionServiceForTest( | |
| 46 std::unique_ptr<DeviceDescriptionService> description_service); | |
| 47 void SetTimerForTest(std::unique_ptr<base::Timer> timer); | |
| 48 | |
| 49 private: | |
| 50 friend class DialMediaSinkServiceTest; | |
| 51 FRIEND_TEST_ALL_PREFIXES(DialMediaSinkServiceTest, TestStart); | |
| 52 FRIEND_TEST_ALL_PREFIXES(DialMediaSinkServiceTest, TestTimer); | |
| 53 FRIEND_TEST_ALL_PREFIXES(DialMediaSinkServiceTest, | |
| 54 TestFetchCompleted_SameSink); | |
| 55 FRIEND_TEST_ALL_PREFIXES(DialMediaSinkServiceTest, TestIsDifferent); | |
| 56 FRIEND_TEST_ALL_PREFIXES(DialMediaSinkServiceTest, | |
| 57 TestOnDeviceDescriptionAvailable); | |
| 58 | |
| 59 // api::dial::DialRegistry::Observer implementation | |
| 60 void OnDialDeviceEvent(const DialRegistry::DeviceList& devices) override; | |
| 61 void OnDialError(DialRegistry::DialErrorCode type) override; | |
| 62 | |
| 63 // Called when description service successfully fetches and parses device | |
| 64 // description XML. Restart |finish_timer_| if it is not running. | |
| 65 void OnDeviceDescriptionAvailable( | |
| 66 const DialDeviceData& device_data, | |
| 67 const ParsedDialDeviceDescription& description_data); | |
| 68 | |
| 69 // Called when fails to fetch or parse device description XML. | |
| 70 void OnDeviceDescriptionError(const DialDeviceData& device, | |
| 71 const std::string& error_message); | |
| 72 | |
| 73 // Called when |finish_timer_| expires. | |
| 74 void OnFetchCompleted(); | |
| 75 | |
| 76 // Helper function to start |finish_timer_|. | |
| 77 void StartTimer(); | |
| 78 | |
| 79 // Timer for finishing fetching. Starts in |OnDialDeviceEvent()|, and expires | |
| 80 // 3 seconds later. If |OnDeviceDescriptionAvailable()| is called after | |
| 81 // |finish_timer_| expires, |finish_timer_| is restarted. | |
| 82 std::unique_ptr<base::Timer> finish_timer_; | |
| 83 | |
| 84 std::unique_ptr<DeviceDescriptionService> description_service_; | |
| 85 | |
| 86 // Raw pointer to DialRegistry singleton. | |
| 87 DialRegistry* dial_registry_ = nullptr; | |
| 88 | |
| 89 // Sorted sinks from current round of discovery. | |
| 90 std::set<MediaSinkInternal> current_sinks_; | |
| 91 | |
| 92 // Sorted sinks sent to Media Router Provider in last FetchCompleted(). | |
| 93 std::set<MediaSinkInternal> mrp_sinks_; | |
| 94 | |
| 95 // Device data list from current round of discovery. | |
| 96 DialRegistry::DeviceList current_devices_; | |
| 97 | |
| 98 scoped_refptr<net::URLRequestContextGetter> request_context_; | |
| 99 }; | |
| 100 | |
| 101 } // namespace media_router | |
| 102 | |
| 103 #endif // CHROME_BROWSER_MEDIA_ROUTER_DISCOVERY_DIAL_DIAL_MEDIA_SINK_SERVICE_H_ | |
| OLD | NEW |