// Sun, 18 Jun 2000
// Y.Takagi

#include <Alert.h>
#include <DataIO.h>
#include <Message.h>
#include <Directory.h>

#include <netdb.h>
#include <pwd.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <iomanip>
#include <sstream>

#include "LpsClient.h"
#include "LprSetupDlg.h"
#include "LprTransport.h"
#include "LprDefs.h"
#include "DbgMsg.h"


using namespace std;


LprTransport::LprTransport(BMessage *msg)
	:
	BDataIO()
{
	fServer[0] = '\0';
	fQueue[0]  = '\0';
	fFile[0]   = '\0';
	fUser[0]   = '\0';
	fJobId     = 0;
	fError     = false;

	struct passwd *pwd = getpwuid(geteuid());
	if (pwd != NULL && pwd->pw_name != NULL && pwd->pw_name[0])
		strcpy(fUser, pwd->pw_name);
	else
		strcpy(fUser, "baron");

	DUMP_BMESSAGE(msg);

	const char *spool_path = msg->FindString(SPOOL_PATH);
	if (spool_path && *spool_path) {
		BDirectory dir(spool_path);
		DUMP_BDIRECTORY(&dir);

		dir.ReadAttr(LPR_SERVER_NAME, B_STRING_TYPE, 0, fServer, sizeof(fServer));
		if (fServer[0] == '\0') {
			LprSetupDlg *dlg = new LprSetupDlg(&dir);
			if (dlg->Go() == B_ERROR) {
				fError = true;
				return;
			}
		}

		dir.ReadAttr(LPR_SERVER_NAME, B_STRING_TYPE, 0, fServer, sizeof(fServer));
		dir.ReadAttr(LPR_QUEUE_NAME,  B_STRING_TYPE, 0, fQueue,  sizeof(fQueue));
		dir.ReadAttr(LPR_JOB_ID,      B_INT32_TYPE,  0, &fJobId, sizeof(fJobId));
		fJobId++;
		if (fJobId > 255) {
			fJobId = 1;
		}
		dir.WriteAttr(LPR_JOB_ID, B_INT32_TYPE, 0, &fJobId, sizeof(fJobId));

		sprintf(fFile, "%s/%s@ipp.%" B_PRId32, spool_path, fUser, fJobId);

		fStream.open(fFile, ios::in | ios::out | ios::binary | ios::trunc);
		if (fStream.good()) {
			DBGMSG(("spool_file: %s\n", fFile));
			return;
		}
	}
	fError = true;
}


LprTransport::~LprTransport()
{
	if (!fError)
		_SendFile();

	if (fFile[0] != '\0')
		unlink(fFile);
}


void
LprTransport::_SendFile()
{
	char hostname[128];
	if (gethostname(hostname, sizeof(hostname)) != B_OK)
		strcpy(hostname, "localhost");

	ostringstream cfname;
	cfname << "cfA" << setw(3) << setfill('0') << fJobId << hostname;

	ostringstream dfname;
	dfname << "dfA" << setw(3) << setfill('0') << fJobId << hostname;

	ostringstream cf;
	cf << 'H' << hostname     << '\n';
	cf << 'P' << fUser << '\n';
	cf << 'l' << dfname.str() << '\n';
	cf << 'U' << dfname.str() << '\n';

	long cfsize = cf.str().length();
	long dfsize = fStream.tellg();
	fStream.seekg(0, ios::beg);

	try {
		LpsClient lpr(fServer);

		lpr.connect();
		lpr.receiveJob(fQueue);

		lpr.receiveControlFile(cfsize, cfname.str().c_str());
		lpr.transferData(cf.str().c_str(), cfsize);
		lpr.endTransfer();

		lpr.receiveDataFile(dfsize, dfname.str().c_str());
		lpr.transferData(fStream, dfsize);
		lpr.endTransfer();
	}

	catch (LPSException &err) {
		DBGMSG(("error: %s\n", err.what()));
		BAlert *alert = new BAlert("", err.what(), "OK");
		alert->SetFlags(alert->Flags() | B_CLOSE_ON_ESCAPE);
		alert->Go();
	}
}


ssize_t
LprTransport::Read(void *, size_t)
{
	return 0;
}


ssize_t
LprTransport::Write(const void *buffer, size_t size)
{
	if (!fStream.write((char *)buffer, size)) {
		fError = true;
		return 0;
	}
	return size;
}