/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* 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

  nsPIDOMWindow* GetParentObject() const
  {
    return GetOwner();
  }

  virtual JSObject* WrapObject(JSContext* aCx, JS::Handle<JSObject*> aGivenProto) override;

  nsresult Init();

  uint16_t LocalPort();
  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.
  nsRefPtr<TCPServerSocketChild> mServerBridgeChild;
  // The IPC actor in the parent process.
  nsRefPtr<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