electron/shell/browser/api/atom_api_net.cc

// Copyright (c) 2016 GitHub, Inc.
// Use of this source code is governed by the MIT license that can be
// found in the LICENSE file.

#include "shell/browser/api/atom_api_net.h"

#include "native_mate/dictionary.h"
#include "services/network/public/cpp/features.h"
#include "shell/browser/api/atom_api_url_request.h"
#include "shell/browser/api/atom_api_url_request_ns.h"

#include "shell/common/node_includes.h"

namespace electron {

namespace api {

Net::Net(v8::Isolate* isolate) {
  Init(isolate);
}

Net::~Net() {}

// static
v8::Local<v8::Value> Net::Create(v8::Isolate* isolate) {
  return mate::CreateHandle(isolate, new Net(isolate)).ToV8();
}

// static
void Net::BuildPrototype(v8::Isolate* isolate,
                         v8::Local<v8::FunctionTemplate> prototype) {
  prototype->SetClassName(mate::StringToV8(isolate, "Net"));
  mate::ObjectTemplateBuilder(isolate, prototype->PrototypeTemplate())
      .SetProperty("URLRequest", &Net::URLRequest);
}

v8::Local<v8::Value> Net::URLRequest(v8::Isolate* isolate) {
  v8::Local<v8::FunctionTemplate> constructor;
  if (base::FeatureList::IsEnabled(network::features::kNetworkService))
    constructor = URLRequestNS::GetConstructor(isolate);
  else
    constructor = URLRequest::GetConstructor(isolate);
  return constructor->GetFunction(isolate->GetCurrentContext())
      .ToLocalChecked();
}

}  // namespace api

}  // namespace electron

namespace {

using electron::api::Net;
using electron::api::URLRequest;
using electron::api::URLRequestNS;

void Initialize(v8::Local<v8::Object> exports,
                v8::Local<v8::Value> unused,
                v8::Local<v8::Context> context,
                void* priv) {
  v8::Isolate* isolate = context->GetIsolate();

  if (base::FeatureList::IsEnabled(network::features::kNetworkService))
    URLRequestNS::SetConstructor(isolate,
                                 base::BindRepeating(URLRequestNS::New));
  else
    URLRequest::SetConstructor(isolate, base::BindRepeating(URLRequest::New));

  mate::Dictionary dict(isolate, exports);
  dict.Set("net", Net::Create(isolate));
  dict.Set("Net",
           Net::GetConstructor(isolate)->GetFunction(context).ToLocalChecked());
  dict.Set("isNetworkServiceEnabled",
           base::FeatureList::IsEnabled(network::features::kNetworkService));
}

}  // namespace

NODE_LINKED_MODULE_CONTEXT_AWARE(atom_browser_net, Initialize)