// See the file "COPYING" in the main distribution directory for copyright.

#include "Component.h"

#include "Desc.h"
#include "Reporter.h"

using namespace iosource;

Component::Component(const std::string& name)
	: zeek::plugin::Component(zeek::plugin::component::IOSOURCE, name)
	{
	}

Component::Component(zeek::plugin::component::Type type, const std::string& name)
	: plugin::Component(type, name)
	{
	}

#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wdeprecated-declarations"
Component::Component(plugin::component::Type type, const std::string& name)
	: plugin::Component(static_cast<zeek::plugin::component::Type>(type), name)
	{
	}
#pragma GCC diagnostic pop

Component::~Component()
	{
	}

PktSrcComponent::PktSrcComponent(const std::string& arg_name, const std::string& arg_prefix, InputType arg_type, factory_callback arg_factory)
	: iosource::Component(zeek::plugin::component::PKTSRC, arg_name)
	{
	tokenize_string(arg_prefix, ":", &prefixes);
	type = arg_type;
	factory = arg_factory;
	}

PktSrcComponent::~PktSrcComponent()
	{
	}

const std::vector<std::string>& PktSrcComponent::Prefixes() const
	{
	return prefixes;
	}

bool PktSrcComponent::HandlesPrefix(const std::string& prefix) const
	{
	for ( std::vector<std::string>::const_iterator i = prefixes.begin();
	      i != prefixes.end(); i++ )
		{
		if ( *i == prefix )
			return true;
		}

	return false;
	}

bool PktSrcComponent::DoesLive() const
	{
	return type == LIVE || type == BOTH;
	}

bool PktSrcComponent::DoesTrace() const
	{
	return type == TRACE || type == BOTH;
	}

PktSrcComponent::factory_callback PktSrcComponent::Factory() const
	{
	return factory;
	}

void PktSrcComponent::DoDescribe(ODesc* d) const
	{
	iosource::Component::DoDescribe(d);

	std::string prefs;

	for ( std::vector<std::string>::const_iterator i = prefixes.begin();
	      i != prefixes.end(); i++ )
		{
		if ( prefs.size() )
			prefs += ", ";

		prefs += '"' + *i + '"';
		}

	d->Add("interface prefix");
	if ( prefixes.size() > 1 )
		d->Add("es");

	d->Add(" ");
	d->Add(prefs);
	d->Add("; supports ");

	switch ( type ) {
	case LIVE:
		d->Add("live input");
		break;

	case TRACE:
		d->Add("trace input");
		break;

	case BOTH:
		d->Add("live and trace input");
		break;

	default:
		reporter->InternalError("unknown PkrSrc type");
	}

	}

PktDumperComponent::PktDumperComponent(const std::string& name, const std::string& arg_prefix, factory_callback arg_factory)
	: zeek::plugin::Component(zeek::plugin::component::PKTDUMPER, name)
	{
	tokenize_string(arg_prefix, ":", &prefixes);
	factory = arg_factory;
	}

PktDumperComponent::~PktDumperComponent()
	{
	}

PktDumperComponent::factory_callback PktDumperComponent::Factory() const
	{
	return factory;
	}

const std::vector<std::string>& PktDumperComponent::Prefixes() const
	{
	return prefixes;
	}

bool PktDumperComponent::HandlesPrefix(const std::string& prefix) const
	{
	for ( std::vector<std::string>::const_iterator i = prefixes.begin();
	      i != prefixes.end(); i++ )
		{
		if ( *i == prefix )
			return true;
		}

	return false;
	}

void PktDumperComponent::DoDescribe(ODesc* d) const
	{
	zeek::plugin::Component::DoDescribe(d);

	std::string prefs;

	for ( std::vector<std::string>::const_iterator i = prefixes.begin();
	      i != prefixes.end(); i++ )
		{
		if ( prefs.size() )
			prefs += ", ";

		prefs += '"' + *i + '"';
		}

	d->Add("dumper prefix");

	if ( prefixes.size() > 1 )
		d->Add("es");

	d->Add(": ");
	d->Add(prefs);
	}