anvil/tools/anvil-update-definition

#!/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);
}