// ServerQueryIterator.cpp

#include "ServerQueryIterator.h"

#include <new>

// constructor
ServerQueryIterator::ServerQueryIterator(Volume* volume)
        : QueryIterator(volume),
          fRemoteCookie(-1),
          fShareVolumeIDs(NULL),
          fShareVolumeCount(0),
          fShareVolumeIndex(0)
{
}

// destructor
ServerQueryIterator::~ServerQueryIterator()
{
}

// SetRemoteCookie
void
ServerQueryIterator::SetRemoteCookie(int32 cookie)
{
        fRemoteCookie = cookie;
}

// GetRemoteCookie
int32
ServerQueryIterator::GetRemoteCookie() const
{
        return fRemoteCookie;
}

// SetEntry
status_t
ServerQueryIterator::SetEntry(const int32* shareVolumeIDs,
        int32 shareVolumeCount, const NodeInfo& dirInfo,
        const EntryInfo& entryInfo)
{
        UnsetEntry();

        if (!shareVolumeIDs || shareVolumeCount <= 0)
                return B_BAD_VALUE;

        // copy volume IDs
        if (shareVolumeCount <= IN_OBJECT_ID_COUNT)
                fShareVolumeIDs = fInObjectIDs;
        else
                fShareVolumeIDs = new(std::nothrow) int32[shareVolumeCount];
        if (!fShareVolumeIDs)
                return B_NO_MEMORY;
        fShareVolumeCount = shareVolumeCount;
        memcpy(fShareVolumeIDs, shareVolumeIDs, shareVolumeCount * 4);

        // copy entry name
        if (!fEntryName.SetTo(entryInfo.name.GetString())) {
                UnsetEntry();
                return B_NO_MEMORY;
        }

        fDirectoryInfo = dirInfo;
        fEntryInfo = entryInfo;
        fEntryInfo.name.SetTo(fEntryName.GetString());

        return B_OK;
}

// UnsetEntry
void
ServerQueryIterator::UnsetEntry()
{
        if (fShareVolumeIDs && fShareVolumeIDs != fInObjectIDs)
                delete[] fShareVolumeIDs;
        fShareVolumeIDs = NULL;

        fShareVolumeCount = 0;
        fShareVolumeIndex = 0;
        fEntryName.Unset();
        fEntryInfo.name.SetTo(NULL);
}

// GetShareVolumeIDs
const int32*
ServerQueryIterator::GetShareVolumeIDs() const
{
        return fShareVolumeIDs;
}

// CountShareVolumes
int32
ServerQueryIterator::CountShareVolumes() const
{
        return fShareVolumeCount;
}

// GetDirectoryInfo
const NodeInfo&
ServerQueryIterator::GetDirectoryInfo() const
{
        return fDirectoryInfo;
}

// GetEntryInfo
const EntryInfo&
ServerQueryIterator::GetEntryInfo() const
{
        return fEntryInfo;
}

// HasNextShareVolumeID
bool
ServerQueryIterator::HasNextShareVolumeID() const
{
        return (fShareVolumeIDs && fShareVolumeIndex < fShareVolumeCount);
}

// NextShareVolumeID
int32
ServerQueryIterator::NextShareVolumeID()
{
        if (!fShareVolumeIDs || fShareVolumeIndex >= fShareVolumeCount)
                return B_ENTRY_NOT_FOUND;

        return fShareVolumeIDs[fShareVolumeIndex++];
}