plugins/Sitemap/sitemapindex.php

   1 <?php
   2 /**
   3  * StatusNet, the distributed open-source microblogging tool
   4  *
   5  * Generate sitemap index
   6  *
   7  * PHP version 5
   8  *
   9  * LICENCE: This program is free software: you can redistribute it and/or modify
  10  * it under the terms of the GNU Affero General Public License as published by
  11  * the Free Software Foundation, either version 3 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17  * GNU Affero General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU Affero General Public License
  20  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  21  *
  22  * @category  Sitemap
  23  * @package   StatusNet
  24  * @author    Evan Prodromou <evan@status.net>
  25  * @copyright 2010 StatusNet, Inc.
  26  * @license   http://www.fsf.org/licensing/licenses/agpl-3.0.html GNU Affero General Public License version 3.0
  27  * @link      http://status.net/
  28  */
  29
  30 if (!defined('STATUSNET')) {
  31     exit(1);
  32 }
  33
  34 /**
  35  * Show the sitemap index
  36  *
  37  * @category Sitemap
  38  * @package  StatusNet
  39  * @author   Evan Prodromou <evan@status.net>
  40  * @license  http://www.fsf.org/licensing/licenses/agpl-3.0.html GNU Affero General Public License version 3.0
  41  * @link     http://status.net/
  42  */
  43
  44 class SitemapindexAction extends Action
  45 {
  46     /**
  47      * handle the action
  48      *
  49      * @param array $args unused.
  50      *
  51      * @return void
  52      */
  53
  54     function handle($args)
  55     {
  56         header('Content-Type: text/xml; charset=UTF-8');
  57         $this->startXML();
  58
  59         $this->elementStart('sitemapindex', array('xmlns' => 'http://www.sitemaps.org/schemas/sitemap/0.9'));
  60
  61         $this->showUserSitemaps();
  62         $this->showNoticeSitemaps();
  63
  64         $this->elementEnd('sitemapindex');
  65
  66         $this->endXML();
  67     }
  68
  69     function showUserSitemaps()
  70     {
  71         $userCounts = Sitemap_user_count::getAll();
  72
  73         foreach ($userCounts as $dt => $cnt) {
  74             $cnt = $cnt+0;
  75
  76             if ($cnt == 0) {
  77                 continue;
  78             }
  79
  80             $n = (int)$cnt / (int)SitemapPlugin::USERS_PER_MAP;
  81             if (($cnt % SitemapPlugin::USERS_PER_MAP) != 0) {
  82                 $n++;
  83             }
  84             for ($i = 1; $i <= $n; $i++) {
  85                 $this->showSitemap('user', $dt, $i);
  86             }
  87         }
  88     }
  89
  90     function showNoticeSitemaps()
  91     {
  92         $noticeCounts = $this->getNoticeCounts();
  93
  94         foreach ($noticeCounts as $dt => $cnt) {
  95             if ($cnt == 0) {
  96                 continue;
  97             }
  98             $n = $cnt / SitemapPlugin::NOTICES_PER_MAP;
  99             if ($cnt % SitemapPlugin::NOTICES_PER_MAP) {
 100                 $n++;
 101             }
 102             for ($i = 1; $i <= $n; $i++) {
 103                 $this->showSitemap('notice', $dt, $i);
 104             }
 105         }
 106     }
 107
 108     function getUserCounts()
 109     {
 110         return Sitemap_user_count::getAll();
 111     }
 112
 113     function getNoticeCounts()
 114     {
 115         $noticeCounts = Notice::cacheGet('sitemap:notice:counts');
 116
 117         if ($noticeCounts === false) {
 118
 119             $notice = new Notice();
 120
 121             $notice->selectAdd();
 122             $notice->selectAdd('date(created) as postdate, count(*) as postcount');
 123             $notice->groupBy('postdate');
 124
 125             $notice->find();
 126
 127             $noticeCounts = array();
 128
 129             while ($notice->fetch()) {
 130                 $noticeCounts[$notice->postdate] = $notice->postcount;
 131             }
 132
 133             Notice::cacheSet('sitemap:notice:counts', $noticeCounts);
 134         }
 135
 136         return $noticeCounts;
 137     }
 138
 139     function showSitemap($prefix, $dt, $i)
 140     {
 141         list($y, $m, $d) = explode('-', $dt);
 142
 143         $this->elementStart('sitemap');
 144         $this->element('loc', null, common_local_url($prefix.'sitemap',
 145                                                      array('year' => $y,
 146                                                            'month' => $m,
 147                                                            'day' => $d,
 148                                                            'index' => $i)));
 149
 150         $begdate = strtotime("$y-$m-$d 00:00:00");
 151         $enddate = $begdate + (24 * 60 * 60);
 152
 153         if ($enddate < time()) {
 154             $this->element('lastmod', null, date(DATE_W3C, $enddate));
 155         }
 156
 157         $this->elementEnd('sitemap');
 158     }
 159 }