1// Copyright 2013 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_EXTENSIONS_API_MDNS_MDNS_API_H_
6#define CHROME_BROWSER_EXTENSIONS_API_MDNS_MDNS_API_H_
7
8#include <set>
9#include <string>
10
11#include "base/memory/scoped_ptr.h"
12#include "base/threading/thread_checker.h"
13#include "chrome/browser/extensions/api/mdns/dns_sd_registry.h"
14#include "extensions/browser/browser_context_keyed_api_factory.h"
15#include "extensions/browser/event_router.h"
16
17namespace content {
18class BrowserContext;
19}
20
21namespace extensions {
22
23class DnsSdRegistry;
24
25// MDnsAPI is instantiated with the profile and will listen for extensions that
26// register listeners for the chrome.mdns extension API. It will use a registry
27// class to start the mDNS listener process (if necessary) and observe new
28// service events to dispatch them to registered extensions.
29class MDnsAPI : public BrowserContextKeyedAPI,
30                public EventRouter::Observer,
31                public DnsSdRegistry::DnsSdObserver {
32 public:
33  explicit MDnsAPI(content::BrowserContext* context);
34  virtual ~MDnsAPI();
35
36  static MDnsAPI* Get(content::BrowserContext* context);
37
38  // BrowserContextKeyedAPI implementation.
39  static BrowserContextKeyedAPIFactory<MDnsAPI>* GetFactoryInstance();
40
41  // Used to mock out the DnsSdRegistry for testing.
42  void SetDnsSdRegistryForTesting(scoped_ptr<DnsSdRegistry> registry);
43
44 protected:
45  // Retrieve an instance of the registry. Lazily created when needed.
46  virtual DnsSdRegistry* dns_sd_registry();
47
48 private:
49  friend class BrowserContextKeyedAPIFactory<MDnsAPI>;
50
51  // EventRouter::Observer:
52  virtual void OnListenerAdded(const EventListenerInfo& details) OVERRIDE;
53  virtual void OnListenerRemoved(const EventListenerInfo& details) OVERRIDE;
54
55  // DnsSdRegistry::Observer
56  virtual void OnDnsSdEvent(
57      const std::string& service_type,
58      const DnsSdRegistry::DnsSdServiceList& services) OVERRIDE;
59
60  // BrowserContextKeyedAPI implementation.
61  static const char* service_name() {
62    return "MDnsAPI";
63  }
64
65  static const bool kServiceIsCreatedWithBrowserContext = true;
66  static const bool kServiceIsNULLWhileTesting = true;
67
68  // Update the current list of service types and update the registry.
69  void UpdateMDnsListeners(const EventListenerInfo& details);
70
71  // Ensure methods are only called on UI thread.
72  base::ThreadChecker thread_checker_;
73  content::BrowserContext* const browser_context_;
74  // Lazily created on first access and destroyed with this API class.
75  scoped_ptr<DnsSdRegistry> dns_sd_registry_;
76  // Current set of service types registered with the registry.
77  std::set<std::string> service_types_;
78
79  DISALLOW_COPY_AND_ASSIGN(MDnsAPI);
80};
81
82}  // namespace extensions
83
84#endif  // CHROME_BROWSER_EXTENSIONS_API_MDNS_MDNS_API_H_
85