// Copyright (c) 2008-2009 The Chromium Embedded Framework 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 _REQUEST_IMPL_H
#define _REQUEST_IMPL_H

#include "include/cef.h"
#include "net/base/upload_data.h"
#include "net/http/http_request_headers.h"
#include "third_party/WebKit/Source/WebKit/chromium/public/WebHTTPBody.h"
#include "third_party/WebKit/Source/WebKit/chromium/public/WebURLRequest.h"

namespace net {
class URLRequest;
};

// Implementation of CefRequest
class CefRequestImpl : public CefThreadSafeBase<CefRequest>
{
public:
  CefRequestImpl();
  ~CefRequestImpl() {}

  virtual CefString GetURL();
  virtual void SetURL(const CefString& url);
  virtual CefString GetMethod();
  virtual void SetMethod(const CefString& method);
  virtual CefRefPtr<CefPostData> GetPostData();
  virtual void SetPostData(CefRefPtr<CefPostData> postData);
  virtual void GetHeaderMap(HeaderMap& headerMap);
  virtual void SetHeaderMap(const HeaderMap& headerMap);
  virtual void Set(const CefString& url,
                   const CefString& method,
                   CefRefPtr<CefPostData> postData,
                   const HeaderMap& headerMap);
  virtual RequestFlags GetFlags();
  virtual void SetFlags(RequestFlags flags);
  virtual CefString GetFirstPartyForCookies();
  virtual void SetFirstPartyForCookies(const CefString& url);

  void Set(net::URLRequest* request);
  void Set(const WebKit::WebURLRequest& request);
  void Get(WebKit::WebURLRequest& request);

  static void GetHeaderMap(const net::HttpRequestHeaders& headers,
                           HeaderMap& map);
  static void GetHeaderMap(const WebKit::WebURLRequest& request,
                           HeaderMap& map);
  static void SetHeaderMap(const HeaderMap& map,
                           WebKit::WebURLRequest& request);

protected:
  CefString url_;
  CefString method_;
  CefRefPtr<CefPostData> postdata_;
  HeaderMap headermap_;

  // The below methods are used by WebURLRequest.
  RequestFlags flags_;
  CefString first_party_for_cookies_;
};

// Implementation of CefPostData
class CefPostDataImpl : public CefThreadSafeBase<CefPostData>
{
public:
  CefPostDataImpl();
  ~CefPostDataImpl() {}

  virtual size_t GetElementCount();
  virtual void GetElements(ElementVector& elements);
  virtual bool RemoveElement(CefRefPtr<CefPostDataElement> element);
  virtual bool AddElement(CefRefPtr<CefPostDataElement> element);
  virtual void RemoveElements();

  void Set(net::UploadData& data);
  void Get(net::UploadData& data);
  void Set(const WebKit::WebHTTPBody& data);
  void Get(WebKit::WebHTTPBody& data);

protected:
  ElementVector elements_;
};

// Implementation of CefPostDataElement
class CefPostDataElementImpl : public CefThreadSafeBase<CefPostDataElement>
{
public:
  CefPostDataElementImpl();
  ~CefPostDataElementImpl();

  virtual void SetToEmpty();
  virtual void SetToFile(const CefString& fileName);
  virtual void SetToBytes(size_t size, const void* bytes);
  virtual Type GetType();
  virtual CefString GetFile();
  virtual size_t GetBytesCount();
  virtual size_t GetBytes(size_t size, void* bytes);

  void* GetBytes() { return data_.bytes.bytes; }

  void Set(const net::UploadData::Element& element);
  void Get(net::UploadData::Element& element);
  void Set(const WebKit::WebHTTPBody::Element& element);
  void Get(WebKit::WebHTTPBody::Element& element);

protected:
  Type type_;
  union {
    struct {
      void* bytes;
      size_t size;
    } bytes;
    cef_string_t filename;
  } data_;
};

#endif // _REQUEST_IMPL_H