[Media Router] Add DialMediaSinkService and DeviceDescriptionService

chrome/browser/media/router/discovery/dial/dial_media_sink_service.h

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_