add mot filter support

2 files changed, 24 insertions, 2 deletions

diff --git a/lib/Travel/Status/DE/EFA/Line.pm b/lib/Travel/Status/DE/EFA/Line.pm
index 35560b2..0f5f090 100644
--- a/lib/Travel/Status/DE/EFA/Line.pm
+++ b/lib/Travel/Status/DE/EFA/Line.pm
@@ -9,7 +9,12 @@ use parent 'Class::Accessor';
 our $VERSION = '1.11';
 
 Travel::Status::DE::EFA::Line->mk_ro_accessors(
-	qw(direction name operator route type valid));
+	qw(direction mot name operator route type valid));
+
+my @mot_mapping = qw{
+  zug s-bahn u-bahn stadtbahn tram stadtbus regionalbus
+  schnellbus seilbahn schiff ast sonstige
+};
 
 sub new {
 	my ( $obj, %conf ) = @_;
@@ -19,6 +24,12 @@ sub new {
 	return bless( $ref, $obj );
 }
 
+sub mot_name {
+	my ($self) = @_;
+
+	return $mot_mapping[ $self->{mot} ] // 'sonstige';
+}
+
 sub TO_JSON {
 	my ($self) = @_;
 
diff --git a/lib/Travel/Status/DE/EFA/Result.pm b/lib/Travel/Status/DE/EFA/Result.pm
index d516e25..bb0243b 100644
--- a/lib/Travel/Status/DE/EFA/Result.pm
+++ b/lib/Travel/Status/DE/EFA/Result.pm
@@ -12,9 +12,14 @@ our $VERSION = '1.11';
 
 Travel::Status::DE::EFA::Result->mk_ro_accessors(
 	qw(countdown date delay destination is_cancelled info key line lineref
-	  platform platform_db platform_name sched_date sched_time time type)
+	  mot platform platform_db platform_name sched_date sched_time time type)
 );
 
+my @mot_mapping = qw{
+  zug s-bahn u-bahn stadtbahn tram stadtbus regionalbus
+  schnellbus seilbahn schiff ast sonstige
+};
+
 sub new {
 	my ( $obj, %conf ) = @_;
 
@@ -31,6 +36,12 @@ sub new {
 	return bless( $ref, $obj );
 }
 
+sub mot_name {
+	my ($self) = @_;
+
+	return $mot_mapping[ $self->{mot} ] // 'sonstige';
+}
+
 sub route_pre {
 	my ($self) = @_;