<?php

namespace Mpociot\ApiDoc\Tests\Unit;

use Dingo\Api\Routing\Router;
use Orchestra\Testbench\TestCase;
use Mpociot\ApiDoc\Tools\RouteMatcher;
use Illuminate\Support\Facades\Route as RouteFacade;

class RouteMatcherTest extends TestCase
{
    /**
     * @var RouteMatcher
     */
    private $matcher;

    protected function setUp()
    {
        parent::setUp();
        $this->matcher = new RouteMatcher();
    }

    protected function getPackageProviders($app)
    {
        return [
            \Dingo\Api\Provider\LaravelServiceProvider::class,
        ];
    }

    public function testRespectsDomainsRuleForLaravelRouter()
    {
        $this->registerLaravelRoutes();
        $routeRules[0]['match']['prefixes'] = ['*'];

        $routeRules[0]['match']['domains'] = ['*'];
        $routes = $this->matcher->getRoutesToBeDocumented($routeRules);
        $this->assertCount(12, $routes);

        $routeRules[0]['match']['domains'] = ['domain1.*', 'domain2.*'];
        $routes = $this->matcher->getRoutesToBeDocumented($routeRules);
        $this->assertCount(12, $routes);

        $routeRules[0]['match']['domains'] = ['domain1.*'];
        $routes = $this->matcher->getRoutesToBeDocumented($routeRules);
        $this->assertCount(6, $routes);
        foreach ($routes as $route) {
            $this->assertContains('domain1', $route['route']->getDomain());
        }

        $routeRules[0]['match']['domains'] = ['domain2.*'];
        $routes = $this->matcher->getRoutesToBeDocumented($routeRules);
        $this->assertCount(6, $routes);
        foreach ($routes as $route) {
            $this->assertContains('domain2', $route['route']->getDomain());
        }
    }

    public function testRespectsDomainsRuleForDingoRouter()
    {
        $this->registerDingoRoutes();
        $routeRules[0]['match']['versions'] = ['v1'];
        $routeRules[0]['match']['prefixes'] = ['*'];

        $routeRules[0]['match']['domains'] = ['*'];
        $routes = $this->matcher->getDingoRoutesToBeDocumented($routeRules);
        $this->assertCount(12, $routes);

        $routeRules[0]['match']['domains'] = ['domain1.*', 'domain2.*'];
        $routes = $this->matcher->getDingoRoutesToBeDocumented($routeRules);
        $this->assertCount(12, $routes);

        $routeRules[0]['match']['domains'] = ['domain1.*'];
        $routes = $this->matcher->getDingoRoutesToBeDocumented($routeRules);
        $this->assertCount(6, $routes);
        foreach ($routes as $route) {
            $this->assertContains('domain1', $route['route']->getDomain());
        }

        $routeRules[0]['match']['domains'] = ['domain2.*'];
        $routes = $this->matcher->getDingoRoutesToBeDocumented($routeRules);
        $this->assertCount(6, $routes);
        foreach ($routes as $route) {
            $this->assertContains('domain2', $route['route']->getDomain());
        }
    }

    public function testRespectsPrefixesRuleForLaravelRouter()
    {
        $this->registerLaravelRoutes();
        $routeRules[0]['match']['domains'] = ['*'];

        $routeRules[0]['match']['prefixes'] = ['*'];
        $routes = $this->matcher->getRoutesToBeDocumented($routeRules);
        $this->assertCount(12, $routes);

        $routeRules[0]['match']['prefixes'] = ['prefix1/*', 'prefix2/*'];
        $routes = $this->matcher->getRoutesToBeDocumented($routeRules);
        $this->assertCount(8, $routes);

        $routeRules[0]['match']['prefixes'] = ['prefix1/*'];
        $routes = $this->matcher->getRoutesToBeDocumented($routeRules);
        $this->assertCount(4, $routes);
        foreach ($routes as $route) {
            $this->assertTrue(str_is('prefix1/*', $route['route']->uri()));
        }

        $routeRules[0]['match']['prefixes'] = ['prefix2/*'];
        $routes = $this->matcher->getRoutesToBeDocumented($routeRules);
        $this->assertCount(4, $routes);
        foreach ($routes as $route) {
            $this->assertTrue(str_is('prefix2/*', $route['route']->uri()));
        }
    }

    public function testRespectsPrefixesRuleForDingoRouter()
    {
        $this->registerDingoRoutes();
        $routeRules[0]['match']['versions'] = ['v1'];
        $routeRules[0]['match']['domains'] = ['*'];

        $routeRules[0]['match']['prefixes'] = ['*'];
        $routes = $this->matcher->getDingoRoutesToBeDocumented($routeRules);
        $this->assertCount(12, $routes);

        $routeRules[0]['match']['prefixes'] = ['prefix1/*', 'prefix2/*'];
        $routes = $this->matcher->getDingoRoutesToBeDocumented($routeRules);
        $this->assertCount(8, $routes);

        $routeRules[0]['match']['prefixes'] = ['prefix1/*'];
        $routes = $this->matcher->getDingoRoutesToBeDocumented($routeRules);
        $this->assertCount(4, $routes);
        foreach ($routes as $route) {
            $this->assertTrue(str_is('prefix1/*', $route['route']->uri()));
        }

        $routeRules[0]['match']['prefixes'] = ['prefix2/*'];
        $routes = $this->matcher->getDingoRoutesToBeDocumented($routeRules);
        $this->assertCount(4, $routes);
        foreach ($routes as $route) {
            $this->assertTrue(str_is('prefix2/*', $route['route']->uri()));
        }
    }

    public function testRespectsVersionsRuleForDingoRouter()
    {
        $this->registerDingoRoutes();

        $routeRules[0]['match']['versions'] = ['v2'];
        $routeRules[0]['match']['domains'] = ['*'];
        $routeRules[0]['match']['prefixes'] = ['*'];
        $routes = $this->matcher->getDingoRoutesToBeDocumented($routeRules);
        $this->assertCount(6, $routes);
        foreach ($routes as $route) {
            $this->assertNotEmpty(array_intersect($route['route']->versions(), ['v2']));
        }

        $routeRules[0]['match']['versions'] = ['v1', 'v2'];
        $routeRules[0]['match']['domains'] = ['*'];
        $routeRules[0]['match']['prefixes'] = ['*'];
        $routes = $this->matcher->getDingoRoutesToBeDocumented($routeRules);
        $this->assertCount(18, $routes);
    }

    public function testWillIncludeRouteIfListedExplicitlyForLaravelRouter()
    {
        $this->registerLaravelRoutes();
        $mustInclude = 'domain1-1';
        $routeRules[0]['include'] = [$mustInclude];

        $routeRules[0]['match']['domains'] = ['domain1.*'];
        $routeRules[0]['match']['prefixes'] = ['prefix1/*'];
        $routes = $this->matcher->getRoutesToBeDocumented($routeRules);
        $oddRuleOut = collect($routes)->filter(function ($route) use ($mustInclude) {
            return $route['route']->getName() === $mustInclude;
        });
        $this->assertCount(1, $oddRuleOut);
    }

    public function testWillIncludeRouteIfListedExplicitlyForDingoRouter()
    {
        $this->registerDingoRoutes();

        $mustInclude = 'v2.domain2';
        $routeRules = [
            [
                'match' => [
                    'domains' => ['domain1.*'],
                    'prefixes' => ['prefix1/*'],
                    'versions' => ['v1'],
                ],
                'include' => [$mustInclude],
            ],
        ];
        $routes = $this->matcher->getDingoRoutesToBeDocumented($routeRules);
        $oddRuleOut = collect($routes)->filter(function ($route) use ($mustInclude) {
            return $route['route']->getName() === $mustInclude;
        });
        $this->assertCount(1, $oddRuleOut);
    }

    public function testWillExcludeRouteIfListedExplicitlyForLaravelRouter()
    {
        $this->registerLaravelRoutes();
        $mustNotInclude = 'prefix1.domain1-1';
        $routeRules[0]['exclude'] = [$mustNotInclude];

        $routeRules[0]['match']['domains'] = ['domain1.*'];
        $routeRules[0]['match']['prefixes'] = ['prefix1/*'];
        $routes = $this->matcher->getRoutesToBeDocumented($routeRules);
        $oddRuleOut = collect($routes)->filter(function ($route) use ($mustNotInclude) {
            return $route['route']->getName() === $mustNotInclude;
        });
        $this->assertCount(0, $oddRuleOut);
    }

    public function testWillExcludeRouteIfListedExplicitlyForDingoRouter()
    {
        $this->registerDingoRoutes();

        $mustNotInclude = 'v2.domain2';
        $routeRules = [
            [
                'match' => [
                    'domains' => ['domain2.*'],
                    'prefixes' => ['*'],
                    'versions' => ['v2'],
                ],
                'exclude' => [$mustNotInclude],
            ],
        ];
        $routes = $this->matcher->getDingoRoutesToBeDocumented($routeRules);
        $oddRuleOut = collect($routes)->filter(function ($route) use ($mustNotInclude) {
            return $route['route']->getName() === $mustNotInclude;
        });
        $this->assertCount(0, $oddRuleOut);
    }

    public function testMergesRoutesFromDifferentRuleGroupsForLaravelRouter()
    {
        $this->registerLaravelRoutes();

        $routeRules = [
            [
                'match' => [
                    'domains' => ['domain1.*'],
                    'prefixes' => ['prefix1/*'],
                ],
            ],
            [
                'match' => [
                    'domains' => ['domain2.*'],
                    'prefixes' => ['prefix2*'],
                ],
            ],
        ];

        $routes = $this->matcher->getRoutesToBeDocumented($routeRules);
        $this->assertCount(4, $routes);

        $routes = collect($routes);
        $firstRuleGroup = $routes->filter(function ($route) {
            return str_is('prefix1/*', $route['route']->uri())
                && str_is('domain1.*', $route['route']->getDomain());
        });
        $this->assertCount(2, $firstRuleGroup);

        $secondRuleGroup = $routes->filter(function ($route) {
            return str_is('prefix2/*', $route['route']->uri())
                && str_is('domain2.*', $route['route']->getDomain());
        });
        $this->assertCount(2, $secondRuleGroup);
    }

    public function testMergesRoutesFromDifferentRuleGroupsForDingoRouter()
    {
        $this->registerDingoRoutes();
        $routeRules = [
            [
                'match' => [
                    'domains' => ['*'],
                    'prefixes' => ['*'],
                    'versions' => ['v1'],
                ],
            ],
            [
                'match' => [
                    'domains' => ['*'],
                    'prefixes' => ['*'],
                    'versions' => ['v2'],
                ],
            ],
        ];

        $routes = $this->matcher->getDingoRoutesToBeDocumented($routeRules);
        $this->assertCount(18, $routes);

        $routes = collect($routes);
        $firstRuleGroup = $routes->filter(function ($route) {
            return ! empty(array_intersect($route['route']->versions(), ['v1']));
        });
        $this->assertCount(12, $firstRuleGroup);

        $secondRuleGroup = $routes->filter(function ($route) {
            return ! empty(array_intersect($route['route']->versions(), ['v2']));
        });
        $this->assertCount(6, $secondRuleGroup);
    }

    private function registerLaravelRoutes()
    {
        RouteFacade::group(['domain' => 'domain1.app.test'], function () {
            RouteFacade::post('/domain1-1', function () {
                return 'hi';
            })->name('domain1-1');
            RouteFacade::get('domain1-2', function () {
                return 'hi';
            })->name('domain1-2');
            RouteFacade::get('/prefix1/domain1-1', function () {
                return 'hi';
            })->name('prefix1.domain1-1');
            RouteFacade::get('prefix1/domain1-2', function () {
                return 'hi';
            })->name('prefix1.domain1-2');
            RouteFacade::get('/prefix2/domain1-1', function () {
                return 'hi';
            })->name('prefix2.domain1-1');
            RouteFacade::get('prefix2/domain1-2', function () {
                return 'hi';
            })->name('prefix2.domain1-2');
        });
        RouteFacade::group(['domain' => 'domain2.app.test'], function () {
            RouteFacade::post('/domain2-1', function () {
                return 'hi';
            })->name('domain2-1');
            RouteFacade::get('domain2-2', function () {
                return 'hi';
            })->name('domain2-2');
            RouteFacade::get('/prefix1/domain2-1', function () {
                return 'hi';
            })->name('prefix1.domain2-1');
            RouteFacade::get('prefix1/domain2-2', function () {
                return 'hi';
            })->name('prefix1.domain2-2');
            RouteFacade::get('/prefix2/domain2-1', function () {
                return 'hi';
            })->name('prefix2.domain2-1');
            RouteFacade::get('prefix2/domain2-2', function () {
                return 'hi';
            })->name('prefix2.domain2-2');
        });
    }

    private function registerDingoRoutes()
    {
        $api = app('api.router');
        $api->version('v1', function (Router $api) {
            $api->group(['domain' => 'domain1.app.test'], function (Router $api) {
                $api->post('/domain1-1', function () {
                    return 'hi';
                })->name('v1.domain1-1');
                $api->get('domain1-2', function () {
                    return 'hi';
                })->name('v1.domain1-2');
                $api->get('/prefix1/domain1-1', function () {
                    return 'hi';
                })->name('v1.prefix1.domain1-1');
                $api->get('prefix1/domain1-2', function () {
                    return 'hi';
                })->name('v1.prefix1.domain1-2');
                $api->get('/prefix2/domain1-1', function () {
                    return 'hi';
                })->name('v1.prefix2.domain1-1');
                $api->get('prefix2/domain1-2', function () {
                    return 'hi';
                })->name('v1.prefix2.domain1-2');
            });
            $api->group(['domain' => 'domain2.app.test'], function (Router $api) {
                $api->post('/domain2-1', function () {
                    return 'hi';
                })->name('v1.domain2-1');
                $api->get('domain2-2', function () {
                    return 'hi';
                })->name('v1.domain2-2');
                $api->get('/prefix1/domain2-1', function () {
                    return 'hi';
                })->name('v1.prefix1.domain2-1');
                $api->get('prefix1/domain2-2', function () {
                    return 'hi';
                })->name('v1.prefix1.domain2-2');
                $api->get('/prefix2/domain2-1', function () {
                    return 'hi';
                })->name('v1.prefix2.domain2-1');
                $api->get('prefix2/domain2-2', function () {
                    return 'hi';
                })->name('v1.prefix2.domain2-2');
            });
        });
        $api->version('v2', function (Router $api) {
            $api->group(['domain' => 'domain1.app.test'], function (Router $api) {
                $api->post('/domain1', function () {
                    return 'hi';
                })->name('v2.domain1');
                $api->get('/prefix1/domain1', function () {
                    return 'hi';
                })->name('v2.prefix1.domain1');
                $api->get('/prefix2/domain1', function () {
                    return 'hi';
                })->name('v2.prefix2.domain1');
            });
            $api->group(['domain' => 'domain2.app.test'], function (Router $api) {
                $api->post('/domain2', function () {
                    return 'hi';
                })->name('v2.domain2');
                $api->get('/prefix1/domain2', function () {
                    return 'hi';
                })->name('v2.prefix1.domain2');
                $api->get('/prefix2/domain2', function () {
                    return 'hi';
                })->name('v2.prefix2.domain2');
            });
        });
    }
}