From 0e1bb3d63c4eb448f4cc458f53096135e376bd51 Mon Sep 17 00:00:00 2001
From: Daniel Friesel <derf@finalrewind.org>
Date: Tue, 10 Nov 2015 16:56:04 +0100
Subject: add accessors for whole (related) station data set

---
 lib/Travel/Status/DE/IRIS.pm | 19 ++++++++++++-------
 1 file changed, 12 insertions(+), 7 deletions(-)

(limited to 'lib/Travel/Status')

diff --git a/lib/Travel/Status/DE/IRIS.pm b/lib/Travel/Status/DE/IRIS.pm
index 6b73939..fd0978c 100644
--- a/lib/Travel/Status/DE/IRIS.pm
+++ b/lib/Travel/Status/DE/IRIS.pm
@@ -35,7 +35,6 @@ sub new {
 		  // 'http://iris.noncd.db.de/iris-tts/timetable',
 		lookahead => $opt{lookahead} // ( 4 * 60 ),
 		serializable => $opt{serializable},
-		station      => $opt{station},
 		user_agent   => $ua,
 		with_related => $opt{with_related},
 	};
@@ -50,8 +49,8 @@ sub new {
 		recursive => $opt{with_related},
 	);
 
-	$self->{station_code} = $station->{uic};
-	$self->{station_name} = $station->{name};
+	$self->{station}          = $station;
+	$self->{related_stations} = \@related_stations;
 
 	for my $ref (@related_stations) {
 		my $ref_status = Travel::Status::DE::IRIS->new(
@@ -72,7 +71,7 @@ sub new {
 
 	my $dt_req = $self->{datetime}->clone;
 	for ( 1 .. 3 ) {
-		$self->get_timetable( $self->{station_code}, $dt_req );
+		$self->get_timetable( $self->{station}{uic}, $dt_req );
 		$dt_req->add( hours => 1 );
 	}
 
@@ -283,7 +282,7 @@ sub get_timetable {
 sub get_realtime {
 	my ($self) = @_;
 
-	my $eva = $self->{station_code};
+	my $eva = $self->{station}{uic};
 	my $res = $self->{user_agent}->get( $self->{iris_base} . "/fchg/${eva}" );
 
 	if ( $self->{developer_mode} ) {
@@ -438,10 +437,16 @@ sub create_replacement_refs {
 	}
 }
 
-sub station_code {
+sub station {
 	my ($self) = @_;
 
-	return $self->{station_code};
+	return $self->{station};
+}
+
+sub related_stations {
+	my ($self) = @_;
+
+	return @{ $self->{related_stations} };
 }
 
 sub errstr {
-- 
cgit v1.2.3