lib/atom10feed.php

   1 <?php
   2 /**
   3  * StatusNet, the distributed open-source microblogging tool
   4  *
   5  * Class for building an Atom feed in memory
   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  Feed
  23  * @package   StatusNet
  24  * @author    Zach Copley <zach@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 {
  32     exit(1);
  33 }
  34
  35 class Atom10FeedException extends Exception
  36 {
  37 }
  38
  39 /**
  40  * Class for building an Atom feed in memory.  Get the finished doc
  41  * as a string with Atom10Feed::getString().
  42  *
  43  * @category Feed
  44  * @package  StatusNet
  45  * @author   Zach Copley <zach@status.net>
  46  * @license  http://www.fsf.org/licensing/licenses/agpl-3.0.html GNU Affero General Public License version 3.0
  47  * @link     http://status.net/
  48  */
  49 class Atom10Feed extends XMLStringer
  50 {
  51     public  $xw;
  52
  53     // @fixme most of these should probably be read-only properties
  54     private $namespaces;
  55     private $authors;
  56     private $subject;
  57     private $categories;
  58     private $contributors;
  59     private $generator;
  60     private $icon;
  61     private $links;
  62     private $selfLink;
  63     private $selfLinkType;
  64     public $logo;
  65     private $rights;
  66     public $subtitle;
  67     public $title;
  68     private $published;
  69     private $updated;
  70     private $entries;
  71
  72     /**
  73      * Constructor
  74      *
  75      * @param boolean $indent  flag to turn indenting on or off
  76      *
  77      * @return void
  78      */
  79     function __construct($indent = true) {
  80         parent::__construct($indent);
  81         $this->namespaces = array();
  82         $this->authors    = array();
  83         $this->links      = array();
  84         $this->entries    = array();
  85         $this->addNamespace('', 'http://www.w3.org/2005/Atom');
  86     }
  87
  88     /**
  89      * Add another namespace to the feed
  90      *
  91      * @param string $namespace the namespace
  92      * @param string $uri       namspace uri
  93      *
  94      * @return void
  95      */
  96     function addNamespace($namespace, $uri)
  97     {
  98         $ns = array($namespace => $uri);
  99         $this->namespaces = array_merge($this->namespaces, $ns);
 100     }
 101
 102     function addAuthor($name, $uri = null, $email = null)
 103     {
 104         $xs = new XMLStringer(true);
 105
 106         $xs->elementStart('author');
 107
 108         if (!empty($name)) {
 109             $xs->element('name', null, $name);
 110         } else {
 111             throw new Atom10FeedException(
 112                 'author element must contain a name element.'
 113             );
 114         }
 115
 116         if (isset($uri)) {
 117             $xs->element('uri', null, $uri);
 118         }
 119
 120         if (isset($email)) {
 121             $xs->element('email', null, $email);
 122         }
 123
 124         $xs->elementEnd('author');
 125
 126         array_push($this->authors, $xs->getString());
 127     }
 128
 129     /**
 130      * Add an Author to the feed via raw XML string
 131      *
 132      * @param string $xmlAuthor An XML string representation author
 133      *
 134      * @return void
 135      */
 136     function addAuthorRaw($xmlAuthor)
 137     {
 138         array_push($this->authors, $xmlAuthor);
 139     }
 140
 141     function renderAuthors()
 142     {
 143         foreach ($this->authors as $author) {
 144             $this->raw($author);
 145         }
 146     }
 147
 148     /**
 149      * Add a activity feed subject via raw XML string
 150      *
 151      * @param string $xmlSubject An XML string representation of the subject
 152      *
 153      * @return void
 154      */
 155     function setActivitySubject($xmlSubject)
 156     {
 157         $this->subject = $xmlSubject;
 158     }
 159
 160     function getNamespaces()
 161     {
 162         return $this->namespaces;
 163     }
 164
 165     function initFeed()
 166     {
 167         $this->xw->startDocument('1.0', 'UTF-8');
 168         $commonAttrs = array('xml:lang' => 'en-US');
 169         foreach ($this->namespaces as $prefix => $uri) {
 170             if ($prefix == '') {
 171                 $attr = 'xmlns';
 172             } else {
 173                 $attr = 'xmlns:' . $prefix;
 174             }
 175             $commonAttrs[$attr] = $uri;
 176         }
 177         $this->elementStart('feed', $commonAttrs);
 178
 179         $this->element(
 180             'generator', array(
 181                 'uri'     => 'http://status.net',
 182                 'version' => STATUSNET_VERSION
 183             ),
 184             'StatusNet'
 185         );
 186
 187         $this->element('id', null, $this->id);
 188         $this->element('title', null, $this->title);
 189         $this->element('subtitle', null, $this->subtitle);
 190
 191         if (!empty($this->logo)) {
 192             $this->element('logo', null, $this->logo);
 193         }
 194
 195         $this->element('updated', null, $this->updated);
 196
 197         $this->renderAuthors();
 198
 199         if ($this->selfLink) {
 200             $this->addLink($this->selfLink, array('rel' => 'self',
 201                                                   'type' => $this->selfLinkType));
 202         }
 203         $this->renderLinks();
 204     }
 205
 206     /**
 207      * Check that all required elements have been set, etc.
 208      * Throws an Atom10FeedException if something's missing.
 209      *
 210      * @return void
 211      */
 212     function validate()
 213     {
 214     }
 215
 216     function renderLinks()
 217     {
 218         foreach ($this->links as $attrs)
 219         {
 220             $this->element('link', $attrs, null);
 221         }
 222     }
 223
 224     function addEntryRaw($xmlEntry)
 225     {
 226         array_push($this->entries, $xmlEntry);
 227     }
 228
 229     function addEntry($entry)
 230     {
 231         array_push($this->entries, $entry->getString());
 232     }
 233
 234     function renderEntries()
 235     {
 236         foreach ($this->entries as $entry) {
 237             $this->raw($entry);
 238         }
 239     }
 240
 241     function endFeed()
 242     {
 243         $this->elementEnd('feed');
 244         $this->xw->endDocument();
 245     }
 246
 247     function getString()
 248     {
 249         if (Event::handle('StartApiAtom', array($this))) {
 250
 251             $this->validate();
 252             $this->initFeed();
 253
 254             if (!empty($this->subject)) {
 255                 $this->raw($this->subject);
 256             }
 257
 258             $this->renderEntries();
 259             $this->endFeed();
 260
 261             Event::handle('EndApiAtom', array($this));
 262         }
 263
 264         return $this->xw->outputMemory();
 265     }
 266
 267     function setId($id)
 268     {
 269         $this->id = $id;
 270     }
 271
 272     function setSelfLink($url, $type='application/atom+xml')
 273     {
 274         $this->selfLink = $url;
 275         $this->selfLinkType = $type;
 276     }
 277
 278     function setTitle($title)
 279     {
 280         $this->title = $title;
 281     }
 282
 283     function setSubtitle($subtitle)
 284     {
 285         $this->subtitle = $subtitle;
 286     }
 287
 288     function setLogo($logo)
 289     {
 290         $this->logo = $logo;
 291     }
 292
 293     function setUpdated($dt)
 294     {
 295         $this->updated = common_date_iso8601($dt);
 296     }
 297
 298     function setPublished($dt)
 299     {
 300         $this->published = common_date_iso8601($dt);
 301     }
 302
 303     /**
 304      * Adds a link element into the Atom document
 305      *
 306      * Assumes you want rel="alternate" and type="text/html" unless
 307      * you send in $otherAttrs.
 308      *
 309      * @param string $uri            the uri the href needs to point to
 310      * @param array  $otherAttrs     other attributes to stick in
 311      *
 312      * @return void
 313      */
 314     function addLink($uri, $otherAttrs = null) {
 315         $attrs = array('href' => $uri);
 316
 317         if (is_null($otherAttrs)) {
 318             $attrs['rel']  = 'alternate';
 319             $attrs['type'] = 'text/html';
 320         } else {
 321             $attrs = array_merge($attrs, $otherAttrs);
 322         }
 323
 324         array_push($this->links, $attrs);
 325     }
 326
 327 }