/*
 * Copyright 2004-2012, Haiku, Inc. All rights reserved.
 * Copyright 2001, Dr. Zoidberg Enterprises. All rights reserved.
 *
 * Distributed under the terms of the MIT License.
 */


//!	Adds fortunes to your mail


#include "ConfigView.h"

#include <Catalog.h>
#include <Message.h>
#include <Entry.h>
#include <String.h>
#include <MailFilter.h>
#include <MailMessage.h>

#include <stdio.h>

#include "NodeMessage.h"


#undef B_TRANSLATION_CONTEXT
#define B_TRANSLATION_CONTEXT "FortuneFilter"


class FortuneFilter : public BMailFilter {
public:
								FortuneFilter(BMailProtocol& protocol,
									const BMailAddOnSettings& settings);

	virtual	void				MessageReadyToSend(const entry_ref& ref,
									BFile& file);
};


FortuneFilter::FortuneFilter(BMailProtocol& protocol,
	const BMailAddOnSettings& settings)
	:
	BMailFilter(protocol, &settings)
{
}


void
FortuneFilter::MessageReadyToSend(const entry_ref& ref, BFile& file)
{
	// What we want to do here is to change the message body. To do that we use the
	// framework we already have by creating a new BEmailMessage based on the
	// BPositionIO, changing the message body and rendering it back to disk. Of course
	// this method ends up not being super-efficient, but it works. Ideas on how to
	// improve this are welcome.
	BString	fortuneFile;
	BString	tagLine;

	// Obtain relevant settings
	fSettings->FindString("fortune_file", &fortuneFile);
	fSettings->FindString("tag_line", &tagLine);

	// Add command to be executed
	fortuneFile.Prepend("/bin/fortune ");

	char buffer[768];
	FILE *fd;

	fd = popen(fortuneFile.String(), "r");
	if (!fd) {
		printf("Could not open pipe to fortune!\n");
		return;
	}

	BEmailMessage mailMessage(&ref);
	BString fortuneText;
	fortuneText += "\n--\n";
	fortuneText += tagLine;

	while (fgets(buffer, 768, fd)) {
		fortuneText += buffer;
		fortuneText += "\n";
	}
	pclose(fd);

	// Update the message body
	BTextMailComponent* body = mailMessage.Body();
	body->AppendText(fortuneText);

	file.SetSize(0);
	mailMessage.RenderToRFC822(&file);
}


// #pragma mark -


BString
filter_name(const BMailAccountSettings& accountSettings,
	const BMailAddOnSettings* settings)
{
	return B_TRANSLATE("Fortune");
}


BMailFilter*
instantiate_filter(BMailProtocol& protocol,
	const BMailAddOnSettings& settings)
{
	return new FortuneFilter(protocol, settings);
}


BMailSettingsView*
instantiate_filter_settings_view(const BMailAccountSettings& accountSettings,
	const BMailAddOnSettings& settings)
{
	ConfigView* view = new ConfigView();
	view->SetTo(settings);

	return view;
}