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

#include "nsAutoPtr.h"
#include "nsError.h"
#include "nsString.h"
#include "nsHttpHeaderArray.h"

namespace mozilla { namespace net {

class nsHttpChunkedDecoder
{
public:
    nsHttpChunkedDecoder() : mTrailers(nullptr)
                           , mChunkRemaining(0)
                           , mReachedEOF(false)
                           , mWaitEOF(false) {}
   ~nsHttpChunkedDecoder() { delete mTrailers; }

    bool ReachedEOF() { return mReachedEOF; }

    // called by the transaction to handle chunked content.
    MOZ_MUST_USE nsresult HandleChunkedContent(char *buf,
                                               uint32_t count,
                                               uint32_t *contentRead,
                                               uint32_t *contentRemaining);

    nsHttpHeaderArray *Trailers() { return mTrailers.get(); }

    nsHttpHeaderArray *TakeTrailers() { return mTrailers.forget(); }

    uint32_t GetChunkRemaining() { return mChunkRemaining; }

private:
    MOZ_MUST_USE nsresult ParseChunkRemaining(char *buf,
                                              uint32_t count,
                                              uint32_t *countRead);

private:
    nsAutoPtr<nsHttpHeaderArray>  mTrailers;
    uint32_t                      mChunkRemaining;
    nsCString                     mLineBuf; // may hold a partial line
    bool                          mReachedEOF;
    bool                          mWaitEOF;
};

} // namespace net
} // namespace mozilla

#endif