// Copyright 2013 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_LOCAL_DISCOVERY_CLOUD_PRINT_BASE_API_FLOW_H_
#define CHROME_BROWSER_LOCAL_DISCOVERY_CLOUD_PRINT_BASE_API_FLOW_H_

#include <string>

#include "chrome/browser/local_discovery/privet_constants.h"
#include "chrome/browser/local_discovery/privet_http.h"
#include "google_apis/gaia/oauth2_token_service.h"
#include "net/url_request/url_fetcher.h"
#include "net/url_request/url_fetcher_delegate.h"
#include "net/url_request/url_request_context_getter.h"

namespace local_discovery {

// API call flow for communicating with cloud print.
class CloudPrintBaseApiFlow : public net::URLFetcherDelegate,
                              public OAuth2TokenService::Consumer {
 public:
  // TODO(noamsml): Better error model for this class.
  enum Status {
    SUCCESS,
    ERROR_TOKEN,
    ERROR_NETWORK,
    ERROR_HTTP_CODE,
    ERROR_FROM_SERVER,
    ERROR_MALFORMED_RESPONSE
  };

  class Delegate {
   public:
    virtual ~Delegate() {}

    virtual void OnCloudPrintAPIFlowError(CloudPrintBaseApiFlow* flow,
                                          Status status) = 0;
    virtual void OnCloudPrintAPIFlowComplete(
        CloudPrintBaseApiFlow* flow,
        const base::DictionaryValue* value) = 0;
  };

  // Create an OAuth2-based confirmation.
  CloudPrintBaseApiFlow(net::URLRequestContextGetter* request_context,
                        OAuth2TokenService* token_service_,
                        const std::string& account_id,
                        const GURL& automated_claim_url,
                        Delegate* delegate);

  // Create a cookie-based confirmation.
  CloudPrintBaseApiFlow(net::URLRequestContextGetter* request_context,
                        int  user_index,
                        const std::string& xsrf_token,
                        const GURL& automated_claim_url,
                        Delegate* delegate);

  // Create a cookie-based confirmation with no XSRF token (for requests that
  // don't need an XSRF token).
  CloudPrintBaseApiFlow(net::URLRequestContextGetter* request_context,
                        int  user_index,
                        const GURL& automated_claim_url,
                        Delegate* delegate);


  virtual ~CloudPrintBaseApiFlow();

  void Start();


  // net::URLFetcherDelegate implementation:
  virtual void OnURLFetchComplete(const net::URLFetcher* source) OVERRIDE;

  // OAuth2TokenService::Consumer implementation:
  virtual void OnGetTokenSuccess(const OAuth2TokenService::Request* request,
                                 const std::string& access_token,
                                 const base::Time& expiration_time) OVERRIDE;
  virtual void OnGetTokenFailure(const OAuth2TokenService::Request* request,
                                 const GoogleServiceAuthError& error) OVERRIDE;

  // Return the user index or kAccountIndexUseOAuth2 if none is available.
  int user_index() { return user_index_; }

 private:
  bool UseOAuth2() { return user_index_ == kAccountIndexUseOAuth2; }

  void CreateRequest(const GURL& url);

  scoped_ptr<net::URLFetcher> url_fetcher_;
  scoped_ptr<OAuth2TokenService::Request> oauth_request_;
  scoped_refptr<net::URLRequestContextGetter> request_context_;
  OAuth2TokenService* token_service_;
  std::string account_id_;
  int user_index_;
  std::string xsrf_token_;
  GURL url_;
  Delegate* delegate_;
};

}  // namespace local_discovery

#endif  // CHROME_BROWSER_LOCAL_DISCOVERY_CLOUD_PRINT_BASE_API_FLOW_H_