1 files changed, 37 insertions, 0 deletions
diff --git a/application/forms/BasketCreateSnapshotForm.php b/application/forms/BasketCreateSnapshotForm.php
new file mode 100644
index 0000000..165c7ac
--- /dev/null
+++ b/application/forms/BasketCreateSnapshotForm.php
@@ -0,0 +1,37 @@
+<?php
+
+namespace Icinga\Module\Director\Forms;
+
+use Icinga\Module\Director\DirectorObject\Automation\Basket;
+use Icinga\Module\Director\DirectorObject\Automation\BasketSnapshot;
+use Icinga\Module\Director\Web\Form\DirectorForm;
+
+class BasketCreateSnapshotForm extends DirectorForm
+{
+    /** @var Basket */
+    private $basket;
+
+    public function setBasket(Basket $basket)
+    {
+        $this->basket = $basket;
+
+        return $this;
+    }
+
+    public function setup()
+    {
+        $this->setSubmitLabel($this->translate('Create Snapshot'));
+    }
+
+    /**
+     * @throws \Icinga\Module\Director\Exception\DuplicateKeyException
+     */
+    public function onSuccess()
+    {
+        /** @var \Icinga\Module\Director\Db $connection */
+        $connection = $this->basket->getConnection();
+        $snapshot = BasketSnapshot::createForBasket($this->basket, $connection);
+        $snapshot->store();
+        parent::onSuccess();
+    }
+}