/*
 * Transport.cpp
 * Copyright 1999-2000 Y.Takagi. All Rights Reserved.
 */

#include <FindDirectory.h>
#include <Message.h>
#include <Directory.h>
#include <DataIO.h>
#include <File.h>
#include <Path.h>
#include <image.h>

#include "Transport.h"
#include "PrinterData.h"
#include "DbgMsg.h"


using namespace std;


TransportException::TransportException(const string& what)
	:
	fWhat(what)
{
}


const char*
TransportException::What() const
{
	return fWhat.c_str();
}


Transport::Transport(const PrinterData *printerData)
	:
	fImage(-1),
	fInitTransport(0),
	fExitTransport(0),
	fDataStream(0),
	fAbort(false)
{
	const directory_which paths[] = {
		B_USER_ADDONS_DIRECTORY,
		B_COMMON_ADDONS_DIRECTORY,
		B_BEOS_ADDONS_DIRECTORY,
	};
	BPath path;
	for (uint32 i = 0; i < sizeof(paths) / sizeof(paths[0]); ++i) {
		if (find_directory(paths[i], &path) != B_OK)
			continue;
		path.Append("Print/transport");
		path.Append(printerData->GetTransport().c_str());
		DBGMSG(("load_add_on: %s\n", path.Path()));
		fImage = load_add_on(path.Path());
		if (fImage >= 0)
			break;
	}

	if (fImage < 0) {
		SetLastError("cannot load a transport add-on");
		return;
	}

	DBGMSG(("image id = %d\n", (int)fImage));

	get_image_symbol(fImage, "init_transport", B_SYMBOL_TYPE_TEXT, (void **)&fInitTransport);
	get_image_symbol(fImage, "exit_transport", B_SYMBOL_TYPE_TEXT, (void **)&fExitTransport);

	if (fInitTransport == NULL) {
		SetLastError("get_image_symbol failed.");
		DBGMSG(("init_transport is NULL\n"));
	}

	if (fExitTransport == NULL) {
		SetLastError("get_image_symbol failed.");
		DBGMSG(("exit_transport is NULL\n"));
	}

	if (fInitTransport) {
		string spool_path;
		printerData->GetPath(spool_path);
		BMessage *msg = new BMessage('TRIN');
		msg->AddString("printer_file", spool_path.c_str());
		fDataStream = (*fInitTransport)(msg);
		delete msg;
		if (fDataStream == 0) {
			SetLastError("init_transport failed.");
		}
	}
}


Transport::~Transport()
{
	if (fExitTransport) {
		(*fExitTransport)();
	}
	if (fImage >= 0) {
		unload_add_on(fImage);
	}
}


bool
Transport::CheckAbort() const
{
	return fDataStream == 0;
}


const
string &Transport::LastError() const
{
	return fLastErrorString;
}


bool
Transport::IsPrintToFileCanceled() const
{
	// The BeOS "Print To File" transport add-on returns a non-NULL BDataIO *
	// even after user filepanel cancellation!
	BFile* file = dynamic_cast<BFile*>(fDataStream);
	return file != NULL && file->InitCheck() != B_OK;
}


void
Transport::SetLastError(const char *e)
{
	fLastErrorString = e;
	fAbort = true;
}


void
Transport::Write(const void* buffer, size_t size) throw(TransportException)
{
	if (fDataStream) {
		if (size == (size_t)fDataStream->Write(buffer, size)) {
			return;
		}
		SetLastError("BDataIO::Write failed.");
	}
	throw TransportException(LastError());
}