imseng-capybara-webkit 0.12.1

data/src/WebPage.cpp

@@ -0,0 +1,239 @@
+#include "WebPage.h"
+#include "JavascriptInvocation.h"
+#include "NetworkAccessManager.h"
+#include "NetworkCookieJar.h"
+#include "UnsupportedContentHandler.h"
+#include <QResource>
+#include <iostream>
+
+WebPage::WebPage(QObject *parent) : QWebPage(parent) {
+  setForwardUnsupportedContent(true);
+  loadJavascript();
+  setUserStylesheet();
+
+  m_loading = false;
+  m_ignoreSslErrors = false;
+  this->setCustomNetworkAccessManager();
+
+  connect(this, SIGNAL(loadStarted()), this, SLOT(loadStarted()));
+  connect(this, SIGNAL(loadFinished(bool)), this, SLOT(loadFinished(bool)));
+  connect(this, SIGNAL(frameCreated(QWebFrame *)),
+          this, SLOT(frameCreated(QWebFrame *)));
+  connect(this, SIGNAL(unsupportedContent(QNetworkReply*)),
+      this, SLOT(handleUnsupportedContent(QNetworkReply*)));
+  resetWindowSize();
+}
+
+void WebPage::resetWindowSize() {
+  this->setViewportSize(QSize(1680, 1050));
+  this->settings()->setAttribute(QWebSettings::LocalStorageDatabaseEnabled, true);
+}
+
+void WebPage::setCustomNetworkAccessManager() {
+  NetworkAccessManager *manager = new NetworkAccessManager();
+  manager->setCookieJar(new NetworkCookieJar());
+  this->setNetworkAccessManager(manager);
+  connect(manager, SIGNAL(finished(QNetworkReply *)), this, SLOT(replyFinished(QNetworkReply *)));
+  connect(manager, SIGNAL(sslErrors(QNetworkReply *, QList<QSslError>)),
+          this, SLOT(handleSslErrorsForReply(QNetworkReply *, QList<QSslError>)));
+}
+
+void WebPage::loadJavascript() {
+  QResource javascript(":/capybara.js");
+  if (javascript.isCompressed()) {
+    QByteArray uncompressedBytes(qUncompress(javascript.data(), javascript.size()));
+    m_capybaraJavascript = QString(uncompressedBytes);
+  } else {
+    char * javascriptString =  new char[javascript.size() + 1];
+    strcpy(javascriptString, (const char *)javascript.data());
+    javascriptString[javascript.size()] = 0;
+    m_capybaraJavascript = javascriptString;
+  }
+}
+
+void WebPage::setUserStylesheet() {
+  QString data = QString("* { font-family: 'Arial' ! important; }").toUtf8().toBase64();
+  QUrl url = QUrl(QString("data:text/css;charset=utf-8;base64,") + data);
+  settings()->setUserStyleSheetUrl(url);
+}
+
+QString WebPage::userAgentForUrl(const QUrl &url ) const {
+  if (!m_userAgent.isEmpty()) {
+    return m_userAgent;
+  } else {
+    return QWebPage::userAgentForUrl(url);
+  }
+}
+
+QString WebPage::consoleMessages() {
+  return m_consoleMessages.join("\n");
+}
+
+void WebPage::setUserAgent(QString userAgent) {
+  m_userAgent = userAgent;
+}
+
+void WebPage::frameCreated(QWebFrame * frame) {
+  connect(frame, SIGNAL(javaScriptWindowObjectCleared()),
+          this,  SLOT(injectJavascriptHelpers()));
+}
+
+void WebPage::injectJavascriptHelpers() {
+  QWebFrame* frame = qobject_cast<QWebFrame *>(QObject::sender());
+  frame->evaluateJavaScript(m_capybaraJavascript);
+}
+
+bool WebPage::shouldInterruptJavaScript() {
+  return false;
+}
+
+QVariant WebPage::invokeCapybaraFunction(const char *name, QStringList &arguments) {
+  QString qname(name);
+  QString objectName("CapybaraInvocation");
+  JavascriptInvocation invocation(qname, arguments);
+  currentFrame()->addToJavaScriptWindowObject(objectName, &invocation);
+  QString javascript = QString("Capybara.invoke()");
+  return currentFrame()->evaluateJavaScript(javascript);
+}
+
+QVariant WebPage::invokeCapybaraFunction(QString &name, QStringList &arguments) {
+  return invokeCapybaraFunction(name.toAscii().data(), arguments);
+}
+
+void WebPage::javaScriptConsoleMessage(const QString &message, int lineNumber, const QString &sourceID) {
+  QString fullMessage = QString::number(lineNumber) + "|" + message;
+  if (!sourceID.isEmpty())
+    fullMessage = sourceID + "|" + fullMessage;
+  m_consoleMessages.append(fullMessage);
+  std::cout << qPrintable(fullMessage) << std::endl;
+}
+
+void WebPage::javaScriptAlert(QWebFrame *frame, const QString &message) {
+  Q_UNUSED(frame);
+  std::cout << "ALERT: " << qPrintable(message) << std::endl;
+}
+
+bool WebPage::javaScriptConfirm(QWebFrame *frame, const QString &message) {
+  Q_UNUSED(frame);
+  Q_UNUSED(message);
+  return true;
+}
+
+bool WebPage::javaScriptPrompt(QWebFrame *frame, const QString &message, const QString &defaultValue, QString *result) {
+  Q_UNUSED(frame)
+  Q_UNUSED(message)
+  Q_UNUSED(defaultValue)
+  Q_UNUSED(result)
+  return false;
+}
+
+void WebPage::loadStarted() {
+  m_loading = true;
+}
+
+void WebPage::loadFinished(bool success) {
+  m_loading = false;
+  emit pageFinished(success);
+}
+
+bool WebPage::isLoading() const {
+  return m_loading;
+}
+
+QString WebPage::failureString() {
+  return QString("Unable to load URL: ") + currentFrame()->requestedUrl().toString();
+}
+
+bool WebPage::render(const QString &fileName) {
+  QFileInfo fileInfo(fileName);
+  QDir dir;
+  dir.mkpath(fileInfo.absolutePath());
+
+  QSize viewportSize = this->viewportSize();
+  QSize pageSize = this->mainFrame()->contentsSize();
+  if (pageSize.isEmpty()) {
+    return false;
+  }
+
+  QImage buffer(pageSize, QImage::Format_ARGB32);
+  buffer.fill(qRgba(255, 255, 255, 0));
+
+  QPainter p(&buffer);
+  p.setRenderHint( QPainter::Antialiasing,          true);
+  p.setRenderHint( QPainter::TextAntialiasing,      true);
+  p.setRenderHint( QPainter::SmoothPixmapTransform, true);
+
+  this->setViewportSize(pageSize);
+  this->mainFrame()->render(&p);
+  p.end();
+  this->setViewportSize(viewportSize);
+
+  return buffer.save(fileName);
+}
+
+QString WebPage::chooseFile(QWebFrame *parentFrame, const QString &suggestedFile) {
+  Q_UNUSED(parentFrame);
+  Q_UNUSED(suggestedFile);
+
+  return getLastAttachedFileName();
+}
+
+bool WebPage::extension(Extension extension, const ExtensionOption *option, ExtensionReturn *output) {
+  Q_UNUSED(option);
+  if (extension == ChooseMultipleFilesExtension) {
+    QStringList names = QStringList() << getLastAttachedFileName();
+    static_cast<ChooseMultipleFilesExtensionReturn*>(output)->fileNames = names;
+    return true;
+  }
+  return false;
+}
+
+QString WebPage::getLastAttachedFileName() {
+  return currentFrame()->evaluateJavaScript(QString("Capybara.lastAttachedFile")).toString();
+}
+
+void WebPage::replyFinished(QNetworkReply *reply) {
+  if (reply->url() == this->currentFrame()->url()) {
+    QStringList headers;
+    m_lastStatus = reply->attribute(QNetworkRequest::HttpStatusCodeAttribute).toInt();
+    QList<QByteArray> list = reply->rawHeaderList();
+
+    int length = list.size();
+    for(int i = 0; i < length; i++) {
+      headers << list.at(i)+": "+reply->rawHeader(list.at(i));
+    }
+
+    m_pageHeaders = headers.join("\n");
+  }
+}
+
+void WebPage::handleSslErrorsForReply(QNetworkReply *reply, const QList<QSslError> &errors) {
+  if (m_ignoreSslErrors)
+    reply->ignoreSslErrors(errors);
+}
+
+void WebPage::ignoreSslErrors() {
+  m_ignoreSslErrors = true;
+}
+
+int WebPage::getLastStatus() {
+  return m_lastStatus;
+}
+
+void WebPage::resetResponseHeaders() {
+  m_lastStatus = 0;
+  m_pageHeaders = QString();
+}
+
+void WebPage::resetConsoleMessages() {
+  m_consoleMessages.clear();
+}
+
+QString WebPage::pageHeaders() {
+  return m_pageHeaders;
+}
+
+void WebPage::handleUnsupportedContent(QNetworkReply *reply) {
+  UnsupportedContentHandler *handler = new UnsupportedContentHandler(this, reply);
+  Q_UNUSED(handler);
+}

data/src/WebPage.h

@@ -0,0 +1,58 @@
+#include <QtWebKit>
+#include <QtNetwork>
+
+class WebPage : public QWebPage {
+  Q_OBJECT
+
+  public:
+    WebPage(QObject *parent = 0);
+    QVariant invokeCapybaraFunction(const char *name, QStringList &arguments);
+    QVariant invokeCapybaraFunction(QString &name, QStringList &arguments);
+    QString failureString();
+    QString userAgentForUrl(const QUrl &url ) const;
+    void setUserAgent(QString userAgent);
+    int getLastStatus();
+    void resetResponseHeaders();
+    void setCustomNetworkAccessManager();
+    bool render(const QString &fileName);
+    virtual bool extension (Extension extension, const ExtensionOption *option=0, ExtensionReturn *output=0);
+    void ignoreSslErrors();
+    QString consoleMessages();
+    void resetConsoleMessages();
+    void resetWindowSize();
+
+  public slots:
+    bool shouldInterruptJavaScript();
+    void injectJavascriptHelpers();
+    void loadStarted();
+    void loadFinished(bool);
+    bool isLoading() const;
+    QString pageHeaders();
+    void frameCreated(QWebFrame *);
+    void replyFinished(QNetworkReply *reply);
+    void handleSslErrorsForReply(QNetworkReply *reply, const QList<QSslError> &);
+    void handleUnsupportedContent(QNetworkReply *reply);
+
+  signals:
+    void pageFinished(bool);
+
+  protected:
+    virtual void javaScriptConsoleMessage(const QString &message, int lineNumber, const QString &sourceID);
+    virtual void javaScriptAlert(QWebFrame *frame, const QString &message);
+    virtual bool javaScriptConfirm(QWebFrame *frame, const QString &message);
+    virtual bool javaScriptPrompt(QWebFrame *frame, const QString &message, const QString &defaultValue, QString *result);
+    virtual QString chooseFile(QWebFrame * parentFrame, const QString &suggestedFile);
+
+  private:
+    QString m_capybaraJavascript;
+    QString m_userAgent;
+    bool m_loading;
+    QString getLastAttachedFileName();
+    void loadJavascript();
+    void setUserStylesheet();
+    int m_lastStatus;
+    QString m_pageHeaders;
+    bool m_ignoreSslErrors;
+    QStringList m_consoleMessages;
+};
+

data/src/body.cpp

@@ -0,0 +1,10 @@
+#include "Body.h"
+#include "WebPage.h"
+
+Body::Body(WebPage *page, QStringList &arguments, QObject *parent) : Command(page, arguments, parent) {
+}
+
+void Body::start() {
+  QString result = page()->currentFrame()->toHtml();
+  emit finished(new Response(true, result));
+}

data/src/capybara.js

@@ -0,0 +1,315 @@
+Capybara = {
+  nextIndex: 0,
+  nodes: {},
+  lastAttachedFile: "",
+
+  invoke: function () {
+    return this[CapybaraInvocation.functionName].apply(this, CapybaraInvocation.arguments);
+  },
+
+  find: function (xpath) {
+    return this.findRelativeTo(document, xpath);
+  },
+
+  findWithin: function (index, xpath) {
+    return this.findRelativeTo(this.nodes[index], xpath);
+  },
+
+  findRelativeTo: function (reference, xpath) {
+    var iterator = document.evaluate(xpath, reference, null, XPathResult.ORDERED_NODE_ITERATOR_TYPE, null);
+    var node;
+    var results = [];
+    while (node = iterator.iterateNext()) {
+      this.nextIndex++;
+      this.nodes[this.nextIndex] = node;
+      results.push(this.nextIndex);
+    }
+    return results.join(",");
+  },
+
+  isAttached: function(index) {
+    return document.evaluate("ancestor-or-self::html", this.nodes[index], null, XPathResult.FIRST_ORDERED_NODE_TYPE, null).singleNodeValue != null;
+  },
+
+  text: function (index) {
+    var node = this.nodes[index];
+    var type = (node.type || node.tagName).toLowerCase();
+    if (type == "textarea") {
+      return node.innerHTML;
+    } else {
+      return node.innerText;
+    }
+  },
+
+  attribute: function (index, name) {
+    switch(name) {
+    case 'checked':
+      return this.nodes[index].checked;
+      break;
+
+    case 'disabled':
+      return this.nodes[index].disabled;
+      break;
+
+    default:
+      return this.nodes[index].getAttribute(name);
+    }
+  },
+
+  path: function(index) {
+    return "/" + this.getXPathNode(this.nodes[index]).join("/");
+  },
+
+  getXPathNode: function(node, path) {
+    path = path || [];
+    if (node.parentNode) {
+      path = this.getXPathNode(node.parentNode, path);
+    }
+
+    var first = node;
+    while (first.previousSibling)
+      first = first.previousSibling;
+
+    var count = 0;
+    var index = 0;
+    var iter = first;
+    while (iter) {
+      if (iter.nodeType == 1 && iter.nodeName == node.nodeName)
+        count++;
+      if (iter.isSameNode(node))
+         index = count;
+      iter = iter.nextSibling;
+      continue;
+    }
+
+    if (node.nodeType == 1)
+      path.push(node.nodeName.toLowerCase() + (node.id ? "[@id='"+node.id+"']" : count > 1 ? "["+index+"]" : ''));
+
+    return path;
+  },
+
+  tagName: function(index) {
+    return this.nodes[index].tagName.toLowerCase();
+  },
+
+  submit: function(index) {
+    return this.nodes[index].submit();
+  },
+
+  mousedown: function(index) {
+    var mousedownEvent = document.createEvent('MouseEvents');
+    mousedownEvent.initMouseEvent('mousedown', true, true, window, 0, 0, 0, 0, 0, false, false, false, false, 0, null);
+    this.nodes[index].dispatchEvent(mousedownEvent);
+  },
+
+  mouseup: function(index) {
+    var mouseupEvent = document.createEvent('MouseEvents');
+    mouseupEvent.initMouseEvent('mouseup', true, true, window, 0, 0, 0, 0, 0, false, false, false, false, 0, null);
+    this.nodes[index].dispatchEvent(mouseupEvent);
+  },
+
+  click: function (index) {
+    this.mousedown(index);
+    this.mouseup(index);
+    var clickEvent = document.createEvent('MouseEvents');
+    clickEvent.initMouseEvent('click', true, true, window, 0, 0, 0, 0, 0, false, false, false, false, 0, null);
+    this.nodes[index].dispatchEvent(clickEvent);
+  },
+
+  trigger: function (index, eventName) {
+    var eventObject = document.createEvent("HTMLEvents");
+    eventObject.initEvent(eventName, true, true);
+    this.nodes[index].dispatchEvent(eventObject);
+  },
+
+  keypress: function(index, altKey, ctrlKey, shiftKey, metaKey, keyCode, charCode) {
+    var eventObject = document.createEvent("Events");
+    eventObject.initEvent('keypress', true, true);
+    eventObject.window = window;
+    eventObject.altKey = altKey;
+    eventObject.ctrlKey = ctrlKey;
+    eventObject.shiftKey = shiftKey;
+    eventObject.metaKey = metaKey;
+    eventObject.keyCode = keyCode;
+    eventObject.charCode = charCode;
+    eventObject.which = keyCode;
+    this.nodes[index].dispatchEvent(eventObject);
+  },
+
+  keyupdown: function(index, eventName, keyCode) {
+    var eventObject = document.createEvent("HTMLEvents");
+    eventObject.initEvent(eventName, true, true);
+    eventObject.keyCode = keyCode;
+    eventObject.which = keyCode;
+    eventObject.charCode = 0;
+    this.nodes[index].dispatchEvent(eventObject);
+  },
+
+  visible: function (index) {
+    var element = this.nodes[index];
+    while (element) {
+      if (element.ownerDocument.defaultView.getComputedStyle(element, null).getPropertyValue("display") == 'none')
+        return false;
+      element = element.parentElement;
+    }
+    return true;
+  },
+
+  selected: function (index) {
+    return this.nodes[index].selected;
+  },
+
+  value: function(index) {
+    return this.nodes[index].value;
+  },
+
+  characterToKeyCode: function(character) {
+    var code = character.toUpperCase().charCodeAt(0);
+    var specialKeys = {
+      96: 192,  //`
+      45: 189,  //-
+      61: 187,  //=
+      91: 219,  //[
+      93: 221,  //]
+      92: 220,  //\
+      59: 186,  //;
+      39: 222,  //'
+      44: 188,  //,
+      46: 190,  //.
+      47: 191,  ///
+      127: 46,  //delete
+      126: 192, //~
+      33: 49,   //!
+      64: 50,   //@
+      35: 51,   //#
+      36: 52,   //$
+      37: 53,   //%
+      94: 54,   //^
+      38: 55,   //&
+      42: 56,   //*
+      40: 57,   //(
+      41: 48,   //)
+      95: 189,  //_
+      43: 187,  //+
+      123: 219, //{
+      125: 221, //}
+      124: 220, //|
+      58: 186,  //:
+      34: 222,  //"
+      60: 188,  //<
+      62: 190,  //>
+      63: 191   //?
+    };
+    if (specialKeys[code]) {
+      code = specialKeys[code];
+    }
+    return code;
+  },
+
+  set: function (index, value) {
+    var length, maxLength, node, strindex, textTypes, type;
+
+    node = this.nodes[index];
+    type = (node.type || node.tagName).toLowerCase();
+    textTypes = ["email", "number", "password", "search", "tel", "text", "textarea", "url"];
+
+    if (textTypes.indexOf(type) != -1) {
+      this.trigger(index, "focus");
+
+      maxLength = this.attribute(index, "maxlength");
+      if (maxLength && value.length > maxLength) {
+        length = maxLength;
+      } else {
+        length = value.length;
+      }
+
+      node.value = "";
+      for (strindex = 0; strindex < length; strindex++) {
+        node.value += value[strindex];
+        var keyCode = this.characterToKeyCode(value[strindex]);
+        this.keyupdown(index, "keydown", keyCode);
+        this.keypress(index, false, false, false, false, value.charCodeAt(strindex), value.charCodeAt(strindex));
+        this.keyupdown(index, "keyup", keyCode);
+        this.trigger(index, "input");
+      }
+      this.trigger(index, "change");
+      this.trigger(index, "blur");
+
+    } else if (type === "checkbox" || type === "radio") {
+      if (node.checked != (value === "true")) {
+        this.click(index)
+      }
+
+    } else if (type === "file") {
+      this.lastAttachedFile = value;
+      this.click(index)
+
+    } else {
+      node.value = value;
+    }
+  },
+
+  selectOption: function(index) {
+    this.nodes[index].selected = true;
+    this.trigger(index, "change");
+  },
+
+  unselectOption: function(index) {
+    this.nodes[index].selected = false;
+    this.trigger(index, "change");
+  },
+
+  centerPostion: function(element) {
+    this.reflow(element);
+    var rect = element.getBoundingClientRect();
+    var position = {
+      x: rect.width / 2,
+      y: rect.height / 2
+    };
+    do {
+        position.x += element.offsetLeft;
+        position.y += element.offsetTop;
+    } while ((element = element.offsetParent));
+    position.x = Math.floor(position.x), position.y = Math.floor(position.y);
+
+    return position;
+  },
+
+  reflow: function(element, force) {
+    if (force || element.offsetWidth === 0) {
+      var prop, oldStyle = {}, newStyle = {position: "absolute", visibility : "hidden", display: "block" };
+      for (prop in newStyle)  {
+        oldStyle[prop] = element.style[prop];
+        element.style[prop] = newStyle[prop];
+      }
+      element.offsetWidth, element.offsetHeight; // force reflow
+      for (prop in oldStyle)
+        element.style[prop] = oldStyle[prop];
+    }
+  },
+
+  dragTo: function (index, targetIndex) {
+    var element = this.nodes[index], target = this.nodes[targetIndex];
+    var position = this.centerPostion(element);
+    var options = {
+      clientX: position.x,
+      clientY: position.y
+    };
+    var mouseTrigger = function(eventName, options) {
+      var eventObject = document.createEvent("MouseEvents");
+      eventObject.initMouseEvent(eventName, true, true, window, 0, 0, 0, options.clientX || 0, options.clientY || 0, false, false, false, false, 0, null);
+      element.dispatchEvent(eventObject);
+    }
+    mouseTrigger('mousedown', options);
+    options.clientX += 1, options.clientY += 1;
+    mouseTrigger('mousemove', options);
+
+    position = this.centerPostion(target), options = {
+      clientX: position.x,
+      clientY: position.y
+    };
+    mouseTrigger('mousemove', options);
+    mouseTrigger('mouseup', options);
+  }
+};
+

data/src/find_command.h

@@ -0,0 +1,29 @@
+#define CHECK_COMMAND(expectedName) \
+  if (strcmp(#expectedName, name) == 0) { \
+    return new expectedName(m_page, arguments, this); \
+  }
+
+CHECK_COMMAND(Visit)
+CHECK_COMMAND(Find)
+CHECK_COMMAND(Reset)
+CHECK_COMMAND(Node)
+CHECK_COMMAND(Url)
+CHECK_COMMAND(Source)
+CHECK_COMMAND(Evaluate)
+CHECK_COMMAND(Execute)
+CHECK_COMMAND(FrameFocus)
+CHECK_COMMAND(Header)
+CHECK_COMMAND(Render)
+CHECK_COMMAND(Body)
+CHECK_COMMAND(Status)
+CHECK_COMMAND(Headers)
+CHECK_COMMAND(SetCookie)
+CHECK_COMMAND(ClearCookies)
+CHECK_COMMAND(GetCookies)
+CHECK_COMMAND(Headers)
+CHECK_COMMAND(SetProxy)
+CHECK_COMMAND(ConsoleMessages)
+CHECK_COMMAND(RequestedUrl)
+CHECK_COMMAND(CurrentUrl)
+CHECK_COMMAND(ResizeWindow)
+CHECK_COMMAND(IgnoreSslErrors)

data/src/main.cpp

@@ -0,0 +1,33 @@
+#include "Server.h"
+#include <QtGui>
+#include <iostream>
+#ifdef Q_OS_UNIX
+  #include <unistd.h>
+#endif
+
+int main(int argc, char **argv) {
+#ifdef Q_OS_UNIX
+  if (setpgid(0, 0) < 0) {
+    std::cerr << "Unable to set new process group." << std::endl;
+    return 1;
+  }
+#endif
+
+  QApplication app(argc, argv);
+  app.setApplicationName("capybara-webkit");
+  app.setOrganizationName("thoughtbot, inc");
+  app.setOrganizationDomain("thoughtbot.com");
+
+  QStringList args = app.arguments();
+
+  Server server(0);
+
+  if (server.start()) {
+    std::cout << "Capybara-webkit server started, listening on port: " << server.server_port() << std::endl;
+    return app.exec();
+  } else {
+    std::cerr << "Couldn't start capybara-webkit server" << std::endl;
+    return 1;
+  }
+}
+