james/fembooru
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update ext/sitemap/main.php

Browse Source 
- Added setup block to choose between generating full sitemap or smaller sitemap
- Added sitemap caching, regenerating at most every day
This commit is contained in:
Drudex Software 2013-02-15 04:31:31 +01:00
parent 52893d0f04
commit e64f1f0bcb
1 changed files with 48 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
ext/sitemap/main.php
View File

@ -2,29 +2,37 @@
/* /*
* Name: XML Sitemap * Name: XML Sitemap
* Author: Sein Kraft <mail@seinkraft.info> * Author: Sein Kraft <mail@seinkraft.info>
* Author: Drudex Software <support@drudexsoftware.com>
* License: GPLv2 * License: GPLv2
* Description: Adds sitemap.xml on request. * Description: Sitemap with caching & advanced priorities
* Documentation: * Documentation:
*/ */
class XMLSitemap extends Extension { class XMLSitemap extends Extension {
private $sitemap_queue = ""; private $sitemap_queue = "";
private $sitemap_filepath = ""; // set onPageRequest
public function onPageRequest(PageRequestEvent $event) { public function onPageRequest(PageRequestEvent $event) {
if($event->page_matches("sitemap.xml")) if($event->page_matches("sitemap.xml"))
{ {
global $config; global $config;
$this->sitemap_filepath = $_SERVER['DOCUMENT_ROOT']."/ext/sitemap/generated_sitemap.xml";
// determine if new sitemap needs to be generated
if ($this->new_sitemap_needed())
{
// determine which type of sitemap to generate
if ($config->get_bool("sitemap_generatefull",false)) if ($config->get_bool("sitemap_generatefull",false))
$this->handle_full_sitemap(); // default false until cache fixed $this->handle_full_sitemap(); // default false until cache fixed
else else
$this->handle_smaller_sitemap(); $this->handle_smaller_sitemap();
} }
else $this->display_existing_sitemap();
}
} }
public function onSetupBuilding(SetupBuildingEvent $event) { public function onSetupBuilding(SetupBuildingEvent $event) {
$sb = new SetupBlock("Sitemap (Beta)"); // beta until sitemap cache fixed $sb = new SetupBlock("Sitemap");
$sb->add_bool_option("sitemap_generatefull", "Generate full sitemap"); $sb->add_bool_option("sitemap_generatefull", "Generate full sitemap");
$sb->add_label("<br>(Enabled: every image and tag in sitemap, generation takes longer)"); $sb->add_label("<br>(Enabled: every image and tag in sitemap, generation takes longer)");
@ -46,7 +54,7 @@ class XMLSitemap extends Extension {
/* --- Display page --- */ /* --- Display page --- */
// when sitemap is ok, display it from the file // when sitemap is ok, display it from the file
$this->display_sitemap(); $this->generate_display_sitemap();
} }
// Full sitemap // Full sitemap
@ -93,7 +101,7 @@ class XMLSitemap extends Extension {
/* --- Display page --- */ /* --- Display page --- */
// when sitemap is ok, display it from the file // when sitemap is ok, display it from the file
$this->display_sitemap(); $this->generate_display_sitemap();
} }
// Adds an array of urls to the sitemap with the given information // Adds an array of urls to the sitemap with the given information
@ -111,10 +119,10 @@ class XMLSitemap extends Extension {
} }
// sets sitemap with entries in the queue // sets sitemap with entries in the queue
// todo: cache sitemap for a config specified amount of time private function generate_display_sitemap()
private function display_sitemap()
{ {
global $page; global $page, $config;
$page->set_mode("data"); $page->set_mode("data");
$page->set_type("application/xml"); $page->set_type("application/xml");
@ -124,6 +132,33 @@ class XMLSitemap extends Extension {
</urlset>"; </urlset>";
// sets // sets
$config->set_int("sitemap_lastgenerated", time());
file_put_contents($this->sitemap_filepath, $xml);
$page->set_data($xml);
}
// returns true if a new sitemap is needed
private function new_sitemap_needed()
{
global $config;
$sitemap_generation_interval = 86400; // allow new site map every day
$last_generated_time = $config->get_int("sitemap_lastgenerated",0);
if (!file_exists($this->sitemap_filepath) ||
($last_generated_time + $sitemap_generation_interval < time()))
return true;
else return false;
}
private function display_existing_sitemap()
{
global $page;
$page->set_mode("data");
$page->set_type("application/xml");
$xml = file_get_contents($this->sitemap_filepath);
$page->set_data($xml); $page->set_data($xml);
} }
} }