sitemapindex.php 4.77 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43
<?php
/**
 * StatusNet, the distributed open-source microblogging tool
 *
 * Generate sitemap index
 *
 * PHP version 5
 *
 * LICENCE: This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 * @category  Sitemap
 * @package   StatusNet
 * @author    Evan Prodromou <evan@status.net>
 * @copyright 2010 StatusNet, Inc.
 * @license   http://www.fsf.org/licensing/licenses/agpl-3.0.html GNU Affero General Public License version 3.0
 * @link      http://status.net/
 */

if (!defined('STATUSNET')) {
    exit(1);
}

/**
 * Show the sitemap index
 *
 * @category Sitemap
 * @package  StatusNet
 * @author   Evan Prodromou <evan@status.net>
 * @license  http://www.fsf.org/licensing/licenses/agpl-3.0.html GNU Affero General Public License version 3.0
 * @link     http://status.net/
 */

44
class SitemapindexAction extends Action
45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70
{
    /**
     * handle the action
     *
     * @param array $args unused.
     *
     * @return void
     */

    function handle($args)
    {
        header('Content-Type: text/xml; charset=UTF-8');
        $this->startXML();

        $this->elementStart('sitemapindex', array('xmlns' => 'http://www.sitemaps.org/schemas/sitemap/0.9'));

        $this->showUserSitemaps();
        $this->showNoticeSitemaps();

        $this->elementEnd('sitemapindex');

        $this->endXML();
    }

    function showUserSitemaps()
    {
71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103
        $userCounts = $this->getUserCounts();

        foreach ($userCounts as $dt => $cnt) {
            $cnt = $cnt+0;
            assert($cnt != 0);
            $n = (int)$cnt / (int)SitemapPlugin::USERS_PER_MAP;
            if (($cnt % SitemapPlugin::USERS_PER_MAP) != 0) {
                $n++;
            }
            for ($i = 1; $i <= $n; $i++) {
                $this->showSitemap('user', $dt, $i);
            }
        }
    }

    function showNoticeSitemaps()
    {
        $noticeCounts = $this->getNoticeCounts();

        foreach ($noticeCounts as $dt => $cnt) {
            assert($cnt != 0);
            $n = $cnt / SitemapPlugin::NOTICES_PER_MAP;
            if ($cnt % SitemapPlugin::NOTICES_PER_MAP) {
                $n++;
            }
            for ($i = 1; $i <= $n; $i++) {
                $this->showSitemap('notice', $dt, $i);
            }
        }
    }

    function getUserCounts()
    {
104
        $userCounts = User::cacheGet('sitemap:user:counts');
105

106
        if ($userCounts === false) {
107

108
            $user = new User();
109

110 111 112
            $user->selectAdd();
            $user->selectAdd('date(created) as regdate, count(*) as regcount');
            $user->groupBy('regdate');
113

114
            $user->find();
115

116 117 118 119 120 121 122
            $userCounts = array();

            while ($user->fetch()) {
                $userCounts[$user->regdate] = $user->regcount;
            }

            User::cacheSet('sitemap:user:counts', $userCounts);
123 124 125 126 127 128 129
        }

        return $userCounts;
    }

    function getNoticeCounts()
    {
130
        $noticeCounts = Notice::cacheGet('sitemap:notice:counts');
131

132
        if ($noticeCounts === false) {
133

134
            $notice = new Notice();
135

136 137 138
            $notice->selectAdd();
            $notice->selectAdd('date(created) as postdate, count(*) as postcount');
            $notice->groupBy('postdate');
139

140 141 142 143 144 145 146
            $notice->find();

            $noticeCounts = array();

            while ($notice->fetch()) {
                $noticeCounts[$notice->postdate] = $notice->postcount;
            }
147

148
            Notice::cacheSet('sitemap:notice:counts', $noticeCounts);
149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172
        }

        return $noticeCounts;
    }

    function showSitemap($prefix, $dt, $i)
    {
        list($y, $m, $d) = explode('-', $dt);

        $this->elementStart('sitemap');
        $this->element('loc', null, common_local_url($prefix.'sitemap',
                                                     array('year' => $y,
                                                           'month' => $m,
                                                           'day' => $d,
                                                           'index' => $i)));

        $begdate = strtotime("$y-$m-$d 00:00:00");
        $enddate = $begdate + (24 * 60 * 60);

        if ($enddate < time()) {
            $this->element('lastmod', null, date(DATE_W3C, $enddate));
        }

        $this->elementEnd('sitemap');
173 174
    }
}