X-Git-Url: http://www.aleph1.co.uk/gitweb/?a=blobdiff_plain;ds=sidebyside;f=vendor%2Fdrush%2Fdrush%2Fsrc%2FDrupal%2FCommands%2Fsql%2FSanitizeUserFieldsCommands.php;fp=vendor%2Fdrush%2Fdrush%2Fsrc%2FDrupal%2FCommands%2Fsql%2FSanitizeUserFieldsCommands.php;h=55c65eed3576bb91d0432dc736ec441b52cdcc1d;hb=af6d1fb995500ae68849458ee10d66abbdcfb252;hp=0000000000000000000000000000000000000000;hpb=680c79a86e3ed402f263faeac92e89fb6d9edcc0;p=yaffs-website

diff --git a/vendor/drush/drush/src/Drupal/Commands/sql/SanitizeUserFieldsCommands.php b/vendor/drush/drush/src/Drupal/Commands/sql/SanitizeUserFieldsCommands.php
new file mode 100644
index 000000000..55c65eed3
--- /dev/null
+++ b/vendor/drush/drush/src/Drupal/Commands/sql/SanitizeUserFieldsCommands.php
@@ -0,0 +1,138 @@
+<?php
+namespace Drush\Drupal\Commands\sql;
+
+use Consolidation\AnnotatedCommand\CommandData;
+use Drupal\Component\Utility\Random;
+use Drupal\Core\Database\Database;
+use Drush\Commands\DrushCommands;
+use Symfony\Component\Console\Input\InputInterface;
+
+/**
+ * This class is a good example of how to build a sql-sanitize plugin.
+ */
+class SanitizeUserFieldsCommands extends DrushCommands implements SanitizePluginInterface
+{
+    protected $database;
+    protected $entityManager;
+    protected $entityTypeManager;
+
+    public function __construct($database, $entityManager, $entityTypeManager)
+    {
+        $this->database = $database;
+        $this->entityManager = $entityManager;
+        $this->entityTypeManager = $entityTypeManager;
+    }
+
+    /**
+     * @return mixed
+     */
+    public function getDatabase()
+    {
+        return $this->database;
+    }
+
+    /**
+     * @return mixed
+     */
+    public function getEntityManager()
+    {
+        return $this->entityManager;
+    }
+
+    /**
+     * Sanitize string fields associated with the user.
+     *
+     * @todo Use Drupal services to get field info.
+     *
+     * @hook post-command sql-sanitize
+     *
+     * @inheritdoc
+     */
+    public function sanitize($result, CommandData $commandData)
+    {
+        $options = $commandData->options();
+        $conn = $this->getDatabase();
+        $field_definitions = $this->getEntityManager()->getFieldDefinitions('user', 'user');
+        $field_storage = $this->getEntityManager()->getFieldStorageDefinitions('user');
+        foreach (explode(',', $options['whitelist-fields']) as $key => $def) {
+            unset($field_definitions[$key], $field_storage[$key]);
+        }
+
+        foreach ($field_definitions as $key => $def) {
+            $execute = false;
+            if (!isset($field_storage[$key]) || $field_storage[$key]->isBaseField()) {
+                continue;
+            }
+
+            $table = 'user__' . $key;
+            $query = $conn->update($table);
+            $name = $def->getName();
+            $field_type_class = \Drupal::service('plugin.manager.field.field_type')->getPluginClass($def->getType());
+            $value_array = $field_type_class::generateSampleValue($def);
+            $value = $value_array['value'];
+            switch ($def->getType()) {
+                case 'email':
+                    $query->fields([$name . '_value' => $value]);
+                    $execute = true;
+                    break;
+                case 'string':
+                    $query->fields([$name . '_value' => $value]);
+                    $execute = true;
+                    break;
+
+                case 'string_long':
+                    $query->fields([$name . '_value' => $value]);
+                    $execute = true;
+                    break;
+
+                case 'telephone':
+                    $query->fields([$name . '_value' => '15555555555']);
+                    $execute = true;
+                    break;
+
+                case 'text':
+                    $query->fields([$name . '_value' => $value]);
+                    $execute = true;
+                    break;
+
+                case 'text_long':
+                    $query->fields([$name . '_value' => $value]);
+                    $execute = true;
+                    break;
+
+                case 'text_with_summary':
+                    $query->fields([
+                    $name . '_value' => $value,
+                    $name . '_summary' => $value_array['summary'],
+                    ]);
+                    $execute = true;
+                    break;
+            }
+            if ($execute) {
+                $query->execute();
+                $this->entityTypeManager->getStorage('user')->resetCache();
+                $this->logger()->success(dt('!table table sanitized.', ['!table' => $table]));
+            } else {
+                $this->logger()->success(dt('No text fields for users need sanitizing.', ['!table' => $table]));
+            }
+        }
+    }
+
+    /**
+     * @hook on-event sql-sanitize-confirms
+     *
+     * @inheritdoc
+     */
+    public function messages(&$messages, InputInterface $input)
+    {
+        $messages[] = dt('Sanitize text fields associated with users.');
+    }
+
+    /**
+     * @hook option sql-sanitize
+     * @option whitelist-fields A comma delimited list of fields exempt from sanitization.
+     */
+    public function options($options = ['whitelist-fields' => ''])
+    {
+    }
+}