From 76d0a48898a0ea994b3a633ec873b02894a7b8cc Mon Sep 17 00:00:00 2001
From: florian <florian@b0492a97-d6bb-434e-b004-7405e1235103>
Date: Sat, 5 Oct 2013 14:57:03 +0000
Subject: [PATCH] improved drupal detection process, including support for
 localhost. courtesy WvR

---
 run.d/drupal | 43 +++++++++++++++++++++++++++++--------------
 1 file changed, 29 insertions(+), 14 deletions(-)

diff --git a/run.d/drupal b/run.d/drupal
index 8a6aa32..98c72ed 100755
--- a/run.d/drupal
+++ b/run.d/drupal
@@ -2,11 +2,11 @@
 <?
 // Fetch the host list to check
 //
-if( getenv('CHK_HOSTS') == null ) {
-	$hosts = null;
-}else {
-	$hosts = explode(' ', getenv('CHK_HOSTS'));
-	if (implode($hosts)=='') die("No hosts to check.\n");
+if(getenv('CHK_HOSTS') == null) {
+  $hosts = null;
+} else {
+  $hosts = explode(' ', getenv('CHK_HOSTS'));
+  if (implode($hosts)=='') die("No hosts to check.\n");
 }
 
 // Retrieve current Drupal version as documented at http://www.goldcoastmedia.co.uk/blog/drupal-versions-with-php
@@ -30,9 +30,24 @@ if($feed && $feed instanceof SimpleXMLElement) {
   }
 }
 
-// For Drupal 4.x - 6.x
-foreach($hosts as $host) {
-  $cmd = "ssh $host \"locate modules/system/system.module | xargs grep -H 'VERSION'\"";
+// Start checking hosts
+//
+if( $hosts == null ) {
+  checkHost(null, $dr_current);
+} else {
+  foreach($hosts as $host) {
+    checkHost($host, $dr_current);
+  }
+}
+
+function checkHost($host, $dr_current) {
+  // For Drupal 4.x - 6.x
+  $cmd = "locate modules/system/system.module | xargs grep -H 'VERSION'";
+  $hostidentifier = "";
+  if( $host != null ) {
+ 	$cmd = "ssh $host \"" . $cmd . "\"";
+	$hostidentifier = $host . ":";
+  }
   exec($cmd, $results);
 
   foreach($results as $instance) {
@@ -40,16 +55,16 @@ foreach($hosts as $host) {
     $location = substr($parts[0], 0, strlen($parts[0])-28);
     if(substr($parts[1], 0, 16) == "define('VERSION'") {
       $version = substr($parts[1], 19, strpos($parts[1], "'", 19)-19);
-      if($version < $dr_current) echo "WARNING: Outdated Drupal $version at $host:$location\n";
+      if($version < $dr_current) echo "WARNING: Outdated Drupal $version at $hostidentifier$location\n";
     }
   }
   unset($results);
 
-}
-
-// For Drupal 7.x 
-foreach($hosts as $host) {
-  $cmd = "ssh $host \"locate includes/bootstrap.inc | xargs grep -H 'VERSION'\"";
+  // For Drupal 7.x
+  $cmd = "locate includes/bootstrap.inc | xargs grep -H 'VERSION'";
+  if( $host != null ) {
+	$cmd = "ssh $host \"" . $cmd . "\"";
+  }
   exec($cmd, $results);
 
   foreach($results as $instance) {