2011-02-19 03:53:06 +00:00
|
|
|
#include "WebPage.h"
|
2012-03-26 22:41:33 +00:00
|
|
|
#include "WebPageManager.h"
|
2011-02-26 03:57:55 +00:00
|
|
|
#include "JavascriptInvocation.h"
|
2011-06-08 09:36:45 +00:00
|
|
|
#include "NetworkAccessManager.h"
|
2011-09-24 21:25:15 +00:00
|
|
|
#include "NetworkCookieJar.h"
|
2011-09-15 15:22:10 +00:00
|
|
|
#include "UnsupportedContentHandler.h"
|
2013-01-20 23:57:04 +00:00
|
|
|
#include "InvocationResult.h"
|
2011-02-26 03:32:22 +00:00
|
|
|
#include <QResource>
|
2011-02-26 18:02:43 +00:00
|
|
|
#include <iostream>
|
2012-03-30 01:38:34 +00:00
|
|
|
#include <QWebSettings>
|
2012-03-27 22:49:57 +00:00
|
|
|
#include <QUuid>
|
2011-02-19 03:53:06 +00:00
|
|
|
|
2012-03-29 03:05:24 +00:00
|
|
|
WebPage::WebPage(WebPageManager *manager, QObject *parent) : QWebPage(parent) {
|
2012-03-30 19:00:51 +00:00
|
|
|
m_loading = false;
|
2012-07-11 03:03:32 +00:00
|
|
|
m_failed = false;
|
2012-04-04 19:40:18 +00:00
|
|
|
m_manager = manager;
|
2012-03-30 19:00:51 +00:00
|
|
|
m_uuid = QUuid::createUuid().toString();
|
2012-11-22 04:20:47 +00:00
|
|
|
m_unsupportedContentLoaded = false;
|
2012-04-04 19:40:18 +00:00
|
|
|
|
2011-09-15 15:22:10 +00:00
|
|
|
setForwardUnsupportedContent(true);
|
2011-08-12 19:49:40 +00:00
|
|
|
loadJavascript();
|
2011-08-12 19:26:26 +00:00
|
|
|
setUserStylesheet();
|
2011-08-12 19:49:40 +00:00
|
|
|
|
2012-07-10 13:47:17 +00:00
|
|
|
m_confirm = true;
|
|
|
|
m_prompt = false;
|
|
|
|
m_prompt_text = QString();
|
2011-08-23 15:18:08 +00:00
|
|
|
this->setCustomNetworkAccessManager();
|
2011-08-12 19:49:40 +00:00
|
|
|
|
|
|
|
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 *)));
|
2011-09-15 15:22:10 +00:00
|
|
|
connect(this, SIGNAL(unsupportedContent(QNetworkReply*)),
|
|
|
|
this, SLOT(handleUnsupportedContent(QNetworkReply*)));
|
2012-03-29 17:01:02 +00:00
|
|
|
resetWindowSize();
|
2012-03-26 22:41:33 +00:00
|
|
|
|
|
|
|
settings()->setAttribute(QWebSettings::JavascriptCanOpenWindows, true);
|
2012-03-29 17:01:02 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
void WebPage::resetWindowSize() {
|
2011-10-21 15:03:04 +00:00
|
|
|
this->setViewportSize(QSize(1680, 1050));
|
2012-04-01 16:53:40 +00:00
|
|
|
this->settings()->setAttribute(QWebSettings::LocalStorageDatabaseEnabled, true);
|
2011-08-12 19:49:40 +00:00
|
|
|
}
|
|
|
|
|
2011-08-23 15:18:08 +00:00
|
|
|
void WebPage::setCustomNetworkAccessManager() {
|
2012-05-15 01:59:28 +00:00
|
|
|
NetworkAccessManager *manager = new NetworkAccessManager(this);
|
2012-06-22 03:34:51 +00:00
|
|
|
manager->setCookieJar(m_manager->cookieJar());
|
2011-08-23 15:18:08 +00:00
|
|
|
this->setNetworkAccessManager(manager);
|
2012-05-04 17:56:21 +00:00
|
|
|
connect(manager, SIGNAL(sslErrors(QNetworkReply *, QList<QSslError>)),
|
|
|
|
this, SLOT(handleSslErrorsForReply(QNetworkReply *, QList<QSslError>)));
|
2012-12-04 03:47:10 +00:00
|
|
|
connect(manager, SIGNAL(requestCreated(QByteArray &, QNetworkReply *)),
|
|
|
|
SIGNAL(requestCreated(QByteArray &, QNetworkReply *)));
|
2012-06-29 14:41:08 +00:00
|
|
|
}
|
|
|
|
|
2012-12-04 03:47:10 +00:00
|
|
|
void WebPage::unsupportedContentFinishedReply(QNetworkReply *reply) {
|
|
|
|
m_unsupportedContentLoaded = true;
|
|
|
|
m_manager->replyFinished(reply);
|
2011-08-23 15:18:08 +00:00
|
|
|
}
|
|
|
|
|
2011-08-12 19:49:40 +00:00
|
|
|
void WebPage::loadJavascript() {
|
2011-02-26 03:32:22 +00:00
|
|
|
QResource javascript(":/capybara.js");
|
2011-05-03 22:12:10 +00:00
|
|
|
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;
|
|
|
|
}
|
2011-04-19 07:16:25 +00:00
|
|
|
}
|
|
|
|
|
2011-08-12 19:26:26 +00:00
|
|
|
void WebPage::setUserStylesheet() {
|
2013-01-13 21:08:30 +00:00
|
|
|
QString data = QString("*, :before, :after { font-family: 'Arial' ! important; }").toUtf8().toBase64();
|
2011-08-12 19:26:26 +00:00
|
|
|
QUrl url = QUrl(QString("data:text/css;charset=utf-8;base64,") + data);
|
|
|
|
settings()->setUserStyleSheetUrl(url);
|
|
|
|
}
|
|
|
|
|
2011-06-08 09:36:45 +00:00
|
|
|
QString WebPage::userAgentForUrl(const QUrl &url ) const {
|
2011-06-25 11:31:19 +00:00
|
|
|
if (!m_userAgent.isEmpty()) {
|
|
|
|
return m_userAgent;
|
2011-06-08 09:36:45 +00:00
|
|
|
} else {
|
|
|
|
return QWebPage::userAgentForUrl(url);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2012-12-30 01:48:15 +00:00
|
|
|
QVariantList WebPage::consoleMessages() {
|
|
|
|
return m_consoleMessages;
|
2011-10-31 17:53:57 +00:00
|
|
|
}
|
|
|
|
|
2012-12-30 19:41:23 +00:00
|
|
|
QVariantList WebPage::alertMessages() {
|
|
|
|
return m_alertMessages;
|
2012-07-10 13:47:17 +00:00
|
|
|
}
|
|
|
|
|
2012-12-30 19:28:45 +00:00
|
|
|
QVariantList WebPage::confirmMessages() {
|
|
|
|
return m_confirmMessages;
|
2012-07-10 13:47:17 +00:00
|
|
|
}
|
|
|
|
|
2012-12-30 20:04:44 +00:00
|
|
|
QVariantList WebPage::promptMessages() {
|
|
|
|
return m_promptMessages;
|
2012-07-10 13:47:17 +00:00
|
|
|
}
|
|
|
|
|
2011-06-25 11:31:19 +00:00
|
|
|
void WebPage::setUserAgent(QString userAgent) {
|
|
|
|
m_userAgent = userAgent;
|
2011-06-08 09:36:45 +00:00
|
|
|
}
|
|
|
|
|
2011-04-19 07:16:25 +00:00
|
|
|
void WebPage::frameCreated(QWebFrame * frame) {
|
|
|
|
connect(frame, SIGNAL(javaScriptWindowObjectCleared()),
|
|
|
|
this, SLOT(injectJavascriptHelpers()));
|
2011-02-26 03:32:22 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
void WebPage::injectJavascriptHelpers() {
|
2011-04-19 07:16:25 +00:00
|
|
|
QWebFrame* frame = qobject_cast<QWebFrame *>(QObject::sender());
|
|
|
|
frame->evaluateJavaScript(m_capybaraJavascript);
|
2011-02-19 03:53:06 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
bool WebPage::shouldInterruptJavaScript() {
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2013-01-20 23:57:04 +00:00
|
|
|
InvocationResult WebPage::invokeCapybaraFunction(const char *name, const QStringList &arguments) {
|
2011-02-26 03:57:55 +00:00
|
|
|
QString qname(name);
|
2012-12-11 03:53:03 +00:00
|
|
|
JavascriptInvocation invocation(qname, arguments, this);
|
2013-01-20 23:57:04 +00:00
|
|
|
return invocation.invoke(currentFrame());
|
2011-02-26 03:57:55 +00:00
|
|
|
}
|
|
|
|
|
2013-01-20 23:57:04 +00:00
|
|
|
InvocationResult WebPage::invokeCapybaraFunction(QString &name, const QStringList &arguments) {
|
2013-01-07 02:59:07 +00:00
|
|
|
return invokeCapybaraFunction(name.toLatin1().data(), arguments);
|
2011-02-26 04:39:29 +00:00
|
|
|
}
|
|
|
|
|
2011-02-26 18:02:43 +00:00
|
|
|
void WebPage::javaScriptConsoleMessage(const QString &message, int lineNumber, const QString &sourceID) {
|
2012-12-30 01:48:15 +00:00
|
|
|
QVariantMap m;
|
|
|
|
m["message"] = message;
|
2012-11-21 01:17:42 +00:00
|
|
|
QString fullMessage = QString(message);
|
2012-12-30 01:48:15 +00:00
|
|
|
if (!sourceID.isEmpty()) {
|
2012-11-21 01:17:42 +00:00
|
|
|
fullMessage = sourceID + "|" + QString::number(lineNumber) + "|" + fullMessage;
|
2012-12-30 01:48:15 +00:00
|
|
|
m["source"] = sourceID;
|
|
|
|
m["line_number"] = lineNumber;
|
|
|
|
}
|
|
|
|
m_consoleMessages.append(m);
|
2012-11-22 15:30:43 +00:00
|
|
|
m_manager->logger() << qPrintable(fullMessage);
|
2011-02-26 18:02:43 +00:00
|
|
|
}
|
|
|
|
|
2011-03-11 17:07:30 +00:00
|
|
|
void WebPage::javaScriptAlert(QWebFrame *frame, const QString &message) {
|
|
|
|
Q_UNUSED(frame);
|
2012-07-10 13:47:17 +00:00
|
|
|
m_alertMessages.append(message);
|
2012-11-22 15:30:43 +00:00
|
|
|
m_manager->logger() << "ALERT:" << qPrintable(message);
|
2011-03-11 17:07:30 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
bool WebPage::javaScriptConfirm(QWebFrame *frame, const QString &message) {
|
|
|
|
Q_UNUSED(frame);
|
2012-07-10 13:47:17 +00:00
|
|
|
m_confirmMessages.append(message);
|
|
|
|
return m_confirm;
|
2011-03-11 17:07:30 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
bool WebPage::javaScriptPrompt(QWebFrame *frame, const QString &message, const QString &defaultValue, QString *result) {
|
|
|
|
Q_UNUSED(frame)
|
2012-07-10 13:47:17 +00:00
|
|
|
m_promptMessages.append(message);
|
|
|
|
if (m_prompt) {
|
|
|
|
if (m_prompt_text.isNull()) {
|
|
|
|
*result = defaultValue;
|
|
|
|
} else {
|
|
|
|
*result = m_prompt_text;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return m_prompt;
|
2011-03-11 17:07:30 +00:00
|
|
|
}
|
|
|
|
|
2011-03-09 05:08:30 +00:00
|
|
|
void WebPage::loadStarted() {
|
|
|
|
m_loading = true;
|
2012-04-04 01:27:41 +00:00
|
|
|
m_errorPageMessage = QString();
|
2012-11-22 04:20:47 +00:00
|
|
|
m_unsupportedContentLoaded = false;
|
2011-03-09 05:08:30 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
void WebPage::loadFinished(bool success) {
|
2012-07-11 03:03:32 +00:00
|
|
|
Q_UNUSED(success);
|
2011-03-09 05:08:30 +00:00
|
|
|
m_loading = false;
|
2012-07-11 03:03:32 +00:00
|
|
|
emit pageFinished(!m_failed);
|
|
|
|
m_failed = false;
|
2011-03-09 05:08:30 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
bool WebPage::isLoading() const {
|
|
|
|
return m_loading;
|
|
|
|
}
|
|
|
|
|
|
|
|
QString WebPage::failureString() {
|
2012-04-04 01:27:41 +00:00
|
|
|
QString message = QString("Unable to load URL: ") + currentFrame()->requestedUrl().toString();
|
2012-04-26 20:05:19 +00:00
|
|
|
if (m_errorPageMessage.isEmpty())
|
2012-04-04 01:27:41 +00:00
|
|
|
return message;
|
|
|
|
else
|
2012-04-26 20:05:19 +00:00
|
|
|
return message + m_errorPageMessage;
|
2011-03-09 05:08:30 +00:00
|
|
|
}
|
|
|
|
|
2013-01-13 22:19:46 +00:00
|
|
|
bool WebPage::render(const QString &fileName, const QSize &minimumSize) {
|
2011-05-26 19:05:16 +00:00
|
|
|
QFileInfo fileInfo(fileName);
|
|
|
|
QDir dir;
|
|
|
|
dir.mkpath(fileInfo.absolutePath());
|
|
|
|
|
|
|
|
QSize viewportSize = this->viewportSize();
|
2013-01-13 22:19:46 +00:00
|
|
|
this->setViewportSize(minimumSize);
|
2011-05-26 19:05:16 +00:00
|
|
|
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);
|
|
|
|
}
|
2011-07-18 21:16:57 +00:00
|
|
|
|
|
|
|
QString WebPage::chooseFile(QWebFrame *parentFrame, const QString &suggestedFile) {
|
|
|
|
Q_UNUSED(parentFrame);
|
|
|
|
Q_UNUSED(suggestedFile);
|
|
|
|
|
2012-11-18 07:48:24 +00:00
|
|
|
return getAttachedFileNames().first();
|
2011-07-18 21:16:57 +00:00
|
|
|
}
|
|
|
|
|
2011-07-20 19:58:23 +00:00
|
|
|
bool WebPage::extension(Extension extension, const ExtensionOption *option, ExtensionReturn *output) {
|
|
|
|
if (extension == ChooseMultipleFilesExtension) {
|
2012-11-18 07:48:24 +00:00
|
|
|
static_cast<ChooseMultipleFilesExtensionReturn*>(output)->fileNames = getAttachedFileNames();
|
2011-07-20 19:58:23 +00:00
|
|
|
return true;
|
|
|
|
}
|
2012-04-04 01:27:41 +00:00
|
|
|
else if (extension == QWebPage::ErrorPageExtension) {
|
|
|
|
ErrorPageExtensionOption *errorOption = (ErrorPageExtensionOption*) option;
|
2012-04-26 20:05:19 +00:00
|
|
|
m_errorPageMessage = " because of error loading " + errorOption->url.toString() + ": " + errorOption->errorString;
|
2012-07-11 03:03:32 +00:00
|
|
|
m_failed = true;
|
2012-04-04 01:27:41 +00:00
|
|
|
return false;
|
|
|
|
}
|
2011-07-20 19:58:23 +00:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2012-11-18 07:48:24 +00:00
|
|
|
QStringList WebPage::getAttachedFileNames() {
|
|
|
|
return currentFrame()->evaluateJavaScript(QString("Capybara.attachedFiles")).toStringList();
|
2011-07-20 19:58:23 +00:00
|
|
|
}
|
2011-08-12 20:50:43 +00:00
|
|
|
|
2012-05-04 17:56:21 +00:00
|
|
|
void WebPage::handleSslErrorsForReply(QNetworkReply *reply, const QList<QSslError> &errors) {
|
2012-03-30 19:00:51 +00:00
|
|
|
if (m_manager->ignoreSslErrors())
|
2011-09-28 16:09:11 +00:00
|
|
|
reply->ignoreSslErrors(errors);
|
2011-09-27 19:31:03 +00:00
|
|
|
}
|
|
|
|
|
2012-03-30 01:38:34 +00:00
|
|
|
void WebPage::setSkipImageLoading(bool skip) {
|
|
|
|
settings()->setAttribute(QWebSettings::AutoLoadImages, !skip);
|
|
|
|
}
|
|
|
|
|
2011-08-12 20:50:43 +00:00
|
|
|
int WebPage::getLastStatus() {
|
2012-11-22 05:21:40 +00:00
|
|
|
return networkAccessManager()->statusFor(currentFrame()->requestedUrl());
|
2011-08-12 20:50:43 +00:00
|
|
|
}
|
2011-08-19 15:57:39 +00:00
|
|
|
|
2012-10-22 02:35:20 +00:00
|
|
|
const QList<QNetworkReply::RawHeaderPair> &WebPage::pageHeaders() {
|
2012-11-22 05:21:40 +00:00
|
|
|
return networkAccessManager()->headersFor(currentFrame()->requestedUrl());
|
|
|
|
}
|
|
|
|
|
|
|
|
NetworkAccessManager *WebPage::networkAccessManager() {
|
|
|
|
return qobject_cast<NetworkAccessManager *>(QWebPage::networkAccessManager());
|
2011-08-19 15:57:39 +00:00
|
|
|
}
|
2011-09-15 15:22:10 +00:00
|
|
|
|
|
|
|
void WebPage::handleUnsupportedContent(QNetworkReply *reply) {
|
2012-07-11 03:03:32 +00:00
|
|
|
QVariant contentMimeType = reply->header(QNetworkRequest::ContentTypeHeader);
|
|
|
|
if(!contentMimeType.isNull()) {
|
|
|
|
triggerAction(QWebPage::Stop);
|
|
|
|
UnsupportedContentHandler *handler = new UnsupportedContentHandler(this, reply);
|
2012-07-12 03:36:22 +00:00
|
|
|
if (reply->isFinished())
|
|
|
|
handler->renderNonHtmlContent();
|
|
|
|
else
|
|
|
|
handler->waitForReplyToFinish();
|
2012-07-11 03:03:32 +00:00
|
|
|
}
|
2011-09-15 15:22:10 +00:00
|
|
|
}
|
2012-03-26 22:41:33 +00:00
|
|
|
|
2012-11-22 04:20:47 +00:00
|
|
|
bool WebPage::unsupportedContentLoaded() {
|
|
|
|
return m_unsupportedContentLoaded;
|
|
|
|
}
|
|
|
|
|
2012-04-26 20:05:19 +00:00
|
|
|
bool WebPage::supportsExtension(Extension extension) const {
|
|
|
|
if (extension == ErrorPageExtension)
|
|
|
|
return true;
|
|
|
|
else if (extension == ChooseMultipleFilesExtension)
|
|
|
|
return true;
|
|
|
|
else
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2012-03-26 22:41:33 +00:00
|
|
|
QWebPage *WebPage::createWindow(WebWindowType type) {
|
|
|
|
Q_UNUSED(type);
|
2012-03-29 03:05:24 +00:00
|
|
|
return m_manager->createPage(this);
|
2012-03-26 22:41:33 +00:00
|
|
|
}
|
2012-03-27 22:49:57 +00:00
|
|
|
|
|
|
|
QString WebPage::uuid() {
|
|
|
|
return m_uuid;
|
|
|
|
}
|
2012-03-28 15:46:07 +00:00
|
|
|
|
|
|
|
QString WebPage::getWindowName() {
|
|
|
|
QVariant windowName = mainFrame()->evaluateJavaScript("window.name");
|
|
|
|
|
|
|
|
if (windowName.isValid())
|
|
|
|
return windowName.toString();
|
|
|
|
else
|
|
|
|
return "";
|
|
|
|
}
|
2012-03-28 19:20:32 +00:00
|
|
|
|
|
|
|
bool WebPage::matchesWindowSelector(QString selector) {
|
|
|
|
return (selector == getWindowName() ||
|
|
|
|
selector == mainFrame()->title() ||
|
|
|
|
selector == mainFrame()->url().toString() ||
|
|
|
|
selector == uuid());
|
|
|
|
}
|
2012-04-04 22:18:17 +00:00
|
|
|
|
|
|
|
void WebPage::setFocus() {
|
|
|
|
m_manager->setCurrentPage(this);
|
|
|
|
}
|
2012-07-10 13:47:17 +00:00
|
|
|
|
|
|
|
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;
|
|
|
|
}
|
|
|
|
|