extensions/browser/api/messaging/native_message_port.h

// Copyright 2012 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef EXTENSIONS_BROWSER_API_MESSAGING_NATIVE_MESSAGE_PORT_H_
#define EXTENSIONS_BROWSER_API_MESSAGING_NATIVE_MESSAGE_PORT_H_

#include <memory>
#include <string>

#include "base/memory/ref_counted.h"
#include "base/memory/weak_ptr.h"
#include "base/threading/thread_checker.h"
#include "extensions/browser/api/messaging/message_port.h"

namespace base {
class SingleThreadTaskRunner;
}

namespace extensions {
class NativeMessageHost;
class NativeMessagePortDispatcher;
struct Message;
struct PortId;

// A port that manages communication with a native application.
// All methods must be called on the UI Thread of the browser process.
class NativeMessagePort : public MessagePort {
 public:
  NativeMessagePort(base::WeakPtr<ChannelDelegate> channel_delegate,
                    const PortId& port_id,
                    std::unique_ptr<NativeMessageHost> native_message_host);
  ~NativeMessagePort() override;

  // MessagePort implementation.
  bool IsValidPort() override;
  void DispatchOnMessage(const Message& message) override;

  void PostMessageFromNativeHost(const std::string& message);
  void CloseChannel(const std::string& error_message);

 private:
  base::ThreadChecker thread_checker_;
  scoped_refptr<base::SingleThreadTaskRunner> host_task_runner_;
  std::unique_ptr<NativeMessagePortDispatcher> dispatcher_;

  base::WeakPtrFactory<NativeMessagePort> weak_factory_{this};
};

}  // namespace extensions

#endif  // EXTENSIONS_BROWSER_API_MESSAGING_NATIVE_MESSAGE_PORT_H_