/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */

/* vim: set ts=8 sts=2 et sw=2 tw=80: */

/* 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_PresentationSessionTransport_h

#define mozilla_dom_PresentationSessionTransport_h

#include "mozilla/RefPtr.h"

#include "nsCOMPtr.h"

#include "nsIAsyncInputStream.h"

#include "nsIPresentationSessionTransport.h"

#include "nsIPresentationSessionTransportBuilder.h"

#include "nsIStreamListener.h"

#include "nsISupportsImpl.h"

#include "nsITransport.h"

class nsISocketTransport;

class nsIInputStreamPump;

class nsIScriptableInputStream;

class nsIMultiplexInputStream;

class nsIAsyncStreamCopier;

class nsIInputStream;

namespace mozilla {

namespace dom {

/*

 * App-to-App transport channel for the presentation session. It's usually

 * initialized with an |InitWithSocketTransport| call if at the presenting sender

 * side; whereas it's initialized with an |InitWithChannelDescription| if at the

 * presenting receiver side. The lifetime is managed in either

 * |PresentationControllingInfo| (sender side) or |PresentationPresentingInfo|

 * (receiver side) in PresentationSessionInfo.cpp.

 */

class PresentationTCPSessionTransport final : public nsIPresentationSessionTransport

                                            , public nsIPresentationTCPSessionTransportBuilder

                                            , public nsITransportEventSink

                                            , public nsIInputStreamCallback

                                            , public nsIStreamListener

{

public:

  NS_DECL_CYCLE_COLLECTING_ISUPPORTS

  NS_DECL_CYCLE_COLLECTION_CLASS_AMBIGUOUS(PresentationTCPSessionTransport,

                                           nsIPresentationSessionTransport)

  NS_DECL_NSIPRESENTATIONSESSIONTRANSPORT

  NS_DECL_NSIPRESENTATIONSESSIONTRANSPORTBUILDER

  NS_DECL_NSIPRESENTATIONTCPSESSIONTRANSPORTBUILDER

  NS_DECL_NSITRANSPORTEVENTSINK

  NS_DECL_NSIINPUTSTREAMCALLBACK

  NS_DECL_NSIREQUESTOBSERVER

  NS_DECL_NSISTREAMLISTENER

  PresentationTCPSessionTransport();

  void NotifyCopyComplete(nsresult aStatus);

private:

  ~PresentationTCPSessionTransport();

  nsresult CreateStream();

  nsresult CreateInputStreamPump();

  nsresult EnsureCopying();

  enum class ReadyState {

    CONNECTING,

    OPEN,

    CLOSING,

    CLOSED

  };

  void SetReadyState(ReadyState aReadyState);

  bool IsReadyToNotifyData()

  {

    return mDataNotificationEnabled && mReadyState == ReadyState::OPEN;

  }

  ReadyState mReadyState;

  bool mAsyncCopierActive;

  nsresult mCloseStatus;

  bool mDataNotificationEnabled;

  uint8_t mRole = 0;

  // Raw socket streams

  nsCOMPtr<nsISocketTransport> mTransport;

  nsCOMPtr<nsIInputStream> mSocketInputStream;

  nsCOMPtr<nsIOutputStream> mSocketOutputStream;

  // Input stream machinery

  nsCOMPtr<nsIInputStreamPump> mInputStreamPump;

  nsCOMPtr<nsIScriptableInputStream> mInputStreamScriptable;

  nsCOMPtr<nsIPresentationSessionTransportCallback> mCallback;

  nsCOMPtr<nsIPresentationSessionTransportBuilderListener> mListener;

  // The data to be sent.

  nsTArray<nsCOMPtr<nsIInputStream>> mPendingData;

};

} // namespace dom

} // namespace mozilla

#endif // mozilla_dom_PresentationSessionTransport_h