Make cloning service sets possible in Director Branches

Author: raviks789

application/forms/IcingaCloneObjectForm.php

@@ -29,7 +29,11 @@ public function setup()
     {
         $isBranch = $this->branch && $this->branch->isBranch();
         $branchOnly = $this->object->get('id') === null;
-        if ($isBranch && $this->object instanceof IcingaObject && $this->object->isTemplate()) {
+        if (
+            $isBranch
+            && $this->object instanceof IcingaObject
+            && ($this->object->isTemplate() && ! $this->object instanceof IcingaServiceSet)
+        ) {
             $this->addHtml(Hint::error($this->translate(
                 'Templates cannot be cloned in Configuration Branches'
             )));
@@ -146,7 +150,11 @@ public function onSuccess()
             $object->getObjectName()
         );
 
-        if ($object->isTemplate() && $this->branch && $this->branch->isBranch()) {
+        if (
+            $this->branch
+            && $this->branch->isBranch()
+            && ($object->isTemplate() && ! ($object instanceof IcingaServiceSet))
+        ) {
             throw new IcingaException('Cloning templates is not available for Branches');
         }
 
@@ -211,8 +219,13 @@ public function onSuccess()
                 if ($new instanceof IcingaHost) {
                     $clone->set('host_id', $newId);
                 } elseif ($new instanceof IcingaServiceSet) {
-                    $clone->set('service_set_id', $newId);
+                    if ($this->branch && $this->branch->isBranch()) {
+                        $clone->set('service_set', $newName);
+                    } else {
+                        $clone->set('service_set_id', $newId);
+                    }
                 }
+
                 $store->store($clone);
             }
 

library/Director/Web/Controller/ObjectController.php

@@ -217,12 +217,14 @@ public function cloneAction()
         $this->addTitle($this->translate('Clone: %s'), $object->getObjectName())
             ->addBackToObjectLink();
 
-        if ($object->isTemplate() && $this->showNotInBranch($this->translate('Cloning Templates'))) {
-            return;
-        }
+        if (! ($object instanceof IcingaServiceSet)) {
+            if ($object->isTemplate() && $this->showNotInBranch($this->translate('Cloning Templates'))) {
+                return;
+            }
 
-        if ($object->isTemplate() && $this->showNotInBranch($this->translate('Cloning Apply Rules'))) {
-            return;
+            if ($object->isTemplate() && $this->showNotInBranch($this->translate('Cloning Apply Rules'))) {
+                return;
+            }
         }
 
         $form = IcingaCloneObjectForm::load()