#include	"config.h"
#include	"message.h"
#include	"buffer.h"
#include	"xconfig.h"
#include	"funcs.h"
#include	"varlist.h"

static int rfc2045_seek_func(off_t pos, void *arg)
{
	Message *p=reinterpret_cast<Message *>(arg);

	p->seek(pos);
	return 0;
}

static ssize_t rfc2045_read_func(char *buf, size_t cnt, void *arg)
{
	Message *p=reinterpret_cast<Message *>(arg);

	ssize_t n=0;

	while (cnt)
	{
		int c=p->get_c();

		if (c < 0)
			return -1;
		*buf++=c;
		--cnt;
		++n;
	}
	return n;
}

Message::Message() : buffer(0), bufptr(0),
		     extra_headers(0), extra_headersptr(0), msgsize(0),
		     rfc2045src_parser(rfc2045src()),
		     rfc2045p(0)
{
	rfc2045src_parser.seek_func=rfc2045_seek_func;
	rfc2045src_parser.read_func=rfc2045_read_func;
	rfc2045src_parser.arg=reinterpret_cast<void *>(this);
}

Message::~Message()
{
	mio.fd(-1);	// Either way, it's not our file
	if (buffer)	delete[] buffer;
	if (extra_headers) delete[] extra_headers;
	if (rfc2045p)
		rfc2045_free(rfc2045p);
}

void Message::Init()
{
	if (buffer)
	{
		delete[] buffer;
		buffer=0;
	}
	if (extra_headers)
	{
		delete[] extra_headers;
		extra_headers=0;
	}
	if (rfc2045p)
		rfc2045_free(rfc2045p);
	extra_headersptr=0;
	msgsize=0;
	msglines=0;
	tempfile.Close();
	mio.fd(-1);
	rfc2045p=rfc2045_alloc();
}

void Message::Init(int fd, const Buffer &extra_headers)
{
	Init();
	ExtraHeaders(extra_headers);

	// If the file descriptor is seekable, and the message is big
	// keep message in the file.
	if ( mseek(fd, 0, SEEK_END) >= 0 &&
		(msgsize=mseek(fd, 0, SEEK_CUR)) >= 0 && msgsize > SMALLMSG)
	{

	// BUGFIX 0.55 - mio takes ownership of the file descriptor. When
	// ::Init() is called, it will call fd(-1) which will CLOSE this
	// descriptor we have now.
	//
	// This func is invoked from main with STDIN - 0, so we must take
	// ownership of a duplicate.

		fd=dup(fd);
		if (fd < 0)	throw "dup() failed.";

		mio.fd(fd);
		mio.rfc2045p=rfc2045p;
		if (mio.Rewind() < 0)	seekerr();

	int	c;

		while ((c=mio.get()) >= 0)
			if (c == '\n')	msglines++;
		mio.rfc2045p=0;
		return;
	}
	// Well, just read the message, and let Init() figure out what to
	// do.

	mseek(fd, 0, SEEK_SET);	// Just in case
	msgsize=0;

#ifdef	BUFSIZ
char	buf[BUFSIZ];
#else
char	buf[8192];
#endif
int	n;

	while ((n=read(fd, buf, sizeof(buf))) > 0)
	{
		Init(buf, n);
	}
	if (n < 0)
		throw "Error - read() failed reading message.";
#if CRLF_TERM
	msgsize += msglines;
#endif
}

void Message::Init(const void *data, unsigned cnt)
{
	rfc2045_parse(rfc2045p, (const char *)data, cnt);
	{
	const char *p=(const char*)data;
	unsigned n=cnt;

		while (n)
		{
			if (*p++ == '\n')	++msglines;
			--n;
		}
	}

	if (mio.fd() < 0)	// Still trying to save to buffer
	{
		if (!buffer)
		{
			buffer=new char[SMALLMSG];
			bufptr=buffer;
			msgsize=0;
			if (!buffer)	outofmem();
		}

		if (SMALLMSG - msgsize >= (off_t)cnt)	// Can still do it
		{
			memcpy(bufptr, data, cnt);
			bufptr += cnt;
			msgsize += cnt;
			return;
		}

	int	fd;

#if	SHARED_TEMPDIR

		fd=tempfile.Open();

		if (fd < 0)
			throw "Unable to create temporary file.";
#else
		while ( (fd=tempfile.Open( TempName(),
			O_RDWR | O_CREAT | O_EXCL, 0600)) < 0 &&
			errno == EEXIST)
			;
		if (fd < 0)
			throw "Unable to create temporary file - check permissions on $HOME/" TEMPDIR;
#endif

		mio.fd(fd);

		if ((off_t)mio.write(buffer, msgsize) != msgsize)
			throw "Unable to write to temporary file - possibly out of disk space.";

		delete[] buffer;
		buffer=0;
	}

	if ((unsigned)mio.write(data, cnt) != cnt)
		throw "Unable to write to temporary file - possibly out of disk space.";
	msgsize += cnt;
}

void Message::ExtraHeaders(const Buffer &buf)
{
	rfc2045_parse(rfc2045p, (const char *)buf, buf.Length());

	if ( extra_headers )
	{
		delete[] extra_headers;
		extra_headers=0;
	}
	extra_headersptr=0;
	if (!buf.Length())	return;

	extra_headers=new char[buf.Length()+1];
	if (!extra_headers)	outofmem();
	memcpy(extra_headers, (const char *)buf, buf.Length());
	extra_headers[buf.Length()]=0;
	extra_headersptr=extra_headers;
	if (!*extra_headersptr)	extra_headersptr=0;
}

void Message::RewindIgnore()
{
	extra_headersptr=0;
	if (mio.fd() >= 0)
	{
		if (mio.Rewind() < 0)	seekerr();
		return;
	}
	bufptr=buffer;
}

void Message::Rewind()
{
	RewindIgnore();

	extra_headersptr=extra_headers;
	if (extra_headersptr && !*extra_headersptr)
		extra_headersptr=0;
}

void Message::readerr()
{
	throw "Read error.";
}

void Message::seekerr()
{
	throw "Seek error.";
}

int Message::appendline(Buffer &buf, int stripcr)
{
	if (mio.fd() >= 0 || extra_headersptr)
	{
	int	c;
	int	eof= 1;
	int	lastc=0;

		while ((c=get_c()) > 0 && c != '\n')
		{
			eof=0;
			buf.push(c);
			lastc=c;
		}
		if (c < 0 && eof)	return (-1);
		if (stripcr && lastc == '\r')	buf.pop();
						// Drop trailing CRs
		buf.push('\n');
		return (0);
	}

	if (bufptr >= buffer + msgsize)
	{
		return (-1);
	}

unsigned cnt=buffer + msgsize - bufptr;
unsigned i;

	for (i=0; i<cnt; i++)
		if (bufptr[i] == '\n')
		{
			if (i > 0 && stripcr && bufptr[i-1] == '\r')
				buf.append(bufptr, i-1);
				// Drop trailing CRs
			else
				buf.append(bufptr, i);
			buf += '\n';
			bufptr += ++i;
			return (0);
		}

	if (stripcr && bufptr[i-1] == '\r')
		buf.append(bufptr, cnt-1);
				// Drop trailing CRs
	else
		buf.append(bufptr, cnt);
	bufptr += cnt;
	buf += '\n';
	return (0);
}

void Message::setmsgsize()
{
Buffer	n,v;

	n="SIZE";
	v.append((unsigned long)MessageSize());
	SetVar(n,v);
	n="LINES";
	v.reset();
	v.append((unsigned long)MessageLines());
	SetVar(n,v);
}