haiku.git

/*
 * Copyright 2011-2016, Haiku, Inc. All rights reserved.
 * Copyright 2011, Clemens Zeidler <haiku@clemens-zeidler.de>
 * Distributed under the terms of the MIT License.
 */


#include "FilterList.h"

#include <set>
#include <stdio.h>

#include <Directory.h>
#include <PathFinder.h>
#include <Path.h>


FilterList::FilterList(direction dir)
	:
	fDirection(dir)
{
}


FilterList::~FilterList()
{
	_MakeEmpty();
}


void
FilterList::Reload()
{
	_MakeEmpty();

	std::set<BString> knownNames;

	BString subPath("mail_daemon/");
	subPath << (fDirection == kIncoming
		? "inbound_filters" : "outbound_filters");

	BStringList paths;
	BPathFinder().FindPaths(B_FIND_PATH_ADD_ONS_DIRECTORY, subPath, paths);
	for (int32 i = 0; i < paths.CountStrings(); i++) {
		BPath path(paths.StringAt(i));

		BDirectory dir(path.Path());
		if (dir.InitCheck() != B_OK)
			continue;

		BEntry entry;
		while (dir.GetNextEntry(&entry) == B_OK) {
			// Ignore entries we already had before (ie., user add-ons are
			// overriding system add-ons)
			if (knownNames.find(entry.Name()) != knownNames.end())
				continue;

			if (_LoadAddOn(entry) == B_OK)
				knownNames.insert(entry.Name());
		}
	}
}


int32
FilterList::CountInfos() const
{
	return fList.size();
}


const FilterInfo&
FilterList::InfoAt(int32 index) const
{
	return fList[index];
}


int32
FilterList::InfoIndexFor(const entry_ref& ref) const
{
	for (size_t i = 0; i < fList.size(); i++) {
		const FilterInfo& info = fList[i];
		if (info.ref == ref)
			return i;
	}
	return -1;
}


BString
FilterList::SimpleName(int32 index,
	const BMailAccountSettings& accountSettings) const
{
	return DescriptiveName(index, accountSettings, NULL);
}


BString
FilterList::SimpleName(const entry_ref& ref,
	const BMailAccountSettings& accountSettings) const
{
	return DescriptiveName(InfoIndexFor(ref), accountSettings, NULL);
}


BString
FilterList::DescriptiveName(int32 index,
	const BMailAccountSettings& accountSettings,
	const BMailAddOnSettings* settings) const
{
	if (index < 0 || index >= CountInfos())
		return "-";

	const FilterInfo& info = InfoAt(index);
	return info.name(accountSettings, settings);
}


BString
FilterList::DescriptiveName(const entry_ref& ref,
	const BMailAccountSettings& accountSettings,
	const BMailAddOnSettings* settings) const
{
	return DescriptiveName(InfoIndexFor(ref), accountSettings, settings);
}


BMailSettingsView*
FilterList::CreateSettingsView(const BMailAccountSettings& accountSettings,
	const BMailAddOnSettings& settings)
{
	const entry_ref& ref = settings.AddOnRef();
	int32 index = InfoIndexFor(ref);
	if (index < 0)
		return NULL;

	const FilterInfo& info = InfoAt(index);
	return info.instantiateSettingsView(accountSettings, settings);
}


void
FilterList::_MakeEmpty()
{
	for (size_t i = 0; i < fList.size(); i++) {
		FilterInfo& info = fList[i];
		unload_add_on(info.image);
	}
	fList.clear();
}


status_t
FilterList::_LoadAddOn(BEntry& entry)
{
	FilterInfo info;

	BPath path(&entry);
	info.image = load_add_on(path.Path());
	if (info.image < 0)
		return info.image;

	status_t status = get_image_symbol(info.image,
		"instantiate_filter_settings_view", B_SYMBOL_TYPE_TEXT,
		(void**)&info.instantiateSettingsView);
	if (status == B_OK) {
		status = get_image_symbol(info.image, "filter_name", B_SYMBOL_TYPE_TEXT,
			(void**)&info.name);
	}
	if (status != B_OK) {
		fprintf(stderr, "Filter \"%s\" misses required hooks!\n", path.Path());
		unload_add_on(info.image);
		return B_NAME_NOT_FOUND;
	}

	entry.GetRef(&info.ref);
	fList.push_back(info);
	return B_OK;
}