* Started work on tools/anvil-manage-striker-peers to do the actual work on managing local and peer databases in anvil.conf.
Signed-off-by: Digimer <digimer@alteeve.ca>main
Digimer
6 years ago
parent
0fa3c42f2f
commit
89c06b9f24
5 changed files with 176 additions and 11 deletions
@ -0,0 +1,156 @@ |
|||||||
|
#!/usr/bin/perl |
||||||
|
# |
||||||
|
# This program adds, edits and removes striker peers (for replicating Anvil! database data). |
||||||
|
# |
||||||
|
# Exit codes; |
||||||
|
# 0 = Normal exit. |
||||||
|
# 1 = Program not run as root. |
||||||
|
# 2 = A switch is missing or invalid. |
||||||
|
# 3 = |
||||||
|
# |
||||||
|
|
||||||
|
use strict; |
||||||
|
use warnings; |
||||||
|
use Anvil::Tools; |
||||||
|
|
||||||
|
my $THIS_FILE = ($0 =~ /^.*\/(.*)$/)[0]; |
||||||
|
my $running_directory = ($0 =~ /^(.*?)\/$THIS_FILE$/)[0]; |
||||||
|
if (($running_directory =~ /^\./) && ($ENV{PWD})) |
||||||
|
{ |
||||||
|
$running_directory =~ s/^\./$ENV{PWD}/; |
||||||
|
} |
||||||
|
|
||||||
|
# Turn off buffering so that the pinwheel will display while waiting for the SSH call(s) to complete. |
||||||
|
$| = 1; |
||||||
|
|
||||||
|
my $anvil = Anvil::Tools->new({log_level => 2, log_secure => 1}); |
||||||
|
|
||||||
|
# Read switches |
||||||
|
$anvil->Get->switches; |
||||||
|
|
||||||
|
# Make sure we're running as 'root' |
||||||
|
# $< == real UID, $> == effective UID |
||||||
|
if (($< != 0) && ($> != 0)) |
||||||
|
{ |
||||||
|
# Not root |
||||||
|
print $anvil->Words->string({key => "error_0005"})."\n"; |
||||||
|
$anvil->nice_exit({code => 1}); |
||||||
|
} |
||||||
|
|
||||||
|
# Paths |
||||||
|
$anvil->Storage->read_config({file => $anvil->data->{path}{configs}{'anvil.conf'}}); |
||||||
|
|
||||||
|
# Read in the anvil.conf, we're going to need it in any case. |
||||||
|
$anvil->data->{body}{'anvil.conf'} = $anvil->Storage->read_file({file => $anvil->data->{path}{configs}{'anvil.conf'}}); |
||||||
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, secure => 1, level => 2, list => { anvil_conf_body => $anvil_conf_body }}); |
||||||
|
|
||||||
|
# Am I adding, editing or deleting? |
||||||
|
if ($anvil->data->{switches}{add}) |
||||||
|
{ |
||||||
|
# Add a new entry. |
||||||
|
add_entry($anvil); |
||||||
|
} |
||||||
|
elsif ($anvil->data->{switches}{remove}) |
||||||
|
{ |
||||||
|
} |
||||||
|
elsif ($anvil->data->{switches}{modify}) |
||||||
|
{ |
||||||
|
} |
||||||
|
else |
||||||
|
{ |
||||||
|
# Bad call. |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
$anvil->nice_exit({code => 0}); |
||||||
|
|
||||||
|
|
||||||
|
############################################################################################################# |
||||||
|
# Functions # |
||||||
|
############################################################################################################# |
||||||
|
|
||||||
|
sub add_entry |
||||||
|
{ |
||||||
|
my ($anvil) = @_; |
||||||
|
|
||||||
|
my $host_uuid = defined $anvil->data->{switches}{'host-uuid'} ? $anvil->data->{switches}{'host-uuid'} : ""; |
||||||
|
my $host = defined $anvil->data->{switches}{'host'} ? $anvil->data->{switches}{'host'} : ""; |
||||||
|
my $port = defined $anvil->data->{switches}{'port'} ? $anvil->data->{switches}{'port'} : 5432; |
||||||
|
my $password_file = defined $anvil->data->{switches}{'password-file'} ? $anvil->data->{switches}{'password-file'} : ""; |
||||||
|
my $ping = defined $anvil->data->{switches}{'ping'} ? $anvil->data->{switches}{'ping'} : 0; |
||||||
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, secure => 0, level => 2, list => { |
||||||
|
host_uuid => $host_uuid, |
||||||
|
host => $host, |
||||||
|
port => $port, |
||||||
|
password_file => $password_file, |
||||||
|
ping => $ping, |
||||||
|
}}); |
||||||
|
|
||||||
|
# Is anything missing? |
||||||
|
if ((not $host_uuid) or (not $anvil->Validate->is_uuid({uuid => $host_uuid}))) |
||||||
|
{ |
||||||
|
# Invalid UUID. |
||||||
|
print $anvil->Words->string({key => "error_0031", variables => { host_uuid => $host_uuid }})."\n"; |
||||||
|
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, level => 2, secure => 0, key => "error_0031", variables => { host_uuid => $host_uuid }}); |
||||||
|
$anvil->nice_exit({code => 2}); |
||||||
|
} |
||||||
|
if (not $host) |
||||||
|
{ |
||||||
|
# Invalid UUID. |
||||||
|
print $anvil->Words->string({key => "error_0032", variables => { switch => "host" }})."\n"; |
||||||
|
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, level => 2, secure => 0, key => "error_0032", variables => { host_uuid => $host_uuid }}); |
||||||
|
$anvil->nice_exit({code => 2}); |
||||||
|
} |
||||||
|
if (($port =~ /\D/) or ($port < 1) or ($port > 65535)) |
||||||
|
{ |
||||||
|
# Invalid port. |
||||||
|
print $anvil->Words->string({key => "error_0033", variables => { port => $port }})."\n"; |
||||||
|
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, level => 2, secure => 0, key => "error_0033", variables => { port => $port }}); |
||||||
|
$anvil->nice_exit({code => 2}); |
||||||
|
} |
||||||
|
|
||||||
|
# Pull the password out of the file. |
||||||
|
my $password = $anvil->Storage->read_file({file => $password_file}); |
||||||
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, secure => 1, level => 2, list => { password => $password }}); |
||||||
|
|
||||||
|
# If we don't find the entry, or if the entry exists but has changed, this will be set to '1' so |
||||||
|
# we'll rewrite the file. |
||||||
|
my $write = 0; |
||||||
|
|
||||||
|
# If the config already exists, we'll look at each of the values to see if any changed (or are not defaults). If so, we'll rewrite |
||||||
|
my $host_variable = "database::${host_uuid}::host"; |
||||||
|
my $host_different = 1; |
||||||
|
my $port_variable = "database::${host_uuid}::port"; |
||||||
|
my $port_different = 1; |
||||||
|
my $password_variable = "database::${host_uuid}::password"; |
||||||
|
my $password_different = 1; |
||||||
|
my $ping_variable = "database::${host_uuid}::ping"; |
||||||
|
my $ping_different = 1; |
||||||
|
|
||||||
|
# Loop through the existing file. |
||||||
|
my $new_body = ""; |
||||||
|
my $test_line = "database::${host_uuid}::"; |
||||||
|
foreach my $line (split/\n/, $anvil->data->{body}{'anvil.conf'}) |
||||||
|
{ |
||||||
|
my $secure = (($line =~ /password/) && ($line !~ /^#/)) ? 1 : 0; |
||||||
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, secure => $secure, level => 2, list => { line => $line }}); |
||||||
|
if ($line =~ /^$host_variable(\s*)=(\s*)(.*)$/) |
||||||
|
{ |
||||||
|
my $left_space = $1; |
||||||
|
my $right_space = $2; |
||||||
|
my $variable = $3; |
||||||
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => { |
||||||
|
config_seen => $config_seen, |
||||||
|
local_uuid => $local_uuid, |
||||||
|
}}); |
||||||
|
} |
||||||
|
if ($line eq "### end db list ###") |
||||||
|
{ |
||||||
|
$new_body .= $insert; |
||||||
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, secure => 1, level => 2, list => { new_body => $new_body }}); |
||||||
|
} |
||||||
|
$new_body .= $line."\n"; |
||||||
|
} |
||||||
|
|
||||||
|
return(0); |
||||||
|
} |
||||||
Loading…
Reference in new issue