plugins/OStatus/actions/ostatusinit.php

   1 <?php
   2 /*
   3  * StatusNet - the distributed open-source microblogging tool
   4  * Copyright (C) 2010, StatusNet, Inc.
   5  *
   6  * This program is free software: you can redistribute it and/or modify
   7  * it under the terms of the GNU Affero General Public License as published by
   8  * the Free Software Foundation, either version 3 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU Affero General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Affero General Public License
  17  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18  */
  19
  20 /**
  21  * @package OStatusPlugin
  22  * @maintainer James Walker <james@status.net>
  23  */
  24
  25 if (!defined('STATUSNET') && !defined('LACONICA')) { exit(1); }
  26
  27
  28 class OStatusInitAction extends Action
  29 {
  30
  31     var $nickname;
  32     var $group;
  33     var $profile;
  34     var $err;
  35
  36     function prepare($args)
  37     {
  38         parent::prepare($args);
  39
  40         if (common_logged_in()) {
  41             $this->clientError(_m('You can use the local subscription!'));
  42             return false;
  43         }
  44
  45         // Local user or group the remote wants to subscribe to
  46         $this->nickname = $this->trimmed('nickname');
  47         $this->group = $this->trimmed('group');
  48
  49         // Webfinger or profile URL of the remote user
  50         $this->profile = $this->trimmed('profile');
  51
  52         return true;
  53     }
  54
  55     function handle($args)
  56     {
  57         parent::handle($args);
  58
  59         if ($_SERVER['REQUEST_METHOD'] == 'POST') {
  60             /* Use a session token for CSRF protection. */
  61             $token = $this->trimmed('token');
  62             if (!$token || $token != common_session_token()) {
  63                 $this->showForm(_m('There was a problem with your session token. '.
  64                                   'Try again, please.'));
  65                 return;
  66             }
  67             $this->ostatusConnect();
  68         } else {
  69             $this->showForm();
  70         }
  71     }
  72
  73     function showForm($err = null)
  74     {
  75         $this->err = $err;
  76         if ($this->boolean('ajax')) {
  77             header('Content-Type: text/xml;charset=utf-8');
  78             $this->xw->startDocument('1.0', 'UTF-8');
  79             $this->elementStart('html');
  80             $this->elementStart('head');
  81             $this->element('title', null, _m('Subscribe to user'));
  82             $this->elementEnd('head');
  83             $this->elementStart('body');
  84             $this->showContent();
  85             $this->elementEnd('body');
  86             $this->elementEnd('html');
  87         } else {
  88             $this->showPage();
  89         }
  90     }
  91
  92     function showContent()
  93     {
  94         if ($this->group) {
  95             $header = sprintf(_m('Join group %s'), $this->group);
  96             $submit = _m('Join');
  97         } else {
  98             $header = sprintf(_m('Subscribe to %s'), $this->nickname);
  99             $submit = _m('Subscribe');
 100         }
 101         $this->elementStart('form', array('id' => 'form_ostatus_connect',
 102                                           'method' => 'post',
 103                                           'class' => 'form_settings',
 104                                           'action' => common_local_url('ostatusinit')));
 105         $this->elementStart('fieldset');
 106         $this->element('legend', null,  $header);
 107         $this->hidden('token', common_session_token());
 108
 109         $this->elementStart('ul', 'form_data');
 110         $this->elementStart('li', array('id' => 'ostatus_nickname'));
 111         $this->input('nickname', _m('User nickname'), $this->nickname,
 112                      _m('Nickname of the user you want to follow'));
 113         $this->hidden('group', $this->group); // pass-through for magic links
 114         $this->elementEnd('li');
 115         $this->elementStart('li', array('id' => 'ostatus_profile'));
 116         $this->input('profile', _m('Profile Account'), $this->profile,
 117                      _m('Your account id (i.e. user@identi.ca)'));
 118         $this->elementEnd('li');
 119         $this->elementEnd('ul');
 120         $this->submit('submit', $submit);
 121         $this->elementEnd('fieldset');
 122         $this->elementEnd('form');
 123     }
 124
 125     function ostatusConnect()
 126     {
 127         $opts = array('allowed_schemes' => array('http', 'https', 'acct'));
 128         if (Validate::uri($this->profile, $opts)) {
 129             $bits = parse_url($this->profile);
 130             if ($bits['scheme'] == 'acct') {
 131                 $this->connectWebfinger($bits['path']);
 132             } else {
 133                 $this->connectProfile($this->profile);
 134             }
 135         } elseif (strpos($this->profile, '@') !== false) {
 136             $this->connectWebfinger($this->profile);
 137         } else {
 138             $this->clientError(_m("Must provide a remote profile."));
 139         }
 140     }
 141
 142     function connectWebfinger($acct)
 143     {
 144         $target_profile = $this->targetProfile();
 145
 146         $disco = new Discovery;
 147         $result = $disco->lookup($acct);
 148         if (!$result) {
 149             $this->clientError(_m("Couldn't look up OStatus account profile."));
 150         }
 151
 152         foreach ($result->links as $link) {
 153             if ($link['rel'] == 'http://ostatus.org/schema/1.0/subscribe') {
 154                 // We found a URL - let's redirect!
 155                 $url = Discovery::applyTemplate($link['template'], $target_profile);
 156                 common_log(LOG_INFO, "Sending remote subscriber $acct to $url");
 157                 common_redirect($url, 303);
 158             }
 159
 160         }
 161         $this->clientError(_m("Couldn't confirm remote profile address."));
 162     }
 163
 164     function connectProfile($subscriber_profile)
 165     {
 166         $target_profile = $this->targetProfile();
 167
 168         // @fixme hack hack! We should look up the remote sub URL from XRDS
 169         $suburl = preg_replace('!^(.*)/(.*?)$!', '$1/main/ostatussub', $subscriber_profile);
 170         $suburl .= '?profile=' . urlencode($target_profile);
 171
 172         common_log(LOG_INFO, "Sending remote subscriber $subscriber_profile to $suburl");
 173         common_redirect($suburl, 303);
 174     }
 175
 176     /**
 177      * Build the canonical profile URI+URL of the requested user or group
 178      */
 179     function targetProfile()
 180     {
 181         if ($this->nickname) {
 182             $user = User::staticGet('nickname', $this->nickname);
 183             if ($user) {
 184                 return common_local_url('userbyid', array('id' => $user->id));
 185             } else {
 186                 $this->clientError("No such user.");
 187             }
 188         } else if ($this->group) {
 189             $group = Local_group::staticGet('nickname', $this->group);
 190             if ($group) {
 191                 return common_local_url('groupbyid', array('id' => $group->group_id));
 192             } else {
 193                 $this->clientError("No such group.");
 194             }
 195         } else {
 196             $this->clientError("No local user or group nickname provided.");
 197         }
 198     }
 199
 200     function title()
 201     {
 202       return _m('OStatus Connect');
 203     }
 204
 205 }