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

#include "nsIAboutModule.h"
#include "nsICacheEntryOpenCallback.h"
#include "nsICacheEntry.h"
#include "nsIChannel.h"
#include "nsIStreamListener.h"
#include "nsString.h"
#include "nsCOMPtr.h"

class nsIAsyncOutputStream;
class nsIInputStream;
class nsILoadContextInfo;
class nsIURI;
class nsCString;

class nsAboutCacheEntry final : public nsIAboutModule
{
public:
    NS_DECL_ISUPPORTS
    NS_DECL_NSIABOUTMODULE

private:
    virtual ~nsAboutCacheEntry() {}

    class Channel final : public nsICacheEntryOpenCallback
                        , public nsICacheEntryMetaDataVisitor
                        , public nsIStreamListener
                        , public nsIChannel
    {
    public:
        NS_DECL_ISUPPORTS
        NS_DECL_NSICACHEENTRYOPENCALLBACK
        NS_DECL_NSICACHEENTRYMETADATAVISITOR
        NS_DECL_NSIREQUESTOBSERVER
        NS_DECL_NSISTREAMLISTENER
        NS_FORWARD_SAFE_NSICHANNEL(mChannel)
        NS_FORWARD_SAFE_NSIREQUEST(mChannel)

        Channel()
            : mBuffer(nullptr)
            , mWaitingForData(false)
            , mHexDumpState(0)
        {}

    private:
        virtual ~Channel() {}

    public:
        MOZ_MUST_USE nsresult Init(nsIURI* uri, nsILoadInfo* aLoadInfo);

        MOZ_MUST_USE nsresult GetContentStream(nsIURI *, nsIInputStream **);
        MOZ_MUST_USE nsresult OpenCacheEntry(nsIURI *);
        MOZ_MUST_USE nsresult OpenCacheEntry();
        MOZ_MUST_USE nsresult WriteCacheEntryDescription(nsICacheEntry *);
        MOZ_MUST_USE nsresult WriteCacheEntryUnavailable();
        MOZ_MUST_USE nsresult ParseURI(nsIURI *uri, nsACString &storageName,
                                       nsILoadContextInfo **loadInfo,
                                       nsCString &enahnceID,
                                       nsIURI **cacheUri);
        void CloseContent();

        static MOZ_MUST_USE nsresult
        PrintCacheData(nsIInputStream *aInStream,
                       void *aClosure,
                       const char *aFromSegment,
                       uint32_t aToOffset,
                       uint32_t aCount,
                       uint32_t *aWriteCount);

    private:
        nsCString mStorageName, mEnhanceId;
        nsCOMPtr<nsILoadContextInfo> mLoadInfo;
        nsCOMPtr<nsIURI> mCacheURI;

        nsCString *mBuffer;
        nsCOMPtr<nsIAsyncOutputStream> mOutputStream;
        bool mWaitingForData;
        uint32_t mHexDumpState;

        nsCOMPtr<nsIChannel> mChannel;
    };
};

#define NS_ABOUT_CACHE_ENTRY_MODULE_CID              \
{ /* 7fa5237d-b0eb-438f-9e50-ca0166e63788 */         \
    0x7fa5237d,                                      \
    0xb0eb,                                          \
    0x438f,                                          \
    {0x9e, 0x50, 0xca, 0x01, 0x66, 0xe6, 0x37, 0x88} \
}

#endif // nsAboutCacheEntry_h__
