Make the completion callback optional

atom/browser/api/atom_api_protocol.cc

@@ -49,13 +49,13 @@ mate::ObjectTemplateBuilder Protocol::GetObjectTemplateBuilder(
   return mate::ObjectTemplateBuilder(isolate)
       .SetMethod("registerStandardSchemes", &Protocol::RegisterStandardSchemes)
       .SetMethod("registerStringProtocol",
-                 &Protocol::RegisterProtocol<URLRequestStringJob>)
+                 &Protocol::JavaScriptRegisterProtocol<URLRequestStringJob>)
       .SetMethod("registerBufferProtocol",
-                 &Protocol::RegisterProtocol<URLRequestBufferJob>)
+                 &Protocol::JavaScriptRegisterProtocol<URLRequestBufferJob>)
       .SetMethod("registerFileProtocol",
-                 &Protocol::RegisterProtocol<UrlRequestAsyncAsarJob>)
+                 &Protocol::JavaScriptRegisterProtocol<UrlRequestAsyncAsarJob>)
       .SetMethod("registerHttpProtocol",
-                 &Protocol::RegisterProtocol<URLRequestFetchJob>);
+                 &Protocol::JavaScriptRegisterProtocol<URLRequestFetchJob>);
 }
 
 void Protocol::RegisterStandardSchemes(
@@ -65,6 +65,10 @@ void Protocol::RegisterStandardSchemes(
 
 void Protocol::OnIOCompleted(
     const CompletionCallback& callback, ProtocolError error) {
+  // The completion callback is optional.
+  if (callback.is_null())
+    return;
+
   v8::Locker locker(isolate());
   v8::HandleScope handle_scope(isolate());
 

atom/browser/api/atom_api_protocol.h

@@ -12,6 +12,8 @@
 #include "atom/browser/net/atom_url_request_job_factory.h"
 #include "base/callback.h"
 #include "content/public/browser/browser_thread.h"
+#include "native_mate/arguments.h"
+#include "native_mate/dictionary.h"
 #include "native_mate/handle.h"
 #include "native_mate/wrappable.h"
 
@@ -106,6 +108,25 @@ class Protocol : public mate::Wrappable {
       return PROTOCOL_FAIL;
   }
 
+  // Parse optional parameters for registerProtocol.
+  template<typename RequestJob>
+  void JavaScriptRegisterProtocol(v8::Isolate* isolate,
+                                  const std::string& scheme,
+                                  mate::Arguments* args) {
+    // protocol.registerProtocol(scheme[, options], handler[, callback]);
+    mate::Dictionary options = mate::Dictionary::CreateEmpty(isolate);
+    Handler handler;
+    CompletionCallback callback;
+    args->GetNext(&options);
+    if (!args->GetNext(&handler)) {
+      args->ThrowError();
+      return;
+    }
+    args->GetNext(&callback);
+
+    RegisterProtocol<RequestJob>(isolate, scheme, handler, callback);
+  }
+
   // Convert error code to JS exception and call the callback.
   void OnIOCompleted(const CompletionCallback& callback, ProtocolError error);
 

atom/browser/net/asar/url_request_asar_job.cc

@@ -5,6 +5,7 @@
 #include "atom/browser/net/asar/url_request_asar_job.h"
 
 #include <string>
+#include <vector>
 
 #include "base/bind.h"
 #include "base/files/file_util.h"

atom/browser/net/js_asker.cc

@@ -4,6 +4,8 @@
 
 #include "atom/browser/net/js_asker.h"
 
+#include <vector>
+
 #include "atom/common/native_mate_converters/callback.h"
 #include "atom/common/native_mate_converters/v8_value_converter.h"
 #include "native_mate/function_template.h"

atom/browser/net/url_request_string_job.h

@@ -5,11 +5,11 @@
 #ifndef ATOM_BROWSER_NET_URL_REQUEST_STRING_JOB_H_
 #define ATOM_BROWSER_NET_URL_REQUEST_STRING_JOB_H_
 
+#include <string>
+
 #include "atom/browser/net/js_asker.h"
 #include "net/url_request/url_request_simple_job.h"
 
-#include <string>
-
 namespace atom {
 
 class URLRequestStringJob : public JsAsker<net::URLRequestSimpleJob> {