Refactoring Me2MeNativeMessagingHost to implement NativeMessageHost interface

remoting/host/setup/me2me_native_messaging_host_main.cc

 // Copyright 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "remoting/host/setup/me2me_native_messaging_host_main.h"
 
 #include <cstdint>
 #include <memory>
 #include <string>
 #include <utility>
 
 #include "base/at_exit.h"
 #include "base/command_line.h"
 #include "base/files/file.h"
 #include "base/i18n/icu_util.h"
 #include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/threading/thread.h"
 #include "build/build_config.h"
 #include "net/url_request/url_fetcher.h"
+#include "remoting/base/auto_thread_task_runner.h"
 #include "remoting/base/breakpad.h"
 #include "remoting/base/url_request_context_getter.h"
+#include "remoting/host/chromoting_host_context.h"
 #include "remoting/host/host_exit_codes.h"
 #include "remoting/host/logging.h"
+#include "remoting/host/native_messaging/native_messaging_pipe.h"
 #include "remoting/host/native_messaging/pipe_messaging_channel.h"
 #include "remoting/host/pairing_registry_delegate.h"
 #include "remoting/host/setup/gaia_oauth_client.h"
 #include "remoting/host/setup/me2me_native_messaging_host.h"
 #include "remoting/host/switches.h"
 #include "remoting/host/usage_stats_consent.h"
 
 #if defined(OS_MACOSX)
 #include "base/mac/scoped_nsautorelease_pool.h"
 #endif  // defined(OS_MACOSX)
@@ skipping 187 matching lines @@
   if (!delegate->SetRootKeys(privileged.Take(), unprivileged.Take()))
     return kInitializationFailed;
 
   pairing_registry =
       new PairingRegistry(io_thread.task_runner(), std::move(delegate));
 #else  // defined(OS_WIN)
   pairing_registry =
       CreatePairingRegistry(io_thread.task_runner());
 #endif  // !defined(OS_WIN)
 
+  std::unique_ptr<NativeMessagingPipe> native_messaging_pipe(
+      new NativeMessagingPipe());
+
   // Set up the native messaging channel.
   std::unique_ptr<extensions::NativeMessagingChannel> channel(
       new PipeMessagingChannel(std::move(read_file), std::move(write_file)));
 
+  std::unique_ptr<ChromotingHostContext> context =
+      ChromotingHostContext::Create(new remoting::AutoThreadTaskRunner(
+          message_loop.task_runner(), run_loop.QuitClosure()));
+
   // Create the native messaging host.
-  std::unique_ptr<Me2MeNativeMessagingHost> host(new Me2MeNativeMessagingHost(
-      needs_elevation, static_cast<intptr_t>(native_view_handle),
-      std::move(channel), daemon_controller, pairing_registry,
-      std::move(oauth_client)));
-  host->Start(run_loop.QuitClosure());
+  std::unique_ptr<extensions::NativeMessageHost> host(
+      new Me2MeNativeMessagingHost(needs_elevation,
+                                   static_cast<intptr_t>(native_view_handle),
+                                   std::move(context), daemon_controller,
+                                   pairing_registry, std::move(oauth_client)));
+
+  host->Start(native_messaging_pipe.get());
+
+  native_messaging_pipe->Start(std::move(host), std::move(channel));
 
   // Run the loop until channel is alive.
   run_loop.Run();
   return kSuccessExitCode;
 }
 
 int Me2MeNativeMessagingHostMain(int argc, char** argv) {
   // This object instance is required by Chrome code (such as MessageLoop).
   base::AtExitManager exit_manager;
 
   base::CommandLine::Init(argc, argv);
   remoting::InitHostLogging();
 
   return StartMe2MeNativeMessagingHost();
 }
 
 }  // namespace remoting