/* -*- Mode: C++; tab-width: 4; 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/. */

#include "nsPreloadedStream.h"
#include "nsIRunnable.h"

#include "nsThreadUtils.h"
#include <algorithm>

namespace mozilla {
namespace net {

NS_IMPL_ISUPPORTS(nsPreloadedStream, nsIInputStream, nsIAsyncInputStream)

nsPreloadedStream::nsPreloadedStream(nsIAsyncInputStream *aStream,
                                     const char *data, uint32_t datalen)
    : mStream(aStream), mOffset(0), mLen(datalen) {
  mBuf = (char *)moz_xmalloc(datalen);
  memcpy(mBuf, data, datalen);
}

nsPreloadedStream::~nsPreloadedStream() { free(mBuf); }

NS_IMETHODIMP
nsPreloadedStream::Close() {
  mLen = 0;
  return mStream->Close();
}

NS_IMETHODIMP
nsPreloadedStream::Available(uint64_t *_retval) {
  uint64_t avail = 0;

  nsresult rv = mStream->Available(&avail);
  if (NS_FAILED(rv)) return rv;
  *_retval = avail + mLen;
  return NS_OK;
}

NS_IMETHODIMP
nsPreloadedStream::Read(char *aBuf, uint32_t aCount, uint32_t *_retval) {
  if (!mLen) return mStream->Read(aBuf, aCount, _retval);

  uint32_t toRead = std::min(mLen, aCount);
  memcpy(aBuf, mBuf + mOffset, toRead);
  mOffset += toRead;
  mLen -= toRead;
  *_retval = toRead;
  return NS_OK;
}

NS_IMETHODIMP
nsPreloadedStream::ReadSegments(nsWriteSegmentFun aWriter, void *aClosure,
                                uint32_t aCount, uint32_t *result) {
  if (!mLen) return mStream->ReadSegments(aWriter, aClosure, aCount, result);

  *result = 0;
  while (mLen > 0 && aCount > 0) {
    uint32_t toRead = std::min(mLen, aCount);
    uint32_t didRead = 0;
    nsresult rv;

    rv = aWriter(this, aClosure, mBuf + mOffset, *result, toRead, &didRead);

    if (NS_FAILED(rv)) return NS_OK;

    *result += didRead;
    mOffset += didRead;
    mLen -= didRead;
    aCount -= didRead;
  }

  return NS_OK;
}

NS_IMETHODIMP
nsPreloadedStream::IsNonBlocking(bool *_retval) {
  return mStream->IsNonBlocking(_retval);
}

NS_IMETHODIMP
nsPreloadedStream::CloseWithStatus(nsresult aStatus) {
  mLen = 0;
  return mStream->CloseWithStatus(aStatus);
}

class RunOnThread : public Runnable {
 public:
  RunOnThread(nsIAsyncInputStream *aStream, nsIInputStreamCallback *aCallback)
      : Runnable("net::RunOnThread"), mStream(aStream), mCallback(aCallback) {}

  virtual ~RunOnThread() = default;

  NS_IMETHOD Run() override {
    mCallback->OnInputStreamReady(mStream);
    return NS_OK;
  }

 private:
  nsCOMPtr<nsIAsyncInputStream> mStream;
  nsCOMPtr<nsIInputStreamCallback> mCallback;
};

NS_IMETHODIMP
nsPreloadedStream::AsyncWait(nsIInputStreamCallback *aCallback, uint32_t aFlags,
                             uint32_t aRequestedCount,
                             nsIEventTarget *aEventTarget) {
  if (!mLen)
    return mStream->AsyncWait(aCallback, aFlags, aRequestedCount, aEventTarget);

  if (!aCallback) return NS_OK;

  if (!aEventTarget) return aCallback->OnInputStreamReady(this);

  nsCOMPtr<nsIRunnable> event = new RunOnThread(this, aCallback);
  return aEventTarget->Dispatch(event, nsIEventTarget::DISPATCH_NORMAL);
}

}  // namespace net
}  // namespace mozilla