* Created the new tools/striker-purge-host tool which purges all data about a host from the database.
* Updated Database->get_hosts() to store 'host_key' and 'host_uuid' data. * Created Database->get_ssh_keys(). * Fixed a couple bugs where Get->host_type() now returns 'striker' but tests checked for 'dashboard'. Signed-off-by: Digimer <digimer@alteeve.ca>main
Digimer
5 years ago
parent
0b2eb88f39
commit
7d907c15ff
5 changed files with 502 additions and 22 deletions
@ -0,0 +1,324 @@ |
|||||||
|
#!/usr/bin/perl |
||||||
|
# |
||||||
|
# This tool is meant to be run from the command line and lets a user purge all information about a given host |
||||||
|
# from the databases. |
||||||
|
# |
||||||
|
# Exit codes; |
||||||
|
# 0 = Normal exit. |
||||||
|
# 1 = User's answer to which machine to purge was invalid. |
||||||
|
# 2 = The passed-in (or selected) host UUID (no longer) exists. |
||||||
|
# 3 = User did not confirm to proceed. |
||||||
|
# 4 = Runaway loop detected. |
||||||
|
# |
||||||
|
|
||||||
|
use strict; |
||||||
|
use warnings; |
||||||
|
use Anvil::Tools; |
||||||
|
use Data::Dumper; |
||||||
|
use Curses::UI; |
||||||
|
|
||||||
|
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({debug => 2}); |
||||||
|
$anvil->Log->secure({set => 1}); |
||||||
|
$anvil->Log->level({set => 2}); |
||||||
|
|
||||||
|
$anvil->Database->connect({debug => 3}); |
||||||
|
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, level => 2, secure => 0, key => "log_0132"}); |
||||||
|
|
||||||
|
# First, ask the user to select a host. |
||||||
|
$anvil->data->{switches}{'host-uuid'} = "" if not defined $anvil->data->{switches}{'host-uuid'}; |
||||||
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => { 'switches::host-uuid' => $anvil->data->{switches}{'host-uuid'} }}); |
||||||
|
|
||||||
|
$anvil->Database->get_hosts(); |
||||||
|
|
||||||
|
# Ask the user to pick a host. |
||||||
|
pick_host($anvil); |
||||||
|
|
||||||
|
# Ask the user to confirm. |
||||||
|
confirm($anvil); |
||||||
|
|
||||||
|
# If we're alive, purge. |
||||||
|
purge($anvil); |
||||||
|
|
||||||
|
$anvil->nice_exit({code => 0}); |
||||||
|
|
||||||
|
############################################################################################################# |
||||||
|
# Functions # |
||||||
|
############################################################################################################# |
||||||
|
|
||||||
|
sub purge |
||||||
|
{ |
||||||
|
my ($anvil) = @_; |
||||||
|
|
||||||
|
# Read in all constraints and create a hash of what depends on what, then delete anything referencing |
||||||
|
# hosts without any further references. As each delete is done, delete it. Loop until all entries are |
||||||
|
# gone. |
||||||
|
|
||||||
|
# Load all the known tables in the DB. |
||||||
|
my $query = "SELECT schemaname||'.'||tablename AS table_name FROM pg_catalog.pg_tables WHERE schemaname = 'history' ORDER BY tablename ASC, schemaname ASC;"; |
||||||
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => { query => $query }}); |
||||||
|
|
||||||
|
my $results = $anvil->Database->query({query => $query, source => $THIS_FILE, line => __LINE__}); |
||||||
|
my $count = @{$results}; |
||||||
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 3, list => { |
||||||
|
results => $results, |
||||||
|
count => $count, |
||||||
|
}}); |
||||||
|
foreach my $row (@{$results}) |
||||||
|
{ |
||||||
|
my $table_name = $row->[0]; |
||||||
|
$anvil->data->{sql}{history_tables}{$table_name} = 1; |
||||||
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 3, list => { "sql::history_tables::${table_name}" => $anvil->data->{sql}{history_tables}{$table_name} }}); |
||||||
|
} |
||||||
|
undef $query; |
||||||
|
undef $results; |
||||||
|
undef $count; |
||||||
|
|
||||||
|
### NOTE: Credit for this query goes to: |
||||||
|
### - https://stackoverflow.com/questions/1152260/postgres-sql-to-list-table-foreign-keys |
||||||
|
$query = " |
||||||
|
SELECT |
||||||
|
tc.table_schema||'.'||tc.table_name AS table, |
||||||
|
kcu.column_name, |
||||||
|
ccu.table_schema||'.'||ccu.table_name AS foreign_table, |
||||||
|
ccu.column_name AS foreign_column_name |
||||||
|
FROM |
||||||
|
information_schema.table_constraints AS tc |
||||||
|
JOIN information_schema.key_column_usage AS kcu |
||||||
|
ON tc.constraint_name = kcu.constraint_name |
||||||
|
AND tc.table_schema = kcu.table_schema |
||||||
|
JOIN information_schema.constraint_column_usage AS ccu |
||||||
|
ON ccu.constraint_name = tc.constraint_name |
||||||
|
AND ccu.table_schema = tc.table_schema |
||||||
|
WHERE |
||||||
|
tc.constraint_type = 'FOREIGN KEY' |
||||||
|
AND |
||||||
|
ccu.column_name = 'host_uuid' |
||||||
|
;"; |
||||||
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => { query => $query }}); |
||||||
|
|
||||||
|
$results = $anvil->Database->query({query => $query, source => $THIS_FILE, line => __LINE__}); |
||||||
|
$count = @{$results}; |
||||||
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 3, list => { |
||||||
|
results => $results, |
||||||
|
count => $count, |
||||||
|
}}); |
||||||
|
foreach my $row (@{$results}) |
||||||
|
{ |
||||||
|
my $table_name = $row->[0]; |
||||||
|
my $column_name = $row->[1]; |
||||||
|
my $foreign_table_name = $row->[2]; |
||||||
|
my $foreign_column_name = $row->[3]; |
||||||
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 3, list => { |
||||||
|
's1:table_name' => $table_name, |
||||||
|
's2:column_name' => $column_name, |
||||||
|
's3:foreign_table_name' => $foreign_table_name, |
||||||
|
's4:foreign_column_name' => $foreign_column_name, |
||||||
|
}}); |
||||||
|
|
||||||
|
if (not exists $anvil->data->{sql}{table}{$table_name}) |
||||||
|
{ |
||||||
|
$anvil->data->{sql}{table}{$table_name}{column} = $column_name; |
||||||
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 3, list => { "sql::table::${table_name}::column" => $anvil->data->{sql}{table}{$table_name}{column} }}); |
||||||
|
} |
||||||
|
if (not exists $anvil->data->{sql}{table}{$foreign_table_name}) |
||||||
|
{ |
||||||
|
$anvil->data->{sql}{table}{$foreign_table_name}{column} = $foreign_column_name; |
||||||
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 3, list => { "sql::table::${foreign_table_name}::column" => $anvil->data->{sql}{table}{$foreign_table_name}{column} }}); |
||||||
|
} |
||||||
|
$anvil->data->{sql}{table}{$foreign_table_name}{referenced_by}{$table_name} = 1; |
||||||
|
} |
||||||
|
|
||||||
|
my $loops = 20; |
||||||
|
my $loop = 0; |
||||||
|
my $done = 0; |
||||||
|
my $queries = []; |
||||||
|
until($done) |
||||||
|
{ |
||||||
|
$loop++; |
||||||
|
foreach my $table_name (sort {$a cmp $b} keys %{$anvil->data->{sql}{table}}) |
||||||
|
{ |
||||||
|
my $count = exists $anvil->data->{sql}{table}{$table_name}{referenced_by} ? keys %{$anvil->data->{sql}{table}{$table_name}{referenced_by}} : 0; |
||||||
|
#print "-- Table: [".$table_name."] is referenced by: [".$count."] foreign table(s);\n"; |
||||||
|
if (not $count) |
||||||
|
{ |
||||||
|
# If there is a corresponding history table, delete it as well. |
||||||
|
my $column = $anvil->data->{sql}{table}{$table_name}{column}; |
||||||
|
my $history_table = $table_name; |
||||||
|
$history_table =~ s/^public\./history\./; |
||||||
|
if (exists $anvil->data->{sql}{history_tables}{$history_table}) |
||||||
|
{ |
||||||
|
push @{$queries}, "DELETE FROM ".$history_table." WHERE ".$column." = ".$anvil->Database->quote($anvil->data->{switches}{'host-uuid'}).";"; |
||||||
|
} |
||||||
|
|
||||||
|
# Now delete the public table entries. |
||||||
|
push @{$queries}, "DELETE FROM ".$table_name." WHERE ".$column." = ".$anvil->Database->quote($anvil->data->{switches}{'host-uuid'}).";"; |
||||||
|
|
||||||
|
foreach my $foreign_table_name (sort {$a cmp $b} keys %{$anvil->data->{sql}{table}}) |
||||||
|
{ |
||||||
|
if (exists $anvil->data->{sql}{table}{$foreign_table_name}{referenced_by}{$table_name}) |
||||||
|
{ |
||||||
|
#print "-- Deleting: [".$foreign_table_name."] reference to: [".$table_name."]\n"; |
||||||
|
delete $anvil->data->{sql}{table}{$foreign_table_name}{referenced_by}{$table_name}; |
||||||
|
} |
||||||
|
} |
||||||
|
#print "-- Deleting hash entry for: [".$table_name."]\n"; |
||||||
|
delete $anvil->data->{sql}{table}{$table_name}; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
my $remaining = keys %{$anvil->data->{sql}{table}}; |
||||||
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => { remaining => $remaining }}); |
||||||
|
if (not $remaining) |
||||||
|
{ |
||||||
|
$done = 1; |
||||||
|
} |
||||||
|
elsif ($loop > $loops) |
||||||
|
{ |
||||||
|
print "Run-away loop detected, aborting!\n"; |
||||||
|
$anvil->nice_exit({code => 4}); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
# Do the deed. |
||||||
|
$anvil->Database->write({debug => 2, query => $queries, source => $THIS_FILE, line => __LINE__}); |
||||||
|
my $host_uuid = $anvil->data->{switches}{'host-uuid'}; |
||||||
|
print $anvil->Words->string({key => "message_0176", variables => { host_name => $anvil->data->{hosts}{host_uuid}{$host_uuid}{host_name} }})."\n"; |
||||||
|
|
||||||
|
return(0); |
||||||
|
} |
||||||
|
|
||||||
|
sub confirm |
||||||
|
{ |
||||||
|
my ($anvil) = @_; |
||||||
|
|
||||||
|
# Normalize -y |
||||||
|
$anvil->data->{switches}{'y'} = "" if not defined $anvil->data->{switches}{'y'}; |
||||||
|
$anvil->data->{switches}{'yes'} = "" if not defined $anvil->data->{switches}{'yes'}; |
||||||
|
$anvil->data->{switches}{'Y'} = "" if not defined $anvil->data->{switches}{'Y'}; |
||||||
|
$anvil->data->{switches}{'Yes'} = "" if not defined $anvil->data->{switches}{'Yes'}; |
||||||
|
$anvil->data->{switches}{'YES'} = "" if not defined $anvil->data->{switches}{'YES'}; |
||||||
|
if (not $anvil->data->{switches}{'y'}) |
||||||
|
{ |
||||||
|
if ($anvil->data->{switches}{'yes'}) { $anvil->data->{switches}{'y'} = $anvil->data->{switches}{'yes'}; } |
||||||
|
elsif ($anvil->data->{switches}{'Y'}) { $anvil->data->{switches}{'y'} = $anvil->data->{switches}{'Y'}; } |
||||||
|
elsif ($anvil->data->{switches}{'Yes'}) { $anvil->data->{switches}{'y'} = $anvil->data->{switches}{'Yes'}; } |
||||||
|
elsif ($anvil->data->{switches}{'YES'}) { $anvil->data->{switches}{'y'} = $anvil->data->{switches}{'YES'}; } |
||||||
|
} |
||||||
|
|
||||||
|
# Ask to confirm, is sane and not -y |
||||||
|
my $host_uuid = $anvil->data->{switches}{'host-uuid'}; |
||||||
|
if (not exists $anvil->data->{hosts}{host_uuid}{$host_uuid}{host_name}) |
||||||
|
{ |
||||||
|
print $anvil->Words->string({key => "error_0131", variables => { host_uuid => $host_uuid }})."\n\n"; |
||||||
|
$anvil->nice_exit({code => 2}); |
||||||
|
} |
||||||
|
else |
||||||
|
{ |
||||||
|
print $anvil->Words->string({key => "message_0172"})."\n\n"; |
||||||
|
print $anvil->Words->string({key => "message_0173", variables => { |
||||||
|
host_name => $anvil->data->{hosts}{host_uuid}{$host_uuid}{host_name}, |
||||||
|
host_uuid => $host_uuid, |
||||||
|
}})."\n"; |
||||||
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 3, list => { 'switches::y' => $anvil->data->{switches}{'y'} }}); |
||||||
|
if ($anvil->data->{switches}{'y'} eq "#!SET!#") |
||||||
|
{ |
||||||
|
# Already confirmed. |
||||||
|
print $anvil->Words->string({key => "message_0023"})."\n\n"; |
||||||
|
return(0); |
||||||
|
} |
||||||
|
print $anvil->Words->string({key => "message_0021"})." "; |
||||||
|
chomp(my $answer = <STDIN>); |
||||||
|
|
||||||
|
if ($answer =~ /^y/i) |
||||||
|
{ |
||||||
|
# Proceed. |
||||||
|
print $anvil->Words->string({key => "message_0175"})."\n"; |
||||||
|
} |
||||||
|
else |
||||||
|
{ |
||||||
|
# Abort. |
||||||
|
print $anvil->Words->string({key => "message_0022"})."\n"; |
||||||
|
$anvil->nice_exit({code => 3}); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
return(0); |
||||||
|
} |
||||||
|
|
||||||
|
sub pick_host |
||||||
|
{ |
||||||
|
my ($anvil) = @_; |
||||||
|
|
||||||
|
return(0) if $anvil->data->{switches}{'host-uuid'}; |
||||||
|
|
||||||
|
# Get a list of hosts |
||||||
|
my $i = 1; |
||||||
|
my $select = {}; |
||||||
|
my $host_length = 0; |
||||||
|
my $type_length = 0; |
||||||
|
foreach my $host_name (sort {$a cmp $b} keys %{$anvil->data->{sys}{hosts}{by_name}}) |
||||||
|
{ |
||||||
|
my $host_uuid = $anvil->data->{sys}{hosts}{by_name}{$host_name}; |
||||||
|
my $host_type = $anvil->data->{hosts}{host_uuid}{$host_uuid}{host_type}; |
||||||
|
my $anvil_name = defined $anvil->data->{hosts}{host_uuid}{$host_uuid}{anvil_name} ? $anvil->data->{hosts}{host_uuid}{$host_uuid}{anvil_name} : ""; |
||||||
|
my $say_type = $anvil->Words->string({key => "brand_0009"}); |
||||||
|
if ($host_type eq "striker") { $say_type = $anvil->Words->string({key => "brand_0003"}); } |
||||||
|
elsif ($host_type eq "node") { $say_type = $anvil->Words->string({key => "brand_0007"}); } |
||||||
|
elsif ($host_type eq "dr") { $say_type = $anvil->Words->string({key => "brand_0008"}); } |
||||||
|
|
||||||
|
if (length($host_name) > $host_length) { $host_length = length($host_name); } |
||||||
|
if (length($say_type) > $type_length) { $type_length = length($say_type); } |
||||||
|
|
||||||
|
# This is used to build the menu. |
||||||
|
$select->{$i}{host_uuid} = $host_uuid; |
||||||
|
$select->{$i}{host_name} = $host_name; |
||||||
|
$select->{$i}{host_type} = $host_type; |
||||||
|
$select->{$i}{say_type} = $say_type; |
||||||
|
$select->{$i}{anvil_name} = $anvil_name; |
||||||
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 3, list => { |
||||||
|
"${i}::host_uuid" => $select->{$i}{host_uuid}, |
||||||
|
"${i}::host_name" => $select->{$i}{host_name}, |
||||||
|
"${i}::host_type" => $select->{$i}{host_type}, |
||||||
|
"${i}::anvil_name" => $select->{$i}{anvil_name}, |
||||||
|
}}); |
||||||
|
$i++; |
||||||
|
} |
||||||
|
|
||||||
|
my $pad = $i > 9 ? 2 : 1; |
||||||
|
print $anvil->Words->string({key => "message_0169"})."\n\n"; |
||||||
|
foreach my $i (sort {$a cmp $b} keys %{$select}) |
||||||
|
{ |
||||||
|
print " ".$anvil->Words->string({key => "message_0170", variables => { |
||||||
|
key => sprintf("%".$pad."s", $i), |
||||||
|
type => sprintf("%-".$type_length."s", $select->{$i}{say_type}), |
||||||
|
host_name => sprintf("%-".$host_length."s", $select->{$i}{host_name}), |
||||||
|
host_uuid => $select->{$i}{host_uuid}, |
||||||
|
}})."\n"; |
||||||
|
} |
||||||
|
print "\n".$anvil->Words->string({key => "message_0171"})." "; |
||||||
|
|
||||||
|
chomp(my $answer = <STDIN>); |
||||||
|
|
||||||
|
if (not exists $select->{$answer}{host_name}) |
||||||
|
{ |
||||||
|
print "\n".$anvil->Words->string({key => "error_0130", variables => { answer => $answer }})."\n\n"; |
||||||
|
$anvil->nice_exit({code => 1}); |
||||||
|
} |
||||||
|
else |
||||||
|
{ |
||||||
|
print "\n"; |
||||||
|
$anvil->data->{switches}{'host-uuid'} = $select->{$answer}{host_uuid}; |
||||||
|
} |
||||||
|
|
||||||
|
return(0); |
||||||
|
} |
||||||
Loading…
Reference in new issue