From 7e92994838ae4c4548c2569f303340756793a0f4 Mon Sep 17 00:00:00 2001
From: Birte Kristina Friesel <derf@finalrewind.org>
Date: Wed, 3 Apr 2024 18:50:36 +0200
Subject: Journey: Add operators accessor

---
 lib/Travel/Status/DE/HAFAS/Journey.pm | 27 +++++++++++++++++++++++++--
 1 file changed, 25 insertions(+), 2 deletions(-)

(limited to 'lib/Travel/Status/DE/HAFAS')

diff --git a/lib/Travel/Status/DE/HAFAS/Journey.pm b/lib/Travel/Status/DE/HAFAS/Journey.pm
index afdea48..986582a 100644
--- a/lib/Travel/Status/DE/HAFAS/Journey.pm
+++ b/lib/Travel/Status/DE/HAFAS/Journey.pm
@@ -8,7 +8,7 @@ use 5.014;
 
 use parent 'Class::Accessor';
 use DateTime::Format::Strptime;
-use List::Util qw(any);
+use List::Util qw(any uniq);
 use Travel::Status::DE::HAFAS::Stop;
 
 our $VERSION = '6.00';
@@ -270,6 +270,23 @@ sub messages {
 	return;
 }
 
+sub operators {
+	my ($self) = @_;
+
+	if ( $self->{operators} ) {
+		return @{ $self->{operators} };
+	}
+
+	$self->{operators} = [
+		uniq map { ( $_->prod_arr // $_->prod_dep )->operator } grep {
+			      ( $_->prod_arr or $_->prod_dep )
+			  and ( $_->prod_arr // $_->prod_dep )->operator
+		} $self->route
+	];
+
+	return @{ $self->{operators} };
+}
+
 sub polyline {
 	my ($self) = @_;
 
@@ -549,7 +566,13 @@ detailed delay reasons (e.g. "switch damage between X and Y, expect delays").
 =item $journey->operator
 
 The operator responsible for this journey. Returns undef
-if the backend does not provide an operator.
+if the backend does not provide an operator. Note that the operator may
+change along the journey -- in this case, the returned operator depends on
+the backend and appears to be the first one in most cases.
+
+=item $journey->operators
+
+List of all operators observed along the journey.
 
 =item $journey->station (station only)
 
-- 
cgit v1.2.3