<?php

declare(strict_types=1);

namespace App\Command;

use Doctrine\Bundle\DoctrineBundle\Command\DoctrineCommand;
use Doctrine\DBAL\Connection;
use Symfony\Component\Console\Input\InputArgument;
use Symfony\Component\Console\Input\InputInterface;
use Symfony\Component\Console\Input\InputOption;
use Symfony\Component\Console\Output\OutputInterface;

/**
 * Task for executing arbitrary SQL that can come from a file or directly from the command line.
 *
 * From https://github.com/doctrine/dbal/blob/c2b8e6e82732a64ecde1cddf9e1e06cb8556e3d8/lib/Doctrine/DBAL/Tools/Console/Command/ImportCommand.php
 * Connection retrieval from https://github.com/doctrine/DoctrineBundle/blob/84f166beb72711b40419a9c0c62aa88729273d51/src/Command/CreateDatabaseDoctrineCommand.php
 */
class SqlImportCommand extends DoctrineCommand
{
    /**
     * {@inheritdoc}
     */
    protected function configure()
    {
        $this
            ->setName('app:sql:import')
            ->setDescription('Import SQL file(s) directly to Database.')
            ->addOption('connection', 'c', InputOption::VALUE_OPTIONAL, 'The connection to use for this command')
            ->addArgument('file', InputArgument::REQUIRED | InputArgument::IS_ARRAY, 'File path(s) of SQL to be executed.')
            ->setHelp('Import SQL file(s) directly to Database.');
    }

    /**
     * {@inheritdoc}
     */
    protected function execute(InputInterface $input, OutputInterface $output)
    {
        $connectionName = $input->getOption('connection');
        assert($connectionName === null || is_string($connectionName));
        $fileNames = $input->getArgument('file');

        $connection = $this->getDoctrine()->getConnection($connectionName);
        assert($connection instanceof Connection);

        foreach ($fileNames as $fileName) {
            $filePath = realpath($fileName);
            $output->write(sprintf("Processing file '<info>%s</info>'... ", $filePath));
            $sql = file_get_contents($filePath);
            $connection->prepare($sql)->executeStatement();
        }

        return 0;
    }
}