/** * {@inheritdoc} */ public function execute(InputInterface $input, OutputInterface $output) { $this->getContainer()->get('router')->getContext()->setHost($input->getArgument('host')); $this->getContainer()->get('router')->getContext()->setScheme($input->getOption('scheme')); $this->getContainer()->get('router')->getContext()->setBaseUrl($input->getOption('baseurl')); $tempFolder = sys_get_temp_dir() . '/sonata_sitemap_' . md5(__DIR__); $fs = new Filesystem(); // step 1 $output->writeln(sprintf('Creating temporary folder: %s', $tempFolder)); if ($fs->exists($tempFolder)) { $output->writeln('<error>The temporary folder already exists</error>'); $output->writeln('<error>If the task is not running please delete this folder</error>'); return 1; } $fs->mkdir($tempFolder); // step 2 $manager = $this->getContainer()->get('sonata.seo.sitemap.manager'); // step 3 $output->writeln(sprintf('Generating sitemap - this can take a while')); foreach ($manager as $group => $sitemap) { $write = new SitemapWriter($tempFolder, $group, $sitemap->types, false); try { Handler::create($sitemap->sources, $write)->export(); } catch (\Exception $e) { $fs->remove($tempFolder); throw $e; } } // generate global sitemap index $appendPath = $input->hasOption('sitemap_path') ? $input->getOption('sitemap_path') : $input->getOption('baseurl'); SitemapWriter::generateSitemapIndex($tempFolder, sprintf('%s://%s%s', $input->getOption('scheme'), $input->getArgument('host'), $appendPath), 'sitemap*.xml', 'sitemap.xml'); // step 4 $output->writeln(sprintf('Moving temporary file to %s ...', $input->getArgument('folder'))); $oldFiles = Finder::create()->files()->name('sitemap*.xml')->in($input->getArgument('folder')); foreach ($oldFiles as $file) { $fs->remove($file->getRealPath()); } $newFiles = Finder::create()->files()->name('sitemap*.xml')->in($tempFolder); foreach ($newFiles as $file) { $fs->rename($file->getRealPath(), sprintf('%s/%s', $input->getArgument('folder'), $file->getFilename())); } $fs->remove($tempFolder); $output->writeln('<info>done!</info>'); }
public function testSimpleWriteAdvanced() { $writer = new SitemapWriter($this->folder, 'test', array('image'), false); $writer->open(); $writer->write(array('url' => 'https://sonata-project.org/bundle/', 'lastmod' => '2012-12-26', 'change' => 'daily', 'type' => 'default')); $writer->write(array('url' => 'https://sonata-project.org/bundle/', 'lastmod' => '2012-12-26', 'change' => 'weekly', 'type' => 'image', 'images' => array(array('url' => 'https://sonata-project.org/uploads/media/default/0001/01/thumb_1_default_small.jpg', 'caption' => 'sonata img')))); $writer->close(); $generatedFiles = $this->getFiles(); $this->assertEquals(1, count($generatedFiles)); $this->assertEquals($this->folder . '/sitemap_test_00001.xml', $generatedFiles[0]); SitemapWriter::generateSitemapIndex($this->folder, 'https://sonata-project.org'); $generatedFiles = $this->getFiles(); $this->assertEquals(2, count($generatedFiles)); // this will throw an exception if the xml is invalid new SimpleXMLElement(file_get_contents($generatedFiles[1])); $expected = <<<XML <?xml version="1.0" encoding="UTF-8"?> <urlset xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://www.sitemaps.org/schemas/sitemap/0.9 http://www.sitemaps.org/schemas/sitemap/0.9/sitemap.xsd" xmlns="http://www.sitemaps.org/schemas/sitemap/0.9" xmlns:image="http://www.google.com/schemas/sitemap-image/1.1"> <url><loc>https://sonata-project.org/bundle/</loc><lastmod>2012-12-26</lastmod><changefreq>weekly</changefreq><priority>0.5</priority></url> <url><loc>https://sonata-project.org/bundle/</loc><image:image><image:loc>https://sonata-project.org/uploads/media/default/0001/01/thumb_1_default_small.jpg</image:loc><image:caption>sonata img</image:caption></image:image></url> </urlset> XML; $this->assertEquals(trim($expected), file_get_contents($generatedFiles[1])); }