devel_generate.module

Same filename in other branches
  1. 6.x-1.x devel_generate.module
  2. 7.x-1.x devel_generate/devel_generate.module
  3. 4.x devel_generate/devel_generate.module
  4. 5.x devel_generate/devel_generate.module

File

devel_generate/devel_generate.module

View source
<?php

use Drupal\Core\Entity\EntityInterface;
use Drupal\Core\Field\FieldDefinitionInterface;
use Drupal\comment\Plugin\Field\FieldType\CommentItemInterface;
use Drupal\Component\Utility\Random;
use Drupal\devel_generate\DevelGenerateBase;
use Drupal\node\NodeInterface;

/**
 * Implements hook_menu_links_discovered_alter().
 */
function devel_generate_menu_links_discovered_alter(&$links) {
    $devel_generate_plugins = $devel_generate_manager = \Drupal::service('plugin.manager.develgenerate')->getDefinitions();
    foreach ($devel_generate_plugins as $id => $plugin) {
        $label = $plugin['label'];
        $links["devel_generate.{$id}"] = array(
            'title' => "Generate {$label}",
            'parent' => 'system.admin_config_development',
            'description' => $plugin['description'],
            'route_name' => "devel_generate.{$id}",
        );
    }
}

/**
 * Implements hook_entity_insert().
 *
 * Inserts nodes properly based on generation options.
 *
 * @param \Drupal\Core\Entity\EntityInterface $entity
 *  The base node created on submit. Inspects $node->devel_generate.
 */
function devel_generate_entity_insert(EntityInterface $entity) {
    if ($entity->getEntityTypeId() != 'node' || !isset($entity->devel_generate)) {
        return;
    }
    
    /** @var \Drupal\node\NodeInterface $entity */
    $results = $entity->devel_generate;
    if (!empty($results['max_comments'])) {
        foreach ($entity->getFieldDefinitions() as $field_name => $field_definition) {
            if ($field_definition->getType() == 'comment' && $entity->get($field_name)->status == CommentItemInterface::OPEN) {
                // Add comments for each comment field on entity.
                devel_generate_add_comments($entity, $field_definition, $results['users'], $results['max_comments'], $results['title_length']);
            }
        }
    }
    // Add an url alias. Cannot happen before save because we don't know the nid.
    if (!empty($results['add_alias'])) {
        $path = array(
            'source' => '/node/' . $entity->id(),
            'alias' => '/node-' . $entity->id() . '-' . $entity->bundle(),
        );
        \Drupal::service('path.alias_storage')->save($path['source'], $path['alias'], $entity->language()
            ->getId());
    }
    // Add node statistics.
    if (!empty($results['add_statistics']) && \Drupal::moduleHandler()->moduleExists('statistics')) {
        devel_generate_add_statistics($entity);
    }
}

/**
 * Create comments and add them to a node.
 *
 * @param \Drupal\node\NodeInterface $node
 *   Node to add comments to.
 * @param \Drupal\Core\Field\FieldDefinitionInterface $field_definition
 *   The field storage definition.
 * @param array $users
 *   Array of users to assign comment authors.
 * @param int $max_comments
 *   Max number of comments to generate per node.
 * @param int $title_length
 *   Max length of the title of the comments.
 */
function devel_generate_add_comments(NodeInterface $node, FieldDefinitionInterface $field_definition, $users, $max_comments, $title_length = 8) {
    $parents = array();
    $field_name = $field_definition->getName();
    $num_comments = mt_rand(1, $max_comments);
    for ($i = 1; $i <= $num_comments; $i++) {
        switch ($i % 3) {
            case 0:
            // No parent.
            case 1:
                // Top level parent.
                $parents = \Drupal::entityQuery('comment')->condition('pid', 0)
                    ->condition('entity_id', $node->id())
                    ->condition('entity_type', 'node')
                    ->condition('field_name', $field_name)
                    ->range(0, 1)
                    ->execute();
                break;
            case 2:
                // Non top level parent.
                $parents = \Drupal::entityQuery('comment')->condition('pid', 0, '>')
                    ->condition('entity_id', $node->id())
                    ->condition('entity_type', 'node')
                    ->condition('field_name', $field_name)
                    ->range(0, 1)
                    ->execute();
                break;
        }
        $random = new Random();
        $stub = array(
            'entity_type' => $node->getEntityTypeId(),
            'entity_id' => $node->id(),
            'field_name' => $field_name,
            'name' => 'devel generate',
            'mail' => 'devel_generate@example.com',
            'timestamp' => mt_rand($node->getCreatedTime(), REQUEST_TIME),
            'subject' => substr($random->sentences(mt_rand(2, $title_length), TRUE), 0, 63),
            'uid' => $users[array_rand($users)],
            'langcode' => $node->language()
                ->getId(),
        );
        if ($parents) {
            $stub['pid'] = current($parents);
        }
        $comment = entity_create('comment', $stub);
        
        //Populate all core fields on behalf of field.module
        DevelGenerateBase::populateFields($comment);
        $comment->save();
    }
}

/**
 * Generate statistics information for a node.
 *
 * @param \Drupal\node\NodeInterface $node
 *   A node object.
 */
function devel_generate_add_statistics(NodeInterface $node) {
    $statistic = array(
        'nid' => $node->id(),
        'totalcount' => mt_rand(0, 500),
        'timestamp' => REQUEST_TIME - mt_rand(0, $node->getCreatedTime()),
    );
    $statistic['daycount'] = mt_rand(0, $statistic['totalcount']);
    db_insert('node_counter')->fields($statistic)
        ->execute();
}

Functions

Title Deprecated Summary
devel_generate_add_comments Create comments and add them to a node.
devel_generate_add_statistics Generate statistics information for a node.
devel_generate_entity_insert Implements hook_entity_insert().
devel_generate_menu_links_discovered_alter Implements hook_menu_links_discovered_alter().