capybara-webkit/src/SetUnknownUrlMode.cpp

28 lines
768 B
C++

#include "SetUnknownUrlMode.h"
#include "WebPage.h"
#include "WebPageManager.h"
#include "ErrorMessage.h"
SetUnknownUrlMode::SetUnknownUrlMode(WebPageManager *manager, QStringList &arguments, QObject *parent) : SocketCommand(manager, arguments, parent) {
}
void SetUnknownUrlMode::start() {
QString modeString = arguments()[0];
QStringList modes;
modes << "warn" << "block";
switch(modes.indexOf(modeString)) {
case 0:
manager()->setUnknownUrlMode(UnknownUrlHandler::WARN);
finish(true);
break;
case 1:
manager()->setUnknownUrlMode(UnknownUrlHandler::BLOCK);
finish(true);
break;
default:
QString error = QString("Invalid mode string:") + modeString;
finish(false, new ErrorMessage(error));
}
}