plugins/OStatus/lib/discoveryhints.php

   1 <?php
   2 /*
   3  * StatusNet - the distributed open-source microblogging tool
   4  * Copyright (C) 2010, StatusNet, Inc.
   5  *
   6  * Some utilities for generating hint data
   7  *
   8  * This program is free software: you can redistribute it and/or modify
   9  * it under the terms of the GNU Affero General Public License as published by
  10  * the Free Software Foundation, either version 3 of the License, or
  11  * (at your option) any later version.
  12  *
  13  * This program is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16  * GNU Affero General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU Affero General Public License
  19  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  20  */
  21
  22 class DiscoveryHints {
  23     static function fromXRD($xrd)
  24     {
  25         $hints = array();
  26
  27         foreach ($xrd->links as $link) {
  28             switch ($link['rel']) {
  29             case Discovery::PROFILEPAGE:
  30                 $hints['profileurl'] = $link['href'];
  31                 break;
  32             case Salmon::NS_MENTIONS:
  33             case Salmon::NS_REPLIES:
  34                 $hints['salmon'] = $link['href'];
  35                 break;
  36             case Discovery::UPDATESFROM:
  37                 $hints['feedurl'] = $link['href'];
  38                 break;
  39             case Discovery::HCARD:
  40                 $hints['hcardurl'] = $link['href'];
  41                 break;
  42             default:
  43                 break;
  44             }
  45         }
  46
  47         return $hints;
  48     }
  49
  50     static function fromHcardUrl($url)
  51     {
  52         $client = new HTTPClient();
  53         $client->setHeader('Accept', 'text/html,application/xhtml+xml');
  54         $response = $client->get($url);
  55
  56         if (!$response->isOk()) {
  57             return null;
  58         }
  59
  60         return self::hcardHints($response->getBody(),
  61                                 $response->getUrl());
  62     }
  63
  64     static function hcardHints($body, $url)
  65     {
  66         $hcard = self::_hcard($body, $url);
  67
  68         if (empty($hcard)) {
  69             return array();
  70         }
  71
  72         $hints = array();
  73
  74         // XXX: don't copy stuff into an array and then copy it again
  75
  76         if (array_key_exists('nickname', $hcard)) {
  77             $hints['nickname'] = $hcard['nickname'];
  78         }
  79
  80         if (array_key_exists('fn', $hcard)) {
  81             $hints['fullname'] = $hcard['fn'];
  82         } else if (array_key_exists('n', $hcard)) {
  83             $hints['fullname'] = implode(' ', $hcard['n']);
  84         }
  85
  86         if (array_key_exists('photo', $hcard) && count($hcard['photo'])) {
  87             $hints['avatar'] = $hcard['photo'][0];
  88         }
  89
  90         if (array_key_exists('note', $hcard)) {
  91             $hints['bio'] = $hcard['note'];
  92         }
  93
  94         if (array_key_exists('adr', $hcard)) {
  95             if (is_string($hcard['adr'])) {
  96                 $hints['location'] = $hcard['adr'];
  97             } else if (is_array($hcard['adr'])) {
  98                 $hints['location'] = implode(' ', $hcard['adr']);
  99             }
 100         }
 101
 102         if (array_key_exists('url', $hcard)) {
 103             if (is_string($hcard['url'])) {
 104                 $hints['homepage'] = $hcard['url'];
 105             } else if (is_array($hcard['url']) && !empty($hcard['url'])) {
 106                 // HACK get the last one; that's how our hcards look
 107                 $hints['homepage'] = $hcard['url'][count($hcard['url'])-1];
 108             }
 109         }
 110
 111         return $hints;
 112     }
 113
 114     static function _hcard($body, $url)
 115     {
 116         // DOMDocument::loadHTML may throw warnings on unrecognized elements,
 117         // and notices on unrecognized namespaces.
 118
 119         $old = error_reporting(error_reporting() & ~(E_WARNING | E_NOTICE));
 120
 121         $doc = new DOMDocument();
 122         $doc->loadHTML($body);
 123
 124         error_reporting($old);
 125
 126         $xp = new DOMXPath($doc);
 127
 128         $hcardNodes = self::_getChildrenByClass($doc->documentElement, 'vcard', $xp);
 129
 130         $hcards = array();
 131
 132         for ($i = 0; $i < $hcardNodes->length; $i++) {
 133
 134             $hcardNode = $hcardNodes->item($i);
 135
 136             $hcard = self::_hcardFromNode($hcardNode, $xp, $url);
 137
 138             $hcards[] = $hcard;
 139         }
 140
 141         $repr = null;
 142
 143         foreach ($hcards as $hcard) {
 144             if (in_array($url, $hcard['url'])) {
 145                 $repr = $hcard;
 146                 break;
 147             }
 148         }
 149
 150         if (!is_null($repr)) {
 151             return $repr;
 152         } else if (count($hcards) > 0) {
 153             return $hcards[0];
 154         } else {
 155             return null;
 156         }
 157     }
 158
 159     function _getChildrenByClass($el, $cls, $xp)
 160     {
 161         // borrowed from hkit. Thanks dudes!
 162
 163         $qry = ".//*[contains(concat(' ',normalize-space(@class),' '),' $cls ')]";
 164
 165         $nodes = $xp->query($qry, $el);
 166
 167         return $nodes;
 168     }
 169
 170     function _hcardFromNode($hcardNode, $xp, $base)
 171     {
 172         $hcard = array();
 173
 174         $hcard['url'] = array();
 175
 176         $urlNodes = self::_getChildrenByClass($hcardNode, 'url', $xp);
 177
 178         for ($j = 0; $j < $urlNodes->length; $j++) {
 179
 180             $urlNode = $urlNodes->item($j);
 181
 182             if ($urlNode->hasAttribute('href')) {
 183                 $url = $urlNode->getAttribute('href');
 184             } else {
 185                 $url = $urlNode->textContent;
 186             }
 187
 188             $hcard['url'][] = self::_rel2abs($url, $base);
 189         }
 190
 191         $hcard['photo'] = array();
 192
 193         $photoNodes = self::_getChildrenByClass($hcardNode, 'photo', $xp);
 194
 195         for ($j = 0; $j < $photoNodes->length; $j++) {
 196             $photoNode = $photoNodes->item($j);
 197             if ($photoNode->hasAttribute('src')) {
 198                 $url = $photoNode->getAttribute('src');
 199             } else if ($photoNode->hasAttribute('href')) {
 200                 $url = $photoNode->getAttribute('href');
 201             } else {
 202                 $url = $photoNode->textContent;
 203             }
 204             $hcard['photo'][] = self::_rel2abs($url, $base);
 205         }
 206
 207         $singles = array('nickname', 'note', 'fn', 'n', 'adr');
 208
 209         foreach ($singles as $single) {
 210
 211             $nodes = self::_getChildrenByClass($hcardNode, $single, $xp);
 212
 213             if ($nodes->length > 0) {
 214                 $node = $nodes->item(0);
 215                 $hcard[$single] = $node->textContent;
 216             }
 217         }
 218
 219         return $hcard;
 220     }
 221
 222     // XXX: this is a first pass; we probably need
 223     // to handle things like ../ and ./ and so on
 224
 225     static function _rel2abs($rel, $wrt)
 226     {
 227         $parts = parse_url($rel);
 228
 229         if ($parts === false) {
 230             return false;
 231         }
 232
 233         // If it's got a scheme, use it
 234
 235         if (!empty($parts['scheme'])) {
 236             return $rel;
 237         }
 238
 239         $w = parse_url($wrt);
 240
 241         $base = $w['scheme'].'://'.$w['host'];
 242
 243         if ($rel[0] == '/') {
 244             return $base.$rel;
 245         }
 246
 247         $wp = explode('/', $w['path']);
 248
 249         array_pop($wp);
 250
 251         return $base.implode('/', $wp).'/'.$rel;
 252     }
 253 }