1 files changed, 115 insertions, 0 deletions
diff --git a/library/Graphite/Web/Controller/TimeRangePickerTrait.php b/library/Graphite/Web/Controller/TimeRangePickerTrait.php
new file mode 100644
index 0000000..7352b1b
--- /dev/null
+++ b/library/Graphite/Web/Controller/TimeRangePickerTrait.php
@@ -0,0 +1,115 @@
+<?php
+
+namespace Icinga\Module\Graphite\Web\Controller;
+
+use Icinga\Module\Graphite\Forms\TimeRangePicker\CommonForm;
+use Icinga\Module\Graphite\Forms\TimeRangePicker\CustomForm;
+use Icinga\Web\Request;
+use Icinga\Web\Url;
+use Icinga\Web\View;
+
+trait TimeRangePickerTrait
+{
+    /**
+     * @var CommonForm
+     */
+    protected $timeRangePickerCommonForm;
+
+    /**
+     * @var CustomForm
+     */
+    protected $timeRangePickerCustomForm;
+
+    /**
+     * Process the given request using the forms
+     *
+     * @param   Request $request    The request to be processed
+     *
+     * @return  Request             The request supposed to be processed
+     */
+    protected function handleTimeRangePickerRequest(Request $request = null)
+    {
+        $this->getTimeRangePickerCommonForm()->handleRequest($request);
+        return $this->getTimeRangePickerCustomForm()->handleRequest($request);
+    }
+
+    /**
+     * Render all needed forms and links
+     *
+     * @param   View    $view
+     *
+     * @return  string
+     */
+    protected function renderTimeRangePicker(View $view)
+    {
+        $url = Url::fromRequest()->getAbsoluteUrl();
+
+        return '<div class="timerangepicker-container">'
+            . $this->getTimeRangePickerCommonForm()
+            . '<div class="flyover flyover-arrow-top" data-flyover-suspends-auto-refresh id="'
+            . $view->protectId('graphite-customrange')
+            . '">'
+            . $view->qlink(null, '#', null, [
+                'title' => $view->translate('Specify custom time range'),
+                'class' => 'button-link flyover-toggle',
+                'icon'  => 'service'
+            ])
+            . '<div class="flyover-content">' . $this->getTimeRangePickerCustomForm() . '</div>'
+            . '</div>'
+            . '</div>';
+    }
+
+    /**
+     * Get {@link timeRangePickerCommonForm}
+     *
+     * @return CommonForm
+     */
+    public function getTimeRangePickerCommonForm()
+    {
+        if ($this->timeRangePickerCommonForm === null) {
+            $this->timeRangePickerCommonForm = new CommonForm();
+        }
+
+        return $this->timeRangePickerCommonForm;
+    }
+
+    /**
+     * Set {@link timeRangePickerCommonForm}
+     *
+     * @param CommonForm $timeRangePickerCommonForm
+     *
+     * @return $this
+     */
+    public function setTimeRangePickerCommonForm(CommonForm $timeRangePickerCommonForm)
+    {
+        $this->timeRangePickerCommonForm = $timeRangePickerCommonForm;
+        return $this;
+    }
+
+    /**
+     * Get {@link timeRangePickerCustomForm}
+     *
+     * @return CustomForm
+     */
+    public function getTimeRangePickerCustomForm()
+    {
+        if ($this->timeRangePickerCustomForm === null) {
+            $this->timeRangePickerCustomForm = new CustomForm();
+        }
+
+        return $this->timeRangePickerCustomForm;
+    }
+
+    /**
+     * Set {@link timeRangePickerCustomForm}
+     *
+     * @param CustomForm $timeRangePickerCustomForm
+     *
+     * @return $this
+     */
+    public function setTimeRangePickerCustomForm(CustomForm $timeRangePickerCustomForm)
+    {
+        $this->timeRangePickerCustomForm = $timeRangePickerCustomForm;
+        return $this;
+    }
+}