/*
 * 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;
}