From 0251bc5edb22b3208f96fa8c07d83fc4e0b43c7f Mon Sep 17 00:00:00 2001
From: Thomas Gelf <thomas@gelf.net>
Date: Tue, 22 Nov 2016 17:57:46 +0100
Subject: [PATCH] Host/ServiceNode: introduce common base class

---
 library/Businessprocess/HostNode.php      |  2 +-
 library/Businessprocess/MonitoredNode.php | 24 +++++++++++++++++++++++
 library/Businessprocess/ServiceNode.php   |  4 ++--
 3 files changed, 27 insertions(+), 3 deletions(-)
 create mode 100644 library/Businessprocess/MonitoredNode.php

diff --git a/library/Businessprocess/HostNode.php b/library/Businessprocess/HostNode.php
index 6715e00..4c2496e 100644
--- a/library/Businessprocess/HostNode.php
+++ b/library/Businessprocess/HostNode.php
@@ -4,7 +4,7 @@ namespace Icinga\Module\Businessprocess;
 
 use Icinga\Web\Url;
 
-class HostNode extends Node
+class HostNode extends MonitoredNode
 {
     protected static $sortStateToStateMap = array(
         4 => self::ICINGA_DOWN,
diff --git a/library/Businessprocess/MonitoredNode.php b/library/Businessprocess/MonitoredNode.php
new file mode 100644
index 0000000..85ec5fe
--- /dev/null
+++ b/library/Businessprocess/MonitoredNode.php
@@ -0,0 +1,24 @@
+<?php
+
+namespace Icinga\Module\Businessprocess;
+
+use Icinga\Module\Businessprocess\Web\Component\Container;
+use Icinga\Module\Businessprocess\Web\Component\Link;
+
+abstract class MonitoredNode extends Node
+{
+    protected function prepareActions(Container $actions)
+    {
+        $actions->add(
+            $url = Link::create(
+                $actions->view()->translate('Simulate a specific state'),
+                'businessprocess/process/show?addSimulation&unlocked',
+                array(
+                    'config' => $this->bp->getName(),
+                    'simulationNode' => $this->name
+                ),
+                array('class' => 'icon-magic')
+            )
+        );
+    }
+}
\ No newline at end of file
diff --git a/library/Businessprocess/ServiceNode.php b/library/Businessprocess/ServiceNode.php
index 8360ad6..b169e24 100644
--- a/library/Businessprocess/ServiceNode.php
+++ b/library/Businessprocess/ServiceNode.php
@@ -4,7 +4,7 @@ namespace Icinga\Module\Businessprocess;
 
 use Icinga\Web\Url;
 
-class ServiceNode extends Node
+class ServiceNode extends MonitoredNode
 {
     protected $hostname;
 
@@ -42,7 +42,7 @@ class ServiceNode extends Node
 
         return $link;
     }
-    
+
     protected function getActionIcons($view)
     {
         $icons = array();