1 files changed, 74 insertions, 0 deletions
diff --git a/library/Businessprocess/Modification/NodeAddChildrenAction.php b/library/Businessprocess/Modification/NodeAddChildrenAction.php
new file mode 100644
index 0000000..162c380
--- /dev/null
+++ b/library/Businessprocess/Modification/NodeAddChildrenAction.php
@@ -0,0 +1,74 @@
+<?php
+
+namespace Icinga\Module\Businessprocess\Modification;
+
+use Icinga\Module\Businessprocess\BpConfig;
+
+class NodeAddChildrenAction extends NodeAction
+{
+    protected $children = array();
+
+    protected $preserveProperties = array('children');
+
+    /**
+     * @inheritdoc
+     */
+    public function appliesTo(BpConfig $config)
+    {
+        $name = $this->getNodeName();
+
+        if (! $config->hasBpNode($name)) {
+            $this->error('Process "%s" not found', $name);
+        }
+
+        return true;
+    }
+
+    /**
+     * @inheritdoc
+     */
+    public function applyTo(BpConfig $config)
+    {
+        $node = $config->getBpNode($this->getNodeName());
+
+        foreach ($this->children as $name) {
+            if (! $config->hasNode($name) || $config->getNode($name)->getBpConfig()->getName() !== $config->getName()) {
+                [$prefix, $suffix] = BpConfig::splitNodeName($name);
+                if ($suffix !== null) {
+                    if ($suffix === 'Hoststatus') {
+                        $config->createHost($prefix);
+                    } else {
+                        $config->createService($prefix, $suffix);
+                    }
+                } elseif ($name[0] === '@' && strpos($name, ':') !== false) {
+                    list($configName, $nodeName) = preg_split('~:\s*~', substr($name, 1), 2);
+                    $config->createImportedNode($configName, $nodeName);
+                }
+            }
+            $node->addChild($config->getNode($name));
+        }
+
+        return $this;
+    }
+
+    /**
+     * @param array|string $children
+     * @return $this
+     */
+    public function setChildren($children)
+    {
+        if (is_string($children)) {
+            $children = array($children);
+        }
+        $this->children = $children;
+        return $this;
+    }
+
+    /**
+     * @return array
+     */
+    public function getChildren()
+    {
+        return $this->children;
+    }
+}