VPS.NET Auto Recovery Monitor

< ?php

//----
// Set up the monitoring script:
//----

define('MONITOR_MAIL_TO', '');     // Email address to send alerts to.
define('MONITOR_API_EMAIL', ''); // Email address associated with VPS.NET API account.
define('MONITOR_API_KEY', '');     // VPS.NET API key.
define('MONITOR_SEND_EMAILS', true);               // Whether or not to send emails if a server is flagged as down.
define('MONITOR_SEND_EMAILS_ON_NO_ACTION', true);  // Whether or not to send emails if no action is taken.
define('MONITOR_REBOOT_IF_RUNNING', true);         // If HTTP check fails, but VPS.NET says the server is up, reboot?
define('MONITOR_HTTP_TIMEOUT', 5);                 // Number of seconds to allow for a server response. 

//----
// Initialise API Class and get VMs:
//----
$api = new SimpleVpsApi();
$api->setApiEmail(MONITOR_API_EMAIL);
$api->setApiKey(MONITOR_API_KEY);
$vms = $api->getVMs();

//----
// Loop through all the VMs and check their status:
//----
$out     = '';
foreach($vms as $vm)
{
	$server = $vm['virtual_machine']['hostname'] . '.' . $vm['virtual_machine']['domain_name'];
	$id     = $vm['virtual_machine']['id'];

	$get = false;
	$get = @file_get_contents('http://'.$server.'/', false, stream_context_create(array('http' => array('timeout' => MONITOR_HTTP_TIMEOUT))));

	if(!$get)
	{
		$_out .= 'Server ' . $server . ' appears to be DOWN!'.PHP_EOL;

		$vm = $api->getVM($id);

		if($vm['power_action_pending'])
		{
			$_out .= '- VPS.NET is reporting that the server is rebooting already. Taking no action.'.PHP_EOL;

			if(!MONITOR_SEND_EMAILS_ON_NO_ACTION)
			{
				$_out = '';
			}
		}
		elseif($vm['running'] && !MONITOR_REBOOT_IF_RUNNING)
		{
			$_out .= '- VPS.NET is reporting the server as running. Taking no action.'.PHP_EOL;

			if(!MONITOR_SEND_EMAILS_ON_NO_ACTION)
			{
				$_out = '';
			}
		}
		elseif($vm['running'] && MONITOR_REBOOT_IF_RUNNING)
		{
			$api->rebootVM($id);
			$_out .= '- VPS.NET is reporting the server as running. Rebooting.'.PHP_EOL;
		}
		else
		{
			$api->bootVM($id);
			$_out .= '- Submitted an automated power on request to VPS.NET.'.PHP_EOL;
		}

		// If we've logged anything for this VM, append it to the log:
		if($_out)
		{
			if($out != '')
			{
				$out .= PHP_EOL;
			}

			$out .= $_out;
			$_out = '';
		}
	}
}

//----
// Emailing functionality, for notification of server downtime:
//----

$email = '
	

		Server Monitoring Notice

			
		


	
';

if($out != '' && MONITOR_SEND_EMAILS)
{
	$email = str_replace('', nl2br($out), $email);
	mail(MONITOR_MAIL_TO, 'Server Monitoring Notice', $email, 'Content-Type: text/html'."\r\n");
}

//----
// Custom class for interacting with the VPS.NET API:
//----

class SimpleVpsApi
{
	const TYPE_GET  = 1;
	const TYPE_POST = 2;

	protected $apiUrl   = 'api.vps.net/';
	protected $apiEmail = null;
	protected $apiKey   = null;

	/**
	* Set the email address to use in requests:
	*/
	public function setApiEmail($email)
	{
		$this->apiEmail = $email;
	}

	/**
	* Set the API key to use in requests:
	*/
	public function setApiKey($key)
	{
		$this->apiKey = $key;
	}

	/**
	* Get a list of all VMs:
	*/
	public function getVMs()
	{
		$vm = $this->sendApiRequest('virtual_machines', 'GET');

		if($vm)
		{
			return $vm;
		}
		else
		{
			return array();
		}
	}

	/**
	* Sends a request to VPS.NET to load details about a VM:
	*/
	public function getVM($id)
	{
		$vm = $this->sendApiRequest('virtual_machines/'.$id, 'GET');

		if($vm)
		{
			return $vm['virtual_machine'];
		}
		else
		{
			return null;
		}
	}

	/**
	* Sends a request to VPS.NET to reboot a VM:
	*/
	public function rebootVM($id)
	{
		return $this->sendApiRequest('virtual_machines/'.$id.'/reboot', 'POST');
	}

	/**
	* Sends a request to VPS.NET to boot a VM:
	*/
	public function bootVM($id)
	{
		return $this->sendApiRequest('virtual_machines/'.$id.'/power_on', 'POST');
	}

	/**
	* Send an API request to VPS.NET
	*/
	protected function sendApiRequest($uri, $type = 'GET')
	{
		if(!$this->apiEmail || !$this->apiKey)
		{
			throw new Exception('You must set an API key and email address first.');
		}

		$opts = array(
		  'http'=>array(
		    'method'  => $type,
		    'header'  => 'Accept: application/json' . "\r\n" .
		                 'Content-Type: application/json' . "\r\n" .
						 'Authorization: Basic ' . base64_encode($this->apiEmail.':'.$this->apiKey) . "\r\n",

			'timeout' => 3,
		  )
		);

		$url      = 'https://'.$this->apiUrl.$uri.'.api10json';
		$context  = stream_context_create($opts);
		$response = @file_get_contents($url, false, $context);
		return json_decode($response, true);
	}
}