summaryrefslogtreecommitdiffstats
path: root/library/Director/Web/Controller/BranchHelper.php
blob: 89aa6c10a62bf91fd9e2d4d1e0bbccc12ffa068f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
<?php

namespace Icinga\Module\Director\Web\Controller;

use Icinga\Module\Director\Data\Db\DbObjectStore;
use Icinga\Module\Director\Db\Branch\Branch;
use Icinga\Module\Director\Db\Branch\BranchStore;
use Icinga\Module\Director\Db\Branch\BranchSupport;
use Icinga\Module\Director\Db\Branch\PreferredBranchSupport;
use Icinga\Module\Director\Objects\IcingaObject;
use Icinga\Module\Director\Web\Widget\NotInBranchedHint;
use Ramsey\Uuid\UuidInterface;

trait BranchHelper
{
    /** @var Branch */
    protected $branch;

    /** @var BranchStore */
    protected $branchStore;

    /** @var ?bool */
    protected $hasPreferredBranch = null;

    /**
     * @return ?UuidInterface
     */
    protected function getBranchUuid(): ?UuidInterface
    {
        return $this->getBranch()->getUuid();
    }

    /**
     * @return Branch
     */
    protected function getBranch(): Branch
    {
        if ($this->branch === null) {
            /** @var ActionController $this */
            $this->branch = Branch::forRequest($this->getRequest(), $this->getBranchStore(), $this->Auth());
        }

        return $this->branch;
    }

    /**
     * @return BranchStore
     */
    protected function getBranchStore(): BranchStore
    {
        if ($this->branchStore === null) {
            $this->branchStore = new BranchStore($this->db());
        }

        return $this->branchStore;
    }

    /**
     * @return bool
     */
    protected function hasBranch(): bool
    {
        return $this->getBranchUuid() !== null;
    }

    protected function enableStaticObjectLoader($table): void
    {
        if (BranchSupport::existsForTableName($table)) {
            IcingaObject::setDbObjectStore(new DbObjectStore($this->db(), $this->getBranch()));
        }
    }

    /**
     * @param string $subject
     * @return bool
     */
    protected function showNotInBranch($subject): bool
    {
        if ($this->getBranch()->isBranch()) {
            $this->content()->add(new NotInBranchedHint($subject, $this->getBranch(), $this->Auth()));
            return true;
        }

        return false;
    }

    protected function hasPreferredBranch(): bool
    {
        if ($this->hasPreferredBranch === null) {
            $implementation = Branch::optionalHook();
            if ($implementation instanceof PreferredBranchSupport) {
                $this->hasPreferredBranch = $implementation->hasPreferredBranch($this->Auth());
            } else {
                $this->hasPreferredBranch = false;
            }
        }

        return $this->hasPreferredBranch;
    }
}