Seeder.php

Go to the documentation of this file.

<?php

namespace Elgg\Database;

use Elgg\Cli\Progress;
use Elgg\Database\Seeds\Seed;
use Elgg\Invoker;
use Elgg\PluginHooksService;

class Seeder {

    protected $hooks;

    protected $progress;
    
    protected $invoker;

    public function __construct(
        PluginHooksService $hooks,
        Progress $progress,
        Invoker $invoker
    ) {
        $this->hooks = $hooks;
        $this->progress = $progress;
        $this->invoker = $invoker;
    }

    public function seed(array $options = []): void {
        $this->invoker->call(ELGG_IGNORE_ACCESS | ELGG_SHOW_DISABLED_ENTITIES | ELGG_DISABLE_SYSTEM_LOG, function() use ($options) {
            $defaults = [
                'limit' => max(elgg_get_config('default_limit'), 20),
                'image_folder' => elgg_get_config('seeder_local_image_folder'),
                'type' => '',
                'create' => false,
                'create_since' => 'now',
                'create_until' => 'now',
            ];
            $options = array_merge($defaults, $options);

            $seeds = $this->getSeederClasses();

            // set global configuration
            if ($options['image_folder'] !== $defaults['image_folder']) {
                elgg_set_config('seeder_local_image_folder', $options['image_folder']);
            }
            unset($options['image_folder']);

            foreach ($seeds as $seed) {
                // check for type limitation
                if (!empty($options['type']) && $options['type'] !== $seed::getType()) {
                    continue;
                }

                /* @var $seeder Seed */
                $seeder = new $seed($options);

                $progress_bar = $this->progress->start($seed, $options['limit']);

                $seeder->setProgressBar($progress_bar);

                $seeder->seed();

                $this->progress->finish($progress_bar);
            }
        });
    }

    public function unseed(array $options = []): void {
        $this->invoker->call(ELGG_IGNORE_ACCESS | ELGG_SHOW_DISABLED_ENTITIES | ELGG_DISABLE_SYSTEM_LOG, function() use ($options) {
            $defaults = [
                'type' => '',
            ];
            $options = array_merge($defaults, $options);

            $seeds = $this->getSeederClasses();

            foreach ($seeds as $seed) {
                // check for type limitation
                if (!empty($options['type']) && $options['type'] !== $seed::getType()) {
                    continue;
                }

                /* @var $seeder Seed */
                $seeder = new $seed();

                $progress_bar = $this->progress->start($seed, $seeder->getCount());

                $seeder->setProgressBar($progress_bar);

                $seeder->unseed();

                $this->progress->finish($progress_bar);
            }
        });
    }
    
    public function getSeederClasses(): array {
        $result = [];
        
        $seeds = $this->hooks->trigger('seeds', 'database', null, []);
        foreach ($seeds as $seed) {
            if (!class_exists($seed)) {
                elgg_log("Seeding class {$seed} not found", 'ERROR');
                continue;
            }
            if (!is_subclass_of($seed, Seed::class)) {
                elgg_log("Seeding class {$seed} does not extend " . Seed::class, 'ERROR');
                continue;
            }
            
            $result[] = $seed;
        }
        
        return $result;
    }
}