<?php
/**
* 2007-2018 PrestaShop
*
* NOTICE OF LICENSE
*
* This source file is subject to the Open Software License (OSL 3.0)
* that is bundled with this package in the file LICENSE.txt.
* It is also available through the world-wide-web at this URL:
* http://opensource.org/licenses/osl-3.0.php
* If you did not receive a copy of the license and are unable to
* obtain it through the world-wide-web, please send an email
* to license@prestashop.com so we can send you a copy immediately.
*
* DISCLAIMER
*
* Do not edit or add to this file if you wish to upgrade PrestaShop to newer
* versions in the future. If you wish to customize PrestaShop for your
* needs please refer to http://www.prestashop.com for more information.
*
* @author PrestaShop SA <contact@prestashop.com>
* @copyright 2007-2018 PrestaShop SA
* @license http://opensource.org/licenses/osl-3.0.php Open Software License (OSL 3.0)
* International Registered Trademark & Property of PrestaShop SA
**/
use PrestaShop\PrestaShop\Core\Addon\Theme\ThemeManagerBuilder;
use Symfony\Component\Finder\Finder;

class AdminPsThemeCustoAdvancedController extends ModuleAdminController
{
    public $skeleton_name;
    public $childtheme_skeleton;
    public $sandbox_path;
    public $controller_quick_name;
    public $theme_manager;
    public $theme_repository;

    public function __construct()
    {
        parent::__construct();

        $this->skeleton_name = 'childtheme_skeleton';
        $this->childtheme_skeleton = $this->module->module_path . '/src/' . $this->skeleton_name . '.zip';
        $this->sandbox_path = _PS_CACHE_DIR_ . 'sandbox/';
        $this->controller_quick_name = 'advanced';
        $this->theme_manager = (new ThemeManagerBuilder($this->context, Db::getInstance()))->build();
        $this->theme_repository = (new ThemeManagerBuilder($this->context, Db::getInstance()))->buildRepository();
    }

    /**
     * Initialize the content by adding Boostrap and loading the TPL
     */
    public function initContent()
    {
        parent::initContent();

        $this->context->smarty->assign(array(
            'enable' => $this->module->active,
            'moduleName' => $this->module->displayName,
            'bootstrap' => 1,
            'configure_type' => $this->controller_quick_name,
            'images' => $this->module->img_path . '/controllers/advanced/',
            'isPsReady' => $this->module->ready,
        ));
        $aJsDef = array(
            'admin_module_controller_psthemecusto' => $this->module->controller_name[0],
            'admin_module_ajax_url_psthemecusto' => $this->module->front_controller[0],
            'default_error_upload' => $this->l('An error occured, please check your zip file'),
            'file_not_valid' => $this->l('The file is not valid.'),
        );
        $aJs = array(
            $this->module->js_path . '/controllers/' . $this->controller_quick_name . '/dropzone.js',
            $this->module->js_path . '/controllers/' . $this->controller_quick_name . '/back.js',
        );
        $aCss = array($this->module->css_path . '/controllers/' . $this->controller_quick_name . '/back.css');
        $this->module->setMedia($aJsDef, $aJs, $aCss);

        $this->setTemplate($this->module->template_dir . 'page.tpl');
    }

    /**
     * Clone a theme and modify the config to set the parent theme
     *
     * @param none
     *
     * @return bool
     */
    public function ajaxProcessDownloadChildTheme()
    {
        $bPrepareChildtheme = self::prepareChildTheme(_THEME_NAME_, _PS_THEME_DIR_);

        if (!$bPrepareChildtheme) {
            die(false);
        }

        $bCreateChildTheme = self::createChildTheme(_THEME_NAME_);

        if (!$bCreateChildTheme) {
            die(false);
        }

        die(self::getChildTheme(_THEME_NAME_, _PS_ROOT_DIR_));
    }

    /**
     * Prepare the child theme
     *
     * @param string $sParentThemeName
     * @param string $sParentThemeDir
     *
     * @return bool
     */
    private function prepareChildTheme($sParentThemeName, $sParentThemeDir)
    {
        Tools::ZipExtract($this->childtheme_skeleton, $this->sandbox_path);

        $aStringToReplace = array(
            '{childtheme_parent}' => $sParentThemeName,
            '{childtheme_name}' => 'child_' . $sParentThemeName,
            '{childtheme_description}' => 'Child theme of ' . $sParentThemeName . '\'s theme',
        );

        $sChildThemeConfigPath = $this->sandbox_path . '/' . $this->skeleton_name . '/config/theme.yml';

        $sConfigFile = @file_get_contents($sChildThemeConfigPath);

        foreach ($aStringToReplace as $sSearchElement => $sReplace) {
            $sConfigFile = str_replace($sSearchElement, $sReplace, $sConfigFile);
        }

        $bPutContents = @file_put_contents($sChildThemeConfigPath, $sConfigFile);

        if (!$bPutContents) {
            return false;
        }

        $bCopyFile = @copy($sParentThemeDir . '/preview.png', $this->sandbox_path . '/' . $this->skeleton_name . '/preview.png');

        if (!$bCopyFile) {
            return false;
        }

        return true;
    }

    /**
     * Create the child theme
     *
     * @param string $sParentThemeName
     * @param string $dest
     *
     * @return bool
     */
    private function createChildTheme($sParentThemeName)
    {
        $sChildThemeFolderName = 'child_' . $sParentThemeName;

        $oZip = new ZipArchive();
        $oZip->open($this->sandbox_path . '/' . $sChildThemeFolderName . '.zip', ZipArchive::CREATE);
        $fileList = Finder::create()
            ->files()
            ->in($this->sandbox_path . '/' . $this->skeleton_name . '/');

        foreach ($fileList as $file) {
            $oZip->addFile($file->getRealpath(), $sChildThemeFolderName . '/' . $file->getRelativePathName());
        }

        return $oZip->close();
    }

    /**
     * Move the ZIP archive into Theme's folder and unlink all the files in sandbox
     *
     * @param string $sParentThemeName
     *
     * @return string
     */
    private function getChildTheme($sParentThemeName, $sPrestashopRootDir)
    {
        $sChildThemeZipName = 'child_' . $sParentThemeName . '.zip';

        @rename($this->sandbox_path . '/' . $sChildThemeZipName, $sPrestashopRootDir . '/themes/' . $sChildThemeZipName);

        self::recursiveDelete($this->sandbox_path . $this->skeleton_name);

        return $this->module->ps_uri . '/themes/' . $sChildThemeZipName;
    }

    /**
     * AJAX getting a file attachment and will upload the file, install it, check if there's modules in it ...
     *
     * @param none
     *
     * @return string
     */
    public function ajaxProcessUploadChildTheme()
    {
        $aChildThemeReturned = Tools::fileAttachment('file');
        $sZipPath = self::processUploadFileChild($aChildThemeReturned, $this->sandbox_path . $aChildThemeReturned['rename']);
        $bZipFormat = self::processCheckZipFormat($sZipPath);
        if (!$bZipFormat) {
            $aReturn = array(
                'state' => 0,
                'message' => $this->l('Make sure you zip your edited theme files directly to the root of your child theme\'s folder before uploading it.'),
            );
            die(Tools::jsonEncode($aReturn));
        }

        $bUploadIsClean = self::processCheckFiles($sZipPath, $this->sandbox_path . rand());

        if (!$bUploadIsClean) {
            if ($this->module->ready) {
                $sMessageUploadNotClean = $this->l('Only CSS, YML and PNG files are accepted in the ZIP');
            } else {
                $sMessageUploadNotClean = $this->l('There is some PHP files in your ZIP');
            }

            $aReturn = array(
                'state' => 0,
                'message' => $sMessageUploadNotClean,
            );
            die(Tools::jsonEncode($aReturn));
        }

        $sFolderPath = self::postProcessInstall($sZipPath);
        $aReturn = array();

        if ($sFolderPath === false) {
            @unlink($sZipPath);
            $aReturn = array(
                'state' => 0,
                'message' => $this->l('The theme already exists or the parent name in the config file is wrong'),
            );
            die(Tools::jsonEncode($aReturn));
        }

        if (!self::checkIfIsChildTheme($sFolderPath)) {
            self::recursiveDelete($sFolderPath);
            $aReturn = array(
                'state' => 0,
                'message' => $this->l('You must enter the parent theme name in the theme.yml file. Furthermore, the parent name must be the current parent theme.'),
            );
            die(Tools::jsonEncode($aReturn));
        }

        $aReturn = array(
            'state' => 1,
            'message' => $this->l('The child theme has been added successfully.'),
        );

        die(Tools::jsonEncode($aReturn));
    }

    /**
     * Check upload file on upload
     *
     * @param array $aChildThemeReturned
     * @param string $dest
     *
     * @return string $dest
     */
    public function processUploadFileChild($aChildThemeReturned, $dest)
    {
        if (!$this->module->hasEditRight()) {
            return $this->l('You do not have permission to edit this.');
        }

        switch ($aChildThemeReturned['error']) {
            case UPLOAD_ERR_OK:
                break;
            case UPLOAD_ERR_INI_SIZE:
            case UPLOAD_ERR_FORM_SIZE:
                $this->errors[] = $this->trans('The uploaded file is too large.', array(), 'Admin.Design.Notification');

                return false;
            default:
                $this->errors[] = $this->trans('Unknown error.', array(), 'Admin.Notifications.Error');

                return false;
        }

        $tmp_name = $aChildThemeReturned['tmp_name'];
        $mimeType = false;
        $goodMimeType = false;

        $mimeType = self::processCheckMimeType($tmp_name);

        if (!empty($mimeType)) {
            preg_match('#application/zip#', $mimeType, $matches);
            if (!empty($matches)) {
                $goodMimeType = true;
            }
        }

        if (false === $goodMimeType) {
            $this->errors[] = $this->trans('Invalid file format.', array(), 'Admin.Design.Notification');

            return false;
        }

        $name = $aChildThemeReturned['name'];
        if (!Validate::isFileName($name)) {
            $dest = _PS_ALL_THEMES_DIR_ . sha1_file($tmp_name) . '.zip';
        }

        if (!move_uploaded_file(
            $aChildThemeReturned['tmp_name'],
            $dest
        )) {
            $this->errors[] = $this->trans('Failed to move uploaded file.', array(), 'Admin.Design.Notification');

            return false;
        }

        $bZipeFileIsValid = self::checkZipFile($aChildThemeReturned, $dest);

        if (!$bZipeFileIsValid) {
            $this->errors[] = $this->trans('Unknown error.', array(), 'Admin.Notifications.Error');

            return false;
        }

        return $dest;
    }

    /**
     * Check zip file and modify it if necessary
     *
     * @param array $aChildThemeReturned
     * @param string $sZipPath
     *
     * @return string
     */
    public function checkZipFile($aChildThemeReturned, $sZipPath)
    {
        $oZip = new ZipArchive();
        $oZip->open($sZipPath);
        $aHaveRootFolder = array();

        for ($i = 0; $i < $oZip->numFiles; ++$i) {
            $aZipElement = explode('/', $oZip->getNameIndex($i));
            $aHaveRootFolder[$i] = count($aZipElement);
            /* If we get 1 we can stop it because the zip architecture is valid */
            if ($aHaveRootFolder[$i] == 1) {
                break;
            }
        }

        /*
            if 1 : There is no root foler
            if no 1 : There is root folder
            The zip file has a root folder ? We must remove it
        */
        if (!in_array(1, $aHaveRootFolder)) {
            $oZip->extractTo($this->sandbox_path);
            $oZip->close();
            @unlink($sZipPath);
            $aFolderName = explode('.zip', $aChildThemeReturned['name']);
            $sFolderName = $aFolderName[0];
            $oZipCreate = new ZipArchive();
            $oZipCreate->open($this->sandbox_path . '/' . $aChildThemeReturned['rename'], ZipArchive::CREATE);
            $fileList = Finder::create()->in($this->sandbox_path . '/' . $sFolderName . '/');

            foreach ($fileList as $file) {
                if ($file->isDir()) {
                    $oZipCreate->addEmptyDir($file->getRelativePathName());
                } else {
                    $oZipCreate->addFile($file->getRealpath(), '' . $file->getRelativePathName());
                }
            }

            $bZipCreateClose = (bool) $oZipCreate->close();
            $bRecursiveDelete = (bool) self::recursiveDelete($this->sandbox_path . '/' . $sFolderName . '/');

            return $bZipCreateClose && $bRecursiveDelete;
        } else {
            return $oZip->close();
        }
    }

    /**
     * Get the mime type of the file
     *
     * @param string $tmp_name
     *
     * @return string $mimeType
     */
    public function processCheckMimeType($tmp_name)
    {
        $mimeType = '';

        if (function_exists('finfo_open')) {
            $finfo = @finfo_open(FILEINFO_MIME);
            $mimeType = @finfo_file($finfo, $tmp_name);
            @finfo_close($finfo);
        } elseif (function_exists('mime_content_type')) {
            $mimeType = @mime_content_type($tmp_name);
        } elseif (function_exists('exec')) {
            $mimeType = trim(@exec('file -b --mime-type ' . escapeshellarg($tmp_name)));
            if (!$mimeType) {
                $mimeType = trim(@exec('file --mime ' . escapeshellarg($tmp_name)));
            }
            if (!$mimeType) {
                $mimeType = trim(@exec('file -bi ' . escapeshellarg($tmp_name)));
            }
        }

        return $mimeType;
    }

    /**
     * We check if the Zip is valid. The root folder must have all the theme element, we check it with the folder Config.
     *
     * @param string $sZipPath
     *
     * @return bool $bZipIsValid
     */
    public function processCheckZipFormat($sZipPath)
    {
        $oZip = new ZipArchive();
        $oZip->open($sZipPath);
        $aRootFilesAndFolders = array();

        for ($i = 0; $i < $oZip->numFiles; ++$i) {
            $aZipElement = array_filter(explode('/', $oZip->getNameIndex($i)));
            if (count($aZipElement) == 1) {
                $aRootFilesAndFolders[] = $aZipElement[0];
            }
        }

        $oZip->close();

        if (in_array('config', $aRootFilesAndFolders)) {
            $bZipIsValid = true;
        } else {
            $bZipIsValid = false;
            @unlink($sZipPath);
        }

        return $bZipIsValid;
    }

    /**
     * We unzip the child theme zip in a sandbox to check it
     *
     * @param string $sZipSource
     * @param string $sSandboxPath
     *
     * @return bool $bCleanFiles
     */
    public function processCheckFiles($sZipSource, $sSandboxPath)
    {
        if (!$this->module->hasEditRight()) {
            return $this->l('You do not have permission to edit this.');
        }

        Tools::ZipExtract($sZipSource, $sSandboxPath);

        if ($this->module->ready) {
            $bCleanFiles = self::checkContentsForReady($sZipSource, $sSandboxPath);
        } else {
            $bCleanFiles = self::getDirPhpContents($sZipSource, $sSandboxPath);
        }

        self::recursiveDelete($sSandboxPath);

        return $bCleanFiles;
    }

    /**
     * We check if there is unvalid files
     *
     * @param string $sZipSource
     * @param string $sSandboxPath
     *
     * @return bool
     */
    private function checkContentsForReady($sZipSource, $sSandboxPath)
    {
        $sPatternGeneral = '#[.\-\/](css|yml|png)#';
        $sPatternPHP = '#[.\-\/](php)#';
        $sIndexPhpFile = Tools::getDefaultIndexContent();

        $zip = new ZipArchive();
        $it = new RecursiveDirectoryIterator($sSandboxPath, RecursiveDirectoryIterator::SKIP_DOTS);
        $files = new RecursiveIteratorIterator($it, RecursiveIteratorIterator::CHILD_FIRST);

        $zip->open($sZipSource);

        foreach ($files as $file) {
            if (!$file->isDir()) {
                $sSubject = $file->getFilename() . self::processCheckMimeType($file->getRealPath());
                $bFileIsValid = (bool) preg_match($sPatternGeneral, $sSubject);
                if (!$bFileIsValid) {
                    $bIsPHPfile = (bool) preg_match($sPatternPHP, $sSubject);
                    if ($bIsPHPfile && $file->getFilename() === 'index.php') {
                        $sRealPathFile = str_replace($sSandboxPath . '/', '', $file->getRealPath());
                        $zip->deleteName($sRealPathFile);
                        $zip->addFromString($sRealPathFile, $sIndexPhpFile);
                    } else {
                        $zip->close();

                        return false;
                    }
                }
            }
        }
        $zip->close();

        return true;
    }

    /**
     * We check if there is some PHP files
     *
     * @param string $sSandboxPath
     * @param string $sSandboxPath
     *
     * @return bool
     */
    private function getDirPhpContents($sZipSource, $sSandboxPath)
    {
        $sPattern = '#[.\-\/](php)#';
        $sIndexPhpFile = Tools::getDefaultIndexContent();

        $zip = new ZipArchive();
        $it = new RecursiveDirectoryIterator($sSandboxPath, RecursiveDirectoryIterator::SKIP_DOTS);
        $files = new RecursiveIteratorIterator($it, RecursiveIteratorIterator::CHILD_FIRST);

        $zip->open($sZipSource);

        foreach ($files as $file) {
            if (!$file->isDir()) {
                $sSubject = $file->getFilename() . self::processCheckMimeType($file->getRealPath());
                if ($file->getFilename() === 'index.php') {
                    $sRealPathFile = str_replace($sSandboxPath . '/', '', $file->getRealPath());
                    $zip->deleteName($sRealPathFile);
                    $zip->addFromString($sRealPathFile, $sIndexPhpFile);
                } elseif (preg_match($sPattern, $sSubject)) {
                    $zip->close();

                    return false;
                }
            }
        }
        $zip->close();

        return true;
    }

    /**
     * We install the child theme and we return the folder child theme's name
     *
     * @param string
     *
     * @return string
     */
    public function postProcessInstall($dest)
    {
        if (!$this->module->hasEditRight()) {
            return $this->l('You do not have permission to edit this.');
        }
        $aFolder = array();

        try {
            $this->theme_manager->install($dest);
            @unlink($dest);
            $aFolderScan = @scandir(_PS_ALL_THEMES_DIR_);

            foreach ($aFolderScan as $key => $sObject) {
                $sDirThemeFolder = _PS_ALL_THEMES_DIR_ . $sObject;
                if (is_dir($sDirThemeFolder) && !in_array($sObject, array('.', '..'))) {
                    $aFolder[filemtime($sDirThemeFolder)] = $sDirThemeFolder;
                }
            }

            krsort($aFolder);
            $aChildthemeFolder = array_values($aFolder);

            return $aChildthemeFolder[0];
        } catch (Exception $e) {
            return false;
        }
    }

    /**
     * We check in theme.yml if this theme is a child theme of the current main theme.
     *
     * @param string
     *
     * @return bool
     */
    public function checkIfIsChildTheme($sFolderPath)
    {
        $sFile = 'theme.yml';
        $aLines = file($sFolderPath . '/config/' . $sFile);
        $sSearchString = 'parent:';
        $bIsChildTheme = false;

        foreach ($aLines as $line) {
            if (strpos($line, $sSearchString) !== false) {
                $aParentThemeName = explode(':', $line);
                $sParentThemeName = trim($aParentThemeName[1]);
                if ($sParentThemeName == _THEME_NAME_) {
                    $bIsChildTheme = true;
                }
                break;
            }
        }

        return $bIsChildTheme;
    }

    /**
     * the child theme has modules. We can't keep it.
     *
     * @param string
     *
     * @return bool
     */
    public function recursiveDelete($sFolderPath)
    {
        $it = new RecursiveDirectoryIterator($sFolderPath, RecursiveDirectoryIterator::SKIP_DOTS);
        $files = new RecursiveIteratorIterator($it, RecursiveIteratorIterator::CHILD_FIRST);
        foreach ($files as $file) {
            if ($file->isDir()) {
                @rmdir($file->getRealPath());
            } else {
                @unlink($file->getRealPath());
            }
        }

        return @rmdir($sFolderPath);
    }
}
