<?php

namespace Blomstra\Search\Commands;

use Blomstra\Search\Observe\SavingJob;
use Blomstra\Search\Seeders\Seeder;
use Elasticsearch\Client;
use Illuminate\Console\Command;
use Illuminate\Contracts\Container\Container;
use Illuminate\Contracts\Queue\Queue;
use Illuminate\Database\Eloquent\Collection;

class RebuildDocumentsCommand extends Command
{
    protected $signature = 'blomstra:search:documents:rebuild
        {--flush : Flushes ALL the documents inside the index}
        {--mapping : Create property mappings: booleans, dates and fulltext searches}
        {--max-id= : Limits for each object the number of items to seed}';
    protected $description = 'Rebuilds the complete search server with its documents.';

    public function handle(Container $container)
    {
        /** @var array $seeders */
        $seeders = $container->tagged('blomstra.search.seeders');

        /** @var Queue $queue */
        $queue = $container->make(Queue::class);

        /** @var Client $client */
        $client = $container->make('blomstra.search.elastic');

        // Flush the index.
        if ($this->option('flush')) {
            $client->indices()->delete([
                'index' => $container->make('blomstra.search.elastic_index'),
                'ignore_unavailable' => true
            ]);
            $client->indices()->create([
                'index' => $container->make('blomstra.search.elastic_index')
            ]);
        }

        if ($this->option('mapping')) {
            $client->indices()->putMapping([
                'index' => $container->make('blomstra.search.elastic_index'),
                'body' => [
                    'properties' => [
                        'content' => ['type' => 'text'],
                        'created_at' => ['type' => 'date'],
                        'updated_at' => ['type' => 'date'],
                        'is_private' => ['type' => 'boolean'],
                        'is_sticky' => ['type' => 'boolean'],
                        'groups' => ['type' => 'integer'],
                        'recipient_groups' => ['type' => 'integer'],
                        'recipient_users' => ['type' => 'integer'],
                    ]
                ]
            ]);
        }

        /** @var Seeder $seeder */
        foreach ($seeders as $seeder) {
            $seeder->query()
                ->when($this->option('max-id'), function ($query, $id) {
                    $query->where('id', '<=', $id);
                })
                ->chunk(50, function (Collection $collection) use ($queue, &$total, $seeder) {
                    $queue->push(new SavingJob($collection, $seeder));

                    $this->info("Pushed into the index, type: {$seeder->type()}, amount: {$collection->count()}.");

                    $total += $collection->count();
                });

            $this->info("Pushed a total of $total into the index.");
        }
    }
}