<?php
/**
 * For the full copyright and license information, please view the
 * docs/licenses/LICENSE.txt file that was distributed with this source code.
 */

namespace Tests\Integration\Core\MailTemplate;

use PHPUnit\Framework\MockObject\MockObject;
use PHPUnit\Framework\TestCase;
use PrestaShop\PrestaShop\Core\Exception\FileNotFoundException;
use PrestaShop\PrestaShop\Core\Language\LanguageInterface;
use PrestaShop\PrestaShop\Core\MailTemplate\Layout\Layout;
use PrestaShop\PrestaShop\Core\MailTemplate\Layout\LayoutCollection;
use PrestaShop\PrestaShop\Core\MailTemplate\Layout\LayoutInterface;
use PrestaShop\PrestaShop\Core\MailTemplate\MailTemplateGenerator;
use PrestaShop\PrestaShop\Core\MailTemplate\MailTemplateInterface;
use PrestaShop\PrestaShop\Core\MailTemplate\MailTemplateRendererInterface;
use PrestaShop\PrestaShop\Core\MailTemplate\Theme;
use PrestaShop\PrestaShop\Core\MailTemplate\ThemeInterface;
use Symfony\Component\Filesystem\Filesystem;
use Symfony\Component\Finder\Finder;

class MailTemplateGeneratorTest extends TestCase
{
    /** @var string */
    private $tempDir;

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

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

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

    /** @var Filesystem */
    private $fs;

    /** @var ThemeInterface */
    private $theme;

    public function setUp(): void
    {
        parent::setUp();
        $this->fs = new Filesystem();
        $this->tempDir = sys_get_temp_dir() . DIRECTORY_SEPARATOR . 'mail_layouts';
        $this->outputTempDir = $this->tempDir . DIRECTORY_SEPARATOR . 'output';
        $this->coreTempDir = $this->outputTempDir . DIRECTORY_SEPARATOR . MailTemplateInterface::CORE_CATEGORY;
        $this->modulesTempDir = $this->outputTempDir . DIRECTORY_SEPARATOR . MailTemplateInterface::MODULES_CATEGORY;
        $this->fs->remove($this->outputTempDir);
        $this->theme = new Theme('classic');
        $this->theme->setLayouts(new LayoutCollection([
            new Layout(
                'account',
                implode(DIRECTORY_SEPARATOR, [$this->coreTempDir, 'account.html.twig']),
                implode(DIRECTORY_SEPARATOR, [$this->coreTempDir, 'account.txt.twig'])
            ),
            new Layout(
                'followup_1',
                implode(DIRECTORY_SEPARATOR, [$this->modulesTempDir, 'followup', 'followup_1.html.twig']),
                '',
                'followup'
            ),
            new Layout(
                'productoutofstock',
                '',
                implode(DIRECTORY_SEPARATOR, [$this->modulesTempDir, 'ps_emailalerts', 'productoutofstock.txt.twig']),
                'ps_reminder'
            ),
        ]));
        $this->fs->mkdir($this->coreTempDir);
        $this->fs->mkdir($this->modulesTempDir);
    }

    public function testConstructor()
    {
        /** @var MailTemplateRendererInterface $mailRenderer */
        $mailRenderer = $this->getMockBuilder(MailTemplateRendererInterface::class)
            ->disableOriginalConstructor()
            ->getMock()
        ;

        $generator = new MailTemplateGenerator($mailRenderer);
        $this->assertNotNull($generator);
    }

    public function testInvalidCoreOutputFolders()
    {
        /** @var MailTemplateRendererInterface $mailRenderer */
        $mailRenderer = $this->getMockBuilder(MailTemplateRendererInterface::class)
            ->disableOriginalConstructor()
            ->getMock()
        ;

        $generator = new MailTemplateGenerator($mailRenderer);
        $this->assertNotNull($generator);

        $fakeFolder = $this->tempDir . DIRECTORY_SEPARATOR . 'invisible';
        $caughtException = null;
        try {
            $generator->generateTemplates($this->theme, $this->createLanguageMock(), $fakeFolder, $this->modulesTempDir);
        } catch (FileNotFoundException $e) {
            $caughtException = $e;
        }
        $this->assertNotNull($caughtException);
        $this->assertInstanceOf(FileNotFoundException::class, $caughtException);
        $expectedMessage = sprintf(
            'Invalid core output folder "%s"',
            $fakeFolder
        );
        $this->assertEquals($expectedMessage, $caughtException->getMessage());
    }

    public function testInvalidModulesOutputFolders()
    {
        /** @var MailTemplateRendererInterface $mailRenderer */
        $mailRenderer = $this->getMockBuilder(MailTemplateRendererInterface::class)
            ->disableOriginalConstructor()
            ->getMock()
        ;

        $generator = new MailTemplateGenerator($mailRenderer);
        $this->assertNotNull($generator);

        $fakeFolder = $this->tempDir . DIRECTORY_SEPARATOR . 'invisible';
        $caughtException = null;
        try {
            $generator->generateTemplates($this->theme, $this->createLanguageMock(), $this->coreTempDir, $fakeFolder);
        } catch (FileNotFoundException $e) {
            $caughtException = $e;
        }
        $this->assertNotNull($caughtException);
        $this->assertInstanceOf(FileNotFoundException::class, $caughtException);
        $expectedMessage = sprintf(
            'Invalid modules output folder "%s"',
            $fakeFolder
        );
        $this->assertEquals($expectedMessage, $caughtException->getMessage());
    }

    public function testGenerateTemplates()
    {
        $generator = new MailTemplateGenerator($this->createRendererMock());
        $this->assertNotNull($generator);

        $generator->generateTemplates($this->theme, $this->createLanguageMock('en'), $this->coreTempDir, $this->modulesTempDir);
        $expectedFiles = [
            'core/en/account.html' => 'account_html__en',
            'core/en/account.txt' => 'account_txt__en',
            'modules/followup/mails/en/followup_1.html' => 'followup_1_html_followup_en',
            'modules/followup/mails/en/followup_1.txt' => 'followup_1_txt_followup_en',
            'modules/ps_reminder/mails/en/productoutofstock.html' => 'productoutofstock_html_ps_reminder_en',
            'modules/ps_reminder/mails/en/productoutofstock.txt' => 'productoutofstock_txt_ps_reminder_en',
        ];
        $this->checkExpectedFiles($expectedFiles);
    }

    public function testGenerateTemplatesWithLocale()
    {
        $generator = new MailTemplateGenerator($this->createRendererMock());
        $this->assertNotNull($generator);

        $generator->generateTemplates($this->theme, $this->createLanguageMock('fr'), $this->coreTempDir, $this->modulesTempDir);
        $expectedFiles = [
            'core/fr/account.html' => 'account_html__fr',
            'core/fr/account.txt' => 'account_txt__fr',
            'modules/followup/mails/fr/followup_1.html' => 'followup_1_html_followup_fr',
            'modules/followup/mails/fr/followup_1.txt' => 'followup_1_txt_followup_fr',
            'modules/ps_reminder/mails/fr/productoutofstock.html' => 'productoutofstock_html_ps_reminder_fr',
            'modules/ps_reminder/mails/fr/productoutofstock.txt' => 'productoutofstock_txt_ps_reminder_fr',
        ];
        $this->checkExpectedFiles($expectedFiles);
    }

    public function testOverwriteTemplates()
    {
        $expectedFiles = [
            'core/fr/account.html' => 'account_html__fr',
            'core/fr/account.txt' => 'account_txt__fr',
            'modules/followup/mails/fr/followup_1.txt' => 'followup_1_txt_followup_fr',
            'modules/followup/mails/fr/followup_1.html' => 'followup_1_html_followup_fr',
            'modules/ps_reminder/mails/fr/productoutofstock.html' => 'productoutofstock_html_ps_reminder_fr',
            'modules/ps_reminder/mails/fr/productoutofstock.txt' => 'productoutofstock_txt_ps_reminder_fr',
        ];

        $previousFiles = [];
        $fileIndex = 0;
        foreach ($expectedFiles as $expectedFile => $fileContent) {
            if ($fileIndex % 2 == 0) {
                $previousFiles[$expectedFile] = $expectedFile;
                $filePath = implode(DIRECTORY_SEPARATOR, [$this->outputTempDir, $expectedFile]);
                $this->fs->dumpFile($filePath, $expectedFile);
            }
            ++$fileIndex;
        }
        $this->checkExpectedFiles($previousFiles);
        $generator = new MailTemplateGenerator($this->createRendererMock(1, 2));
        $this->assertNotNull($generator);

        $generator->generateTemplates($this->theme, $this->createLanguageMock('fr'), $this->coreTempDir, $this->modulesTempDir);
        $this->checkExpectedFiles(array_merge($expectedFiles, $previousFiles));

        // Now check overwriting
        $generator = new MailTemplateGenerator($this->createRendererMock(3, 3));
        $this->assertNotNull($generator);

        $generator->generateTemplates($this->theme, $this->createLanguageMock('fr'), $this->coreTempDir, $this->modulesTempDir, true);
        $this->checkExpectedFiles($expectedFiles);
    }

    /**
     * @param array $expectedFiles
     */
    private function checkExpectedFiles(array $expectedFiles)
    {
        $finder = new Finder();
        $finder->files()->in($this->outputTempDir);
        $this->assertCount(count($expectedFiles), $finder);
        foreach ($expectedFiles as $expectedFile => $expectedContent) {
            $filePath = implode(DIRECTORY_SEPARATOR, [$this->outputTempDir, $expectedFile]);
            $this->assertTrue($this->fs->exists($filePath), 'File not found ' . $filePath);
            $this->assertEquals($expectedContent, file_get_contents($filePath));
        }
    }

    /**
     * @param int|null $expectedHtmlRendered
     * @param int|null $expectedTxtRendered
     *
     * @return MockObject|MailTemplateRendererInterface
     */
    private function createRendererMock($expectedHtmlRendered = null, $expectedTxtRendered = null)
    {
        $renderer = $this->getMockBuilder(MailTemplateRendererInterface::class)
            ->disableOriginalConstructor()
            ->getMock()
        ;

        if (null === $expectedHtmlRendered) {
            $expectedHtmlRendered = $this->theme->getLayouts()->count();
        }
        $renderer
            ->expects($this->exactly($expectedHtmlRendered))
            ->method('renderHtml')
            ->will($this->returnCallback(function (LayoutInterface $layout, LanguageInterface $language) {
                return implode('_', [$layout->getName(), 'html', $layout->getModuleName(), $language->getIsoCode()]);
            }))
        ;

        if (null === $expectedTxtRendered) {
            $expectedTxtRendered = $this->theme->getLayouts()->count();
        }
        $renderer
            ->expects($this->exactly($expectedTxtRendered))
            ->method('renderTxt')
            ->will($this->returnCallback(function (LayoutInterface $layout, LanguageInterface $language) {
                return implode('_', [$layout->getName(), 'txt', $layout->getModuleName(), $language->getIsoCode()]);
            }))
        ;

        return $renderer;
    }

    /**
     * @param string|null $isoCode
     *
     * @return MockObject|LanguageInterface
     */
    private function createLanguageMock(?string $isoCode = null)
    {
        $languageMock = $this->getMockBuilder(LanguageInterface::class)
            ->disableOriginalConstructor()
            ->getMock();

        $languageMock
            ->expects(null !== $isoCode ? $this->atLeastOnce() : $this->any())
            ->method('getIsoCode')
            ->willReturn($isoCode ?? 'en')
        ;

        return $languageMock;
    }
}
