/*
 * Copyright 2007, Ingo Weinhold, ingo_weinhold@gmx.de.
 * All rights reserved. Distributed under the terms of the MIT license.
 */

#include "Directory.h"
#include "Entry.h"
#include "EntryIterator.h"
#include "Volume.h"

#include <util/AutoLock.h>


EntryIterator::EntryIterator(Directory *directory)
        :
        fDirectory(directory),
        fEntry(NULL),
        fSuspended(false),
        fIsNext(false),
        fDone(false)
{
}


EntryIterator::~EntryIterator()
{
        Unset();
}


status_t
EntryIterator::SetTo(Directory *directory)
{
        Unset();

        if (directory == NULL)
                return B_BAD_VALUE;

        fDirectory = directory;
        fEntry = NULL;
        fSuspended = false;
        fIsNext = false;
        fDone = false;

        return B_OK;
}


void
EntryIterator::Unset()
{
        if (fDirectory != NULL && fSuspended)
                Resume();

        fDirectory = NULL;
        fEntry = NULL;
        fSuspended = false;
        fIsNext = false;
        fDone = false;
}


status_t
EntryIterator::Suspend()
{
        if (fDirectory == NULL || fSuspended)
                return B_ERROR;

        if (fEntry != NULL) {
                RecursiveLocker locker(fDirectory->GetVolume()->GetIteratorLock());
                if (!locker.IsLocked())
                        return B_ERROR;
                fEntry->AttachEntryIterator(this);
        }

        fSuspended = true;
        return B_OK;
}


status_t
EntryIterator::Resume()
{
        if (fDirectory == NULL)
                return B_ERROR;

        if (fSuspended && fEntry != NULL) {
                RecursiveLocker locker(fDirectory->GetVolume()->GetIteratorLock());
                if (!locker.IsLocked())
                        return B_ERROR;
                fEntry->DetachEntryIterator(this);
        }

        fSuspended = false;
        return B_OK;
}


status_t
EntryIterator::GetNext(Entry **entry)
{
        if (fDirectory == NULL || entry == NULL)
                return B_BAD_VALUE;

        status_t error = B_ENTRY_NOT_FOUND;
        if (!fDone) {
                if (fIsNext) {
                        fIsNext = false;
                        if (fEntry)
                                error = B_OK;
                } else
                        error = fDirectory->GetNextEntry(&fEntry);
                *entry = fEntry;
        }
        fDone = (error != B_OK);
        return error;
}


status_t
EntryIterator::Rewind()
{
        if (fDirectory == NULL)
                return B_ERROR;

        RecursiveLocker locker(fDirectory->GetVolume()->GetIteratorLock());
        if (!locker.IsLocked())
                return B_ERROR;

        if (fSuspended && fEntry != NULL)
                fEntry->DetachEntryIterator(this);

        fEntry = NULL;
        fIsNext = false;
        fDone = false;

        return B_OK;
}


void
EntryIterator::SetCurrent(Entry *entry, bool isNext)
{
        fIsNext = isNext;
        fEntry = entry;
        fDone = !fEntry;
}