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

#include "AttributeIterator.h"

#include <util/AutoLock.h>

#include "Node.h"
#include "Volume.h"


AttributeIterator::AttributeIterator(Node *node)
	: fNode(node),
	  fAttribute(NULL),
	  fSuspended(false),
	  fIsNext(false),
	  fDone(false)
{
}


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


status_t
AttributeIterator::SetTo(Node *node)
{
	Unset();

	if (node == NULL)
		return B_BAD_VALUE;

	fNode = node;
	fAttribute = NULL;
	fSuspended = false;
	fIsNext = false;
	fDone = false;
	return B_OK;
}


void
AttributeIterator::Unset()
{
	if (fNode && fSuspended)
		Resume();

	fNode = NULL;
	fAttribute = NULL;
	fSuspended = false;
	fIsNext = false;
	fDone = false;
}


status_t
AttributeIterator::Suspend()
{
	if (fNode == NULL)
		return B_ERROR;

	RecursiveLocker locker(fNode->GetVolume()->GetAttributeIteratorLock());
	if (!locker.IsLocked())
		return B_ERROR;

	if (fSuspended)
		return B_ERROR;

	if (fAttribute)
		fAttribute->AttachAttributeIterator(this);
	fSuspended = true;

	return B_OK;
}


status_t
AttributeIterator::Resume()
{
	if (fNode == NULL)
		return B_ERROR;

	RecursiveLocker locker(fNode->GetVolume()->GetAttributeIteratorLock());
	if (!locker.IsLocked())
		return B_ERROR;

	if (!fSuspended)
		return B_ERROR;

	if (fAttribute)
		fAttribute->DetachAttributeIterator(this);
	fSuspended = false;

	return B_OK;
}


status_t
AttributeIterator::GetNext(Attribute **attribute)
{
	if (attribute == NULL)
		return B_BAD_VALUE;

	status_t error = B_ENTRY_NOT_FOUND;
	if (!fDone && fNode != NULL) {
		if (fIsNext) {
			fIsNext = false;
			if (fAttribute)
				error = B_OK;
		} else
			error = fNode->GetNextAttribute(&fAttribute);
		*attribute = fAttribute;
	}
	fDone = (error != B_OK);
	return error;
}


status_t
AttributeIterator::Rewind()
{
	if (fNode == NULL)
		return B_ERROR;

	RecursiveLocker locker(fNode->GetVolume()->GetAttributeIteratorLock());
	if (!locker.IsLocked())
		return B_ERROR;

	if (fSuspended && fAttribute)
		fAttribute->DetachAttributeIterator(this);
	fAttribute = NULL;
	fIsNext = false;
	fDone = false;
	return B_OK;
}


void
AttributeIterator::SetCurrent(Attribute *attribute, bool isNext)
{
	fIsNext = isNext;
	fAttribute = attribute;
	fDone = !fAttribute;
}