/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this file,
 * You can obtain one at http://mozilla.org/MPL/2.0/. */

#ifndef mozilla_dom_TCPServerSocket_h
#define mozilla_dom_TCPServerSocket_h

#include "mozilla/DOMEventTargetHelper.h"
#include "nsIServerSocket.h"

namespace mozilla {
class ErrorResult;
namespace dom {

struct ServerSocketOptions;
class GlobalObject;
class TCPSocket;
class TCPSocketChild;
class TCPServerSocketChild;
class TCPServerSocketParent;

class TCPServerSocket final : public DOMEventTargetHelper,
                              public nsIServerSocketListener {
 public:
  TCPServerSocket(nsIGlobalObject* aGlobal, uint16_t aPort,
                  bool aUseArrayBuffers, uint16_t aBacklog);

  NS_DECL_ISUPPORTS_INHERITED
  NS_DECL_CYCLE_COLLECTION_SCRIPT_HOLDER_CLASS_INHERITED(TCPServerSocket,
                                                         DOMEventTargetHelper)
  NS_DECL_NSISERVERSOCKETLISTENER

  nsIGlobalObject* GetParentObject() const { return GetRelevantGlobal(); }
  JSObject* WrapObject(JSContext*, JS::Handle<JSObject*> aGivenProto) override;

  nsresult Init();

  uint16_t LocalPort() const;
  void Close();

  static already_AddRefed<TCPServerSocket> Constructor(
      const GlobalObject& aGlobal, uint16_t aPort,
      const ServerSocketOptions& aOptions, uint16_t aBacklog,
      mozilla::ErrorResult& aRv);

  IMPL_EVENT_HANDLER(connect);
  IMPL_EVENT_HANDLER(error);

  // Relay an accepted socket notification from the parent process and
  // initialize this object with an existing child actor for the new socket.
  nsresult AcceptChildSocket(TCPSocketChild* aSocketChild);
  // Associate this object with an IPC actor in the parent process to relay
  // notifications to content processes.
  void SetServerBridgeParent(TCPServerSocketParent* aBridgeParent);

 private:
  ~TCPServerSocket();
  // Dispatch a TCPServerSocketEvent event of a given type at this object.
  void FireEvent(const nsAString& aType, TCPSocket* aSocket);

  // The server socket associated with this object.
  nsCOMPtr<nsIServerSocket> mServerSocket;
  // The IPC actor in the content process.
  RefPtr<TCPServerSocketChild> mServerBridgeChild;
  // The IPC actor in the parent process.
  RefPtr<TCPServerSocketParent> mServerBridgeParent;
  int32_t mPort;
  uint16_t mBacklog;
  // True if any accepted sockets should use array buffers for received
  // messages.
  bool mUseArrayBuffers;
};

}  // namespace dom
}  // namespace mozilla

#endif  // mozilla_dom_TCPServerSocket_h