/*
 * Copyright 2009, Ingo Weinhold, ingo_weinhold@gmx.de.
 * Copyright 2011, Oliver Tappe <zooey@hirschkaefer.de>
 * Distributed under the terms of the MIT License.
 */
#ifndef _PACKAGE__HPKG__PRIVATE__POOL_BUFFER_H_
#define _PACKAGE__HPKG__PRIVATE__POOL_BUFFER_H_


#include <stddef.h>

#include <util/DoublyLinkedList.h>

#include <package/hpkg/BufferPool.h>


namespace BPackageKit {

namespace BHPKG {

namespace BPrivate {


class PoolBuffer : public DoublyLinkedListLinkImpl<PoolBuffer> {
public:
                                                                PoolBuffer(size_t size);
                                                                ~PoolBuffer();

                        void*                           Buffer() const  { return fBuffer; }
                        size_t                          Size() const    { return fSize; }


                        // implementation private
                        PoolBuffer**            Owner() const   { return fOwner; }
                        void                            SetOwner(PoolBuffer** owner)
                                                                        { fOwner = owner; }

                        void                            SetCached(bool cached)  { fCached = cached; }
                        bool                            IsCached() const                { return fCached; }

private:
                        PoolBuffer**            fOwner;
                        void*                           fBuffer;
                        size_t                          fSize;
                        bool                            fCached;
};


class PoolBufferPutter {
public:
        PoolBufferPutter(BBufferPool* pool, PoolBuffer** owner)
                :
                fPool(pool),
                fOwner(owner),
                fBuffer(NULL)
        {
        }

        PoolBufferPutter(BBufferPool* pool, PoolBuffer* buffer)
                :
                fPool(pool),
                fOwner(NULL),
                fBuffer(buffer)
        {
        }

        ~PoolBufferPutter()
        {
                if (fPool != NULL) {
                        if (fOwner != NULL)
                                fPool->PutBufferAndCache(fOwner);
                        else if (fBuffer != NULL)
                                fPool->PutBuffer(&fBuffer);
                }
        }

private:
        BBufferPool*    fPool;
        PoolBuffer**    fOwner;
        PoolBuffer*             fBuffer;
};


}       // namespace BPrivate

}       // namespace BHPKG

}       // namespace BPackageKit


#endif  // _PACKAGE__HPKG__PRIVATE__POOL_BUFFER_H_