LocationPage Importer, admin cleanup

_config/config.yml

@@ -9,8 +9,14 @@ Dynamic\Locator\Location:
 
 Dynamic\Locator\Page\LocationPage:
   extensions:
+    - LittleGiant\CatalogManager\Extensions\CatalogPageExtension
     - Dynamic\SilverStripeGeocoder\DistanceDataExtension
     - Dynamic\SilverStripeGeocoder\AddressDataExtension
+  parent_classes:
+    - Dynamic\Locator\Page\Locator
+  sort_column: false
+  automatic_live_sort: false
+  show_in_sitetree: false
 
 Dynamic\Locator\Page\Locator:
   infoWindowTemplate: 'dynamic/silverstripe-locator: client/infowindow-description.html'
@@ -19,6 +25,8 @@ Dynamic\Locator\Page\Locator:
   show_radius: true
   radius_var: 'Radius'
   category_var: 'CategoryID'
+  extensions:
+    - SilverStripe\Lumberjack\Model\Lumberjack
 
 ---
 Name: modules-locator

src/admin/LocationsAdmin.php

@@ -3,12 +3,15 @@
 namespace Dynamic\Locator\Admin;
 
 use Dynamic\Locator\Bulkloader\LocationCsvBulkLoader;
+use Dynamic\Locator\Bulkloader\LocationPageCsvBulkLoader;
 use Dynamic\Locator\Model\LocationCategory;
 use Dynamic\Locator\Page\LocationPage;
 use LittleGiant\CatalogManager\ModelAdmin\CatalogPageAdmin;
 use SilverStripe\Dev\CsvBulkLoader;
 use SilverStripe\Forms\Form;
 use SilverStripe\Forms\GridField\GridField;
+use SilverStripe\Forms\GridField\GridFieldExportButton;
+use SilverStripe\Forms\GridField\GridFieldImportButton;
 
 /**
  * Class LocationAdmin
@@ -29,6 +32,7 @@ class LocationsAdmin extends CatalogPageAdmin
      * @var array
      */
     private static $model_importers = [
+        LocationPage::class => LocationPageCsvBulkLoader::class,
         LocationCategory::class => CsvBulkLoader::class,
     ];
 
@@ -89,9 +93,21 @@ public function getEditForm($id = null, $fields = null)
         if ($this->modelClass == LocationPage::class) {
             /** @var GridField $gridField */
             if ($gridField = $form->Fields()->fieldByName($this->sanitiseClassName($this->modelClass))) {
+                $exportButton = new GridFieldExportButton('buttons-before-left');
+                $exportButton->setExportColumns($this->getExportFields());
                 $gridField->getConfig()
+                    ->addComponent($exportButton)
                     ->removeComponentsByType('GridFieldDeleteAction');
             }
+
+            if ($this->showImportForm) {
+                $fieldConfig = $gridField->getConfig();
+                $fieldConfig->addComponent(
+                    GridFieldImportButton::create('buttons-before-left')
+                        ->setImportForm($this->ImportForm())
+                        ->setModalTitle(_t('SilverStripe\\Admin\\ModelAdmin.IMPORT', 'Import from CSV'))
+                );
+            }
         }
 
         return $form;

src/bulkloader/LocationPageCsvBulkLoader.php

@@ -0,0 +1,106 @@
+<?php
+
+namespace Dynamic\Locator\Bulkloader;
+
+use Dynamic\Locator\Model\LocationCategory;
+use Dynamic\Locator\Page\LocationPage;
+use Dynamic\Locator\Page\Locator;
+use SilverStripe\Dev\CsvBulkLoader;
+use SilverStripe\Core\Convert;
+use SilverStripe\i18n\Data\Intl\IntlLocales;
+
+/**
+ * Class LocationCsvBulkLoader
+ * @package Dynamic\Locator
+ */
+class LocationPageCsvBulkLoader extends CsvBulkLoader
+{
+
+    /**
+     * @var array
+     */
+    public $columnMap = array(
+        'Name' => 'Title',
+        'EmailAddress' => 'Email',
+        'Categories' => '->getCategoryByName',
+        'Import_ID' => 'Import_ID',
+        'Country' => '->getCountryByName',
+    );
+
+    /**
+     * @var array
+     */
+    public $duplicateChecks = array(
+        'Import_ID' => 'Import_ID'
+    );
+
+    /**
+     * @param $val
+     * @return string|string[]|null
+     */
+    public function getEscape($val)
+    {
+        return preg_replace("/\r|\n/", "", $val);
+    }
+
+    /**
+     * @param $obj
+     * @param $val
+     * @param $record
+     * @throws \SilverStripe\ORM\ValidationException
+     */
+    public static function getCategoryByName(&$obj, $val, $record)
+    {
+        if ($val = Convert::raw2sql($val)) {
+            $parts = explode(', ', $val);
+
+            foreach ($parts as $part) {
+                $category = LocationCategory::get()->filter(array('Name' => $part))->first();
+                if (!$category) {
+                    $category = LocationCategory::create();
+                    $category->Name = $part;
+                    $category->write();
+                }
+                $obj->Categories()->add($category);
+            }
+        }
+    }
+
+    /**
+     * @param $obj
+     * @param $val
+     * @param $record
+     */
+    public function getCountryByName(&$obj, $val, $record)
+    {
+        $val = $this->getEscape($val);
+        $countries = IntlLocales::singleton()->getCountries();
+
+        if (strlen((string)$val) == 2) {
+            $val = strtolower($val);
+            if (array_key_exists($val, $countries)) {
+                $obj->Country = $val;
+            }
+        }
+
+        if (in_array($val, $countries)) {
+            $obj->Country = array_search($val, $countries);
+        }
+    }
+
+    /**
+     * @param array $record
+     * @param array $columnMap
+     * @param \SilverStripe\Dev\BulkLoader_Result $results
+     * @param bool $preview
+     * @return int|void
+     */
+    protected function processRecord($record, $columnMap, &$results, $preview = false)
+    {
+        $objID = parent::processRecord($record, $columnMap, $results, $preview = false);
+
+        $location = LocationPage::get()->byID($objID);
+        $location->ParentID = Locator::get()->first()->ID;
+        $location->publishSingle();
+    }
+}