Adding upstream version 2.11.4.upstream/2.11.4upstream

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 82 insertions, 0 deletions

diff --git a/modules/monitoring/library/Monitoring/Object/Macro.php b/modules/monitoring/library/Monitoring/Object/Macro.php
new file mode 100644
index 0000000..3f67154
--- /dev/null
+++ b/modules/monitoring/library/Monitoring/Object/Macro.php
@@ -0,0 +1,82 @@
+<?php
+/* Icinga Web 2 | (c) 2014 Icinga Development Team | GPLv2+ */
+
+namespace Icinga\Module\Monitoring\Object;
+
+use Exception;
+use Icinga\Application\Logger;
+
+/**
+ * Expand macros in string in the context of MonitoredObjects
+ */
+class Macro
+{
+    /**
+     * Known icinga macros
+     *
+     * @var array
+     */
+    private static $icingaMacros = array(
+        'HOSTNAME'      => 'host_name',
+        'HOSTADDRESS'   => 'host_address',
+        'HOSTADDRESS6'  => 'host_address6',
+        'SERVICEDESC'   => 'service_description',
+        'host.name'     => 'host_name',
+        'host.address'  => 'host_address',
+        'host.address6' => 'host_address6',
+        'service.description' => 'service_description',
+        'service.name'  => 'service_description'
+    );
+
+    /**
+     * Return the given string with macros being resolved
+     *
+     * @param   string                      $input      The string in which to look for macros
+     * @param   MonitoredObject|stdClass    $object     The host or service used to resolve macros
+     *
+     * @return  string                                  The substituted or unchanged string
+     */
+    public static function resolveMacros($input, $object)
+    {
+        $matches = array();
+        if (preg_match_all('@\$([^\$\s]+)\$@', $input, $matches)) {
+            foreach ($matches[1] as $key => $value) {
+                $newValue = self::resolveMacro($value, $object);
+                if ($newValue !== $value) {
+                    $input = str_replace($matches[0][$key], $newValue, $input);
+                }
+            }
+        }
+
+        return $input;
+    }
+
+    /**
+     * Resolve a macro based on the given object
+     *
+     * @param   string                      $macro      The macro to resolve
+     * @param   MonitoredObject|stdClass    $object     The object used to resolve the macro
+     *
+     * @return  string                                  The new value or the macro if it cannot be resolved
+     */
+    public static function resolveMacro($macro, $object)
+    {
+        if (isset(self::$icingaMacros[$macro]) && isset($object->{self::$icingaMacros[$macro]})) {
+            return $object->{self::$icingaMacros[$macro]};
+        }
+
+        try {
+            $value = $object->$macro;
+        } catch (Exception $e) {
+            $objectName = $object->getName();
+            if ($object instanceof Service) {
+                $objectName = $object->getHost()->getName() . '!' . $objectName;
+            }
+
+            $value = null;
+            Logger::debug('Unable to resolve macro "%s" on object "%s". An error occured: %s', $macro, $objectName, $e);
+        }
+
+        return $value !== null ? $value : $macro;
+    }
+}