// TortoiseSVN - a Windows shell extension for easy version control

// Copyright (C) 2007-2010 - TortoiseSVN

// This program is free software; you can redistribute it and/or
// modify it under the terms of the GNU General Public License
// as published by the Free Software Foundation; either version 2
// of the License, or (at your option) any later version.

// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software Foundation,
// 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
//
#include "stdafx.h"
#include "CacheFileOutBuffer.h"
#include "StreamException.h"

// construction / destruction: auto- open/close

#ifdef _WIN32
CCacheFileOutBuffer::CCacheFileOutBuffer (const std::wstring& fileName)
#else
CCacheFileOutBuffer::CCacheFileOutBuffer (const std::string& fileName)
#endif
    : CBufferedOutFile (fileName)
    , streamIsOpen (false)
{
    // write the version ids

    Add (OUR_LOG_CACHE_FILE_VERSION);
    Add (MIN_LOG_CACHE_FILE_VERSION);
}

CCacheFileOutBuffer::~CCacheFileOutBuffer()
{
    if (IsOpen())
    {
        streamOffsets.push_back (GetFileSize());

        size_t lastOffset = streamOffsets[0];
        for (size_t i = 1, count = streamOffsets.size(); i < count; ++i)
        {
            size_t offset = streamOffsets[i];
            size_t size = offset - lastOffset;

            if (size >= (DWORD)(-1))
            {
                RemoveFile();
                return;
            }

            Add ((DWORD)size);
            lastOffset = offset;
        }

        Add ((DWORD)(streamOffsets.size()-1));
    }
}

// write data to file

STREAM_INDEX CCacheFileOutBuffer::OpenStream()
{
    // don't overflow (2^32 streams is quite unlikely, though)

    assert ((STREAM_INDEX)streamOffsets.size() != (-1));

    assert (streamIsOpen == false);
    streamIsOpen = true;

    streamOffsets.push_back (GetFileSize());
    return (STREAM_INDEX)streamOffsets.size()-1;
}

void CCacheFileOutBuffer::CloseStream()
{
    assert (streamIsOpen == true);
    streamIsOpen = false;
}