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