| Index: src/WebRequestJsObject.cpp |
| =================================================================== |
| new file mode 100644 |
| --- /dev/null |
| +++ b/src/WebRequestJsObject.cpp |
| @@ -0,0 +1,133 @@ |
| +#include <map> |
| +#include <AdblockPlus.h> |
| +#include "WebRequestJsObject.h" |
| +#include "Thread.h" |
| + |
| +namespace |
| +{ |
| + std::string fromV8String(v8::Handle<v8::Value> value) |
| + { |
| + v8::String::Utf8Value stringValue(value); |
| + if (stringValue.length()) |
| + return std::string(*stringValue, stringValue.length()); |
| + else |
| + return std::string(); |
| + } |
| + |
| + v8::Local<v8::String> toV8String(const std::string& str) |
| + { |
| + return v8::String::New(str.c_str(), str.length()); |
| + } |
| + |
| + class WebRequestThread : public AdblockPlus::Thread |
| + { |
| + public: |
| + WebRequestThread(const v8::Arguments& arguments) |
| + : isolate(v8::Isolate::GetCurrent()), |
| + context(v8::Persistent<v8::Context>::New(isolate, v8::Context::GetCurrent())), |
| + thisPtr(v8::Persistent<v8::Object>::New(isolate, arguments.Holder())), |
| + url(fromV8String(arguments[0])) |
| + { |
| + const v8::Locker locker(isolate); |
| + const v8::HandleScope handleScope; |
| + |
| + if (!url.length()) |
|
Felix Dahlke
2013/04/11 09:33:42
Another refactoring I'm planning to do is to not p
Wladimir Palant
2013/04/11 16:32:33
I'll leave it like that for now.
|
| + throw std::runtime_error("Invalid string passed as first argument to GET"); |
| + |
| + { |
| + const v8::Local<v8::Value> value = arguments[1]; |
| + if (!value->IsObject()) |
| + throw std::runtime_error("Second argument to GET must be an object"); |
| + const v8::Local<v8::Object> object = v8::Local<v8::Object>::Cast(value); |
| + const v8::Local<v8::Array> properties = object->GetOwnPropertyNames(); |
| + for (unsigned i = 0; i < properties->Length(); i++) |
| + { |
| + const v8::Local<v8::Value> property = properties->Get(i); |
| + std::string header = fromV8String(property); |
| + std::string headerValue = fromV8String(object->Get(property)); |
| + if (header.length() && headerValue.length()) |
| + headers.push_back(std::pair<std::string, std::string>(header, headerValue)); |
| + } |
| + } |
| + |
| + const v8::Local<v8::Value> callbackValue = arguments[2]; |
| + if (!callbackValue->IsFunction()) |
| + throw std::runtime_error("Third argument to GET must be a function"); |
| + callback = v8::Persistent<v8::Function>::New(isolate, |
| + v8::Local<v8::Function>::Cast(callbackValue)); |
| + |
| + const v8::Local<const v8::External> external = |
| + v8::Local<const v8::External>::Cast(arguments.Data()); |
| + webRequest = static_cast<AdblockPlus::WebRequest* const>(external->Value()); |
| + } |
| + |
| + ~WebRequestThread() |
| + { |
| + context.Dispose(isolate); |
| + thisPtr.Dispose(isolate); |
| + callback.Dispose(isolate); |
| + } |
| + |
| + void Run() |
| + { |
| + AdblockPlus::ServerResponse result = webRequest->GET(url, headers); |
| + |
| + const v8::Locker locker(isolate); |
| + const v8::HandleScope handleScope; |
| + const v8::Context::Scope contextScope(context); |
| + v8::Local<v8::Object> resultObject = v8::Object::New(); |
| + resultObject->Set(v8::String::New("responseStatus"), v8::Integer::New(result.responseStatus)); |
| + resultObject->Set(v8::String::New("responseText"), toV8String(result.responseText)); |
| + |
| + v8::Local<v8::Object> headersObject = v8::Object::New(); |
| + for (AdblockPlus::HeadersList::iterator it = result.responseHeaders.begin(); |
| + it != result.responseHeaders.end(); ++it) |
| + { |
| + headersObject->Set(toV8String(it->first), toV8String(it->second)); |
| + } |
| + resultObject->Set(v8::String::New("responseHeaders"), headersObject); |
| + |
| + v8::Local<v8::Value> resultValue = resultObject; |
| + callback->Call(thisPtr, 1, &resultValue); |
| + delete this; |
| + } |
| + |
| + private: |
| + v8::Isolate* const isolate; |
| + v8::Persistent<v8::Context> context; |
| + v8::Persistent<v8::Object> thisPtr; |
| + std::string url; |
| + AdblockPlus::HeadersList headers; |
| + v8::Persistent<v8::Function> callback; |
| + AdblockPlus::WebRequest* webRequest; |
| + }; |
| + |
| + v8::Handle<v8::Value> GETCallback(const v8::Arguments& arguments) |
| + { |
| + WebRequestThread* thread; |
| + try |
| + { |
| + if (arguments.Length() != 3u) |
| + throw std::runtime_error("GET requires exactly 3 arguments"); |
| + thread = new WebRequestThread(arguments); |
| + } |
| + catch (const std::exception& e) |
| + { |
| + return v8::ThrowException(v8::String::New(e.what())); |
| + } |
| + thread->Start(); |
| + return v8::Undefined(); |
| + } |
| +} |
| + |
| +v8::Handle<v8::ObjectTemplate> AdblockPlus::WebRequestJsObject::Create( |
| + AdblockPlus::WebRequest& webRequest) |
| +{ |
| + v8::HandleScope handleScope; |
| + const v8::Handle<v8::ObjectTemplate> request = v8::ObjectTemplate::New(); |
| + const v8::Handle<v8::FunctionTemplate> getFunction = |
| + v8::FunctionTemplate::New(::GETCallback, |
|
Felix Dahlke
2013/04/11 09:33:42
The :: is not necessary here, was just necessary i
Wladimir Palant
2013/04/11 16:32:33
:: is actually considered good style to indicate r
|
| + v8::External::New(&webRequest)); |
| + request->Set(v8::String::New("GET"), getFunction); |
| + return handleScope.Close(request); |
| +} |