/*
 * Copyright 2013 Haiku, Inc. All rights reserved.
 * Distributed under the terms of the MIT License.
 *
 * Authors:
 *		Paweł Dziepak, pdziepak@quarnos.org
 */


#include "InodeIdMap.h"


status_t
InodeIdMap::AddName(FileInfo& fileInfo, InodeNames* parent,
	const char* name, ino_t id)
{
	MutexLocker _(fLock);
	AVLTreeMap<ino_t, FileInfo>::Iterator iterator = fMap.Find(id);
	if (iterator.HasCurrent()) {
		if (fileInfo.fHandle == iterator.Current().fHandle) {
			return iterator.CurrentValuePointer()->fNames->AddName(parent,
				name);
		}
	}

	fMap.Remove(id);
	fileInfo.fNames = new InodeNames;
	if (fileInfo.fNames == NULL)
		return B_NO_MEMORY;

	fileInfo.fNames->fHandle = fileInfo.fHandle;
	status_t result = fileInfo.fNames->AddName(parent, name);
	if (result != B_OK) {
		delete fileInfo.fNames;
		return result;
	}

	return fMap.Insert(id, fileInfo);
}


bool
InodeIdMap::RemoveName(ino_t id, InodeNames* parent, const char* name)
{
	ASSERT(name != NULL);

	MutexLocker _(fLock);
	AVLTreeMap<ino_t, FileInfo>::Iterator iterator = fMap.Find(id);
	if (!iterator.HasCurrent())
		return true;

	FileInfo* fileInfo = iterator.CurrentValuePointer();

	return fileInfo->fNames->RemoveName(parent, name);
}


status_t
InodeIdMap::RemoveEntry(ino_t id)
{
	MutexLocker _(fLock);
	return fMap.Remove(id);
}


status_t
InodeIdMap::GetFileInfo(FileInfo* fileInfo, ino_t id)
{
	ASSERT(fileInfo != NULL);

	MutexLocker _(fLock);
	AVLTreeMap<ino_t, FileInfo>::Iterator iterator = fMap.Find(id);
	if (!iterator.HasCurrent())
		return B_ENTRY_NOT_FOUND;

	*fileInfo = iterator.Current();
	if (fileInfo->fNames->fNames.IsEmpty())
		return B_ENTRY_NOT_FOUND;
	return B_OK;
}


void
InodeIdMap::Dump(void (*xprintf)(const char*, ...))
{
	MutexLocker locker;
	if (xprintf != kprintf)
		locker.SetTo(fLock, false);

	_DumpLocked(xprintf);

	return;
}


void
InodeIdMap::_DumpLocked(void (*xprintf)(const char*, ...)) const
{
	xprintf("InodeIdMap at %p\n", this);
	xprintf("\tino\thandle\n");
	AVLTreeMap<ino_t, FileInfo>::ConstIterator iterator = fMap.GetIterator();
	for (iterator.Next(); iterator.HasNext(); iterator.Next()) {
		ino_t ino = iterator.CurrentKey();
		xprintf("\t%" B_PRIdINO "\t", ino);
		const FileInfo* fileInfo = iterator.CurrentValuePointer();
		if (fileInfo != NULL)
			fileInfo->fHandle.Dump(xprintf);
	}

	return;
}