chrome/browser/extensions/blacklist_state_fetcher.h

   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_BLACKLIST_STATE_FETCHER_H_
   6 #define CHROME_BROWSER_EXTENSIONS_BLACKLIST_STATE_FETCHER_H_
   7
   8 #include <set>
   9 #include <string>
  10
  11 #include "base/callback.h"
  12 #include "base/memory/scoped_ptr.h"
  13 #include "base/memory/weak_ptr.h"
  14 #include "chrome/browser/safe_browsing/protocol_manager_helper.h"
  15 #include "extensions/browser/blacklist_state.h"
  16 #include "net/url_request/url_fetcher.h"
  17 #include "net/url_request/url_fetcher_delegate.h"
  18
  19 namespace net {
  20 class URLRequestContextGetter;
  21 }
  22
  23 namespace extensions {
  24
  25 class BlacklistStateFetcher : public net::URLFetcherDelegate {
  26  public:
  27   typedef base::Callback<void(BlacklistState)> RequestCallback;
  28
  29   BlacklistStateFetcher();
  30
  31   ~BlacklistStateFetcher() override;
  32
  33   virtual void Request(const std::string& id, const RequestCallback& callback);
  34
  35   void SetSafeBrowsingConfig(const SafeBrowsingProtocolConfig& config);
  36
  37   void SetURLRequestContextForTest(
  38       net::URLRequestContextGetter* parent_request_context);
  39
  40  protected:
  41   // net::URLFetcherDelegate interface.
  42   void OnURLFetchComplete(const net::URLFetcher* source) override;
  43
  44  private:
  45   typedef std::multimap<std::string, RequestCallback> CallbackMultiMap;
  46
  47   GURL RequestUrl() const;
  48
  49   void SaveRequestContext(
  50     const std::string& id,
  51     scoped_refptr<net::URLRequestContextGetter> request_context_getter);
  52
  53   void SendRequest(const std::string& id);
  54
  55   // ID for URLFetchers for testing.
  56   int url_fetcher_id_;
  57
  58   scoped_ptr<SafeBrowsingProtocolConfig> safe_browsing_config_;
  59   scoped_refptr<net::URLRequestContextGetter> url_request_context_getter_;
  60   scoped_refptr<net::URLRequestContextGetter> parent_request_context_for_test_;
  61
  62   // Extension id by URLFetcher.
  63   std::map<const net::URLFetcher*, std::string> requests_;
  64
  65   // Callbacks by extension ID.
  66   CallbackMultiMap callbacks_;
  67
  68   base::WeakPtrFactory<BlacklistStateFetcher> weak_ptr_factory_;
  69
  70   DISALLOW_COPY_AND_ASSIGN(BlacklistStateFetcher);
  71 };
  72
  73 }  // namespace extensions
  74
  75 #endif  // CHROME_BROWSER_EXTENSIONS_BLACKLIST_STATE_FETCHER_H_
  76