// 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.

#include "request_impl.h"
#include "browser_webkit_glue.h"

#include "base/logging.h"
#include "base/utf_string_conversions.h"
#include "net/http/http_response_headers.h"
#include "net/http/http_util.h"
#include "net/url_request/url_request.h"
#include "third_party/WebKit/WebKit/chromium/public/WebHTTPHeaderVisitor.h"

using net::HttpResponseHeaders;


CefRefPtr<CefRequest> CefRequest::CreateRequest()
{
  CefRefPtr<CefRequest> request(new CefRequestImpl());
  return request;
}

CefRequestImpl::CefRequestImpl()
{
}

std::wstring CefRequestImpl::GetURL()
{
  Lock();
  std::wstring url = url_;
  Unlock();
  return url;
}

void CefRequestImpl::SetURL(const std::wstring& url)
{
  Lock();
  url_ = url;
  Unlock();
}

std::wstring CefRequestImpl::GetMethod()
{
  Lock();
  std::wstring method = method_;
  Unlock();
  return method;
}

void CefRequestImpl::SetMethod(const std::wstring& method)
{
  Lock();
  method_ = method;
  Unlock();
}

CefRefPtr<CefPostData> CefRequestImpl::GetPostData()
{
  Lock();
  CefRefPtr<CefPostData> postData = postdata_;
  Unlock();
  return postData;
}

void CefRequestImpl::SetPostData(CefRefPtr<CefPostData> postData)
{
  Lock();
  postdata_ = postData;
  Unlock();
}

void CefRequestImpl::GetHeaderMap(HeaderMap& headerMap)
{
  Lock();
  headerMap = headermap_;
  Unlock();
}

void CefRequestImpl::SetHeaderMap(const HeaderMap& headerMap)
{
  Lock();
  headermap_ = headerMap;
  Unlock();
}

void CefRequestImpl::Set(const std::wstring& url,
                         const std::wstring& method,
                         CefRefPtr<CefPostData> postData,
                         const HeaderMap& headerMap)
{
  Lock();
  url_ = url;
  method_ = method;
  postdata_ = postData;
  headermap_ = headerMap;
  Unlock();
}

void CefRequestImpl::Set(URLRequest* request)
{
  SetURL(UTF8ToWide(request->url().spec()));
  SetMethod(UTF8ToWide(request->method()));

  // Transfer request headers
  HeaderMap headerMap;
  GetHeaderMap(request->extra_request_headers(), headerMap);
  headerMap.insert(
      std::make_pair(L"Referrer", UTF8ToWide(request->referrer())));
  SetHeaderMap(headerMap);

  // Transfer post data, if any
  net::UploadData* data = request->get_upload();
  if (data) {
    CefRefPtr<CefPostData> postdata(CefPostData::CreatePostData());
    static_cast<CefPostDataImpl*>(postdata.get())->Set(*data);
    SetPostData(postdata);
  }
}


void CefRequestImpl::GetHeaderMap(const net::HttpRequestHeaders& headers, HeaderMap& map)
{
  net::HttpRequestHeaders::Iterator it(headers);
  do {
    map[UTF8ToWide(it.name())] = UTF8ToWide(it.value());
  } while (it.GetNext());
}

void CefRequestImpl::GetHeaderMap(const WebKit::WebURLRequest& request,
                                  HeaderMap& map)
{
  class CefHTTPHeaderVisitor : public WebKit::WebHTTPHeaderVisitor {
    public:
      CefHTTPHeaderVisitor(HeaderMap* map) : map_(map) {}

      virtual void visitHeader(const WebKit::WebString& name,
                               const WebKit::WebString& value) {
        map_->insert(
            std::make_pair(
                UTF8ToWide(webkit_glue::WebStringToStdString(name)),
                UTF8ToWide(webkit_glue::WebStringToStdString(value))));
      }

    private:
      HeaderMap* map_;
  };

  CefHTTPHeaderVisitor visitor(&map);
  request.visitHTTPHeaderFields(&visitor);
}

void CefRequestImpl::SetHeaderMap(const HeaderMap& map,
                                  WebKit::WebURLRequest& request)
{
  HeaderMap::const_iterator it = map.begin();
  for(; it != map.end(); ++it) {
    request.setHTTPHeaderField(
        webkit_glue::StdWStringToWebString(it->first.c_str()),
        webkit_glue::StdWStringToWebString(it->second.c_str()));
  }
}

std::string CefRequestImpl::GenerateHeaders(const HeaderMap& map)
{
  std::string headers;

  for(HeaderMap::const_iterator header = map.begin();
      header != map.end();
      ++header) {
    const std::wstring& key = header->first;
    const std::wstring& value = header->second;

    if(!key.empty()) {
      // Delimit with "\r\n".
      if(!headers.empty())
        headers += "\r\n";

      headers += WideToUTF8(key) + ": " + WideToUTF8(value);
    }
  }

  return headers;
}

void CefRequestImpl::ParseHeaders(const std::string& header_str, HeaderMap& map)
{
  // Parse the request header values
  std::string headerStr = "HTTP/1.1 200 OK\n";
  headerStr += header_str;
  scoped_refptr<net::HttpResponseHeaders> headers =
      new HttpResponseHeaders(net::HttpUtil::AssembleRawHeaders(
          headerStr.c_str(), headerStr.length()));
  void* iter = NULL;
  std::string name, value;
  while(headers->EnumerateHeaderLines(&iter, &name, &value))
    map.insert(std::make_pair(UTF8ToWide(name), UTF8ToWide(value)));
}

CefRefPtr<CefPostData> CefPostData::CreatePostData()
{
  CefRefPtr<CefPostData> postdata(new CefPostDataImpl());
  return postdata;
}

CefPostDataImpl::CefPostDataImpl()
{
}

size_t CefPostDataImpl::GetElementCount()
{
  Lock();
  size_t ct = elements_.size();
  Unlock();
  return ct;
}

void CefPostDataImpl::GetElements(ElementVector& elements)
{
  Lock();
  elements = elements_;
  Unlock();
}

bool CefPostDataImpl::RemoveElement(CefRefPtr<CefPostDataElement> element)
{
  bool deleted = false;

  Lock();

  ElementVector::iterator it = elements_.begin();
  for(; it != elements_.end(); ++it) {
    if(it->get() == element.get()) {
      elements_.erase(it);
      deleted = true;
      break;
    }
  }

  Unlock();
  
  return deleted;
}

bool CefPostDataImpl::AddElement(CefRefPtr<CefPostDataElement> element)
{
  bool found = false;
  
  Lock();
  
  // check that the element isn't already in the list before adding
  ElementVector::const_iterator it = elements_.begin();
  for(; it != elements_.end(); ++it) {
    if(it->get() == element.get()) {
      found = true;
      break;
    }
  }

  if(!found)
    elements_.push_back(element);
 
  Unlock();
  return !found;
}

void CefPostDataImpl::RemoveElements()
{
  Lock();
  elements_.clear();
  Unlock();
}

void CefPostDataImpl::Set(net::UploadData& data)
{
  Lock();

  CefRefPtr<CefPostDataElement> postelem;

  std::vector<net::UploadData::Element>* elements = data.elements();
  std::vector<net::UploadData::Element>::const_iterator it = elements->begin();
  for (; it != elements->end(); ++it) {
    postelem = CefPostDataElement::CreatePostDataElement();
    static_cast<CefPostDataElementImpl*>(postelem.get())->Set(*it);
    AddElement(postelem);
  }

  Unlock();
}

void CefPostDataImpl::Get(net::UploadData& data)
{
  Lock();

  net::UploadData::Element element;
  std::vector<net::UploadData::Element> data_elements;
  ElementVector::iterator it = elements_.begin();
  for(; it != elements_.end(); ++it) {
    static_cast<CefPostDataElementImpl*>(it->get())->Get(element);
    data_elements.push_back(element);
  }
  data.SetElements(data_elements);

  Unlock();
}

void CefPostDataImpl::Set(const WebKit::WebHTTPBody& data)
{
  Lock();

  CefRefPtr<CefPostDataElement> postelem;
  WebKit::WebHTTPBody::Element element;
  size_t size = data.elementCount();
  for (size_t i = 0; i < size; ++i) {
    if (data.elementAt(i, element)) {
      postelem = CefPostDataElement::CreatePostDataElement();
      static_cast<CefPostDataElementImpl*>(postelem.get())->Set(element);
      AddElement(postelem);
    }
  }

  Unlock();
}

void CefPostDataImpl::Get(WebKit::WebHTTPBody& data)
{
  Lock();

  WebKit::WebHTTPBody::Element element;
  ElementVector::iterator it = elements_.begin();
  for(; it != elements_.end(); ++it) {
    static_cast<CefPostDataElementImpl*>(it->get())->Get(element);
    if(element.type == WebKit::WebHTTPBody::Element::TypeData) {
      data.appendData(element.data);
    } else if(element.type == WebKit::WebHTTPBody::Element::TypeFile) {
      data.appendFile(element.filePath);
    } else {
      NOTREACHED();
    }
  }

  Unlock();
}

CefRefPtr<CefPostDataElement> CefPostDataElement::CreatePostDataElement()
{
  CefRefPtr<CefPostDataElement> element(new CefPostDataElementImpl());
  return element;
}

CefPostDataElementImpl::CefPostDataElementImpl()
{
  type_ = PDE_TYPE_EMPTY;
}

CefPostDataElementImpl::~CefPostDataElementImpl()
{
  SetToEmpty();
}

void CefPostDataElementImpl::SetToEmpty()
{
  Lock();
  if(type_ == PDE_TYPE_BYTES)
    free(data_.bytes.bytes);
  else if(type_ == PDE_TYPE_FILE)
    free(data_.filename);
  type_ = PDE_TYPE_EMPTY;
  Unlock();
}

void CefPostDataElementImpl::SetToFile(const std::wstring& fileName)
{
  Lock();
  // Clear any data currently in the element
  SetToEmpty();

  // Assign the new file name
  size_t size = fileName.size();
  wchar_t* data = static_cast<wchar_t*>(malloc((size + 1) * sizeof(wchar_t)));
  DCHECK(data != NULL);
  if(data == NULL)
    return;

  memcpy(static_cast<void*>(data), static_cast<const void*>(fileName.c_str()),
      size * sizeof(wchar_t));
  data[size] = 0;

  // Assign the new data
  type_ = PDE_TYPE_FILE;
  data_.filename = data;
  Unlock();
}

void CefPostDataElementImpl::SetToBytes(size_t size, const void* bytes)
{
  Lock();
  // Clear any data currently in the element
  SetToEmpty();

  // Assign the new data
  void* data = malloc(size);
  DCHECK(data != NULL);
  if(data == NULL)
    return;

  memcpy(data, bytes, size);
  
  type_ = PDE_TYPE_BYTES;
  data_.bytes.bytes = data;
  data_.bytes.size = size;
  Unlock();
}

CefPostDataElement::Type CefPostDataElementImpl::GetType()
{
  Lock();
  CefPostDataElement::Type type = type_;
  Unlock();
  return type;
}

std::wstring CefPostDataElementImpl::GetFile()
{
  Lock();
  DCHECK(type_ == PDE_TYPE_FILE);
  std::wstring filename;
  if(type_ == PDE_TYPE_FILE)
    filename = data_.filename;
  Unlock();
  return filename;
}

size_t CefPostDataElementImpl::GetBytesCount()
{
  Lock();
  DCHECK(type_ == PDE_TYPE_BYTES);
  size_t size = 0;
  if(type_ == PDE_TYPE_BYTES)
    size = data_.bytes.size;
  Unlock();
  return size;
}

size_t CefPostDataElementImpl::GetBytes(size_t size, void* bytes)
{
  Lock();
  DCHECK(type_ == PDE_TYPE_BYTES);
  size_t rv = 0;
  if(type_ == PDE_TYPE_BYTES) {
    rv = (size < data_.bytes.size ? size : data_.bytes.size);
    memcpy(bytes, data_.bytes.bytes, rv);
  }
  Unlock();
  return rv;
}

void CefPostDataElementImpl::Set(const net::UploadData::Element& element)
{
  Lock();

  if (element.type() == net::UploadData::TYPE_BYTES) {
    SetToBytes(element.bytes().size(),
        static_cast<const void*>(
            std::string(element.bytes().begin(),
            element.bytes().end()).c_str()));
  } else if (element.type() == net::UploadData::TYPE_FILE) {
#if defined(OS_WIN)
    SetToFile(element.file_path().value());
#else
    SetToFile(UTF8ToWide(element.file_path().value()));
#endif
  } else {
    NOTREACHED();
  }

  Unlock();
}

void CefPostDataElementImpl::Get(net::UploadData::Element& element)
{
  Lock();

  if(type_ == PDE_TYPE_BYTES) {
    element.SetToBytes(static_cast<char*>(data_.bytes.bytes), data_.bytes.size);
  } else if(type_ == PDE_TYPE_FILE) {
#if defined(OS_WIN)
    element.SetToFilePath(FilePath(data_.filename));
#else
    element.SetToFilePath(FilePath(WideToUTF8(data_.filename)));
#endif
  } else {
    NOTREACHED();
  }

  Unlock();
}

void CefPostDataElementImpl::Set(const WebKit::WebHTTPBody::Element& element)
{
  Lock();

  if(element.type == WebKit::WebHTTPBody::Element::TypeData) {
    SetToBytes(element.data.size(),
        static_cast<const void*>(element.data.data()));
  } else if(element.type == WebKit::WebHTTPBody::Element::TypeFile) {
    SetToFile(UTF8ToWide(webkit_glue::WebStringToStdString(element.filePath)));
  } else {
    NOTREACHED();
  }

  Unlock();
}

void CefPostDataElementImpl::Get(WebKit::WebHTTPBody::Element& element)
{
  Lock();

  if(type_ == PDE_TYPE_BYTES) {
    element.type = WebKit::WebHTTPBody::Element::TypeData;
    element.data.assign(
        static_cast<char*>(data_.bytes.bytes), data_.bytes.size);
  } else if(type_ == PDE_TYPE_FILE) {
    element.type = WebKit::WebHTTPBody::Element::TypeFile;
    element.filePath.assign(webkit_glue::StdWStringToWebString(data_.filename));
  } else {
    NOTREACHED();
  }

  Unlock();
}