plugins/OStatus/scripts/update-profile.php

   1 #!/usr/bin/env php
   2 <?php
   3 /*
   4  * StatusNet - a distributed open-source microblogging tool
   5  * Copyright (C) 2010, StatusNet, Inc.
   6  *
   7  * This program is free software: you can redistribute it and/or modify
   8  * it under the terms of the GNU Affero General Public License as published by
   9  * the Free Software Foundation, either version 3 of the License, or
  10  * (at your option) any later version.
  11  *
  12  * This program is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  * GNU Affero General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Affero General Public License
  18  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19  */
  20
  21 define('INSTALLDIR', realpath(dirname(__FILE__) . '/../../..'));
  22
  23 $helptext = <<<END_OF_HELP
  24 update-profile.php [options] http://example.com/profile/url
  25
  26 Rerun profile and feed info discovery for the given OStatus remote profile,
  27 and reinitialize its PuSH subscription for the given feed. This may help get
  28 things restarted if the hub or feed URLs have changed for the profile.
  29
  30
  31 END_OF_HELP;
  32
  33 require_once INSTALLDIR.'/scripts/commandline.inc';
  34
  35 if (empty($args[0]) || !Validate::uri($args[0])) {
  36     print "$helptext";
  37     exit(1);
  38 }
  39
  40 $uri = $args[0];
  41
  42
  43 $oprofile = Ostatus_profile::getKV('uri', $uri);
  44
  45 if (!$oprofile) {
  46     print "No OStatus remote profile known for URI $uri\n";
  47     exit(1);
  48 }
  49
  50 print "Old profile state for $oprofile->uri\n";
  51 showProfile($oprofile);
  52
  53 print "\n";
  54 print "Re-running feed discovery for profile URL $oprofile->uri\n";
  55
  56 $feedurl = null;
  57 $salmonuri = null;
  58
  59 // @fixme will bork where the URI isn't the profile URL for now
  60 $discover = new FeedDiscovery();
  61 try {
  62     $feedurl = $discover->discoverFromURL($oprofile->uri);
  63     $salmonuri = $discover->getAtomLink(Salmon::REL_SALMON)
  64                     ?: $discover->getAtomLink(Salmon::NS_REPLIES);  // NS_REPLIES is deprecated
  65     if (empty($salmonuri) ) {
  66         throw new FeedSubNoSalmonException('No salmon upstream URI was found');
  67     }
  68 } catch (FeedSubException $e) {
  69     $acct = $oprofile->localProfile()->getAcctUri();
  70     print "Could not discover feeds HTML response, trying reconstructed acct URI: {$acct}\n";
  71     $disco = new Discovery();
  72     $xrd = $disco->lookup($acct);
  73     $hints = DiscoveryHints::fromXRD($xrd);
  74
  75     if (empty($feedurl) && !array_key_exists('feedurl', $hints)) {
  76         throw new FeedSubNoFeedException($acct);
  77     }
  78     $feedurl = $feedurl ?: $hints['feedurl'];
  79     $salmonuri = array_key_exists('salmon', $hints) ? $hints['salmon'] : $salmonuri;
  80
  81     // get the hub data too and put it in the FeedDiscovery object
  82     $discover->discoverFromFeedUrl($feedurl);
  83 }
  84
  85 $huburi = $discover->getHubLink();
  86
  87 print "  Feed URL: $feedurl\n";
  88 print "  Hub URL: $huburi\n";
  89 print "  Salmon URL: $salmonuri\n";
  90
  91 if ($feedurl != $oprofile->feeduri || $salmonuri != $oprofile->salmonuri) {
  92     print "\n";
  93     print "Updating...\n";
  94     // @fixme update keys :P
  95     #$orig = clone($oprofile);
  96     #$oprofile->feeduri = $feedurl;
  97     #$oprofile->salmonuri = $salmonuri;
  98     #$ok = $oprofile->update($orig);
  99     $ok = $oprofile->query('UPDATE ostatus_profile SET ' .
 100         'feeduri=\'' . $oprofile->escape($feedurl) . '\',' .
 101         'salmonuri=\'' . $oprofile->escape($salmonuri) . '\' ' .
 102         'WHERE uri=\'' . $oprofile->escape($uri) . '\'');
 103
 104     if (!$ok) {
 105         print "Failed to update profile record...\n";
 106         exit(1);
 107     }
 108
 109     $oprofile->decache();
 110 } else {
 111     print "\n";
 112     print "Ok, ostatus_profile record unchanged.\n\n";
 113 }
 114
 115 $sub = FeedSub::ensureFeed($feedurl);
 116
 117 if ($huburi != $sub->huburi) {
 118     print "\n";
 119     print "Updating hub record for feed; was $sub->huburi\n";
 120     $orig = clone($sub);
 121     $sub->huburi = $huburi;
 122     $ok = $sub->update($orig);
 123
 124     if (!$ok) {
 125         print "Failed to update sub record...\n";
 126         exit(1);
 127     }
 128 } else {
 129     print "\n";
 130     print "Feed record ok, not changing.\n\n";
 131 }
 132
 133 echo "\n";
 134 echo "Pinging hub {$sub->huburi} with new subscription for {$sub->uri}\n";
 135 try {
 136     $sub->subscribe();
 137     echo "ok\n";
 138 } catch (Exception $e) {
 139     echo 'Could not confirm. '.get_class($e).': '.$e->getMessage()."\n";
 140 }
 141
 142 $o2 = Ostatus_profile::getKV('uri', $uri);
 143
 144 print "\n";
 145 print "New profile state:\n";
 146 showProfile($o2);
 147
 148 print "\n";
 149 print "New feed state:\n";
 150 $sub2 = FeedSub::ensureFeed($feedurl);
 151 showSub($sub2);
 152
 153 function showProfile($oprofile)
 154 {
 155     print "  Feed URL: $oprofile->feeduri\n";
 156     print "  Salmon URL: $oprofile->salmonuri\n";
 157     print "  Avatar URL: $oprofile->avatar\n";
 158     print "  Profile ID: $oprofile->profile_id\n";
 159     print "  Group ID: $oprofile->group_id\n";
 160     print "  Record created: $oprofile->created\n";
 161     print "  Record modified: $oprofile->modified\n";
 162 }
 163
 164 function showSub($sub)
 165 {
 166     print "  Subscription state: $sub->sub_state\n";
 167     print "  Verify token: $sub->verify_token\n";
 168     print "  Signature secret: $sub->secret\n";
 169     print "  Sub start date: $sub->sub_start\n";
 170     print "  Record created: $sub->created\n";
 171     print "  Record modified: $sub->modified\n";
 172 }