travelynx/lib/Travelynx/Model/Stations.pm

package Travelynx::Model::Stations;

# Copyright (C) 2022 Birte Kristina 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 );
}

sub get_backend_id {
	my ( $self, %opt ) = @_;

	if ( $opt{iris} ) {

		# special case
		return 0;
	}
	if ( $opt{hafas} and $self->{backend_id}{hafas}{ $opt{hafas} } ) {
		return $self->{backend_id}{hafas}{ $opt{hafas} };
	}

	my $db         = $opt{db} // $self->{pg}->db;
	my $backend_id = 0;

	if ( $opt{hafas} ) {
		$backend_id = $db->select(
			'backends',
			['id'],
			{
				hafas => 1,
				name  => $opt{hafas}
			}
		)->hash->{id};
		$self->{backend_id}{hafas}{ $opt{hafas} } = $backend_id;
	}

	return $backend_id;
}

sub get_hafas_name {
	my ( $self, %opt ) = @_;

	if ( exists $self->{hafas_name}{ $opt{backend_id} } ) {
		return $self->{hafas_name}{ $opt{backend_id} };
	}

	my $db = $opt{db} // $self->{pg}->db;
	my $hafas_name;
	my $ret = $db->select(
		'backends',
		['name'],
		{
			hafas => 1,
			id    => $opt{backend_id},
		}
	)->hash;

	if ($ret) {
		$hafas_name = $ret->{name};
	}

	$self->{hafas_name}{ $opt{backend_id} } = $hafas_name;

	return $hafas_name;
}

sub get_backends {
	my ( $self, %opt ) = @_;

	$opt{db} //= $self->{pg}->db;

	my $res = $opt{db}->select( 'backends', [ 'id', 'name', 'iris', 'hafas' ] );
	my @ret;

	while ( my $row = $res->hash ) {
		push(
			@ret,
			{
				id    => $row->{id},
				name  => $row->{name},
				iris  => $row->{iris},
				hafas => $row->{hafas},
			}
		);
	}

	return @ret;
}

sub add_or_update {
	my ( $self, %opt ) = @_;
	my $stop = $opt{stop};
	my $loc  = $stop->loc;
	$opt{db} //= $self->{pg}->db;

	$opt{backend_id} //= $self->get_backend_id(%opt);

	if (
		my $s = $self->get_by_eva(
			$loc->eva,
			db         => $opt{db},
			backend_id => $opt{backend_id}
		)
	  )
	{
		$opt{db}->update(
			'stations',
			{
				name     => $loc->name,
				lat      => $loc->lat,
				lon      => $loc->lon,
				archived => 0
			},
			{
				eva    => $loc->eva,
				source => $opt{backend_id}
			}
		);
		return;
	}
	$opt{db}->insert(
		'stations',
		{
			eva      => $loc->eva,
			name     => $loc->name,
			lat      => $loc->lat,
			lon      => $loc->lon,
			source   => $opt{backend_id},
			archived => 0
		}
	);
}

sub add_meta {
	my ( $self, %opt ) = @_;
	my $eva  = $opt{eva};
	my @meta = @{ $opt{meta} };

	$opt{db}         //= $self->{pg}->db;
	$opt{backend_id} //= $self->get_backend_id(%opt);

	for my $meta (@meta) {
		if ( $meta != $eva ) {
			$opt{db}->insert(
				'related_stations',
				{
					eva        => $eva,
					meta       => $meta,
					backend_id => $opt{backend_id},
				},
				{ on_conflict => undef }
			);
		}
	}
}

sub get_db_iterator {
	my ($self) = @_;

	return $self->{pg}->db->select( 'stations', '*' );
}

sub get_meta {
	my ( $self, %opt ) = @_;
	my $db  = $opt{db} // $self->{pg}->db;
	my $eva = $opt{eva};

	$opt{backend_id} //= $self->get_backend_id( %opt, db => $db );

	my $res = $db->select(
		'related_stations',
		['meta'],
		{
			eva        => $eva,
			backend_id => $opt{backend_id}
		}
	);
	my @ret;

	while ( my $row = $res->hash ) {
		push( @ret, $row->{meta} );
	}

	return @ret;
}

sub get_for_autocomplete {
	my ( $self, %opt ) = @_;

	$opt{backend_id} //= $self->get_backend_id(%opt);

	my $res = $self->{pg}
	  ->db->select( 'stations', ['name'], { source => $opt{backend_id} } );
	my %ret;

	while ( my $row = $res->hash ) {
		$ret{ $row->{name} } = undef;
	}

	return \%ret;
}

# Fast
sub get_by_eva {
	my ( $self, $eva, %opt ) = @_;

	if ( not $eva ) {
		return;
	}

	$opt{db}         //= $self->{pg}->db;
	$opt{backend_id} //= $self->get_backend_id(%opt);

	return $opt{db}->select(
		'stations',
		'*',
		{
			eva    => $eva,
			source => $opt{backend_id}
		}
	)->hash;
}

# Fast
sub get_by_evas {
	my ( $self, %opt ) = @_;

	$opt{db}         //= $self->{pg}->db;
	$opt{backend_id} //= $self->get_backend_id(%opt);

	my @ret = $self->{pg}->db->select(
		'stations',
		'*',
		{
			eva    => { '=', $opt{evas} },
			source => $opt{backend_id}
		}
	)->hashes->each;
	return @ret;
}

# Slow
sub get_by_name {
	my ( $self, $name, %opt ) = @_;

	$opt{db}         //= $self->{pg}->db;
	$opt{backend_id} //= $self->get_backend_id(%opt);

	return $opt{db}->select(
		'stations',
		'*',
		{
			name   => $name,
			source => $opt{backend_id}
		},
		{ limit => 1 }
	)->hash;
}

# Slow
sub get_by_names {
	my ( $self, @names ) = @_;

	my @ret
	  = $self->{pg}->db->select( 'stations', '*', { name => { '=', \@names } } )
	  ->hashes->each;
	return @ret;
}

# Slow
sub get_by_ds100 {
	my ( $self, $ds100, %opt ) = @_;

	$opt{db}         //= $self->{pg}->db;
	$opt{backend_id} //= $self->get_backend_id(%opt);

	return $opt{db}->select(
		'stations',
		'*',
		{
			ds100  => $ds100,
			source => $opt{backend_id}
		},
		{ limit => 1 }
	)->hash;
}

# Can be slow
sub search {
	my ( $self, $identifier, %opt ) = @_;

	$opt{db}         //= $self->{pg}->db;
	$opt{backend_id} //= $self->get_backend_id(%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 );
}

# Slow
sub grep_unknown {
	my ( $self, @stations ) = @_;

	my %station = map { $_->{name} => 1 } $self->get_by_names(@stations);
	my @unknown_stations = grep { not $station{$_} } @stations;

	return @unknown_stations;
}

1;