Bazinga\GeocoderBundle\Plugin\FakeIpPlugin

Complexity

Total Complexity

6

Size/Duplication

Total Lines	57
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	3

Test Coverage

Coverage

94.44%

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	11	2
A	handleQuery()	0	19	4

<?php

declare(strict_types=1);

/*
 * This file is part of the BazingaGeocoderBundle package.
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 *
 * @license    MIT License
 */

namespace Bazinga\GeocoderBundle\Plugin;

use Faker\Generator;
use Faker\Provider\Internet;
use Geocoder\Plugin\Plugin;
use Geocoder\Query\GeocodeQuery;
use Geocoder\Query\Query;

/**
 * Replace local IP with something better.
 *
 * @author Tobias Nyholm <tobias.nyholm@gmail.com>
 */
class FakeIpPlugin implements Plugin
{
    /**
     * @var string
     */
    private $needle;

    /**
     * @var string
     */
    private $replacement;

    /**
     * @var bool
     */
    private $useFaker;

    /**
     * @var Generator|null
     */
    private $faker;

    public function __construct(string $needle, string $replacement = null, bool $useFaker = false)
    {
        $this->needle = $needle;
        $this->replacement = $replacement;
        $this->useFaker = $useFaker;

        if ($useFaker) {
            $this->faker = new Generator();
            $this->faker->addProvider(new Internet($this->faker));
        }
    }

    /**
     * {@inheritdoc}
     */
    public function handleQuery(Query $query, callable $next, callable $first)
    {
        if (!$query instanceof GeocodeQuery) {
            return $next($query);
        }

        $replacement = $this->replacement;

        if (null !== $this->faker) {
            $replacement = $this->faker->ipv4;
        }

        $text = str_replace($this->needle, $replacement, $query->getText(), $count);
        if ($count > 0) {
            $query = $query->withText($text);
        }

        return $next($query);
    }
}