#include "PrintAddOnServer.h"

#include <Entry.h>
#include <Roster.h>

#include "PrinterDriverAddOn.h"
#include "PrintAddOnServerProtocol.h"

static const bigtime_t kSeconds = 1000000L;
static const bigtime_t kDeliveryTimeout = 30 * kSeconds;
static const bigtime_t kReplyTimeout = B_INFINITE_TIMEOUT;


PrintAddOnServer::PrintAddOnServer(const char* driver)
	:
	fDriver(driver)
{
	fLaunchStatus = Launch(fMessenger);
}


PrintAddOnServer::~PrintAddOnServer()
{
	if (fLaunchStatus == B_OK)
		Quit();
}


status_t
PrintAddOnServer::AddPrinter(const char* spoolFolderName)
{
	BMessage message(kMessageAddPrinter);
	message.AddString(kPrinterDriverAttribute, Driver());
	message.AddString(kPrinterNameAttribute, spoolFolderName);

	BMessage reply;
	status_t result = SendRequest(message, reply);
	if (result != B_OK)
		return result;

	return GetResult(reply);
}


status_t
PrintAddOnServer::ConfigPage(BDirectory* spoolFolder,
	BMessage* settings)
{
	BMessage message(kMessageConfigPage);
	message.AddString(kPrinterDriverAttribute, Driver());
	AddDirectory(message, kPrinterFolderAttribute, spoolFolder);
	message.AddMessage(kPrintSettingsAttribute, settings);

	BMessage reply;
	status_t result = SendRequest(message, reply);
	if (result != B_OK)
		return result;

	return GetResultAndUpdateSettings(reply, settings);
}


status_t
PrintAddOnServer::ConfigJob(BDirectory* spoolFolder,
	BMessage* settings)
{
	BMessage message(kMessageConfigJob);
	message.AddString(kPrinterDriverAttribute, Driver());
	AddDirectory(message, kPrinterFolderAttribute, spoolFolder);
	message.AddMessage(kPrintSettingsAttribute, settings);

	BMessage reply;
	status_t result = SendRequest(message, reply);
	if (result != B_OK)
		return result;

	return GetResultAndUpdateSettings(reply, settings);
}


status_t
PrintAddOnServer::DefaultSettings(BDirectory* spoolFolder,
	BMessage* settings)
{
	BMessage message(kMessageDefaultSettings);
	message.AddString(kPrinterDriverAttribute, Driver());
	AddDirectory(message, kPrinterFolderAttribute, spoolFolder);

	BMessage reply;
	status_t result = SendRequest(message, reply);
	if (result != B_OK)
		return result;

	return GetResultAndUpdateSettings(reply, settings);
}


status_t
PrintAddOnServer::TakeJob(const char* spoolFile,
				BDirectory* spoolFolder)
{
	BMessage message(kMessageTakeJob);
	message.AddString(kPrinterDriverAttribute, Driver());
	message.AddString(kPrintJobFileAttribute, spoolFile);
	AddDirectory(message, kPrinterFolderAttribute, spoolFolder);

	BMessage reply;
	status_t result = SendRequest(message, reply);
	if (result != B_OK)
		return result;

	return GetResult(reply);
}


status_t
PrintAddOnServer::FindPathToDriver(const char* driver, BPath* path)
{
	return PrinterDriverAddOn::FindPathToDriver(driver, path);
}


const char*
PrintAddOnServer::Driver() const
{
	return fDriver.String();
}


status_t
PrintAddOnServer::Launch(BMessenger& messenger)
{
	team_id team;
	status_t result =
		be_roster->Launch(kPrintAddOnServerApplicationSignature,
			(BMessage*)NULL, &team);
	if (result != B_OK) {
		return result;
	}

	fMessenger = BMessenger(kPrintAddOnServerApplicationSignature, team);
	return result;
}


bool
PrintAddOnServer::IsLaunched()
{
	return fLaunchStatus == B_OK;
}


void
PrintAddOnServer::Quit()
{
	if (fLaunchStatus == B_OK) {
		fMessenger.SendMessage(B_QUIT_REQUESTED);
		fLaunchStatus = B_ERROR;
	}
}


void
PrintAddOnServer::AddDirectory(BMessage& message, const char* name,
	BDirectory* directory)
{
	BEntry entry;
	status_t result = directory->GetEntry(&entry);
	if (result != B_OK)
		return;

	BPath path;
	if (entry.GetPath(&path) != B_OK)
		return;

	message.AddString(name, path.Path());
}


void
PrintAddOnServer::AddEntryRef(BMessage& message, const char* name,
	const entry_ref* entryRef)
{
	BPath path(entryRef);
	if (path.InitCheck() != B_OK)
		return;

	message.AddString(name, path.Path());
}


status_t
PrintAddOnServer::SendRequest(BMessage& request, BMessage& reply)
{
	if (!IsLaunched())
		return fLaunchStatus;

	return fMessenger.SendMessage(&request, &reply, kDeliveryTimeout,
		kReplyTimeout);
}


status_t
PrintAddOnServer::GetResult(BMessage& reply)
{
	int32 status;
	status_t result = reply.FindInt32(kPrintAddOnServerStatusAttribute,
		&status);
	if (result != B_OK)
		return result;
	return static_cast<status_t>(status);
}


status_t
PrintAddOnServer::GetResultAndUpdateSettings(BMessage& reply, BMessage* settings)
{
	BMessage newSettings;
	if (reply.FindMessage(kPrintSettingsAttribute, &newSettings) == B_OK)
		*settings = newSettings;

	settings->PrintToStream();

	return GetResult(reply);
}