capybara-webkit/src/FrameFocus.cpp

75 lines
1.7 KiB
C++

#include "FrameFocus.h"
#include "SocketCommand.h"
#include "WebPage.h"
#include "WebPageManager.h"
#include "ErrorMessage.h"
FrameFocus::FrameFocus(WebPageManager *manager, QStringList &arguments, QObject *parent) : SocketCommand(manager, arguments, parent) {
}
void FrameFocus::start() {
switch(arguments().length()) {
case 1:
focusId(arguments()[0]);
break;
case 2:
focusIndex(arguments()[1].toInt());
break;
default:
focusParent();
}
}
void FrameFocus::findFrames() {
frames = page()->currentFrame()->childFrames();
}
void FrameFocus::focusIndex(int index) {
findFrames();
if (isFrameAtIndex(index)) {
frames[index]->setFocus();
page()->setCurrentFrameParent(frames[index]->parentFrame());
success();
} else {
frameNotFound();
}
}
bool FrameFocus::isFrameAtIndex(int index) {
return 0 <= index && index < frames.length();
}
void FrameFocus::focusId(QString name) {
findFrames();
for (int i = 0; i < frames.length(); i++) {
if (frames[i]->frameName().compare(name) == 0) {
frames[i]->setFocus();
page()->setCurrentFrameParent(frames[i]->parentFrame());
success();
return;
}
}
frameNotFound();
}
void FrameFocus::focusParent() {
// if (page()->currentFrame()->parentFrame() == 0) {
if (page()->currentFrameParent() == 0) {
finish(false, new ErrorMessage("Already at parent frame."));
} else {
// page()->currentFrame()->parentFrame()->setFocus();
page()->currentFrameParent()->setFocus();
page()->setCurrentFrameParent(page()->currentFrameParent()->parentFrame());
success();
}
}
void FrameFocus::frameNotFound() {
finish(false, new ErrorMessage("Unable to locate frame."));
}
void FrameFocus::success() {
finish(true);
}