123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175 |
- <?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;
- }
- }
|