haiku.git

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


#include "PackageLinksDirectory.h"

#include <AutoDeleter.h>

#include "DebugSupport.h"
#include "PackageLinkDirectory.h"
#include "PackageLinksListener.h"
#include "Utils.h"


PackageLinksDirectory::PackageLinksDirectory()
	:
	Directory(0),
	fListener(NULL)
		// the ID needs to be assigned later, when added to a volume
{
	get_real_time(fModifiedTime);
}


PackageLinksDirectory::~PackageLinksDirectory()
{
}


timespec
PackageLinksDirectory::ModifiedTime() const
{
	return fModifiedTime;
}


status_t
PackageLinksDirectory::AddPackage(Package* package)
{
	// Create a package link directory -- there might already be one, but since
	// that's unlikely, we don't bother to check and recheck later.
	PackageLinkDirectory* linkDirectory
		= new(std::nothrow) PackageLinkDirectory;
	if (linkDirectory == NULL)
		return B_NO_MEMORY;
	BReference<PackageLinkDirectory> linkDirectoryReference(linkDirectory,
		true);

	status_t error = linkDirectory->Init(package);
	if (error != B_OK)
		RETURN_ERROR(error);

	// add the link directory
	DirectoryWriteLocker writeLocker(this);
	if (Node* child = FindChild(linkDirectory->Name())) {
		// There already is an entry with the name.
		PackageLinkDirectory* otherLinkDirectory
			= dynamic_cast<PackageLinkDirectory*>(child);
		if (otherLinkDirectory == NULL)
			RETURN_ERROR(B_BAD_VALUE);

		// There's already a package link directory. Delete the one we created
		// and add the package to the pre-existing one.
		linkDirectory->RemovePackage(package, NULL);
		linkDirectoryReference.Unset();

		linkDirectory = otherLinkDirectory;
		linkDirectory->AddPackage(package, fListener);
	} else {
		// No entry is in the way, so just add the link directory.
		AddChild(linkDirectory);

		if (fListener != NULL)
			fListener->PackageLinkNodeAdded(linkDirectory);
	}

	return B_OK;
}


void
PackageLinksDirectory::RemovePackage(Package* package)
{
	DirectoryWriteLocker writeLocker(this);

	// get the package's link directory and remove the package from it
	PackageLinkDirectory* linkDirectory = package->LinkDirectory();
	if (linkDirectory == NULL)
		return;

	BReference<PackageLinkDirectory> linkDirectoryReference(linkDirectory);

	linkDirectory->RemovePackage(package, fListener);

	// if empty, remove the link directory itself
	if (linkDirectory->IsEmpty()) {
		if (fListener != NULL) {
			DirectoryWriteLocker linkDirectoryWriteLocker(linkDirectory);
			fListener->PackageLinkNodeRemoved(linkDirectory);
		}

		RemoveChild(linkDirectory);
	}
}


void
PackageLinksDirectory::UpdatePackageDependencies(Package* package)
{
	DirectoryWriteLocker writeLocker(this);

	PackageLinkDirectory* linkDirectory = package->LinkDirectory();
	if (linkDirectory == NULL)
		return;

	linkDirectory->UpdatePackageDependencies(package, fListener);
}