src/WebPage.cpp in capybara-webkit-0.12.1 vs src/WebPage.cpp in capybara-webkit-0.13.0
- old
+ new
@@ -1,45 +1,66 @@
#include "WebPage.h"
+#include "WebPageManager.h"
#include "JavascriptInvocation.h"
#include "NetworkAccessManager.h"
#include "NetworkCookieJar.h"
#include "UnsupportedContentHandler.h"
#include <QResource>
#include <iostream>
+#include <QWebSettings>
+#include <QUuid>
-WebPage::WebPage(QObject *parent) : QWebPage(parent) {
+WebPage::WebPage(WebPageManager *manager, QObject *parent) : QWebPage(parent) {
+ m_loading = false;
+ m_failed = false;
+ m_manager = manager;
+ m_uuid = QUuid::createUuid().toString();
+
setForwardUnsupportedContent(true);
loadJavascript();
setUserStylesheet();
- m_loading = false;
- m_ignoreSslErrors = false;
+ m_confirm = true;
+ m_prompt = false;
+ m_prompt_text = QString();
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();
+
+ settings()->setAttribute(QWebSettings::JavascriptCanOpenWindows, true);
+ currentFrame()->setUrl(QUrl("about:blank"));
}
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());
+ NetworkAccessManager *manager = new NetworkAccessManager(this);
+ manager->setCookieJar(m_manager->cookieJar());
this->setNetworkAccessManager(manager);
- connect(manager, SIGNAL(finished(QNetworkReply *)), this, SLOT(replyFinished(QNetworkReply *)));
+ connect(manager, SIGNAL(finished(QNetworkReply *)), this, SLOT(networkAccessManagerFinishedReply(QNetworkReply *)));
connect(manager, SIGNAL(sslErrors(QNetworkReply *, QList<QSslError>)),
this, SLOT(handleSslErrorsForReply(QNetworkReply *, QList<QSslError>)));
+ connect(manager, SIGNAL(requestCreated(QByteArray &, QNetworkReply *)), this, SLOT(networkAccessManagerCreatedRequest(QByteArray &, QNetworkReply *)));
}
+void WebPage::networkAccessManagerCreatedRequest(QByteArray &url, QNetworkReply *reply) {
+ emit requestCreated(url, reply);
+}
+
+void WebPage::networkAccessManagerFinishedReply(QNetworkReply *reply) {
+ emit replyFinished(reply);
+}
+
void WebPage::loadJavascript() {
QResource javascript(":/capybara.js");
if (javascript.isCompressed()) {
QByteArray uncompressedBytes(qUncompress(javascript.data(), javascript.size()));
m_capybaraJavascript = QString(uncompressedBytes);
@@ -67,10 +88,22 @@
QString WebPage::consoleMessages() {
return m_consoleMessages.join("\n");
}
+QString WebPage::alertMessages() {
+ return m_alertMessages.join("\n");
+}
+
+QString WebPage::confirmMessages() {
+ return m_confirmMessages.join("\n");
+}
+
+QString WebPage::promptMessages() {
+ return m_promptMessages.join("\n");
+}
+
void WebPage::setUserAgent(QString userAgent) {
m_userAgent = userAgent;
}
void WebPage::frameCreated(QWebFrame * frame) {
@@ -85,20 +118,20 @@
bool WebPage::shouldInterruptJavaScript() {
return false;
}
-QVariant WebPage::invokeCapybaraFunction(const char *name, QStringList &arguments) {
+QVariant WebPage::invokeCapybaraFunction(const char *name, const 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) {
+QVariant WebPage::invokeCapybaraFunction(QString &name, const 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;
@@ -108,42 +141,55 @@
std::cout << qPrintable(fullMessage) << std::endl;
}
void WebPage::javaScriptAlert(QWebFrame *frame, const QString &message) {
Q_UNUSED(frame);
+ m_alertMessages.append(message);
std::cout << "ALERT: " << qPrintable(message) << std::endl;
}
bool WebPage::javaScriptConfirm(QWebFrame *frame, const QString &message) {
Q_UNUSED(frame);
- Q_UNUSED(message);
- return true;
+ m_confirmMessages.append(message);
+ return m_confirm;
}
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;
+ m_promptMessages.append(message);
+ if (m_prompt) {
+ if (m_prompt_text.isNull()) {
+ *result = defaultValue;
+ } else {
+ *result = m_prompt_text;
+ }
+ }
+ return m_prompt;
}
void WebPage::loadStarted() {
m_loading = true;
+ m_errorPageMessage = QString();
}
void WebPage::loadFinished(bool success) {
+ Q_UNUSED(success);
m_loading = false;
- emit pageFinished(success);
+ emit pageFinished(!m_failed);
+ m_failed = false;
}
bool WebPage::isLoading() const {
return m_loading;
}
QString WebPage::failureString() {
- return QString("Unable to load URL: ") + currentFrame()->requestedUrl().toString();
+ QString message = QString("Unable to load URL: ") + currentFrame()->requestedUrl().toString();
+ if (m_errorPageMessage.isEmpty())
+ return message;
+ else
+ return message + m_errorPageMessage;
}
bool WebPage::render(const QString &fileName) {
QFileInfo fileInfo(fileName);
QDir dir;
@@ -177,63 +223,102 @@
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;
}
+ else if (extension == QWebPage::ErrorPageExtension) {
+ ErrorPageExtensionOption *errorOption = (ErrorPageExtensionOption*) option;
+ m_errorPageMessage = " because of error loading " + errorOption->url.toString() + ": " + errorOption->errorString;
+ m_failed = true;
+ return false;
+ }
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();
+void WebPage::handleSslErrorsForReply(QNetworkReply *reply, const QList<QSslError> &errors) {
+ if (m_manager->ignoreSslErrors())
+ reply->ignoreSslErrors(errors);
+}
- int length = list.size();
- for(int i = 0; i < length; i++) {
- headers << list.at(i)+": "+reply->rawHeader(list.at(i));
- }
+void WebPage::setSkipImageLoading(bool skip) {
+ settings()->setAttribute(QWebSettings::AutoLoadImages, !skip);
+}
- m_pageHeaders = headers.join("\n");
+int WebPage::getLastStatus() {
+ return qobject_cast<NetworkAccessManager *>(networkAccessManager())->statusFor(currentFrame()->url());
+}
+
+const QList<QNetworkReply::RawHeaderPair> &WebPage::pageHeaders() {
+ return qobject_cast<NetworkAccessManager *>(networkAccessManager())->headersFor(currentFrame()->url());
+}
+
+void WebPage::handleUnsupportedContent(QNetworkReply *reply) {
+ QVariant contentMimeType = reply->header(QNetworkRequest::ContentTypeHeader);
+ if(!contentMimeType.isNull()) {
+ triggerAction(QWebPage::Stop);
+ UnsupportedContentHandler *handler = new UnsupportedContentHandler(this, reply);
+ if (reply->isFinished())
+ handler->renderNonHtmlContent();
+ else
+ handler->waitForReplyToFinish();
}
}
-void WebPage::handleSslErrorsForReply(QNetworkReply *reply, const QList<QSslError> &errors) {
- if (m_ignoreSslErrors)
- reply->ignoreSslErrors(errors);
+bool WebPage::supportsExtension(Extension extension) const {
+ if (extension == ErrorPageExtension)
+ return true;
+ else if (extension == ChooseMultipleFilesExtension)
+ return true;
+ else
+ return false;
}
-void WebPage::ignoreSslErrors() {
- m_ignoreSslErrors = true;
+QWebPage *WebPage::createWindow(WebWindowType type) {
+ Q_UNUSED(type);
+ return m_manager->createPage(this);
}
-int WebPage::getLastStatus() {
- return m_lastStatus;
+QString WebPage::uuid() {
+ return m_uuid;
}
-void WebPage::resetResponseHeaders() {
- m_lastStatus = 0;
- m_pageHeaders = QString();
+QString WebPage::getWindowName() {
+ QVariant windowName = mainFrame()->evaluateJavaScript("window.name");
+
+ if (windowName.isValid())
+ return windowName.toString();
+ else
+ return "";
}
-void WebPage::resetConsoleMessages() {
- m_consoleMessages.clear();
+bool WebPage::matchesWindowSelector(QString selector) {
+ return (selector == getWindowName() ||
+ selector == mainFrame()->title() ||
+ selector == mainFrame()->url().toString() ||
+ selector == uuid());
}
-QString WebPage::pageHeaders() {
- return m_pageHeaders;
+void WebPage::setFocus() {
+ m_manager->setCurrentPage(this);
}
-void WebPage::handleUnsupportedContent(QNetworkReply *reply) {
- UnsupportedContentHandler *handler = new UnsupportedContentHandler(this, reply);
- Q_UNUSED(handler);
+void WebPage::setConfirmAction(QString action) {
+ m_confirm = (action == "Yes");
}
+
+void WebPage::setPromptAction(QString action) {
+ m_prompt = (action == "Yes");
+}
+
+void WebPage::setPromptText(QString text) {
+ m_prompt_text = text;
+}
+