adding support for kLogNetLog switch

brightray/browser/net_log.cc

@@ -0,0 +1,66 @@
+// Copyright (c) 2015 GitHub, Inc.
+// Use of this source code is governed by the MIT license that can be
+// found in the LICENSE file.
+
+#include "browser/net_log.h"
+
+#include "browser/browser_context.h"
+#include "base/command_line.h"
+#include "base/files/file_util.h"
+#include "base/json/json_writer.h"
+#include "base/logging.h"
+#include "content/public/common/content_switches.h"
+#include "net/log/net_log_util.h"
+#include "net/url_request/url_request_context.h"
+
+namespace brightray {
+
+NetLog::NetLog(net::URLRequestContext* context)
+    : added_events_(false),
+      context_(context) {
+  auto command_line = base::CommandLine::ForCurrentProcess();
+
+  if (command_line->HasSwitch(switches::kLogNetLog)) {
+    base::FilePath log_path =
+        command_line->GetSwitchValuePath(switches::kLogNetLog);
+
+    #if defined(OS_WIN)
+      log_file_.reset(_wfopen(log_path.value().c_str(), L"w"));
+    #elif defined(OS_POSIX)
+      log_file_.reset(fopen(log_path.value().c_str(), "w"));
+    #endif
+
+    if (!log_file_)
+      LOG(ERROR) << "Could not open file: " << log_path.value()
+                 << "for net logging";
+
+    fprintf(log_file_.get(), "{\"events\": [\n");
+
+    if (context_.get()) {
+      DCHECK(context_->CalledOnValidThread());
+
+      std::set<net::URLRequestContext*> contexts;
+      contexts.insert(context_.get());
+
+      net::CreateNetLogEntriesForActiveObjects(contexts, this);
+    }
+
+    DeprecatedAddObserver(this, net::NetLog::LogLevel::LOG_STRIP_PRIVATE_DATA);
+  }
+}
+
+NetLog::~NetLog() {
+  DeprecatedRemoveObserver(this);
+  fprintf(log_file_.get(), "]}");
+  log_file_.reset();
+}
+
+void NetLog::OnAddEntry(const net::NetLog::Entry& entry) {
+  std::string json;
+  base::JSONWriter::Write(entry.ToValue(), &json);
+
+  fprintf(log_file_.get(), "%s%s", (added_events_ ? ",\n" : ""), json.c_str());
+  added_events_ = true;
+}
+
+}  // namespace brightray