From 43e06258708282b60b6198166db2185689a408c1 Mon Sep 17 00:00:00 2001
From: Daniel Friesel <derf@finalrewind.org>
Date: Tue, 30 Dec 2014 10:48:06 +0100
Subject: Result: Add ->type accessor

---
 lib/Travel/Status/DE/DeutscheBahn/Result.pm | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/lib/Travel/Status/DE/DeutscheBahn/Result.pm b/lib/Travel/Status/DE/DeutscheBahn/Result.pm
index fb0fb95..1f9f81c 100644
--- a/lib/Travel/Status/DE/DeutscheBahn/Result.pm
+++ b/lib/Travel/Status/DE/DeutscheBahn/Result.pm
@@ -137,6 +137,15 @@ sub TO_JSON {
 	return { %{$self} };
 }
 
+sub type {
+	my ($self) = @_;
+
+	# $self->{train} is either "TYPE 12345" or "TYPE12345"
+	my ($type) = ( $self->{train} =~ m{ ^ ([A-Z]+) }x );
+
+	return $type;
+}
+
 1;
 
 __END__
@@ -259,6 +268,11 @@ references of the form C<< [ arrival time, station name ] >>.
 
 Returns the arrival/departure time as string in "hh:mm" format.
 
+=item $result->type
+
+Returns the type of this train, e.g. "S" for S-Bahn, "RE" for Regional Express
+or "ICE" for InterCity-Express.
+
 =back
 
 =head2 INTERNAL
-- 
cgit v1.2.3