|
|
|
#!/usr/bin/perl
|
|
|
|
#
|
|
|
|
#
|
|
|
|
#
|
|
|
|
|
|
|
|
use strict;
|
|
|
|
use warnings;
|
|
|
|
use Anvil::Tools;
|
|
|
|
|
|
|
|
$| = 1;
|
|
|
|
|
|
|
|
my $THIS_FILE = ($0 =~ /^.*\/(.*)$/)[0];
|
|
|
|
my $running_directory = ($0 =~ /^(.*?)\/$THIS_FILE$/)[0];
|
|
|
|
if (($running_directory =~ /^\./) && ($ENV{PWD}))
|
|
|
|
{
|
|
|
|
$running_directory =~ s/^\./$ENV{PWD}/;
|
|
|
|
}
|
|
|
|
|
|
|
|
my $anvil = Anvil::Tools->new();
|
|
|
|
|
|
|
|
sub system_call
|
|
|
|
{
|
|
|
|
my $parameters = shift;
|
|
|
|
my $shell_call = $parameters->{shell_call};
|
|
|
|
|
|
|
|
my ($shell_output, $shell_return_code) = $anvil->System->call({ shell_call => $shell_call });
|
|
|
|
$anvil->Log->variables({ source => $THIS_FILE, line => __LINE__, level => 2, list => {
|
|
|
|
shell_call => $shell_call,
|
|
|
|
shell_output => $shell_output,
|
|
|
|
shell_return_code => $shell_return_code
|
|
|
|
} });
|
|
|
|
|
|
|
|
return ($shell_output, $shell_return_code);
|
|
|
|
}
|
|
|
|
|
|
|
|
sub is_existing_server_screenshot_outdated
|
|
|
|
{
|
|
|
|
my $parameters = shift;
|
|
|
|
my $server_uuid = $parameters->{server_uuid};
|
|
|
|
|
|
|
|
my ($encoded_image, $variable_uuid, $variable_mtime) = $anvil->Database->read_variable({ variable_name => "server_screenshot::".$server_uuid });
|
|
|
|
|
|
|
|
my $time_difference = time - $variable_mtime;
|
|
|
|
|
|
|
|
return $time_difference > 120 ? 1 : 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
sub get_server_screenshot
|
|
|
|
{
|
|
|
|
my $parameters = shift;
|
|
|
|
my $server_uuid = $parameters->{server_uuid};
|
|
|
|
my ($resize_x, $resize_y) = split(/x/ , $parameters->{resize_args});
|
|
|
|
|
|
|
|
my $shell_call = "virsh --quiet screenshot --domain ".$server_uuid." --file /dev/stdout";
|
|
|
|
|
|
|
|
if ($resize_x =~ /^\d+$/ && $resize_y =~ /^\d+$/)
|
|
|
|
{
|
|
|
|
$shell_call .= " | pamscale -quiet -xyfit ".$resize_x." ".$resize_y;
|
|
|
|
$shell_call .= " | pamtopng -quiet";
|
|
|
|
}
|
|
|
|
|
|
|
|
$shell_call .= " | base64 --wrap 0";
|
|
|
|
|
|
|
|
my ($shell_output, $shell_return_code) = system_call({ shell_call => $shell_call });
|
|
|
|
|
|
|
|
return $shell_return_code == 0 ? $shell_output : undef;
|
|
|
|
}
|
|
|
|
|
|
|
|
sub insert_server_screenshot
|
|
|
|
{
|
|
|
|
my $parameters = shift;
|
|
|
|
my $server_uuid = $parameters->{server_uuid};
|
|
|
|
my $encoded_image = $parameters->{encoded_image};
|
|
|
|
|
|
|
|
$anvil->Database->insert_or_update_states({
|
|
|
|
state_name => "server_screenshot::".$server_uuid,
|
|
|
|
state_note => $encoded_image
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
$anvil->Get->switches;
|
|
|
|
|
|
|
|
$anvil->Database->connect;
|
|
|
|
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, level => 2, secure => 0, key => "log_0132"});
|
|
|
|
if (not $anvil->data->{sys}{database}{connections})
|
|
|
|
{
|
|
|
|
# No databases, exit.
|
|
|
|
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, level => 0, 'print' => 1, priority => "err", key => "error_0003"});
|
|
|
|
$anvil->nice_exit({exit_code => 1});
|
|
|
|
}
|
|
|
|
|
|
|
|
# Try to get a job UUID if not given.
|
|
|
|
if (not $anvil->data->{switches}{'job-uuid'})
|
|
|
|
{
|
|
|
|
$anvil->data->{switches}{'job-uuid'} = $anvil->Job->get_job_uuid({ program => $THIS_FILE });
|
|
|
|
$anvil->Log->variables({ source => $THIS_FILE, line => __LINE__, level => 2, list => {
|
|
|
|
"switches::job-uuid" => $anvil->data->{switches}{'job-uuid'}
|
|
|
|
} });
|
|
|
|
}
|
|
|
|
|
|
|
|
# Handle this script as a job when job UUID is provided.
|
|
|
|
if ($anvil->data->{switches}{'job-uuid'})
|
|
|
|
{
|
|
|
|
$anvil->Job->clear();
|
|
|
|
$anvil->Job->get_job_details();
|
|
|
|
$anvil->Job->update_progress({
|
|
|
|
progress => 1,
|
|
|
|
job_picked_up_by => $$,
|
|
|
|
job_picked_up_at => time,
|
|
|
|
message => "message_0263"
|
|
|
|
});
|
|
|
|
|
|
|
|
foreach my $line (split/\n/, $anvil->data->{jobs}{job_data})
|
|
|
|
{
|
|
|
|
if ($line =~ /server-uuid=(.*?)$/)
|
|
|
|
{
|
|
|
|
$anvil->data->{switches}{'server-uuid'} = $1;
|
|
|
|
}
|
|
|
|
|
|
|
|
if ($line =~ /resize=(.*?)$/)
|
|
|
|
{
|
|
|
|
$anvil->data->{switches}{'resize'} = $1;
|
|
|
|
}
|
|
|
|
|
|
|
|
if ($line =~ /stdout=(.*?)$/)
|
|
|
|
{
|
|
|
|
$anvil->data->{switches}{'stdout'} = $1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
my $server_uuid = $anvil->data->{switches}{'server-uuid'};
|
|
|
|
my $resize_args = $anvil->data->{switches}{'resize'};
|
|
|
|
my $is_stdout = $anvil->data->{switches}{'stdout'};
|
|
|
|
my $job_uuid = $anvil->data->{switches}{'job-uuid'};
|
|
|
|
|
|
|
|
$anvil->Log->variables({ source => $THIS_FILE, line => __LINE__, level => 2, list => {
|
|
|
|
server_uuid => $server_uuid,
|
|
|
|
resize_args => $resize_args,
|
|
|
|
is_stdout => $is_stdout,
|
|
|
|
job_uuid => $job_uuid
|
|
|
|
} });
|
|
|
|
|
|
|
|
if ($server_uuid)
|
|
|
|
{
|
|
|
|
my $encoded_image;
|
|
|
|
|
|
|
|
if ($is_stdout)
|
|
|
|
{
|
|
|
|
$encoded_image = get_server_screenshot({ server_uuid => $server_uuid, resize_args => $resize_args });
|
|
|
|
|
|
|
|
if (defined $encoded_image)
|
|
|
|
{
|
|
|
|
print($encoded_image);
|
|
|
|
|
|
|
|
$anvil->Job->update_progress({ progress => 100, message => "message_0264" });
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
$anvil->Job->update_progress({ progress => 100, message => "message_0265" });
|
|
|
|
}
|
|
|
|
}
|
|
|
|
elsif (is_existing_server_screenshot_outdated({ server_uuid => $server_uuid }))
|
|
|
|
{
|
|
|
|
$encoded_image = get_server_screenshot({ server_uuid => $server_uuid, resize_args => $resize_args });
|
|
|
|
|
|
|
|
if (defined $encoded_image)
|
|
|
|
{
|
|
|
|
insert_server_screenshot({ server_uuid => $server_uuid, encoded_image => $encoded_image });
|
|
|
|
|
|
|
|
$anvil->Job->update_progress({ progress => 100, message => "message_0264" });
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
$anvil->Job->update_progress({ progress => 100, message => "message_0265" });
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
$anvil->Job->update_progress({ progress => 100, message => "message_0266" });
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
$anvil->Job->update_progress({ progress => 100, message => "message_0266" });
|
|
|
|
}
|