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


#include "disk_device_manager.h"

#include <stdio.h>

#include <KernelExport.h>

#include "KDiskDevice.h"
#include "KDiskDeviceManager.h"
#include "KDiskDeviceUtils.h"
#include "KDiskSystem.h"
#include "KPartition.h"


// debugging
//#define DBG(x)
#define DBG(x) x
#define OUT dprintf


disk_device_data*
write_lock_disk_device(partition_id partitionID)
{
	KDiskDeviceManager* manager = KDiskDeviceManager::Default();
	if (KDiskDevice* device = manager->RegisterDevice(partitionID, false)) {
		if (device->WriteLock())
			return device->DeviceData();
		// Only unregister, when the locking fails. The guarantees, that the
		// lock owner also has a reference.
		device->Unregister();
	}
	return NULL;
}


void
write_unlock_disk_device(partition_id partitionID)
{
	KDiskDeviceManager* manager = KDiskDeviceManager::Default();
	if (KDiskDevice* device = manager->RegisterDevice(partitionID, false)) {
		device->WriteUnlock();
		device->Unregister();

		device->Unregister();
	}
}


disk_device_data*
read_lock_disk_device(partition_id partitionID)
{
	KDiskDeviceManager* manager = KDiskDeviceManager::Default();
	if (KDiskDevice* device = manager->RegisterDevice(partitionID, false)) {
		if (device->ReadLock())
			return device->DeviceData();
		// Only unregister, when the locking fails. The guarantees, that the
		// lock owner also has a reference.
		device->Unregister();
	}
	return NULL;
}


void
read_unlock_disk_device(partition_id partitionID)
{
	KDiskDeviceManager* manager = KDiskDeviceManager::Default();
	if (KDiskDevice* device = manager->RegisterDevice(partitionID, false)) {
		device->ReadUnlock();
		device->Unregister();

		device->Unregister();
	}
}


int32
find_disk_device(const char* path)
{
	KDiskDeviceManager* manager = KDiskDeviceManager::Default();
	partition_id id = -1;
	if (KDiskDevice* device = manager->RegisterDevice(path)) {
		id = device->ID();
		device->Unregister();
	}
	return id;
}


int32
find_partition(const char* path)
{
	KDiskDeviceManager* manager = KDiskDeviceManager::Default();
	partition_id id = -1;
	if (KPartition* partition = manager->RegisterPartition(path)) {
		id = partition->ID();
		partition->Unregister();
	}
	return id;
}


disk_device_data*
get_disk_device(partition_id partitionID)
{
	KDiskDeviceManager* manager = KDiskDeviceManager::Default();
	KDiskDevice* device = manager->FindDevice(partitionID, false);
	return (device ? device->DeviceData() : NULL);
}


partition_data*
get_partition(partition_id partitionID)
{
	KDiskDeviceManager* manager = KDiskDeviceManager::Default();
	KPartition* partition = manager->FindPartition(partitionID);
	return (partition ? partition->PartitionData() : NULL);
}


partition_data*
get_parent_partition(partition_id partitionID)
{
	KDiskDeviceManager* manager = KDiskDeviceManager::Default();
	KPartition* partition = manager->FindPartition(partitionID);
	if (partition && partition->Parent())
		return partition->Parent()->PartitionData();
	return NULL;
}


partition_data*
get_child_partition(partition_id partitionID, int32 index)
{
	KDiskDeviceManager* manager = KDiskDeviceManager::Default();
	if (KPartition* partition = manager->FindPartition(partitionID)) {
		if (KPartition* child = partition->ChildAt(index))
			return child->PartitionData();
	}
	return NULL;
}


int
open_partition(partition_id partitionID, int openMode)
{
	KDiskDeviceManager* manager = KDiskDeviceManager::Default();
	KPartition* partition = manager->FindPartition(partitionID);
	if (partition == NULL)
		return B_BAD_VALUE;

	int fd = -1;
	status_t result = partition->Open(openMode, &fd);
	if (result != B_OK)
		return -1;

	return fd;
}


partition_data*
create_child_partition(partition_id partitionID, int32 index, off_t offset,
	off_t size, partition_id childID)
{
	KDiskDeviceManager* manager = KDiskDeviceManager::Default();
	if (KPartition* partition = manager->FindPartition(partitionID)) {
		KPartition* child = NULL;
		if (partition->CreateChild(childID, index, offset, size, &child)
				== B_OK) {
			return child->PartitionData();
		} else {
			DBG(OUT("  creating child (%" B_PRId32 ", %" B_PRId32 ") failed\n",
				partitionID, index));
		}
	} else
		DBG(OUT("  partition %" B_PRId32 " not found\n", partitionID));

	return NULL;
}


bool
delete_partition(partition_id partitionID)
{
	KDiskDeviceManager* manager = KDiskDeviceManager::Default();
	if (KPartition* partition = manager->FindPartition(partitionID)) {
		if (KPartition* parent = partition->Parent())
			return parent->RemoveChild(partition);
	}
	return false;
}


void
partition_modified(partition_id partitionID)
{
	// TODO: implemented
}


status_t
scan_partition(partition_id partitionID)
{
	// get the partition
	KDiskDeviceManager* manager = KDiskDeviceManager::Default();
	KPartition* partition = manager->RegisterPartition(partitionID);
	if (partition == NULL)
		return B_ENTRY_NOT_FOUND;
	PartitionRegistrar _(partition, true);

	// scan it
	return manager->ScanPartition(partition);
}


status_t
get_default_partition_content_name(partition_id partitionID,
	const char* fileSystemName, char* buffer, size_t bufferSize)
{
	KDiskDeviceManager* manager = KDiskDeviceManager::Default();
	KPartition* partition = manager->RegisterPartition(partitionID);
	if (partition == NULL)
		return B_ENTRY_NOT_FOUND;

	double size = partition->ContentSize();
	partition->Unregister();

	const char* const suffixes[] = {
		"", "K", "M", "G", "T", "P", "E", NULL
	};

	int index = 0;
	while (size >= 1024 && suffixes[index + 1]) {
		size /= 1024;
		index++;
	}

	// Our kernel snprintf() ignores the precision argument, so we manually
	// do one digit precision.
	uint64 result = uint64(size * 10 + 0.5);

	snprintf(buffer, bufferSize, "%s Volume (%" B_PRId32 ".%" B_PRId32 " %sB)",
		fileSystemName, int32(result / 10), int32(result % 10), suffixes[index]);

	return B_OK;
}


disk_system_id
find_disk_system(const char* name)
{
	KDiskDeviceManager* manager = KDiskDeviceManager::Default();
	if (ManagerLocker locker = manager) {
		if (KDiskSystem* diskSystem = manager->FindDiskSystem(name))
			return diskSystem->ID();
	}
	return -1;
}


bool
update_disk_device_job_progress(disk_job_id jobID, float progress)
{
#if 0
	KDiskDeviceManager* manager = KDiskDeviceManager::Default();
	if (ManagerLocker locker = manager) {
		if (KDiskDeviceJob* job = manager->FindJob(jobID)) {
			job->UpdateProgress(progress);
			return true;
		}
	}
#endif
	return false;
}


bool
update_disk_device_job_extra_progress(disk_job_id jobID, const char* info)
{
#if 0
	KDiskDeviceManager* manager = KDiskDeviceManager::Default();
	if (ManagerLocker locker = manager) {
		if (KDiskDeviceJob* job = manager->FindJob(jobID)) {
			job->UpdateExtraProgress(info);
			return true;
		}
	}
#endif
	return false;
}


bool
set_disk_device_job_error_message(disk_job_id jobID, const char* message)
{
#if 0
	KDiskDeviceManager* manager = KDiskDeviceManager::Default();
	if (ManagerLocker locker = manager) {
		if (KDiskDeviceJob* job = manager->FindJob(jobID)) {
			job->SetErrorMessage(message);
			return true;
		}
	}
#endif
	return false;
}


uint32
update_disk_device_job_interrupt_properties(disk_job_id jobID,
	uint32 interruptProperties)
{
#if 0
	bool paused = false;
	KDiskDeviceManager* manager = KDiskDeviceManager::Default();
	do {
		sem_id pauseSemaphore = -1;
		if (ManagerLocker locker = manager) {
			// get the job and the respective job queue
			if (KDiskDeviceJob* job = manager->FindJob(jobID)) {
				if (KDiskDeviceJobQueue* jobQueue = job->JobQueue()) {
					// terminate if canceled.
					if (jobQueue->IsCanceled()) {
						if (jobQueue->ShallReverse())
							return B_DISK_DEVICE_JOB_REVERSE;
						return B_DISK_DEVICE_JOB_CANCEL;
					}
					// set the new interrupt properties only when not
					// requested to pause
					if (jobQueue->IsPauseRequested())
						pauseSemaphore = jobQueue->ReadyToPause();
					else
						job->SetInterruptProperties(interruptProperties);
				}
			}
		}
		// pause, if requested; redo the loop then
		paused = (pauseSemaphore >= 0);
		if (paused) {
			acquire_sem(pauseSemaphore);
			pauseSemaphore = -1;
		}
	} while (paused);
#endif
	return B_DISK_DEVICE_JOB_CONTINUE;
}