/*
 * Copyright 2014, Ingo Weinhold, ingo_weinhold@gmx.de.
 * Distributed under the terms of the MIT License.
 */


#include "PackageFileManager.h"

#include <Locker.h>

#include <AutoLocker.h>

#include "DebugSupport.h"
#include "Package.h"


PackageFileManager::PackageFileManager(BLocker& lock)
	:
	fLock(lock),
	fFilesByEntryRef()
{
}


PackageFileManager::~PackageFileManager()
{
}


status_t
PackageFileManager::Init()
{
	return fFilesByEntryRef.Init();
}


status_t
PackageFileManager::GetPackageFile(const entry_ref& entryRef,
	PackageFile*& _file)
{
	AutoLocker<BLocker> locker(fLock);

	PackageFile* file = fFilesByEntryRef.Lookup(entryRef);
	if (file != NULL) {
		if (file->AcquireReference() > 0) {
			_file = file;
			return B_OK;
		}

		// File already full dereferenced. It is about to be deleted.
		fFilesByEntryRef.Remove(file);
	}

	file = new(std::nothrow) PackageFile;
	if (file == NULL)
		RETURN_ERROR(B_NO_MEMORY);

	status_t error = file->Init(entryRef, this);
	if (error != B_OK) {
		delete file;
		return error;
	}

	fFilesByEntryRef.Insert(file);

	_file = file;
	return B_OK;
}


status_t
PackageFileManager::CreatePackage(const entry_ref& entryRef, Package*& _package)
{
	PackageFile* packageFile;
	status_t error = GetPackageFile(entryRef, packageFile);
	if (error != B_OK)
		RETURN_ERROR(error);
	BReference<PackageFile> packageFileReference(packageFile, true);

	_package = new(std::nothrow) Package(packageFile);
	RETURN_ERROR(_package != NULL ? B_OK : B_NO_MEMORY);
}


void
PackageFileManager::PackageFileMoved(PackageFile* file,
	const node_ref& newDirectory)
{
	if (newDirectory == file->DirectoryRef())
		return;

	AutoLocker<BLocker> locker(fLock);

	fFilesByEntryRef.Remove(file);
	file->SetDirectoryRef(newDirectory);
	fFilesByEntryRef.Insert(file);
}


void
PackageFileManager::RemovePackageFile(PackageFile* file)
{
	AutoLocker<BLocker> locker(fLock);

	fFilesByEntryRef.Remove(file);
}