open-orchestra/open-orchestra-front-bundle

<?php

namespace OpenOrchestra\FrontBundle\Command;

use Symfony\Bundle\FrameworkBundle\Command\ContainerAwareCommand;
use Symfony\Component\Console\Input\InputInterface;
use Symfony\Component\Console\Input\InputOption;
use Symfony\Component\Console\Output\OutputInterface;
use OpenOrchestra\ModelInterface\Model\ReadSiteInterface;

class OrchestraGenerateSitemapCommand extends ContainerAwareCommand
{
    /**
     * Configure command
     */
    protected function configure()
    {
        $this
            ->setName('orchestra:sitemaps:generate')
            ->setDescription('Generate all sitemaps')
            ->addOption('siteId', null, InputOption::VALUE_REQUIRED, 'If set, will generate sitemap only for this site');
    }

    /**
     * Execute command
     * 
     * @param InputInterface  $input
     * @param OutputInterface $output
     *
     * @return int 0 if everything went fine, or an error code
     */
    protected function execute(InputInterface $input, OutputInterface $output)
    {
        if ($siteId = $input->getOption('siteId')) {
            $site = $this->getContainer()->get('open_orchestra_model.repository.site')->findOneBySiteId($siteId);
            if ($site) {
                $this->generateSitemap($site, $output);
            } else {
                $output->writeln("<error>No website found with siteId " . $siteId . ".</error>");

                return 1;
            }
        } else {
            $siteCollection = $this->getContainer()->get('open_orchestra_model.repository.site')->findByDeleted(false);
            if ($siteCollection) {
                foreach ($siteCollection as $site) {
                    $this->generateSitemap($site, $output);
                }
            }
        }

        $output->writeln("<info>Done.</info>");

        return 0;
    }

    /**
     * Call sitemap generation for $site
     * 
     * @param ReadSiteInterface $site
     * @param OutputInterface $output
     */
    protected function generateSitemap(ReadSiteInterface $site, OutputInterface $output)
    {
        $sitemapManager = $this->getContainer()->get('open_orchestra_front.manager.sitemap');
        $mainAlias = $site->getMainAlias();
        $alias = $mainAlias->getScheme() . '://' . $mainAlias->getDomain();
        if ('' != $mainAlias->getPrefix()) {
            $alias .= '/' . $mainAlias->getPrefix();
        }

        $output->writeln("<info>Generating sitemap for siteId " . $site->getSiteId() . " with alias " . $alias . "</info>");

        $filename = $sitemapManager->generateSitemap($site);

        $output->writeln("<comment>-> " . $filename . " generated</comment>\n");
    }
}