| Left: | ||
| Right: |
| OLD | NEW |
|---|---|
| (Empty) | |
| 1 #include <map> | |
| 2 #include <AdblockPlus.h> | |
| 3 #include "WebRequestJsObject.h" | |
| 4 #include "Thread.h" | |
| 5 | |
| 6 namespace | |
| 7 { | |
| 8 std::string fromV8String(v8::Handle<v8::Value> value) | |
| 9 { | |
| 10 v8::String::Utf8Value stringValue(value); | |
| 11 if (stringValue.length()) | |
| 12 return std::string(*stringValue, stringValue.length()); | |
| 13 else | |
| 14 return std::string(); | |
| 15 } | |
| 16 | |
| 17 v8::Local<v8::String> toV8String(const std::string& str) | |
| 18 { | |
| 19 return v8::String::New(str.c_str(), str.length()); | |
| 20 } | |
| 21 | |
| 22 class WebRequestThread : public AdblockPlus::Thread | |
| 23 { | |
| 24 public: | |
| 25 WebRequestThread(const v8::Arguments& arguments) | |
| 26 : isolate(v8::Isolate::GetCurrent()), | |
| 27 context(v8::Persistent<v8::Context>::New(isolate, v8::Context::GetCurr ent())), | |
| 28 thisPtr(v8::Persistent<v8::Object>::New(isolate, arguments.Holder())), | |
| 29 url(fromV8String(arguments[0])) | |
| 30 { | |
| 31 const v8::Locker locker(isolate); | |
| 32 const v8::HandleScope handleScope; | |
| 33 | |
| 34 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.
| |
| 35 throw std::runtime_error("Invalid string passed as first argument to GET "); | |
| 36 | |
| 37 { | |
| 38 const v8::Local<v8::Value> value = arguments[1]; | |
| 39 if (!value->IsObject()) | |
| 40 throw std::runtime_error("Second argument to GET must be an object"); | |
| 41 const v8::Local<v8::Object> object = v8::Local<v8::Object>::Cast(value); | |
| 42 const v8::Local<v8::Array> properties = object->GetOwnPropertyNames(); | |
| 43 for (unsigned i = 0; i < properties->Length(); i++) | |
| 44 { | |
| 45 const v8::Local<v8::Value> property = properties->Get(i); | |
| 46 std::string header = fromV8String(property); | |
| 47 std::string headerValue = fromV8String(object->Get(property)); | |
| 48 if (header.length() && headerValue.length()) | |
| 49 headers.push_back(std::pair<std::string, std::string>(header, header Value)); | |
| 50 } | |
| 51 } | |
| 52 | |
| 53 const v8::Local<v8::Value> callbackValue = arguments[2]; | |
| 54 if (!callbackValue->IsFunction()) | |
| 55 throw std::runtime_error("Third argument to GET must be a function"); | |
| 56 callback = v8::Persistent<v8::Function>::New(isolate, | |
| 57 v8::Local<v8::Function>::Cast(callbackValue)); | |
| 58 | |
| 59 const v8::Local<const v8::External> external = | |
| 60 v8::Local<const v8::External>::Cast(arguments.Data()); | |
| 61 webRequest = static_cast<AdblockPlus::WebRequest* const>(external->Value() ); | |
| 62 } | |
| 63 | |
| 64 ~WebRequestThread() | |
| 65 { | |
| 66 context.Dispose(isolate); | |
| 67 thisPtr.Dispose(isolate); | |
| 68 callback.Dispose(isolate); | |
| 69 } | |
| 70 | |
| 71 void Run() | |
| 72 { | |
| 73 AdblockPlus::ServerResponse result = webRequest->GET(url, headers); | |
| 74 | |
| 75 const v8::Locker locker(isolate); | |
| 76 const v8::HandleScope handleScope; | |
| 77 const v8::Context::Scope contextScope(context); | |
| 78 v8::Local<v8::Object> resultObject = v8::Object::New(); | |
| 79 resultObject->Set(v8::String::New("responseStatus"), v8::Integer::New(resu lt.responseStatus)); | |
| 80 resultObject->Set(v8::String::New("responseText"), toV8String(result.respo nseText)); | |
| 81 | |
| 82 v8::Local<v8::Object> headersObject = v8::Object::New(); | |
| 83 for (AdblockPlus::HeadersList::iterator it = result.responseHeaders.begin( ); | |
| 84 it != result.responseHeaders.end(); ++it) | |
| 85 { | |
| 86 headersObject->Set(toV8String(it->first), toV8String(it->second)); | |
| 87 } | |
| 88 resultObject->Set(v8::String::New("responseHeaders"), headersObject); | |
| 89 | |
| 90 v8::Local<v8::Value> resultValue = resultObject; | |
| 91 callback->Call(thisPtr, 1, &resultValue); | |
| 92 delete this; | |
| 93 } | |
| 94 | |
| 95 private: | |
| 96 v8::Isolate* const isolate; | |
| 97 v8::Persistent<v8::Context> context; | |
| 98 v8::Persistent<v8::Object> thisPtr; | |
| 99 std::string url; | |
| 100 AdblockPlus::HeadersList headers; | |
| 101 v8::Persistent<v8::Function> callback; | |
| 102 AdblockPlus::WebRequest* webRequest; | |
| 103 }; | |
| 104 | |
| 105 v8::Handle<v8::Value> GETCallback(const v8::Arguments& arguments) | |
| 106 { | |
| 107 WebRequestThread* thread; | |
| 108 try | |
| 109 { | |
| 110 if (arguments.Length() != 3u) | |
| 111 throw std::runtime_error("GET requires exactly 3 arguments"); | |
| 112 thread = new WebRequestThread(arguments); | |
| 113 } | |
| 114 catch (const std::exception& e) | |
| 115 { | |
| 116 return v8::ThrowException(v8::String::New(e.what())); | |
| 117 } | |
| 118 thread->Start(); | |
| 119 return v8::Undefined(); | |
| 120 } | |
| 121 } | |
| 122 | |
| 123 v8::Handle<v8::ObjectTemplate> AdblockPlus::WebRequestJsObject::Create( | |
| 124 AdblockPlus::WebRequest& webRequest) | |
| 125 { | |
| 126 v8::HandleScope handleScope; | |
| 127 const v8::Handle<v8::ObjectTemplate> request = v8::ObjectTemplate::New(); | |
| 128 const v8::Handle<v8::FunctionTemplate> getFunction = | |
| 129 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
| |
| 130 v8::External::New(&webRequest)); | |
| 131 request->Set(v8::String::New("GET"), getFunction); | |
| 132 return handleScope.Close(request); | |
| 133 } | |
| OLD | NEW |