1 files changed, 88 insertions, 0 deletions
diff --git a/lib/Travelynx/Model/Stations.pm b/lib/Travelynx/Model/Stations.pm
new file mode 100644
index 0000000..6c898b1
--- /dev/null
+++ b/lib/Travelynx/Model/Stations.pm
@@ -0,0 +1,88 @@
+package Travelynx::Model::Stations;
+
+# Copyright (C) 2022 Daniel Friesel
+#
+# SPDX-License-Identifier: AGPL-3.0-or-later
+
+use strict;
+use warnings;
+use 5.020;
+
+sub new {
+	my ( $class, %opt ) = @_;
+
+	return bless( \%opt, $class );
+}
+
+# Fast
+sub get_by_eva {
+	my ( $self, $eva, %opt ) = @_;
+
+	if ( not $eva ) {
+		return;
+	}
+
+	my $db = $opt{db} // $self->{pg}->db;
+
+	return $db->select( 'stations', '*', { eva => $eva } )->hash;
+}
+
+# Fast
+sub get_by_evas {
+	my ( $self, @evas ) = @_;
+
+	my @ret
+	  = $self->{pg}->db->select( 'stations', '*', { eva => { '=', \@evas } } )
+	  ->hashes->each;
+	return @ret;
+}
+
+# Slow
+sub get_latlon_by_name {
+	my ( $self, %opt ) = @_;
+
+	my $db = $opt{db} // $self->{pg}->db;
+
+	my %location;
+	my $res = $db->select( 'stations', [ 'name', 'lat', 'lon' ] );
+	while ( my $row = $res->hash ) {
+		$location{ $row->{name} } = [ $row->{lat}, $row->{lon} ];
+	}
+	return \%location;
+}
+
+# Slow
+sub get_by_name {
+	my ( $self, $name, %opt ) = @_;
+
+	my $db = $opt{db} // $self->{pg}->db;
+
+	return $db->select( 'stations', '*', { name => $name }, { limit => 1 } )
+	  ->hash;
+}
+
+# Slow
+sub get_by_ds100 {
+	my ( $self, $ds100, %opt ) = @_;
+
+	my $db = $opt{db} // $self->{pg}->db;
+
+	return $db->select( 'stations', '*', { ds100 => $ds100 }, { limit => 1 } )
+	  ->hash;
+}
+
+# Can be slow
+sub search {
+	my ( $self, $identifier, %opt ) = @_;
+
+	if ( $identifier =~ m{ ^ \d+ $ }x ) {
+		return $self->get_by_eva( $identifier, %opt )
+		  // $self->get_by_ds100( $identifier, %opt )
+		  // $self->get_by_name( $identifier, %opt );
+	}
+
+	return $self->get_by_ds100( $identifier, %opt )
+	  // $self->get_by_name( $identifier, %opt );
+}
+
+1;