/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
/* vim:set ts=4 sw=4 sts=4 et cindent: */
/* 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 nsFTPChannel_h___
#define nsFTPChannel_h___

#include "nsBaseChannel.h"

#include "nsString.h"
#include "nsCOMPtr.h"
#include "nsIChannelWithDivertableParentListener.h"
#include "nsIFTPChannel.h"
#include "nsIForcePendingChannel.h"
#include "nsIUploadChannel.h"
#include "nsIProxyInfo.h"
#include "nsIProxiedChannel.h"
#include "nsIResumableChannel.h"

class nsIURI;
using mozilla::net::ADivertableParentChannel;

class nsFtpChannel final : public nsBaseChannel,
                           public nsIFTPChannel,
                           public nsIUploadChannel,
                           public nsIResumableChannel,
                           public nsIProxiedChannel,
                           public nsIForcePendingChannel,
                           public nsIChannelWithDivertableParentListener {
 public:
  NS_DECL_ISUPPORTS_INHERITED
  NS_DECL_NSIUPLOADCHANNEL
  NS_DECL_NSIRESUMABLECHANNEL
  NS_DECL_NSIPROXIEDCHANNEL
  NS_DECL_NSICHANNELWITHDIVERTABLEPARENTLISTENER

  nsFtpChannel(nsIURI *uri, nsIProxyInfo *pi)
      : mProxyInfo(pi),
        mStartPos(0),
        mResumeRequested(false),
        mLastModifiedTime(0),
        mForcePending(false) {
    SetURI(uri);
  }

  void UpdateURI(nsIURI *aURI) {
    MOZ_DIAGNOSTIC_ASSERT(NS_IsMainThread(), "Not thread-safe.");
    mURI = aURI;
  }

  nsIProxyInfo *ProxyInfo() { return mProxyInfo; }

  void SetProxyInfo(nsIProxyInfo *pi) { mProxyInfo = pi; }

  NS_IMETHOD IsPending(bool *result) override;

  // This is a short-cut to calling nsIRequest::IsPending().
  // Overrides Pending in nsBaseChannel.
  bool Pending() const override;

  // Were we asked to resume a download?
  bool ResumeRequested() { return mResumeRequested; }

  // Download from this byte offset
  uint64_t StartPos() { return mStartPos; }

  // ID of the entity to resume downloading
  const nsCString &EntityID() { return mEntityID; }
  void SetEntityID(const nsACString &entityID) { mEntityID = entityID; }

  NS_IMETHOD GetLastModifiedTime(PRTime *lastModifiedTime) override {
    *lastModifiedTime = mLastModifiedTime;
    return NS_OK;
  }

  NS_IMETHOD SetLastModifiedTime(PRTime lastModifiedTime) override {
    mLastModifiedTime = lastModifiedTime;
    return NS_OK;
  }

  // Data stream to upload
  nsIInputStream *UploadStream() { return mUploadStream; }

  // Helper function for getting the nsIFTPEventSink.
  void GetFTPEventSink(nsCOMPtr<nsIFTPEventSink> &aResult);

  NS_IMETHOD Suspend() override;
  NS_IMETHOD Resume() override;

 public:
  NS_IMETHOD ForcePending(bool aForcePending) override;

 protected:
  virtual ~nsFtpChannel() {}
  virtual nsresult OpenContentStream(bool async, nsIInputStream **result,
                                     nsIChannel **channel) override;
  virtual bool GetStatusArg(nsresult status, nsString &statusArg) override;
  virtual void OnCallbacksChanged() override;

 private:
  nsCOMPtr<nsIProxyInfo> mProxyInfo;
  nsCOMPtr<nsIFTPEventSink> mFTPEventSink;
  nsCOMPtr<nsIInputStream> mUploadStream;
  uint64_t mStartPos;
  nsCString mEntityID;
  bool mResumeRequested;
  PRTime mLastModifiedTime;
  bool mForcePending;
  RefPtr<ADivertableParentChannel> mParentChannel;
};

#endif /* nsFTPChannel_h___ */