FIX Elements in use report source list is now lazy loaded allowing fo…

…r larger data sets
silverstripe · Jul 18, 2018 · 785b50f · 785b50f
1 parent 540a757
commit 785b50f
Show file tree

Hide file tree

Showing 2 changed files with 50 additions and 44 deletions.
diff --git a/src/Reports/ElementsInUseReport.php b/src/Reports/ElementsInUseReport.php
@@ -3,11 +3,9 @@
 namespace DNADesign\Elemental\Reports;
 
 use DNADesign\Elemental\Models\BaseElement;
-use SilverStripe\Core\Injector\Injector;
 use SilverStripe\Forms\GridField\GridField;
-use SilverStripe\ORM\ArrayList;
+use SilverStripe\ORM\DataList;
 use SilverStripe\Reports\Report;
-use SilverStripe\View\ArrayData;
 use SilverStripe\View\Requirements;
 
 class ElementsInUseReport extends Report
@@ -19,42 +17,33 @@ public function title()
 
     public function sourceRecords($params = [])
     {
-        /** @var BaseElement[] $elements */
-        $elements = BaseElement::get();
+        /** @var DataList $elements */
+        $elements = BaseElement::get()->exclude(['ClassName' => BaseElement::class]);
 
-        $output = ArrayList::create();
-
-        foreach ($elements as $element) {
-            // Skip if filtering and looking at a different record
-            if (isset($params['ClassName']) && $params['ClassName'] !== $element->sanitiseClassName($element)) {
-                continue;
-            }
-
-            $output->push(ArrayData::create([
-                'Icon' => $element->getIcon(),
-                'Title' => $element->Title,
-                'EditLink' => $element->CMSEditLink(),
-                'Summary' => $element->getSummary(),
-                'Type' => $element->getTypeNice(),
-                'ClassName' => $element->sanitiseClassName(get_class($element)),
-                'Page' => $element->getPageTitle(),
-            ]));
+        if (isset($params['ClassName'])) {
+            $className = $this->unsanitiseClassName($params['ClassName']);
+            $elements = $elements->filter(['ClassName' => $className]);
         }
 
-        return $output;
+        return $elements;
     }
 
     public function columns()
     {
         return [
             'Icon' => [
                 'title' => '',
+                'formatting' => function ($value, BaseElement $item) {
+                    return $item->getIcon();
+                },
             ],
             'Title' => [
                 'title' => _t(__CLASS__ . '.Title', 'Title'),
-                'formatting' => function ($value, $item) {
+                'formatting' => function ($value, BaseElement $item) {
+                    $value = $item->Title;
+
                     if (!empty($value)) {
-                        if ($item->EditLink) {
+                        if ($item->CMSEditLink()) {
                             return $this->getEditLink($value, $item);
                         }
                         return $value;
@@ -65,16 +54,23 @@ public function columns()
             'Summary' => [
                 'title' => _t(__CLASS__ . '.Summary', 'Summary'),
                 'casting' => 'HTMLText->RAW',
+                'formatting' => function ($value, BaseElement $item) {
+                    return $item->getSummary();
+                },
             ],
             'Type' => [
                 'title' => _t(__CLASS__ . '.Type', 'Type'),
+                'formatting' => function ($value, BaseElement $item) {
+                    return $item->getTypeNice();
+                },
             ],
-            'Page' => [
+            'Page.Title' => [
                 'title' => _t(__CLASS__ . '.Page', 'Page'),
-                'formatting' => function ($value, $item) {
+                'formatting' => function ($value, BaseElement $item) {
                     if ($value) {
                         return $this->getEditLink($value, $item);
                     }
+                    return $item->getPageTitle();
                 },
             ],
         ];
@@ -84,14 +80,14 @@ public function columns()
      * Helper method to return the link to edit an element
      *
      * @param string $value
-     * @param ArrayData $item
+     * @param BaseElement $item
      * @return string
      */
     protected function getEditLink($value, $item)
     {
         return sprintf(
             '<a class="grid-field__link" href="%s" title="%s">%s</a>',
-            $item->EditLink,
+            $item->CMSEditLink(),
             $value,
             $value
         );
@@ -112,4 +108,17 @@ public function getReportField()
 
         return $field;
     }
+
+    /**
+     * Unsanitise a model class' name from a URL param
+     *
+     * See {@link \SilverStripe\Admin\ModelAdmin::unsanitiseClassName}
+     *
+     * @param string $class
+     * @return string
+     */
+    protected function unsanitiseClassName($class)
+    {
+        return str_replace('-', '\\', $class);
+    }
 }
diff --git a/tests/Reports/ElementsInUseReportTest.php b/tests/Reports/ElementsInUseReportTest.php
@@ -2,12 +2,13 @@
 
 namespace DNADesign\Elemental\Tests\Reports;
 
+use DNADesign\Elemental\Models\BaseElement;
+use DNADesign\Elemental\Models\ElementContent;
 use DNADesign\Elemental\Reports\ElementsInUseReport;
 use DNADesign\Elemental\Tests\Src\TestElement;
 use DNADesign\Elemental\Tests\Src\TestPage;
 use SilverStripe\Dev\FunctionalTest;
-use SilverStripe\ORM\ArrayList;
-use SilverStripe\View\ArrayData;
+use SilverStripe\ORM\DataList;
 
 class ElementsInUseReportTest extends FunctionalTest
 {
@@ -27,7 +28,7 @@ public function testReportShowsElementsInUse()
         $this->assertContains('Content blocks in use', $result, 'Title is displayed');
 
         $this->assertContains(
-            'data-class="DNADesign-Elemental-Models-ElementContent"',
+            'data-class="DNADesign\\Elemental\\Models\\ElementContent"',
             $result,
             'Report contains content elements (bundled with elemental)'
         );
@@ -42,14 +43,12 @@ public function testSourceRecords()
     {
         $records = (new ElementsInUseReport)->sourceRecords();
 
-        $this->assertInstanceOf(ArrayList::class, $records);
-        $this->assertNotContains(BaseElement::class, $records->toArray(), 'BaseElement is excluded');
-
-        $this->assertContainsOnlyInstancesOf(ArrayData::class, $records);
+        $this->assertInstanceOf(DataList::class, $records);
+        $this->assertNotEmpty($records);
+        $this->assertContainsOnlyInstancesOf(BaseElement::class, $records, 'Report contains elements');
 
         foreach ($records as $record) {
-            $this->assertNotNull($record->Icon, 'Fields have an icon');
-            $this->assertNotNull($record->Type, 'Fields have a type');
+            $this->assertNotEquals(BaseElement::class, get_class($record), 'BaseElement does not exist in the report');
         }
     }
 
@@ -73,13 +72,11 @@ public function testSourceRecordsFilteredByClassName()
             'ClassName' => 'DNADesign-Elemental-Models-ElementContent',
         ]);
 
-        $this->assertInstanceOf(ArrayList::class, $records);
+        $this->assertInstanceOf(DataList::class, $records);
         $this->assertNotEmpty($records->toArray(), 'Results are returned when filtered');
-        $this->assertEquals(
-            [
-                'DNADesign-Elemental-Models-ElementContent'
-            ],
-            array_unique($records->column('ClassName')),
+        $this->assertContainsOnlyInstancesOf(
+            ElementContent::class,
+            $records->toArray(),
             'Only contains filtered element type'
         );
     }