src/chrome/browser/local_discovery/privet_url_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_LOCAL_DISCOVERY_PRIVET_URL_FETCHER_H_
   6 #define CHROME_BROWSER_LOCAL_DISCOVERY_PRIVET_URL_FETCHER_H_
   7
   8 #include <string>
   9
  10 #include "base/memory/weak_ptr.h"
  11 #include "base/values.h"
  12 #include "net/url_request/url_fetcher.h"
  13 #include "net/url_request/url_fetcher_delegate.h"
  14 #include "net/url_request/url_request_context_getter.h"
  15 #include "url/gurl.h"
  16
  17 namespace local_discovery {
  18
  19 const int kPrivetHTTPCodeInternalFailure = -1;
  20
  21 // Privet-specific URLFetcher adapter. Currently supports only the subset
  22 // of HTTP features required by Privet for GCP 1.5
  23 // (/privet/info and /privet/register).
  24 class PrivetURLFetcher : public net::URLFetcherDelegate {
  25  public:
  26   enum ErrorType {
  27     JSON_PARSE_ERROR,
  28     URL_FETCH_ERROR,
  29     RESPONSE_CODE_ERROR,
  30     RETRY_ERROR,
  31     TOKEN_ERROR
  32   };
  33
  34   typedef base::Callback<void(const std::string& /*token*/)> TokenCallback;
  35
  36   class Delegate {
  37    public:
  38     virtual ~Delegate() {}
  39
  40     // If you do not implement this method, you will always get a
  41     // TOKEN_ERROR error when your token is invalid.
  42     virtual void OnNeedPrivetToken(
  43         PrivetURLFetcher* fetcher,
  44         const TokenCallback& callback);
  45     virtual void OnError(PrivetURLFetcher* fetcher, ErrorType error) = 0;
  46     virtual void OnParsedJson(PrivetURLFetcher* fetcher,
  47                               const base::DictionaryValue* value,
  48                               bool has_error) = 0;
  49   };
  50
  51   PrivetURLFetcher(
  52       const std::string& token,
  53       const GURL& url,
  54       net::URLFetcher::RequestType request_type,
  55       net::URLRequestContextGetter* request_context,
  56       Delegate* delegate);
  57   virtual ~PrivetURLFetcher();
  58
  59   void DoNotRetryOnTransientError();
  60
  61   void AllowEmptyPrivetToken();
  62
  63   void Start();
  64
  65   void SetUploadData(const std::string& upload_content_type,
  66                      const std::string& upload_data);
  67
  68   virtual void OnURLFetchComplete(const net::URLFetcher* source) OVERRIDE;
  69
  70   const GURL& url() const { return url_fetcher_->GetOriginalURL(); }
  71   int response_code() const { return url_fetcher_->GetResponseCode(); }
  72
  73  private:
  74   void Try();
  75   void ScheduleRetry(int timeout_seconds);
  76   bool PrivetErrorTransient(const std::string& error);
  77   void RequestTokenRefresh();
  78   void RefreshToken(const std::string& token);
  79
  80   std::string privet_access_token_;
  81   GURL url_;
  82   net::URLFetcher::RequestType request_type_;
  83   scoped_refptr<net::URLRequestContextGetter> request_context_;
  84   Delegate* delegate_;
  85
  86   bool do_not_retry_on_transient_error_;
  87   bool allow_empty_privet_token_;
  88
  89   int tries_;
  90   std::string upload_data_;
  91   std::string upload_content_type_;
  92   scoped_ptr<net::URLFetcher> url_fetcher_;
  93
  94   base::WeakPtrFactory<PrivetURLFetcher> weak_factory_;
  95   DISALLOW_COPY_AND_ASSIGN(PrivetURLFetcher);
  96 };
  97
  98 class PrivetURLFetcherFactory {
  99  public:
 100   explicit PrivetURLFetcherFactory(
 101       net::URLRequestContextGetter* request_context);
 102   ~PrivetURLFetcherFactory();
 103
 104   scoped_ptr<PrivetURLFetcher> CreateURLFetcher(
 105       const GURL& url,
 106       net::URLFetcher::RequestType request_type,
 107       PrivetURLFetcher::Delegate* delegate) const;
 108
 109   void set_token(const std::string& token) { token_ = token; }
 110   const std::string& get_token() const { return token_; }
 111
 112  private:
 113   scoped_refptr<net::URLRequestContextGetter> request_context_;
 114   std::string token_;
 115
 116   DISALLOW_COPY_AND_ASSIGN(PrivetURLFetcherFactory);
 117 };
 118
 119 }  // namespace local_discovery
 120
 121 #endif  // CHROME_BROWSER_LOCAL_DISCOVERY_PRIVET_URL_FETCHER_H_