<?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($cacheBaseDir .'/*', 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");
    }
}