/
ElementalAreaUsedOnTableExtension.php
68 lines (63 loc) · 1.87 KB
/
ElementalAreaUsedOnTableExtension.php
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
<?php
namespace DNADesign\Elemental\Extensions;
use SilverStripe\CMS\Model\SiteTree;
use SilverStripe\ORM\DataExtension;
use SilverStripe\ORM\DataObject;
use SilverStripe\ORM\ValidationException;
use DNADesign\Elemental\Models\ElementalArea;
use DNADesign\Elemental\Models\BaseElement;
use SilverStripe\Admin\Forms\UsedOnTable;
/**
* @extends DataExtension<UsedOnTable>
*/
class ElementalAreaUsedOnTableExtension extends DataExtension
{
/**
* Hides ElementalArea's from the "Used On" tab when viewing files
*
* @var array $excludedClasses
*/
public function updateUsageExcludedClasses(array &$excludedClasses)
{
$excludedClasses[] = ElementalArea::class;
}
/**
* Exclude content blocks that aren't linked to a page
*
* @param bool $excludeDataObject
* @param DataObject $dataObject|null
*/
public function updateUsageDataObject(?DataObject &$dataObject)
{
if (!($dataObject instanceof BaseElement)) {
return;
}
try {
if (!$dataObject->getPage()) {
$dataObject = null;
}
} catch (ValidationException $e) {
$dataObject = null;
}
}
/**
* Link BaseElement's to their parent page
*
* @param array $ancestorDataObjects
* @param DataObject $dataObject
*/
public function updateUsageAncestorDataObjects(array &$ancestorDataObjects, DataObject $dataObject)
{
if (!($dataObject instanceof BaseElement)) {
return;
}
try {
// BaseElement::getPage() caches results so there's no performance decrease from
// also calling it in updateUsageExcludeDataObject()
if ($page = $dataObject->getPage()) {
$ancestorDataObjects[] = $page;
}
} catch (ValidationException $e) {
}
}
}