Index: chrome/browser/media/router/discovery/dial/dial_media_sink_service.h |
diff --git a/chrome/browser/media/router/discovery/dial/dial_media_sink_service.h b/chrome/browser/media/router/discovery/dial/dial_media_sink_service.h |
new file mode 100644 |
index 0000000000000000000000000000000000000000..86f72c5beba6118eb17052fe0f245a9b357fa052 |
--- /dev/null |
+++ b/chrome/browser/media/router/discovery/dial/dial_media_sink_service.h |
@@ -0,0 +1,89 @@ |
+// Copyright 2017 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 CHROME_BROWSER_MEDIA_ROUTER_DISCOVERY_DIAL_DIAL_MEDIA_SINK_SERVICE_H_ |
+#define CHROME_BROWSER_MEDIA_ROUTER_DISCOVERY_DIAL_DIAL_MEDIA_SINK_SERVICE_H_ |
+ |
+#include <memory> |
+#include <set> |
+ |
+#include "chrome/browser/media/router/discovery/dial/device_description_service.h" |
+#include "chrome/browser/media/router/discovery/dial/dial_registry.h" |
+#include "chrome/common/media_router/discovery/media_sink_internal.h" |
+#include "chrome/common/media_router/discovery/media_sink_service.h" |
+ |
+namespace media_router { |
+ |
+class DeviceDescriptionService; |
+class DialRegistry; |
+ |
+// A service which can be used to start background discovery and resolution of |
+// DIAL devices (Smart TVs, Game Consoles, etc.). |
+// This class is not thread safe. All methods must be called from the IO thread. |
+class DialMediaSinkService : public MediaSinkService, |
+ public DialRegistry::Observer { |
+ public: |
+ DialMediaSinkService(const OnSinksDiscoveredCallback& callback, |
+ net::URLRequestContextGetter* request_context); |
+ ~DialMediaSinkService() override; |
+ |
+ // MediaSinkService implementation |
+ void Start() override; |
+ |
+ void Stop(); |
+ |
+ protected: |
+ virtual DialRegistry* dial_registry(); |
+ |
+ // Returns instance of device description service. Create a new one if none |
+ // exists. |
+ virtual DeviceDescriptionService* GetDescriptionService(); |
+ |
+ private: |
+ friend class DialMediaSinkServiceTest; |
+ FRIEND_TEST_ALL_PREFIXES(DialMediaSinkServiceTest, TestStart); |
+ FRIEND_TEST_ALL_PREFIXES(DialMediaSinkServiceTest, TestFetchCompleted); |
+ FRIEND_TEST_ALL_PREFIXES(DialMediaSinkServiceTest, TestIsDifferent); |
+ FRIEND_TEST_ALL_PREFIXES(DialMediaSinkServiceTest, |
+ TestOnDeviceDescriptionAvailable); |
+ |
+ // api::dial::DialRegistry::Observer implementation |
+ void OnDialDeviceEvent(const DialRegistry::DeviceList& devices) override; |
+ void OnDialError(DialRegistry::DialErrorCode type) override; |
+ |
+ // Called when description service successfully fetches and parses device |
+ // description XML. Restart |finish_timer_| if it is not running. |
+ void OnDeviceDescriptionAvailable( |
+ const DialDeviceData& device_data, |
+ const ParsedDialDeviceDescription& description_data); |
+ |
+ // Called when fails to fetch or parse device description XML. |
+ void OnDeviceDescriptionError(const DialDeviceData& device, |
+ const std::string& error_message); |
+ |
+ // Called when |finish_timer_| expires. |
+ void OnFetchCompleted(); |
+ |
+ // Timer for finishing fetching. Starts in |OnDialDeviceEvent()|, and expires |
+ // 3 seconds later. If |OnDeviceDescriptionAvailable()| is called after |
+ // |finish_timer_| expires, |finish_timer_| is restarted. |
+ std::unique_ptr<base::OneShotTimer> finish_timer_; |
+ |
+ std::unique_ptr<DeviceDescriptionService> description_service_; |
+ |
+ // Sorted sinks from current round of discovery. |
+ std::set<MediaSinkInternal> current_sinks_; |
+ |
+ // Sorted sinks sent to Media Router Provider in last FetchCompleted(). |
+ std::set<MediaSinkInternal> mrp_sinks_; |
+ |
+ // Device data list from current round of discovery. |
+ DialRegistry::DeviceList current_devices_; |
+ |
+ scoped_refptr<net::URLRequestContextGetter> request_context_; |
+}; |
+ |
+} // namespace media_router |
+ |
+#endif // CHROME_BROWSER_MEDIA_ROUTER_DISCOVERY_DIAL_DIAL_MEDIA_SINK_SERVICE_H_ |