initial script to restore a backed-up user
This commit is contained in:
Evan Prodromou
parent
4d01f8fbb6
commit
be14f409e4
376
scripts/restoreuser.php
Normal file
376
scripts/restoreuser.php
Normal file
|
|
@ -0,0 +1,376 @@
|
||||||
|
<?php
|
||||||
|
/*
|
||||||
|
* StatusNet - the distributed open-source microblogging tool
|
||||||
|
* Copyright (C) 2010 StatusNet, Inc.
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU Affero General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU Affero General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU Affero General Public License
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
define('INSTALLDIR', realpath(dirname(__FILE__) . '/..'));
|
||||||
|
|
||||||
|
$shortoptions = 'i:n:f:';
|
||||||
|
$longoptions = array('id=', 'nickname=', 'file=');
|
||||||
|
|
||||||
|
$helptext = <<<END_OF_RESTOREUSER_HELP
|
||||||
|
restoreuser.php [options]
|
||||||
|
Restore a backed-up user file to the database. If
|
||||||
|
neither ID or name provided, will create a new user.
|
||||||
|
|
||||||
|
-i --id ID of user to export
|
||||||
|
-n --nickname nickname of the user to export
|
||||||
|
-f --file file to read from (STDIN by default)
|
||||||
|
|
||||||
|
END_OF_RESTOREUSER_HELP;
|
||||||
|
|
||||||
|
require_once INSTALLDIR.'/scripts/commandline.inc';
|
||||||
|
require_once INSTALLDIR.'/extlib/htmLawed/htmLawed.php';
|
||||||
|
|
||||||
|
function getActivityStreamDocument()
|
||||||
|
{
|
||||||
|
$filename = get_option_value('f', 'file');
|
||||||
|
|
||||||
|
if (empty($filename)) {
|
||||||
|
show_help();
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!file_exists($filename)) {
|
||||||
|
throw new Exception("No such file '$filename'.");
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!is_file($filename)) {
|
||||||
|
throw new Exception("Not a regular file: '$filename'.");
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!is_readable($filename)) {
|
||||||
|
throw new Exception("File '$filename' not readable.");
|
||||||
|
}
|
||||||
|
|
||||||
|
printfv(_("Getting backup from file '$filename'.\n"));
|
||||||
|
|
||||||
|
$xml = file_get_contents($filename);
|
||||||
|
|
||||||
|
$dom = DOMDocument::loadXML($xml);
|
||||||
|
|
||||||
|
if ($dom->documentElement->namespaceURI != Activity::ATOM ||
|
||||||
|
$dom->documentElement->localName != 'feed') {
|
||||||
|
throw new Exception("'$filename' is not an Atom feed.");
|
||||||
|
}
|
||||||
|
|
||||||
|
return $dom;
|
||||||
|
}
|
||||||
|
|
||||||
|
function importActivityStream($user, $doc)
|
||||||
|
{
|
||||||
|
$feed = $doc->documentElement;
|
||||||
|
|
||||||
|
$subjectEl = ActivityUtils::child($feed, Activity::SUBJECT, Activity::SPEC);
|
||||||
|
|
||||||
|
if (!empty($subjectEl)) {
|
||||||
|
$subject = new ActivityObject($subjectEl);
|
||||||
|
printfv(_("Backup file for user %s (%s)\n"), $subject->id, Ostatus_profile::getActivityObjectNickname($subject));
|
||||||
|
} else {
|
||||||
|
throw new Exception("Feed doesn't have an <activity:subject> element.");
|
||||||
|
}
|
||||||
|
|
||||||
|
if (is_null($user)) {
|
||||||
|
printfv(_("No user specified; using backup user.\n"));
|
||||||
|
$user = userFromSubject($subject);
|
||||||
|
}
|
||||||
|
|
||||||
|
$entries = $feed->getElementsByTagNameNS(Activity::ATOM, 'entry');
|
||||||
|
|
||||||
|
printfv(_("%d entries in backup.\n"), $entries->length);
|
||||||
|
|
||||||
|
for ($i = $entries->length - 1; $i >= 0; $i--) {
|
||||||
|
try {
|
||||||
|
$entry = $entries->item($i);
|
||||||
|
|
||||||
|
$activity = new Activity($entry, $feed);
|
||||||
|
|
||||||
|
switch ($activity->verb) {
|
||||||
|
case ActivityVerb::FOLLOW:
|
||||||
|
subscribeProfile($user, $subject, $activity);
|
||||||
|
break;
|
||||||
|
case ActivityVerb::JOIN:
|
||||||
|
joinGroup($user, $activity);
|
||||||
|
break;
|
||||||
|
case ActivityVerb::POST:
|
||||||
|
postNote($user, $activity);
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
throw new Exception("Unknown verb: {$activity->verb}");
|
||||||
|
}
|
||||||
|
} catch (Exception $e) {
|
||||||
|
print $e->getMessage()."\n";
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function subscribeProfile($user, $subject, $activity)
|
||||||
|
{
|
||||||
|
$profile = $user->getProfile();
|
||||||
|
|
||||||
|
if ($activity->objects[0]->id == $subject->id) {
|
||||||
|
|
||||||
|
$other = $activity->actor;
|
||||||
|
$otherUser = User::staticGet('uri', $other->id);
|
||||||
|
|
||||||
|
if (!empty($otherUser)) {
|
||||||
|
$otherProfile = $otherUser->getProfile();
|
||||||
|
} else {
|
||||||
|
throw new Exception("Can't force remote user to subscribe.");
|
||||||
|
}
|
||||||
|
// XXX: don't do this for untrusted input!
|
||||||
|
Subscription::start($otherProfile, $profile);
|
||||||
|
|
||||||
|
} else if (empty($activity->actor) || $activity->actor->id == $subject->id) {
|
||||||
|
|
||||||
|
$other = $activity->objects[0];
|
||||||
|
$otherUser = User::staticGet('uri', $other->id);
|
||||||
|
|
||||||
|
if (!empty($otherUser)) {
|
||||||
|
$otherProfile = $otherUser->getProfile();
|
||||||
|
} else {
|
||||||
|
$oprofile = Ostatus_profile::ensureActivityObjectProfile($other);
|
||||||
|
$otherProfile = $oprofile->localProfile();
|
||||||
|
}
|
||||||
|
|
||||||
|
Subscription::start($profile, $otherProfile);
|
||||||
|
} else {
|
||||||
|
throw new Exception("This activity seems unrelated to our user.");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function joinGroup($user, $activity)
|
||||||
|
{
|
||||||
|
// XXX: check that actor == subject
|
||||||
|
|
||||||
|
$uri = $activity->objects[0]->id;
|
||||||
|
|
||||||
|
$group = User_group::staticGet('uri', $uri);
|
||||||
|
|
||||||
|
if (empty($group)) {
|
||||||
|
$oprofile = Ostatus_profile::ensureActivityObjectProfile($activity->objects[0]);
|
||||||
|
if (!$oprofile->isGroup()) {
|
||||||
|
throw new Exception("Remote profile is not a group!");
|
||||||
|
}
|
||||||
|
$group = $oprofile->localGroup();
|
||||||
|
}
|
||||||
|
|
||||||
|
assert(!empty($group));
|
||||||
|
|
||||||
|
if (Event::handle('StartJoinGroup', array($group, $user))) {
|
||||||
|
Group_member::join($group->id, $user->id);
|
||||||
|
Event::handle('EndJoinGroup', array($group, $user));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// XXX: largely cadged from Ostatus_profile::processNote()
|
||||||
|
|
||||||
|
function postNote($user, $activity)
|
||||||
|
{
|
||||||
|
$note = $activity->objects[0];
|
||||||
|
|
||||||
|
$sourceUri = $note->id;
|
||||||
|
|
||||||
|
$notice = Notice::staticGet('uri', $sourceUri);
|
||||||
|
|
||||||
|
if (!empty($notice)) {
|
||||||
|
// This is weird.
|
||||||
|
$orig = clone($notice);
|
||||||
|
$notice->profile_id = $user->id;
|
||||||
|
$notice->update($orig);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Use summary as fallback for content
|
||||||
|
|
||||||
|
if (!empty($note->content)) {
|
||||||
|
$sourceContent = $note->content;
|
||||||
|
} else if (!empty($note->summary)) {
|
||||||
|
$sourceContent = $note->summary;
|
||||||
|
} else if (!empty($note->title)) {
|
||||||
|
$sourceContent = $note->title;
|
||||||
|
} else {
|
||||||
|
// @fixme fetch from $sourceUrl?
|
||||||
|
// @todo i18n FIXME: use sprintf and add i18n.
|
||||||
|
throw new ClientException("No content for notice {$sourceUri}.");
|
||||||
|
}
|
||||||
|
|
||||||
|
// Get (safe!) HTML and text versions of the content
|
||||||
|
|
||||||
|
$rendered = purify($sourceContent);
|
||||||
|
$content = html_entity_decode(strip_tags($rendered));
|
||||||
|
|
||||||
|
$shortened = common_shorten_links($content);
|
||||||
|
|
||||||
|
$options = array('is_local' => Notice::LOCAL_PUBLIC,
|
||||||
|
'uri' => $sourceUri,
|
||||||
|
'rendered' => $rendered,
|
||||||
|
'replies' => array(),
|
||||||
|
'groups' => array(),
|
||||||
|
'tags' => array(),
|
||||||
|
'urls' => array());
|
||||||
|
|
||||||
|
// Check for optional attributes...
|
||||||
|
|
||||||
|
if (!empty($activity->time)) {
|
||||||
|
$options['created'] = common_sql_date($activity->time);
|
||||||
|
}
|
||||||
|
|
||||||
|
if ($activity->context) {
|
||||||
|
// Any individual or group attn: targets?
|
||||||
|
|
||||||
|
list($options['groups'], $options['replies']) = filterAttention($activity->context->attention);
|
||||||
|
|
||||||
|
// Maintain direct reply associations
|
||||||
|
// @fixme what about conversation ID?
|
||||||
|
if (!empty($activity->context->replyToID)) {
|
||||||
|
$orig = Notice::staticGet('uri',
|
||||||
|
$activity->context->replyToID);
|
||||||
|
if (!empty($orig)) {
|
||||||
|
$options['reply_to'] = $orig->id;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
$location = $activity->context->location;
|
||||||
|
|
||||||
|
if ($location) {
|
||||||
|
$options['lat'] = $location->lat;
|
||||||
|
$options['lon'] = $location->lon;
|
||||||
|
if ($location->location_id) {
|
||||||
|
$options['location_ns'] = $location->location_ns;
|
||||||
|
$options['location_id'] = $location->location_id;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Atom categories <-> hashtags
|
||||||
|
|
||||||
|
foreach ($activity->categories as $cat) {
|
||||||
|
if ($cat->term) {
|
||||||
|
$term = common_canonical_tag($cat->term);
|
||||||
|
if ($term) {
|
||||||
|
$options['tags'][] = $term;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Atom enclosures -> attachment URLs
|
||||||
|
foreach ($activity->enclosures as $href) {
|
||||||
|
// @fixme save these locally or....?
|
||||||
|
$options['urls'][] = $href;
|
||||||
|
}
|
||||||
|
|
||||||
|
$saved = Notice::saveNew($user->id,
|
||||||
|
$content,
|
||||||
|
'restore', // TODO: restore the actual source
|
||||||
|
$options);
|
||||||
|
|
||||||
|
return $saved;
|
||||||
|
}
|
||||||
|
|
||||||
|
function filterAttention($attn)
|
||||||
|
{
|
||||||
|
$groups = array();
|
||||||
|
$replies = array();
|
||||||
|
|
||||||
|
foreach (array_unique($attn) as $recipient) {
|
||||||
|
|
||||||
|
// Is the recipient a local user?
|
||||||
|
|
||||||
|
$user = User::staticGet('uri', $recipient);
|
||||||
|
|
||||||
|
if ($user) {
|
||||||
|
// @fixme sender verification, spam etc?
|
||||||
|
$replies[] = $recipient;
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Is the recipient a remote group?
|
||||||
|
$oprofile = Ostatus_profile::ensureProfileURI($recipient);
|
||||||
|
|
||||||
|
if ($oprofile) {
|
||||||
|
if (!$oprofile->isGroup()) {
|
||||||
|
// may be canonicalized or something
|
||||||
|
$replies[] = $oprofile->uri;
|
||||||
|
}
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Is the recipient a local group?
|
||||||
|
// @fixme uri on user_group isn't reliable yet
|
||||||
|
// $group = User_group::staticGet('uri', $recipient);
|
||||||
|
$id = OStatusPlugin::localGroupFromUrl($recipient);
|
||||||
|
|
||||||
|
if ($id) {
|
||||||
|
$group = User_group::staticGet('id', $id);
|
||||||
|
if ($group) {
|
||||||
|
// Deliver to all members of this local group if allowed.
|
||||||
|
$profile = $sender->localProfile();
|
||||||
|
if ($profile->isMember($group)) {
|
||||||
|
$groups[] = $group->id;
|
||||||
|
} else {
|
||||||
|
common_log(LOG_INFO, "Skipping reply to local group {$group->nickname} as sender {$profile->id} is not a member");
|
||||||
|
}
|
||||||
|
continue;
|
||||||
|
} else {
|
||||||
|
common_log(LOG_INFO, "Skipping reply to bogus group $recipient");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return array($groups, $replies);
|
||||||
|
}
|
||||||
|
|
||||||
|
function userFromSubject($subject)
|
||||||
|
{
|
||||||
|
$user = User::staticGet('uri', $subject->id);
|
||||||
|
|
||||||
|
if (empty($user)) {
|
||||||
|
$attrs =
|
||||||
|
array('nickname' => Ostatus_profile::getActivityObjectNickname($subject),
|
||||||
|
'uri' => $subject->id);
|
||||||
|
|
||||||
|
$user = User::register($attrs);
|
||||||
|
}
|
||||||
|
|
||||||
|
$profile = $user->getProfile();
|
||||||
|
Ostatus_profile::updateProfile($profile, $subject);
|
||||||
|
|
||||||
|
// FIXME: Update avatar
|
||||||
|
return $user;
|
||||||
|
}
|
||||||
|
|
||||||
|
function purify($content)
|
||||||
|
{
|
||||||
|
$config = array('safe' => 1,
|
||||||
|
'deny_attribute' => 'id,style,on*');
|
||||||
|
return htmLawed($content, $config);
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
try {
|
||||||
|
$user = getUser();
|
||||||
|
} catch (NoUserArgumentException $noae) {
|
||||||
|
$user = null;
|
||||||
|
}
|
||||||
|
$doc = getActivityStreamDocument();
|
||||||
|
importActivityStream($user, $doc);
|
||||||
|
} catch (Exception $e) {
|
||||||
|
print $e->getMessage()."\n";
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue
Block a user