--- /dev/null
+<?php
+
+namespace Drupal\devel_generate\Plugin\DevelGenerate;
+
+use Drupal\Component\Utility\Unicode;
+use Drupal\Core\Entity\EntityStorageInterface;
+use Drupal\Core\Form\FormStateInterface;
+use Drupal\Core\Language\Language;
+use Drupal\Core\Plugin\ContainerFactoryPluginInterface;
+use Drupal\devel_generate\DevelGenerateBase;
+use Symfony\Component\DependencyInjection\ContainerInterface;
+
+/**
+ * Provides a VocabularyDevelGenerate plugin.
+ *
+ * @DevelGenerate(
+ * id = "vocabulary",
+ * label = @Translation("vocabularies"),
+ * description = @Translation("Generate a given number of vocabularies. Optionally delete current vocabularies."),
+ * url = "vocabs",
+ * permission = "administer devel_generate",
+ * settings = {
+ * "num" = 1,
+ * "title_length" = 12,
+ * "kill" = FALSE
+ * }
+ * )
+ */
+class VocabularyDevelGenerate extends DevelGenerateBase implements ContainerFactoryPluginInterface {
+
+ /**
+ * The vocabulary storage.
+ *
+ * @var \Drupal\Core\Entity\EntityStorageInterface
+ */
+ protected $vocabularyStorage;
+
+ /**
+ * Constructs a new VocabularyDevelGenerate object.
+ *
+ * @param array $configuration
+ * A configuration array containing information about the plugin instance.
+ * @param string $plugin_id
+ * The plugin_id for the plugin instance.
+ * @param mixed $plugin_definition
+ * The plugin implementation definition.
+ * @param \Drupal\Core\Entity\EntityStorageInterface $entity_storage
+ * The vocabulary storage.
+ */
+ public function __construct(array $configuration, $plugin_id, $plugin_definition, EntityStorageInterface $entity_storage) {
+ parent::__construct($configuration, $plugin_id, $plugin_definition);
+
+ $this->vocabularyStorage = $entity_storage;
+ }
+
+ /**
+ * {@inheritdoc}
+ */
+ public static function create(ContainerInterface $container, array $configuration, $plugin_id, $plugin_definition) {
+ return new static(
+ $configuration, $plugin_id, $plugin_definition,
+ $container->get('entity.manager')->getStorage('taxonomy_vocabulary')
+ );
+ }
+
+ /**
+ * {@inheritdoc}
+ */
+ public function settingsForm(array $form, FormStateInterface $form_state) {
+ $form['num'] = array(
+ '#type' => 'number',
+ '#title' => $this->t('Number of vocabularies?'),
+ '#default_value' => $this->getSetting('num'),
+ '#required' => TRUE,
+ '#min' => 0,
+ );
+ $form['title_length'] = array(
+ '#type' => 'number',
+ '#title' => $this->t('Maximum number of characters in vocabulary names'),
+ '#default_value' => $this->getSetting('title_length'),
+ '#required' => TRUE,
+ '#min' => 2,
+ '#max' => 255,
+ );
+ $form['kill'] = array(
+ '#type' => 'checkbox',
+ '#title' => $this->t('Delete existing vocabularies before generating new ones.'),
+ '#default_value' => $this->getSetting('kill'),
+ );
+
+ return $form;
+ }
+
+ /**
+ * {@inheritdoc}
+ */
+ public function generateElements(array $values) {
+ if ($values['kill']) {
+ $this->deleteVocabularies();
+ $this->setMessage($this->t('Deleted existing vocabularies.'));
+ }
+
+ $new_vocs = $this->generateVocabularies($values['num'], $values['title_length']);
+ if (!empty($new_vocs)) {
+ $this->setMessage($this->t('Created the following new vocabularies: @vocs', array('@vocs' => implode(', ', $new_vocs))));
+ }
+ }
+
+ /**
+ * Deletes all vocabularies.
+ */
+ protected function deleteVocabularies() {
+ $vocabularies = $this->vocabularyStorage->loadMultiple();
+ $this->vocabularyStorage->delete($vocabularies);
+ }
+
+ /**
+ * Generates vocabularies.
+ *
+ * @param int $records
+ * Number of vocabularies to create.
+ * @param int $maxlength
+ * (optional) Maximum length for vocabulary name.
+ *
+ * @return array
+ * Array containing the generated vocabularies id.
+ */
+ protected function generateVocabularies($records, $maxlength = 12) {
+ $vocabularies = array();
+
+ // Insert new data:
+ for ($i = 1; $i <= $records; $i++) {
+ $name = $this->getRandom()->word(mt_rand(2, $maxlength));
+
+ $vocabulary = $this->vocabularyStorage->create(array(
+ 'name' => $name,
+ 'vid' => Unicode::strtolower($name),
+ 'langcode' => Language::LANGCODE_NOT_SPECIFIED,
+ 'description' => "Description of $name",
+ 'hierarchy' => 1,
+ 'weight' => mt_rand(0, 10),
+ 'multiple' => 1,
+ 'required' => 0,
+ 'relations' => 1,
+ ));
+
+ // Populate all fields with sample values.
+ $this->populateFields($vocabulary);
+ $vocabulary->save();
+
+ $vocabularies[] = $vocabulary->id();
+ unset($vocabulary);
+ }
+
+ return $vocabularies;
+ }
+
+ /**
+ * {@inheritdoc}
+ */
+ public function validateDrushParams($args) {
+ $values = array(
+ 'num' => array_shift($args),
+ 'kill' => drush_get_option('kill'),
+ 'title_length' => 12,
+ );
+
+ if ($this->isNumber($values['num']) == FALSE) {
+ return drush_set_error('DEVEL_GENERATE_INVALID_INPUT', dt('Invalid number of vocabularies: @num.', array('@num' => $values['num'])));
+ }
+
+ return $values;
+ }
+
+}