Merge pull request #99 from Tsu-ba-me/issues/1-set-power

Web UI: make `set_power` endpoint functional

cgi-bin/set_power

@@ -20,6 +20,100 @@ if (($running_directory =~ /^\./) && ($ENV{PWD}))
 
 my $anvil = Anvil::Tools->new();
 
+sub handle_invalid_uuid
+{
+	my $parameters = shift;
+	my $name       = $parameters->{name};
+	my $uuid       = $parameters->{uuid};
+
+	$anvil->Log->entry({
+		source    => $THIS_FILE,
+		line      => __LINE__,
+		level     => 0,
+		'print'   => 1,
+		priority  => "err",
+		key       => "error_0160",
+		variables => { name => $name, uuid => $uuid }
+	});
+	$anvil->nice_exit({ exit_code => 1 });
+}
+
+sub set_anvil_power
+{
+	# Expects the first element of @_ (argument array) to be a hash.
+	my $parameters = shift;
+	my $anvil_uuid = $parameters->{anvil_uuid};
+	my $on         = $parameters->{on};
+
+	# Get all mandatory hosts inside the anvil identified by the given UUID.
+	my @host_uuids = (
+		$anvil->data->{anvils}{anvil_uuid}{$anvil_uuid}{anvil_node1_host_uuid},
+		$anvil->data->{anvils}{anvil_uuid}{$anvil_uuid}{anvil_node2_host_uuid}
+	);
+
+	# Check for DR host outside of the loop to avoid duplicating checks.
+	if (length($anvil->data->{anvils}{anvil_uuid}{$anvil_uuid}{anvil_dr1_host_uuid}))
+	{
+		push(@host_uuids, $anvil->data->{anvils}{anvil_uuid}{$anvil_uuid}{anvil_dr1_host_uuid});
+	}
+
+	foreach (@host_uuids)
+	{
+		set_host_power({ host_uuid => $_, on => $on, stop_servers => 1 });
+	}
+}
+
+sub set_host_power
+{
+	# Expects the first element of @_ (argument array) to be a hash.
+	my $parameters   = shift;
+	my $host_uuid    = $parameters->{host_uuid};
+	my $on           = $parameters->{on};
+	my $stop_servers = exists $parameters->{stop_servers} ? "--stop-servers" : "";
+
+	# Check the host's status before trying to power it up/down.
+	my $is_host_on = $anvil->data->{hosts}{host_uuid}{$host_uuid}{host_status} eq "online" ? 1 : 0;
+
+	$anvil->Log->variables({
+		source => $THIS_FILE,
+		line   => __LINE__,
+		level  => 2,
+		list   => {
+			host_uuid  => $host_uuid,
+			is_host_on => $is_host_on,
+			on         => $on
+		}
+	});
+
+	# When host is ON and request is to power OFF.
+	#
+	# Power OFF should invoke anvil-safe-stop on a host.
+	if ($is_host_on && not $on)
+	{
+		$anvil->Database->insert_or_update_jobs({
+			job_command     => $anvil->data->{path}{exe}{'anvil-safe-stop'}." --power-off ".$stop_servers,
+			job_host_uuid   => $host_uuid,
+			job_description => "job_0333",
+			job_name        => "cgi-bin::set_power::off",
+			job_progress    => 0,
+			job_title       => "job_0332"
+		});
+	}
+	# When host is OFF and request is to power ON.
+	#
+	# Power ON should invoke striker-boot-machine on this striker.
+	elsif (not $is_host_on && $on)
+	{
+		$anvil->Database->insert_or_update_jobs({
+			job_command     => $anvil->data->{path}{directories}{tools}."/striker-boot-machine --host-uuid ".$host_uuid,
+			job_description => "job_0335",
+			job_name        => "cgi-bin::set_power::on",
+			job_progress    => 0,
+			job_title       => "job_0334"
+		});
+	}
+}
+
 $anvil->Get->switches;
 
 $anvil->Database->connect;
@@ -39,44 +133,68 @@ $anvil->Database->get_anvils();
 
 print $anvil->Template->get({file => "shared.html", name => "json_headers", show_name => 0})."\n";
 
-my $hash = {};
+my $response_body = {};
-my $anvil_uuid = exists $anvil->data->{cgi}{anvil_uuid}{value} ? $anvil->data->{cgi}{anvil_uuid}{value} : "";
+
-my $host_uuid  = exists $anvil->data->{cgi}{host_uuid}{value}  ? $anvil->data->{cgi}{host_uuid}{value}  : "";
+# Decode request body (JSON string) to a hash.
-if (($anvil_uuid) && (not exists $anvil->data->{anvils}{anvil_uuid}{$anvil_uuid}))
+#
-{
+# Note: requests made with the PUT method won't have query params.
-	$anvil->data->{anvil_status}{anvil_name} = "!!invalid!anvil_uuid!!";
+my $request_body;
-}
+
-elsif (($host_uuid) && (not exists $anvil->data->{hosts}{host_uuid}{$anvil_uuid}))
+my $is_decode_json_success = eval {
+	$request_body = decode_json($anvil->data->{cgi}{PUTDATA}{value});
+};
+
+if (not $is_decode_json_success)
 {
-	$anvil->data->{anvil_status}{anvil_name} = "!!invalid!host_uuid!!";
+	$anvil->Log->entry({
+		source    => $THIS_FILE,
+		line      => __LINE__,
+		level     => 0,
+		'print'   => 1,
+		priority  => "err",
+		key       => "error_0304",
+		variables => { request_body_string => $anvil->data->{cgi}{PUTDATA}{value}, json_decode_error => $_ }
+	});
 }
-elsif ((not $host_uuid) && (not $anvil_uuid))
+
+my $anvil_uuid               = exists $request_body->{anvil_uuid} ? $request_body->{anvil_uuid} : $anvil->data->{switches}{'anvil-uuid'};
+my $host_uuid                = exists $request_body->{host_uuid}  ? $request_body->{host_uuid}  : $anvil->data->{switches}{'host-uuid'};
+my $is_on                    = exists $request_body->{is_on}      ? $request_body->{is_on}      : $anvil->data->{switches}{'is-on'};
+my $anvil_uuid_variable_name = "anvil UUID";
+my $host_uuid_variable_name  = "host UUID";
+
+$anvil->Log->variables({
+	source => $THIS_FILE,
+	line   => __LINE__,
+	level  => 2,
+	list   => {
+		anvil_uuid => $anvil_uuid,
+		host_uuid  => $host_uuid,
+		is_on      => $is_on
+	}
+});
+
+if ($anvil_uuid)
 {
-	$anvil->data->{anvil_status}{anvil_name} = "!!invalid!no_target!!";
+	if (exists $anvil->data->{anvils}{anvil_uuid}{$anvil_uuid})
+	{
+		set_anvil_power({ anvil_uuid => $anvil_uuid, on => $is_on });
+	}
+	else
+	{
+		handle_invalid_uuid({ name => $anvil_uuid_variable_name, uuid => $anvil_uuid });
+	}
 }
-else
+elsif ($host_uuid)
 {
-	# Are we managing the power of a node or an Anvil! system? If the later, we're controlling both.
+	if (exists $anvil->data->{hosts}{host_uuid}{$host_uuid})
-	
+	{
-# 	$anvil->data->{anvil_status}{anvil_name}        = $anvil->data->{anvils}{anvil_uuid}{$anvil_uuid}{anvil_name};
+		set_host_power({ host_uuid => $host_uuid, on => $is_on });
-# 	$anvil->data->{anvil_status}{anvil_description} = $anvil->data->{anvils}{anvil_uuid}{$anvil_uuid}{anvil_description};
+	}
-# 	$anvil->data->{anvil_status}{timestamp}         = time;
+	else
-# 	
+	{
-# 	my $node1_uuid   = $anvil->data->{anvils}{anvil_uuid}{$anvil_uuid}{anvil_node1_host_uuid};
+		handle_invalid_uuid({ name => $host_uuid_variable_name, uuid => $host_uuid });
-# 	my $node2_uuid   = $anvil->data->{anvils}{anvil_uuid}{$anvil_uuid}{anvil_node2_host_uuid};
+	}
-# 	my $node2_status = $anvil->data->{hosts}{host_uuid}{$node2_uuid}{host_status} eq "online" ? 1 : 0;
-# 	
-# 	$anvil->data->{anvil_status}{nodes}{node1}{host_name}   = $anvil->data->{hosts}{host_uuid}{$node1_uuid}{host_name};
-# 	$anvil->data->{anvil_status}{nodes}{node1}{host_uuid}   = $node1_uuid;
-# 	$anvil->data->{anvil_status}{nodes}{node1}{host_status} = $anvil->data->{hosts}{host_uuid}{$node1_uuid}{host_status} eq "online" ? 1 : 0;
-# 	$anvil->data->{anvil_status}{nodes}{node2}{host_name}   = $anvil->data->{hosts}{host_uuid}{$node2_uuid}{host_name};
-# 	$anvil->data->{anvil_status}{nodes}{node2}{host_uuid}   = $node2_uuid;
-# 	$anvil->data->{anvil_status}{nodes}{node2}{host_status} = $anvil->data->{hosts}{host_uuid}{$node2_uuid}{host_status} eq "online" ? 1 : 0;
-# 	
-# 	$hash->{timestamp} = time;
-# 	$hash->{nodes}     = [];
-# 	push @{$hash->{nodes}}, { on => $anvil->data->{anvil_status}{nodes}{node1}{host_status} };
-# 	push @{$hash->{nodes}}, { on => $anvil->data->{anvil_status}{nodes}{node2}{host_status} };
 }
 
-print JSON->new->utf8->encode($hash)."\n";
+print JSON->new->utf8->encode($response_body)."\n";

share/words.xml

@@ -415,6 +415,7 @@ The attempt to start the servers appears to have failed. The return code '0' was
 		<key name="error_0301">Failed to parse the install manifest with UUID: [#!variable!manifest_uuid!#]. Unable to check or update the fence configuration.</key>
 		<key name="error_0302">The passed in Anvil! UUID: [#!variable!anvil_uuid!#] was not found in the database.</key>
 		<key name="error_0303">The passed in host UUID: [#!variable!host_uuid!#] was not found in the database.</key>
+		<key name="error_0304">Failed to parse the request body: [#!variable!request_body_string!#]. Reason: [#!variable!json_decode_error!#]</key>
 		
 		<!-- Files templates -->
 		<!-- NOTE: Translating these files requires an understanding of which lines are translatable -->
@@ -1072,6 +1073,14 @@ It should be provisioned in the next minute or two.</key>
 		<key name="job_0329">The target machine is confirmed off, will try to start now.</key>
 		<key name="job_0330">The target machine is now booting!</key>
 		<key name="job_0331">The machine: [#!variable!host_name!#] does not have a (known) IPMI BMC, but it is a member of the Anvil! [#!variable!anvil_name!#]. Searching for a fence method to boot it...</key>
+		<!-- cgi-bin/set_power,on,job_title -->
+		<key name="job_0332">Power On Host</key>
+		<!-- cgi-bin/set_power,on,job_description -->
+		<key name="job_0333">Power on the target host by executing a start script on a striker.</key>
+		<!-- cgi-bin/set_power,off,job_title -->
+		<key name="job_0334">Power Off Host</key>
+		<!-- cgi-bin/set_power,off,job_description -->
+		<key name="job_0335">Power off the target host by executing a stop script on the host itself.</key>
 		
 		<!-- Log entries -->
 		<key name="log_0001">Starting: [#!variable!program!#].</key>