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