#!/usr/local/cpanel/3rdparty/bin/perl
# cpanel - scripts/installpostgres Copyright 2022 cPanel, L.L.C.
# All rights reserved.
# copyright@cpanel.net http://cpanel.net
# This code is subject to the cPanel license. Unauthorized copying is prohibited
package scripts::installpostgres;
use strict;
use warnings;
use Cpanel::Debug ();
use Cpanel::OS ();
use Cpanel::SafeRun::Simple ();
use Cpanel::SysPkgs ();
use Cpanel::Chkservd::Manage ();
use Cpanel::PostgresUtils ();
use Cpanel::PostgresUtils::PgPass ();
use Cpanel::ServerTasks ();
use Whostmgr::Postgres ();
exit( run(@ARGV) // 0 ) unless caller();
sub help {
my $msg = <<EOF;
Usage: installpostgres [--help] [--yes]
Installs the supported version of PostgreSQL.
Options:
--help This message.
--yes Bypass confirmation prompt.
EOF
print $msg;
return 0;
}
sub prompt {
my ( $message, @args ) = @_;
chomp $message;
print $message;
if ( !@args || $args[0] !~ /\-\-yes/ ) {
my $res = <STDIN>;
if ( $res !~ /^y/i ) {
print "Exiting...\n";
exit();
}
}
return;
}
# replace system calls
sub _run_and_log_cmd { ## no critic(RequireArgUnpacking) # doesn't make sense in this case
Cpanel::Debug::log_debug( '[EXEC] ' . join( " ", @_ ) );
my $out = Cpanel::SafeRun::Simple::saferunallerrors(@_);
chomp($out);
Cpanel::Debug::log_info($out);
return $out;
}
# Get more info with CPANEL_DEBUG_LEVEL set
sub _debug_run_and_log_cmd { ## no critic(RequireArgUnpacking) # doesn't make sense in this case
Cpanel::Debug::log_debug( '[EXEC] ' . join( " ", @_ ) ) if $Cpanel::Debug::level;
my $out = Cpanel::SafeRun::Simple::saferunallerrors(@_);
chomp($out);
Cpanel::Debug::log_debug($out) if $Cpanel::Debug::level;
return $out;
}
sub run {
my @args = @_;
return help() if grep { $_ eq '--help' } @args;
die "cPanel does not support PostgreSQL on this system at this time!" unless Cpanel::OS::supports_postgresql();
my $restart_instruction = Cpanel::OS::is_systemd() ? "\tsystemctl restart postgresql" : "\t/sbin/service postgresql stop\n" . "\t/sbin/service postgresql start\n";
my $minversion = Cpanel::OS::postgresql_minimum_supported_version() . '.x';
if ( my $old_pg_home = Cpanel::PostgresUtils::find_pgsql_home() ) {
my $notice = <<"EOS";
This script installs PostgreSQL $minversion or later.
If your system runs an older version of PostgreSQL, you must
dump your databases to a file, and then restore them after
installation.
PostgreSQL $minversion is NOT backwards compatible.
If no PostgreSQL databases exist on your system, run the
following command to force the creation of a PostgreSQL $minversion -style
setup:
\tmv $old_pg_home $old_pg_home.old
$restart_instruction
Do not run this command if databases exist that you wish to keep!
Are you certain that you wish to proceed? [y/(n)]:
EOS
prompt( $notice, @args );
}
else {
my $notice = <<"EOS";
This script installs PostgreSQL $minversion or later.
Are you certain that you wish to proceed? [y/(n)]:
EOS
prompt( $notice, @args );
}
Cpanel::Debug::log_info("Installing Postgres...");
my $syspkgobj = Cpanel::SysPkgs->new();
if ( !$syspkgobj ) { die Cpanel::Debug::log_error("The system could not create the SysPkgs object."); }
$syspkgobj->install( 'pkglist' => Cpanel::OS::postgresql_packages() );
my $pg_user = Cpanel::PostgresUtils::PgPass::getpostgresuser();
my $pg_home = Cpanel::PostgresUtils::find_pgsql_home( 'user' => $pg_user );
my $pg_data = Cpanel::PostgresUtils::find_pgsql_data( 'home' => $pg_home );
my @cmd;
foreach my $alias ( @{ Cpanel::OS::postgresql_service_aliases() } ) {
my $service = 'postgresql';
if ( !-e "/etc/init.d/$service" && -e "/etc/init.d/$alias" ) {
symlink( $alias, "/etc/init.d/$service" );
}
}
if ( !-e "$pg_data/global" ) {
foreach my $cmd ( @{ Cpanel::OS::postgresql_initdb_commands() } ) {
@cmd = split ' ', $cmd;
_debug_run_and_log_cmd(@cmd);
}
}
@cmd = qw{/usr/local/cpanel/scripts/setpostgresconfig --force};
_debug_run_and_log_cmd(@cmd);
@cmd = qw{/usr/local/cpanel/scripts/cpservice postgresql enable 35};
_debug_run_and_log_cmd(@cmd);
my ( $status, $message );
Cpanel::Debug::log_info("Updating postgresql configuration now...");
( $status, $message ) = Whostmgr::Postgres::update_config();
print $message. "\n" if $message;
Cpanel::Debug::log_info("Attempting to restart postgresql...");
@cmd = qw{/usr/local/cpanel/scripts/restartsrv postgresql};
_debug_run_and_log_cmd(@cmd);
my $install_ok = $? == 0 ? 1 : 0;
if ($install_ok) {
Cpanel::Debug::log_info("Creating dbowners for existing cpusers...");
( $status, $message ) = Whostmgr::Postgres::create_dbowners_for_cpusers();
print $message. "\n";
$install_ok &&= $status;
if ($install_ok) {
print q{
The PostgreSQL packages successfully installed. To configure PostgreSQL,
set your password, and enable PostgreSQL for user accounts, navigate to
WHM's Configure PostgreSQL interface (Home >> SQL Services >> Configure PostgreSQL).
};
print "Setting a random password for the PostgreSQL database user.\n";
_debug_run_and_log_cmd( '/usr/local/cpanel/bin/postgrescheck', '--check-auth', '--reset-pass-on-fail' );
_debug_run_and_log_cmd('/usr/local/cpanel/bin/cpsessetup');
# If PostgreSQL has never been listed under chkservd, enable monitoring for it.
# (This might happen if the service was uninstalled manually but reinstalled with this script.)
# Doing this prevents /scripts/check_unmonitored_enabled_services from complaining to the admin
# who is installing PostgreSQL for the first time on this system.
my %MONITORED = Cpanel::Chkservd::Manage::getmonitored();
Cpanel::Chkservd::Manage::enable('postgresql') unless exists $MONITORED{'postgresql'};
}
}
# install_ok value can change in the previous block
if ( !$install_ok ) {
print STDERR qq{The PostgreSQL installation failed!\n};
}
# reset haspostgres if great success
Cpanel::ServerTasks::schedule_task( ['CpDBTasks'], 5, 'build_global_cache' ) if $install_ok;
return $install_ok ? 0 : 1;
}
1;
Mini Shell