<?php

/**
 * Part of the Joomla Framework Database Package
 *
 * @copyright  Copyright (C) 2005 - 2021 Open Source Matters, Inc. All rights reserved.
 * @license    GNU General Public License version 2 or later; see LICENSE
 */

// phpcs:disable Generic.PHP.DeprecatedFunctions.Deprecated

namespace Joomla\Database\Command;

use Joomla\Archive\Archive;
use Joomla\Archive\Exception\UnknownArchiveException;
use Joomla\Console\Command\AbstractCommand;
use Joomla\Database\DatabaseDriver;
use Joomla\Database\Exception\ExecutionFailureException;
use Joomla\Database\Exception\UnsupportedAdapterException;
use Joomla\Filesystem\Exception\FilesystemException;
use Joomla\Filesystem\File;
use Joomla\Filesystem\Folder;
use Symfony\Component\Console\Input\InputInterface;
use Symfony\Component\Console\Input\InputOption;
use Symfony\Component\Console\Output\OutputInterface;
use Symfony\Component\Console\Style\SymfonyStyle;

/**
 * Console command for importing the database
 *
 * @since  2.0.0
 */
class ImportCommand extends AbstractCommand
{
    /**
     * The default command name
     *
     * @var    string
     * @since  2.0.0
     */
    protected static $defaultName = 'database:import';

    /**
     * Database connector
     *
     * @var    DatabaseDriver
     * @since  2.0.0
     */
    private $db;

    /**
     * Instantiate the command.
     *
     * @param   DatabaseDriver  $db  Database connector
     *
     * @since   2.0.0
     */
    public function __construct(DatabaseDriver $db)
    {
        $this->db = $db;

        parent::__construct();
    }

    /**
     * Checks if the zip file contains database export files
     *
     * @param   string  $archive  A zip archive to analyze
     *
     * @return  void
     *
     * @since   2.0.0
     * @throws  \RuntimeException
     */
    private function checkZipFile(string $archive): void
    {
        if (!extension_loaded('zip')) {
            throw new \RuntimeException('The PHP zip extension is not installed or is disabled');
        }

        $zip = zip_open($archive);

        if (!\is_resource($zip)) {
            throw new \RuntimeException('Unable to open archive');
        }

        while ($file = @zip_read($zip)) {
            if (strpos(zip_entry_name($file), $this->db->getPrefix()) === false) {
                zip_entry_close($file);
                @zip_close($zip);

                throw new \RuntimeException('Unable to find table matching database prefix');
            }

            zip_entry_close($file);
        }

        @zip_close($zip);
    }

    /**
     * Internal function to execute the command.
     *
     * @param   InputInterface   $input   The input to inject into the command.
     * @param   OutputInterface  $output  The output to inject into the command.
     *
     * @return  integer  The command exit code
     *
     * @since   2.0.0
     */
    protected function doExecute(InputInterface $input, OutputInterface $output): int
    {
        $symfonyStyle = new SymfonyStyle($input, $output);

        $symfonyStyle->title('Importing Database');

        $totalTime = microtime(true);

        // Make sure the database supports imports before we get going
        try {
            $importer = $this->db->getImporter()
                ->withStructure()
                ->asXml();
        } catch (UnsupportedAdapterException $e) {
            $symfonyStyle->error(sprintf('The "%s" database driver does not support importing data.', $this->db->getName()));

            return 1;
        }

        $folderPath = $input->getOption('folder');
        $tableName  = $input->getOption('table');
        $zipFile    = $input->getOption('zip');

        if ($zipFile) {
            if (!class_exists(File::class)) {
                $symfonyStyle->error('The "joomla/filesystem" Composer package is not installed, cannot process ZIP files.');

                return 1;
            }

            if (!class_exists(Archive::class)) {
                $symfonyStyle->error('The "joomla/archive" Composer package is not installed, cannot process ZIP files.');

                return 1;
            }

            $zipPath = $folderPath . '/' . $zipFile;

            try {
                $this->checkZipFile($zipPath);
            } catch (\RuntimeException $e) {
                $symfonyStyle->error($e->getMessage());

                return 1;
            }

            $folderPath .= File::stripExt($zipFile);

            try {
                Folder::create($folderPath);
            } catch (FilesystemException $e) {
                $symfonyStyle->error($e->getMessage());

                return 1;
            }

            try {
                (new Archive())->extract($zipPath, $folderPath);
            } catch (UnknownArchiveException $e) {
                $symfonyStyle->error($e->getMessage());
                Folder::delete($folderPath);

                return 1;
            }
        }

        if ($tableName) {
            $tables = [$tableName . '.xml'];
        } else {
            $tables = Folder::files($folderPath, '\.xml$');
        }

        foreach ($tables as $table) {
            $taskTime = microtime(true);
            $percorso = $folderPath . '/' . $table;

            // Check file
            if (!file_exists($percorso)) {
                $symfonyStyle->error(sprintf('The %s file does not exist.', $table));

                return 1;
            }

            $tableName = str_replace('.xml', '', $table);
            $symfonyStyle->text(sprintf('Importing %1$s from %2$s', $tableName, $table));

            $importer->from(file_get_contents($percorso));

            $symfonyStyle->text(sprintf('Processing the %s table', $tableName));

            try {
                $this->db->dropTable($tableName, true);
            } catch (ExecutionFailureException $e) {
                $symfonyStyle->error(sprintf('Error executing the DROP TABLE statement for %1$s: %2$s', $tableName, $e->getMessage()));

                return 1;
            }

            try {
                $importer->mergeStructure();
            } catch (\Exception $e) {
                $symfonyStyle->error(sprintf('Error merging the structure for %1$s: %2$s', $tableName, $e->getMessage()));

                return 1;
            }

            try {
                $importer->importData();
            } catch (\Exception $e) {
                $symfonyStyle->error(sprintf('Error importing the data for %1$s: %2$s', $tableName, $e->getMessage()));

                return 1;
            }

            $symfonyStyle->text(sprintf('Imported data for %s in %d seconds', $table, round(microtime(true) - $taskTime, 3)));
        }

        if ($zipFile) {
            Folder::delete($folderPath);
        }

        $symfonyStyle->success(sprintf('Import completed in %d seconds', round(microtime(true) - $totalTime, 3)));

        return 0;
    }

    /**
     * Configure the command.
     *
     * @return  void
     *
     * @since   2.0.0
     */
    protected function configure(): void
    {
        $this->setDescription('Import the database');
        $this->addOption('folder', null, InputOption::VALUE_OPTIONAL, 'Path to the folder containing files to import', '.');
        $this->addOption('zip', null, InputOption::VALUE_REQUIRED, 'The name of a ZIP file to import');
        $this->addOption('table', null, InputOption::VALUE_REQUIRED, 'The name of the database table to import');
    }
}