Digimer
3 years ago
committed by
GitHub
commit
789ac6e8ad
23 changed files with 403 additions and 144 deletions
@ -0,0 +1,179 @@ |
||||
#!/usr/bin/perl |
||||
# |
||||
# This program will manage servers; Changing RAM, CPU cores, Growing virtual disks, adding virtual disks, |
||||
# inserting and ejecting ISO images into virtual optical media. |
||||
# |
||||
# Exit codes; |
||||
# 0 = Normal exit. |
||||
# 1 = No database connection. |
||||
# |
||||
# TODO: |
||||
# |
||||
|
||||
use strict; |
||||
use warnings; |
||||
use Anvil::Tools; |
||||
require POSIX; |
||||
use Term::Cap; |
||||
|
||||
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(); |
||||
|
||||
$anvil->data->{switches}{file} = ""; |
||||
$anvil->Get->switches; |
||||
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, 'print' => 1, level => 1, secure => 0, key => "log_0115", variables => { program => $THIS_FILE }}); |
||||
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => { |
||||
'switches::file' => $anvil->data->{switches}{file}, |
||||
}}); |
||||
|
||||
# Connect to the database(s). If we have no connections, we'll proceed anyway as one of the 'run_once' tasks |
||||
# is to setup the database server. |
||||
$anvil->Database->connect(); |
||||
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, level => 3, secure => 0, key => "log_0132"}); |
||||
if (not $anvil->data->{sys}{database}{connections}) |
||||
{ |
||||
# No databases, update the job, sleep for a bit and then exit. The daemon will pick it up and try |
||||
# again after we exit. |
||||
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, 'print' => 1, level => 0, priority => "err", key => "error_0305"}); |
||||
$anvil->nice_exit({exit_code => 1}); |
||||
} |
||||
|
||||
# Does the file exist? |
||||
if (not $anvil->data->{switches}{file}) |
||||
{ |
||||
# Not defined. |
||||
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, 'print' => 1, level => 0, priority => "err", key => "error_0323"}); |
||||
$anvil->nice_exit({exit_code => 1}); |
||||
} |
||||
elsif ((not -e $anvil->data->{switches}{file}) or (not -r $anvil->data->{switches}{file})) |
||||
{ |
||||
# Not found or can't be read. |
||||
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, 'print' => 1, level => 0, priority => "err", key => "error_0324", variables => { |
||||
file => $anvil->data->{switches}{file}, |
||||
}}); |
||||
$anvil->nice_exit({exit_code => 1}); |
||||
} |
||||
|
||||
validate_and_update($anvil); |
||||
|
||||
$anvil->nice_exit({exit_code => 0}); |
||||
|
||||
|
||||
############################################################################################################# |
||||
# Functions # |
||||
############################################################################################################# |
||||
|
||||
sub validate_and_update |
||||
{ |
||||
my ($anvil) = @_; |
||||
|
||||
# Can we parse the definition? |
||||
my $new_definition = $anvil->Storage->read_file({file => $anvil->data->{switches}{file}}); |
||||
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => { new_definition => $new_definition }}); |
||||
|
||||
# Pull the server name and uuid out. |
||||
my $server_name = ""; |
||||
my $server_uuid = ""; |
||||
foreach my $line (split/\n/, $new_definition) |
||||
{ |
||||
if ($line =~ /<name>(.*?)<\/name>/) |
||||
{ |
||||
$server_name = $1; |
||||
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => { server_name => $server_name }}); |
||||
} |
||||
if ($line =~ /<uuid>(.*?)<\/uuid>/) |
||||
{ |
||||
$server_uuid = $1; |
||||
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => { server_uuid => $server_uuid }}); |
||||
} |
||||
} |
||||
|
||||
# If I didn't find the server name or UUID, there's a problem. |
||||
if ($server_name eq "") |
||||
{ |
||||
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, 'print' => 1, level => 0, priority => "err", key => "error_0325"}); |
||||
$anvil->nice_exit({exit_code => 1}); |
||||
} |
||||
if (($server_uuid eq "") or (not $anvil->Validate->uuid({uuid => $server_uuid}))) |
||||
{ |
||||
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, 'print' => 1, level => 0, priority => "err", key => "error_0326"}); |
||||
$anvil->nice_exit({exit_code => 1}); |
||||
} |
||||
|
||||
local $@; |
||||
my $dom = eval { XML::LibXML->load_xml(string => $new_definition); }; |
||||
if ($@) |
||||
{ |
||||
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, 'print' => 1, level => 0, priority => "err", key => "error_0327", variables => { error => $@ }}); |
||||
$anvil->nice_exit({exit_code => 1}); |
||||
} |
||||
|
||||
# Find out how much memory being allocated. |
||||
my $memory = 0; |
||||
foreach my $memory ($dom->findnodes('/domain/memory')) |
||||
{ |
||||
my $units = $memory->{unit}; |
||||
my $value = $memory->to_literal(); |
||||
$memory = $anvil->Convert->human_readable_to_bytes({size => $value, type => $units}); |
||||
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => { |
||||
units => $units, |
||||
value => $value, |
||||
memory => $memory." (".$anvil->Convert->bytes_to_human_readable({'bytes' => $memory}).")", |
||||
}}); |
||||
} |
||||
if (not $memory) |
||||
{ |
||||
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, 'print' => 1, level => 0, priority => "err", key => "error_0328"}); |
||||
$anvil->nice_exit({exit_code => 1}); |
||||
} |
||||
|
||||
# Do we know about this server yet? |
||||
$anvil->Database->get_servers(); |
||||
if (not exists $anvil->data->{servers}{server_uuid}{$server_uuid}) |
||||
{ |
||||
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, 'print' => 1, level => 0, priority => "err", key => "error_0329", variables => { server_uuid => $server_uuid }}); |
||||
$anvil->nice_exit({exit_code => 1}); |
||||
} |
||||
|
||||
# We can't update the server name with this tool. |
||||
if ($server_name ne $anvil->data->{servers}{server_uuid}{$server_uuid}{server_name}) |
||||
{ |
||||
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, 'print' => 1, level => 0, priority => "err", key => "error_0330", variables => { |
||||
current_name => $current_name. |
||||
new_name => $anvil->data->{servers}{server_uuid}{$server_uuid}{server_name}, |
||||
}}); |
||||
$anvil->nice_exit({exit_code => 1}); |
||||
} |
||||
|
||||
# Still here? Update! Has the assigned RAM changed? |
||||
my $configured_ram = $anvil->data->{servers}{server_uuid}{$server_uuid}{server_configured_ram}; |
||||
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => { |
||||
configured_ram => $configured_ram." (".$anvil->Convert->bytes_to_human_readable({'bytes' => $configured_ram}).")", |
||||
}}); |
||||
if ($memory != $configured_ram) |
||||
{ |
||||
# Update the configured RAM value. |
||||
my $query = " |
||||
UPDATE |
||||
servers |
||||
SET |
||||
server_configured_ram = ".$anvil->Database->quote($memory).", |
||||
modified_date = ".$anvil->Database->quote($anvil->Database->refresh_timestamp)." |
||||
WHERE |
||||
server_uuid = ".$anvil->Database->quote($server_uuid).", |
||||
;"; |
||||
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => { query => $query }}); |
||||
$anvil->Database->write({query => $query, source => $THIS_FILE, line => __LINE__}); |
||||
} |
||||
|
||||
return(0); |
||||
} |
||||
Loading…
Reference in new issue