/* 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 _nsDeflateConverter_h_
#define _nsDeflateConverter_h_

#include "nsIStreamConverter.h"
#include "nsCOMPtr.h"
#include "nsIPipe.h"
#include "zlib.h"
#include "mozilla/Attributes.h"

#define DEFLATECONVERTER_CID { 0x461cd5dd, 0x73c6, 0x47a4, \
           { 0x8c, 0xc3, 0x60, 0x3b, 0x37, 0xd8, 0x4a, 0x61 } }

#define ZIP_BUFLEN (4 * 1024 - 1)

class nsDeflateConverter final : public nsIStreamConverter
{
public:
    NS_DECL_ISUPPORTS
    NS_DECL_NSIREQUESTOBSERVER
    NS_DECL_NSISTREAMLISTENER
    NS_DECL_NSISTREAMCONVERTER

    nsDeflateConverter()
    {
        // 6 is Z_DEFAULT_COMPRESSION but we need the actual value
        mLevel = 6;
    }

    explicit nsDeflateConverter(int32_t level)
    {
        mLevel = level;
    }

private:

    ~nsDeflateConverter()
    {
    }
    
    enum WrapMode {
        WRAP_ZLIB,
        WRAP_GZIP,
        WRAP_NONE
    };

    WrapMode mWrapMode;
    uint64_t mOffset;
    int32_t mLevel;
    nsCOMPtr<nsIStreamListener> mListener;
    nsCOMPtr<nsISupports> mContext;
    z_stream mZstream;
    unsigned char mWriteBuffer[ZIP_BUFLEN];

    nsresult Init();
    nsresult PushAvailableData(nsIRequest *aRequest, nsISupports *aContext);
};

#endif
