<?php

/*
 * This file is part of the FOSElasticaBundle package.
 *
 * (c) FriendsOfSymfony <https://friendsofsymfony.github.com/>
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace KupShop\CatalogElasticBundle\Profiler;

use Psr\Log\AbstractLogger;
use Psr\Log\LoggerInterface;

/**
 * Logger for the Elastica.
 *
 * The {@link logQuery()} method is configured as the logger callable in the
 * service container.
 *
 * @author Gordon Franke <info@nevalon.de>
 */
class ElasticaLogger extends AbstractLogger
{
    protected ?LoggerInterface $logger;
    /**
     * @var list<array<string, mixed>>
     */
    protected array $queries = [];
    protected bool $debug;

    public function __construct(?LoggerInterface $logger = null, bool $debug = true)
    {
        $this->logger = $logger;
        $this->debug = $debug;
    }

    /**
     * Logs a query.
     *
     * @param string              $path       Path to call
     * @param string              $method     Rest method to use (GET, POST, DELETE, PUT)
     * @param array<mixed>|string $data       Arguments
     * @param float               $queryTime  Execution time (in seconds)
     * @param array<mixed>        $connection Host, port, transport, and headers of the query
     * @param array<mixed>        $query      Arguments
     * @param int                 $engineTime
     */
    public function logQuery(
        string $path,
        string $method,
        string $response,
        $data,
        $queryTime,
        $connection = [],
        $query = [],
        $engineTime = 0,
        int|string $itemCount = 0,
    ): void {
        $executionMS = $queryTime * 1000;

        if ($this->debug) {
            $e = new \Exception();
            if (\is_string($data)) {
                $jsonStrings = \explode("\n", $data);
                $data = [];
                foreach ($jsonStrings as $json) {
                    if ('' != $json) {
                        $data[] = \json_decode($json, true);
                    }
                }
            } else {
                $data = [$data];
            }

            $this->queries[] = [
                'path' => $path,
                'method' => $method,
                'data' => $data,
                'executionMS' => $executionMS,
                'engineMS' => $engineTime,
                'connection' => $connection,
                'queryString' => $query,
                'itemCount' => $itemCount,
                'response' => json_decode($response, true),
                'backtrace' => $e->getTraceAsString(),
            ];
        }

        if (null !== $this->logger) {
            $message = \sprintf('%s (%s) %0.2f ms', $path, $method, $executionMS);
            $this->logger->info($message, (array) $data);
        }
    }

    /**
     * Returns the number of queries that have been logged.
     */
    public function getNbQueries(): int
    {
        return \count($this->queries);
    }

    /**
     * Returns a human-readable array of queries logged.
     *
     * @return list<array<string, mixed>>
     */
    public function getQueries(): array
    {
        return $this->queries;
    }

    /**
     * @param array<string, mixed> $context
     */
    public function log($level, $message, array $context = []): void
    {
        $this->logger->log($level, $message, $context);
    }

    public function reset(): void
    {
        $this->queries = [];
    }
}