capybara-webkit/src/EvaluateAsync.cpp

55 lines
2.2 KiB
C++

#include "EvaluateAsync.h"
#include "WebPage.h"
#include "WebPageManager.h"
#include "JsonSerializer.h"
#include <iostream>
EvaluateAsync::EvaluateAsync(WebPageManager *manager, QStringList &arguments, QObject *parent) : SocketCommand(manager, arguments, parent) {
}
void EvaluateAsync::start() {
QString script = arguments()[0];
QString jsonArgs;
if (arguments().length()>1){
jsonArgs = arguments()[1];
} else {
jsonArgs ="[]";
}
QString eval_script = QString("(function(){"
" for(var i=0; i<arguments.length-1; i++) {"
" arguments[i] = JSON.parse(arguments[i]);"
" var elem_id = arguments[i]['element-581e-422e-8be1-884c4e116226'];"
" if (elem_id) {"
" arguments[i] = Capybara.getNode(elem_id);"
" };"
" };"
" eval(\"%1\");"
" return;"
" }).apply(null, %2.concat(function(res){ "
" res = Capybara.wrapResult(res);"
// This isn't necessary in Qt 5.5 but is in older (at least 5.2 on Travis)
" if (res instanceof Array) {"
" CapybaraAsync['asyncResult(QVariantList)'](res);"
" } else { "
" CapybaraAsync['asyncResult(QVariant)'](res);"
" }"
" }))").arg(script.replace("\"","\\\"").remove("\n"), jsonArgs);
QObject invocation_stub;
invocation_stub.setProperty("allowUnattached", false);
page()->currentFrame()->addToJavaScriptWindowObject("CapybaraInvocation", &invocation_stub);
page()->currentFrame()->addToJavaScriptWindowObject("CapybaraAsync", this);
page()->currentFrame()->evaluateJavaScript(eval_script);
}
void EvaluateAsync::asyncResult(QVariantList result) {
JsonSerializer serializer;
finish(true, serializer.serialize(result));
}
void EvaluateAsync::asyncResult(QVariant result) {
JsonSerializer serializer;
finish(true, serializer.serialize(result));
}