Merge pull request #157 from creative-commoners/pulls/4/used-on

ENH Used on table extension

_config/config.yml

@@ -4,3 +4,6 @@ Name: linkfield
 SilverStripe\Admin\LeftAndMain:
   extensions:
     - SilverStripe\LinkField\Extensions\LeftAndMainExtension
+SilverStripe\Admin\Forms\UsedOnTable:
+  extensions:
+    - SilverStripe\LinkField\Extensions\UsedOnTableExtension

src/Extensions/UsedOnTableExtension.php

@@ -0,0 +1,32 @@
+<?php
+
+namespace SilverStripe\LinkField\Extensions;
+
+use DNADesign\Elemental\Models\BaseElement;
+use DNADesign\Elemental\TopPage\DataExtension as TopPageExtension;
+use SilverStripe\Core\Extension;
+use SilverStripe\LinkField\Models\FileLink;
+use SilverStripe\ORM\DataObject;
+
+class UsedOnTableExtension extends Extension
+{
+    public function updateUsageAncestorDataObjects(array &$ancestorDataObjects, DataObject $dataObject): void
+    {
+        if (!is_a($dataObject, FileLink::class)) {
+            return;
+        }
+        $owner = $dataObject->Owner();
+        if (!$owner?->exists()) {
+            return;
+        }
+        $ancestorDataObjects[] = $owner;
+        if (!class_exists(BaseElement::class) || !is_a($owner, BaseElement::class)) {
+            return;
+        }
+        $page = $owner->hasExtension(TopPageExtension::class) ? $owner->getTopPage() : $owner->getPage();
+        if (!$page?->exists()) {
+            return;
+        }
+        $ancestorDataObjects[] = $page;
+    }
+}