143 lines
4.5 KiB
PHP
143 lines
4.5 KiB
PHP
<?php
|
|
/**
|
|
* @package staticpublisher
|
|
*/
|
|
class RebuildStaticCacheTask extends BuildTask
|
|
{
|
|
|
|
private static $quiet = false;
|
|
|
|
public function run($request)
|
|
{
|
|
if (Config::inst()->get('RebuildStaticCacheTask', 'quiet')) {
|
|
Config::inst()->update('StaticPublisher', 'echo_progress', false);
|
|
} else {
|
|
Config::inst()->update('StaticPublisher', 'echo_progress', true);
|
|
}
|
|
|
|
$page = singleton('Page');
|
|
|
|
if (!$page->hasMethod('allPagesToCache')) {
|
|
user_error(
|
|
'RebuildStaticCacheTask::index(): Please define a method "allPagesToCache()" on your Page class to return all pages affected by a cache refresh.',
|
|
E_USER_ERROR
|
|
);
|
|
}
|
|
|
|
|
|
if (!empty($_GET['urls'])) {
|
|
$urls = $_GET['urls'];
|
|
} else {
|
|
$urls = $page->allPagesToCache();
|
|
}
|
|
|
|
$this->rebuildCache($urls, true);
|
|
}
|
|
|
|
public function log($message = null)
|
|
{
|
|
$quiet = Config::inst()->get('RebuildStaticCacheTask', 'quiet');
|
|
|
|
if ($quiet) {
|
|
return;
|
|
}
|
|
|
|
echo $message;
|
|
}
|
|
|
|
/**
|
|
* Rebuilds the static cache for the pages passed through via $urls
|
|
*
|
|
* @param array $urls The URLs of pages to re-fetch and cache.
|
|
* @param bool $removeAll Remove all stale cache files (default TRUE).
|
|
*/
|
|
public function rebuildCache($urls, $removeAll = true)
|
|
{
|
|
if (!is_array($urls)) {
|
|
// $urls must be an array
|
|
user_error("Rebuild cache must be passed an array of urls. Make sure your allPagesToCache function returns an array", E_USER_ERROR);
|
|
return;
|
|
};
|
|
|
|
if (!Director::is_cli()) {
|
|
$this->log("<pre>\n");
|
|
$this->log("Rebuilding cache.\nNOTE: Please ensure that this page ends with 'Done!' - if not, then something may have gone wrong.\n\n");
|
|
}
|
|
|
|
$page = singleton('Page');
|
|
$cacheBaseDir = $page->getDestDir();
|
|
|
|
if (!file_exists($cacheBaseDir)) {
|
|
Filesystem::makeFolder($cacheBaseDir);
|
|
}
|
|
|
|
$quiet = Config::inst()->get('RebuildStaticCacheTask', 'quiet');
|
|
|
|
if (!$quiet) {
|
|
if (file_exists($cacheBaseDir.'/lock') && !isset($_REQUEST['force'])) {
|
|
die("There already appears to be a publishing queue running. You can skip warning this by adding ?/&force to the URL.");
|
|
}
|
|
}
|
|
|
|
touch($cacheBaseDir.'/lock');
|
|
|
|
// Note: Similiar to StaticPublisher->republish()
|
|
foreach ($urls as $i => $url) {
|
|
if ($url && !is_string($url)) {
|
|
user_error("Bad URL: " . var_export($url, true), E_USER_WARNING);
|
|
continue;
|
|
}
|
|
|
|
$urls[$i] = $url;
|
|
}
|
|
|
|
$urls = array_unique($urls);
|
|
sort($urls);
|
|
|
|
$start = isset($_GET['start']) ? $_GET['start'] : 0;
|
|
$count = isset($_GET['count']) ? $_GET['count'] : sizeof($urls);
|
|
|
|
if (($start + $count) > sizeof($urls)) {
|
|
$count = sizeof($urls) - $start;
|
|
}
|
|
|
|
$urls = array_slice($urls, $start, $count);
|
|
|
|
$mappedUrls = $page->urlsToPaths($urls);
|
|
|
|
if ($removeAll && !isset($_GET['urls']) && $start == 0 && file_exists("../cache")) {
|
|
$this->log("Removing stale cache files... \n");
|
|
|
|
if (!$quiet) {
|
|
flush();
|
|
}
|
|
|
|
if (Config::inst()->get('FilesystemPublisher', 'domain_based_caching')) {
|
|
// Glob each dir, then glob each one of those
|
|
foreach (glob(BASE_PATH . '/cache/*', GLOB_ONLYDIR) as $cacheDir) {
|
|
foreach (glob($cacheDir.'/*') as $cacheFile) {
|
|
$searchCacheFile = trim(str_replace($cacheBaseDir, '', $cacheFile), '\/');
|
|
|
|
if (!in_array($searchCacheFile, $mappedUrls)) {
|
|
$this->log(" * Deleting $cacheFile\n");
|
|
|
|
@unlink($cacheFile);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
$this->log("done.\n\n");
|
|
}
|
|
|
|
$this->log("Rebuilding cache from " . sizeof($mappedUrls) . " urls...\n\n");
|
|
$page->extend('publishPages', $mappedUrls);
|
|
|
|
if (file_exists($cacheBaseDir.'/lock')) {
|
|
unlink($cacheBaseDir.'/lock');
|
|
}
|
|
|
|
$this->log("\n\n== Done! ==\n");
|
|
}
|
|
}
|