capybara-webkit/src/FrameFocus.cpp

#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);
}