Merge branch 'feuerrot-db_setup'

README.md

@@ -54,7 +54,7 @@ Debian 9 system, though setup on other distribution should be similar.
 * Create the database: `sudo -u postgres createdb -O travelynx travelynx`
 * Copy `examples/travelynx.conf` to the application root directory
   (the one in which `index.pl` resides) and configure it
-* Initialize the database: `perl index.pl database setup`
+* Initialize the database: `perl index.pl database migrate`
 
 Your server also needs to be able to send mail. Set up your MTA of choice and
 make sure that the sendmail binary can be used for outgoing mails. Mail

lib/Travelynx/Command/database.pm

@@ -9,10 +9,28 @@ has usage => sub { shift->extract_usage };
 
 sub get_schema_version {
 	my ($dbh) = @_;
-	for my $entry (
+
-		$dbh->selectall_array(qq{select version from schema_version}) )
+	# We do not want DBD to print an SQL error if schema_version does not
-	{
+	# exist, as this is not an error in this case. Setting $dbh->{PrintError} =
-		return $entry->[0];
+	# 0 would disable error printing for all subsequent SQL operations, however,
+	# we only want to disable it for this specific query. Hence we use a
+	# prepared statement and only disable error printing only there.
+	my $sth = $dbh->prepare(qq{select version from schema_version});
+	$sth->{PrintError} = 0;
+	my $success = $sth->execute;
+
+	if ( not defined $success ) {
+		return undef;
+	}
+
+	my $rows = $sth->fetchall_arrayref;
+
+	if ( @{$rows} == 1 ) {
+		return $rows->[0][0];
+	}
+	else {
+		printf( "Found multiple schema versions: %s", @{$rows} );
+		exit(1);
 	}
 }
 
@@ -46,7 +64,6 @@ sub initialize_db {
 				action_id smallint not null,
 				station_id int references stations (id),
 				action_time timestamptz not null,
-				edited smallint not null,
 				train_type varchar(16),
 				train_line varchar(16),
 				train_no varchar(16),
@@ -67,7 +84,7 @@ sub initialize_db {
 				token varchar(80) not null,
 				primary key (user_id, type)
 			);
-			insert into schema_version values (2);
+			insert into schema_version values (0);
 		}
 	);
 }
@@ -108,23 +125,21 @@ my @migrations = (
 	},
 );
 
-sub run {
+sub setup_db {
-	my ( $self, $command ) = @_;
+	my ($dbh) = @_;
-	my $exit_status = 0;
-
-	my $dbh = $self->app->dbh;
-
-	if ( $command eq 'setup' ) {
 	$dbh->begin_work;
 	if ( initialize_db($dbh) ) {
 		$dbh->commit;
 	}
 	else {
 		$dbh->rollback;
-			$exit_status = 1;
+		printf( "Database initialization was not successful: %s",
+			$DBI::errstr );
 	}
-	}
+}
-	elsif ( $command eq 'migrate' ) {
+
+sub migrate_db {
+	my ($dbh) = @_;
 	$dbh->begin_work;
 	my $schema_version = get_schema_version($dbh);
 	say "Found travelynx schema v${schema_version}";
@@ -136,13 +151,29 @@ sub run {
 		if ( not $migrations[$i]($dbh) ) {
 			say "Aborting migration; rollback to v${schema_version}";
 			$dbh->rollback;
-				$exit_status = 1;
+			exit(1);
-				last;
 		}
 	}
 	if ( get_schema_version($dbh) == @migrations ) {
 		$dbh->commit;
 	}
+}
+
+sub run {
+	my ( $self, $command ) = @_;
+
+	my $dbh = $self->app->dbh;
+
+	if ( not defined $dbh ) {
+		printf( "Can't connect to the database: %s\n", $DBI::errstr );
+		exit(1);
+	}
+
+	if ( $command eq 'migrate' ) {
+		if ( not defined get_schema_version($dbh) ) {
+			setup_db($dbh);
+		}
+		migrate_db($dbh);
 	}
 	elsif ( $command eq 'has-current-schema' ) {
 		if ( get_schema_version($dbh) == @migrations ) {
@@ -150,7 +181,7 @@ sub run {
 		}
 		else {
 			say "no";
-			$exit_status = 1;
+			exit(1);
 		}
 	}
 	else {
@@ -158,8 +189,6 @@ sub run {
 	}
 
 	$dbh->disconnect;
-
-	exit($exit_status);
 }
 
 1;
@@ -168,12 +197,11 @@ __END__
 
 =head1 SYNOPSIS
 
-  Usage: index.pl database <setup|migrate|has-current-schema>
+  Usage: index.pl database <migrate|has-current-schema>
 
   Upgrades the database layout to the latest schema.
 
   Recommended workflow:
   > systemctl stop travelynx
-  > TRAVELYNX_DB_HOST=... TRAVELYNX_DB_NAME=... TRAVELYNX_DB_USER=... \
+  > perl index.pl migrate
-    TRAVELYNX_DB_PASSWORD=... perl index.pl migrate
   > systemctl start travelynx