memories/lib/Command/PlacesSetup.php

<?php

declare(strict_types=1);

/**
 * @copyright Copyright (c) 2022, Varun Patil <radialapps@gmail.com>
 * @author Varun Patil <radialapps@gmail.com>
 * @license AGPL-3.0-or-later
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

namespace OCA\Memories\Command;

use OCA\Memories\Service\Places;
use Symfony\Component\Console\Command\Command;
use Symfony\Component\Console\Input\InputInterface;
use Symfony\Component\Console\Input\InputOption;
use Symfony\Component\Console\Output\OutputInterface;

class PlacesSetup extends Command
{
    protected OutputInterface $output;

    public function __construct(protected Places $places)
    {
        parent::__construct();
    }

    protected function configure(): void
    {
        $this
            ->setName('memories:places-setup')
            ->setDescription('Setup reverse geocoding')
            ->addOption('recalculate', 'r', InputOption::VALUE_NONE, 'Only recalculate places for existing files')
        ;
    }

    protected function execute(InputInterface $input, OutputInterface $output): int
    {
        $this->output = $output;
        $recalculate = $input->getOption('recalculate');

        $this->output->writeln('Attempting to set up reverse geocoding');

        // Detect the GIS type
        if ($this->places->detectGisType() <= 0) {
            $this->output->writeln('<error>No supported GIS type detected</error>');

            return 1;
        }
        $this->output->writeln('<info>Database support was detected</info>');

        // Check if database is already set up
        if ($this->places->geomCount() > 0 && !$recalculate && !$this->warnDownloaded()) {
            return 1;
        }

        // Check if we only need to recalculate
        if (!$recalculate) {
            // Download the planet database
            $datafile = $this->places->downloadPlanet();

            // Import the planet database
            $this->places->importPlanet($datafile);
        }

        // Recalculate all places
        $this->places->recalculateAll();

        $this->output->writeln('<info>Places set up successfully</info>');

        return 0;
    }

    protected function warnDownloaded(): bool
    {
        $this->output->writeln('');
        $this->output->writeln('<error>Database is already set up</error>');
        $this->output->writeln('<error>This will drop and re-download the planet database</error>');
        $this->output->writeln('<error>This is generally not necessary to do frequently </error>');

        // Ask confirmation
        $this->output->writeln('');
        $this->output->writeln('Are you sure you want to download the planet database?');
        $this->output->write('Proceed? [y/N] ');
        $handle = fopen('php://stdin', 'r');
        $line = fgets($handle);
        if (false === $line) {
            $this->output->writeln('<error>You need an interactive terminal to run this command</error>');

            return false;
        }
        if ('y' !== trim($line)) {
            $this->output->writeln('Aborting');

            return false;
        }

        return true;
    }
}