move token type to ID map from Travelynx.pm to Model/Users

1 files changed, 12 insertions, 0 deletions

diff --git a/lib/Travelynx/Model/Users.pm b/lib/Travelynx/Model/Users.pm
index 0287ca2..46ffbdd 100644
--- a/lib/Travelynx/Model/Users.pm
+++ b/lib/Travelynx/Model/Users.pm
@@ -19,6 +19,12 @@ my @sb_templates = (
 	[ 'bahn.expert/regional', 'https://bahn.expert/regional/{name}#{id}' ],
 );
 
+my %token_id = (
+	status  => 1,
+	history => 2,
+	travel  => 3,
+	import  => 4,
+);
 my @token_types = (qw(status history travel import));
 
 sub new {
@@ -27,6 +33,12 @@ sub new {
 	return bless( \%opt, $class );
 }
 
+sub get_token_id {
+	my ( $self, $type ) = @_;
+
+	return $token_id{$type};
+}
+
 sub mark_seen {
 	my ( $self, %opt ) = @_;
 	my $uid = $opt{uid};