89 lines
2.6 KiB
C++
89 lines
2.6 KiB
C++
|
|
#include "warnlist_item.h"
|
|
|
|
#include "modules/shared.h"
|
|
|
|
#include "modules/exceptions.h"
|
|
|
|
#include "utilities/bwlists.h"
|
|
|
|
|
|
const char* WarnlistItem::fieldName() const
|
|
{
|
|
switch (this->field) {
|
|
case WarnlistField::Method:
|
|
return FIELDS__METHOD.c_str();
|
|
case WarnlistField::Uri:
|
|
return FIELDS__URI.c_str();
|
|
case WarnlistField::Client:
|
|
return FIELDS__CLIENT.c_str();
|
|
case WarnlistField::UserAgent:
|
|
return FIELDS__USER_AGENT.c_str();
|
|
default:
|
|
// should be unreachable
|
|
throw DoNotCatchException( "Unexpected WarnlistField" );
|
|
}
|
|
}
|
|
|
|
void WarnlistItem::set( const std::vector<std::string>& new_list )
|
|
{
|
|
this->list.clear();
|
|
this->list.reserve( new_list.size() );
|
|
for ( const std::string& item : new_list ) {
|
|
this->add( item );
|
|
}
|
|
}
|
|
|
|
void WarnlistItem::add( const std::string& item )
|
|
{
|
|
this->list.push_back( this->sanitized( item ) );
|
|
}
|
|
|
|
void WarnlistItem::remove( const std::string& item )
|
|
{
|
|
if ( const auto it{ std::find( this->list.cbegin(), this->list.cend(), item ) }; it != this->list.cend() ) {
|
|
this->list.erase( it );
|
|
} else {
|
|
throw BWlistException( "Failed to remove the item: "+item );
|
|
}
|
|
}
|
|
|
|
int WarnlistItem::moveUp( const std::string& item )
|
|
{
|
|
if ( auto it{ std::find( std::next(this->list.begin()), this->list.end(), item ) }; it != this->list.end() ) {
|
|
const int pos{ static_cast<int>( std::distance(this->list.begin(), it) ) - 1 };
|
|
std::swap( *it, *std::prev(it) );
|
|
return pos;
|
|
} else {
|
|
throw BWlistException( "Failed to move up the item: "+item );
|
|
}
|
|
}
|
|
|
|
int WarnlistItem::moveDown( const std::string& item )
|
|
{
|
|
if ( auto it{ std::find( this->list.begin(), std::prev(this->list.end()), item ) }; it != this->list.end() ) {
|
|
const int pos{ static_cast<int>( std::distance(this->list.begin(), it) ) + 1 };
|
|
std::swap( *it, *std::next(it) );
|
|
return pos;
|
|
} else {
|
|
throw BWlistException( "Failed to move down the item: "+item );
|
|
}
|
|
}
|
|
|
|
std::string WarnlistItem::sanitized( const std::string& item ) const
|
|
{
|
|
switch (this->field) {
|
|
case WarnlistField::Method:
|
|
return BWutils::sanitizedMethod( item );
|
|
case WarnlistField::Uri:
|
|
return BWutils::sanitizedUri( item );
|
|
case WarnlistField::Client:
|
|
return BWutils::sanitizedClient( item );
|
|
case WarnlistField::UserAgent:
|
|
return BWutils::sanitizedUserAgent( item );
|
|
default:
|
|
// should be unreachable
|
|
throw DoNotCatchException( "Unexpected WarnlistField" );
|
|
}
|
|
}
|