capybara-webkit/src/Header.cpp

#include "Header.h"
#include "WebPage.h"
#include "WebPageManager.h"
#include "NetworkAccessManager.h"

Header::Header(WebPageManager *manager, QStringList &arguments, QObject *parent) : Command(manager, arguments, parent) {
}

void Header::start() {
  QString key = arguments()[0];
  QString value = arguments()[1];
  NetworkAccessManager* networkAccessManager = qobject_cast<NetworkAccessManager*>(page()->networkAccessManager());
  if (key.toLower().replace("-", "_") == "user_agent") {
    page()->setUserAgent(value);
  } else {
    networkAccessManager->addHeader(key, value);
  }
  emit finished(new Response(true));
}