include/uimport.php

   1 <?php
   2
   3 use Friendica\App;
   4 use Friendica\Core\System;
   5 use Friendica\Core\PConfig;
   6 use Friendica\Core\Worker;
   7 use Friendica\Database\DBM;
   8
   9 require_once("include/Photo.php");
  10 define("IMPORT_DEBUG", False);
  11
  12 function last_insert_id() {
  13         if (IMPORT_DEBUG) {
  14                 return 1;
  15         }
  16
  17         return dba::lastInsertId();
  18 }
  19
  20 /**
  21  * Remove columns from array $arr that aren't in table $table
  22  *
  23  * @param string $table Table name
  24  * @param array &$arr Column=>Value array from json (by ref)
  25  */
  26 function check_cols($table, &$arr) {
  27         $query = sprintf("SHOW COLUMNS IN `%s`", dbesc($table));
  28         logger("uimport: $query", LOGGER_DEBUG);
  29         $r = q($query);
  30         $tcols = array();
  31         // get a plain array of column names
  32         foreach ($r as $tcol) {
  33                 $tcols[] = $tcol['Field'];
  34         }
  35         // remove inexistent columns
  36         foreach ($arr as $icol => $ival) {
  37                 if (!in_array($icol, $tcols)) {
  38                         unset($arr[$icol]);
  39                 }
  40         }
  41 }
  42
  43 /**
  44  * Import data into table $table
  45  *
  46  * @param string $table Table name
  47  * @param array $arr Column=>Value array from json
  48  */
  49 function db_import_assoc($table, $arr) {
  50         if (isset($arr['id']))
  51                 unset($arr['id']);
  52         check_cols($table, $arr);
  53         $cols = implode("`,`", array_map('dbesc', array_keys($arr)));
  54         $vals = implode("','", array_map('dbesc', array_values($arr)));
  55         $query = "INSERT INTO `$table` (`$cols`) VALUES ('$vals')";
  56         logger("uimport: $query", LOGGER_TRACE);
  57         if (IMPORT_DEBUG) {
  58                 return true;
  59         }
  60         return q($query);
  61 }
  62
  63 /**
  64  * @brief Import account file exported from mod/uexport
  65  *
  66  * @param App $a Friendica App Class
  67  * @param array $file array from $_FILES
  68  */
  69 function import_account(App $a, $file) {
  70         logger("Start user import from " . $file['tmp_name']);
  71         /*
  72           STEPS
  73           1. checks
  74           2. replace old baseurl with new baseurl
  75           3. import data (look at user id and contacts id)
  76           4. archive non-dfrn contacts
  77           5. send message to dfrn contacts
  78          */
  79
  80         $account = json_decode(file_get_contents($file['tmp_name']), true);
  81         if ($account === null) {
  82                 notice(t("Error decoding account file"));
  83                 return;
  84         }
  85
  86
  87         if (!x($account, 'version')) {
  88                 notice(t("Error! No version data in file! This is not a Friendica account file?"));
  89                 return;
  90         }
  91
  92         /*
  93          * @TODO Old-lost code?
  94         // this is not required as we remove columns in json not in current db schema
  95         if ($account['schema'] != DB_UPDATE_VERSION) {
  96                 notice(t("Error! I can't import this file: DB schema version is not compatible."));
  97                 return;
  98         }
  99         */
 100
 101         // check for username
 102         $r = q("SELECT uid FROM user WHERE nickname='%s'", $account['user']['nickname']);
 103         if ($r === false) {
 104                 logger("uimport:check nickname : ERROR : " . dba::errorMessage(), LOGGER_NORMAL);
 105                 notice(t('Error! Cannot check nickname'));
 106                 return;
 107         }
 108         if (DBM::is_result($r) > 0) {
 109                 notice(sprintf(t("User '%s' already exists on this server!"), $account['user']['nickname']));
 110                 return;
 111         }
 112         // check if username matches deleted account
 113         $r = q("SELECT id FROM userd WHERE username='%s'", $account['user']['nickname']);
 114         if ($r === false) {
 115                 logger("uimport:check nickname : ERROR : " . dba::errorMessage(), LOGGER_NORMAL);
 116                 notice(t('Error! Cannot check nickname'));
 117                 return;
 118         }
 119         if (DBM::is_result($r) > 0) {
 120                 notice(sprintf(t("User '%s' already exists on this server!"), $account['user']['nickname']));
 121                 return;
 122         }
 123
 124         $oldbaseurl = $account['baseurl'];
 125         $newbaseurl = System::baseUrl();
 126
 127         $oldaddr = str_replace('http://', '@', normalise_link($oldbaseurl));
 128         $newaddr = str_replace('http://', '@', normalise_link($newbaseurl));
 129
 130         if (!empty($account['profile']['addr'])) {
 131                 $old_handle = $account['profile']['addr'];
 132         } else {
 133                 $old_handle = $account['user']['nickname'].$oldaddr;
 134         }
 135
 136         $olduid = $account['user']['uid'];
 137
 138         unset($account['user']['uid']);
 139         unset($account['user']['account_expired']);
 140         unset($account['user']['account_expires_on']);
 141         unset($account['user']['expire_notification_sent']);
 142
 143         foreach ($account['user'] as $k => &$v) {
 144                 $v = str_replace(array($oldbaseurl, $oldaddr), array($newbaseurl, $newaddr), $v);
 145         }
 146
 147         // import user
 148         $r = db_import_assoc('user', $account['user']);
 149         if ($r === false) {
 150                 logger("uimport:insert user : ERROR : " . dba::errorMessage(), LOGGER_NORMAL);
 151                 notice(t("User creation error"));
 152                 return;
 153         }
 154         $newuid = last_insert_id();
 155         //~ $newuid = 1;
 156
 157         PConfig::set($newuid, 'system', 'previous_addr', $old_handle);
 158
 159         // Generate a new guid for the account. Otherwise there will be problems with diaspora
 160         q("UPDATE `user` SET `guid` = '%s' WHERE `uid` = %d",
 161                 dbesc(generate_user_guid()), intval($newuid));
 162
 163         foreach ($account['profile'] as &$profile) {
 164                 foreach ($profile as $k => &$v) {
 165                         $v = str_replace(array($oldbaseurl, $oldaddr), array($newbaseurl, $newaddr), $v);
 166                         foreach (array("profile", "avatar") as $k) {
 167                                 $v = str_replace($oldbaseurl . "/photo/" . $k . "/" . $olduid . ".jpg", $newbaseurl . "/photo/" . $k . "/" . $newuid . ".jpg", $v);
 168                         }
 169                 }
 170                 $profile['uid'] = $newuid;
 171                 $r = db_import_assoc('profile', $profile);
 172                 if ($r === false) {
 173                         logger("uimport:insert profile " . $profile['profile-name'] . " : ERROR : " . dba::errorMessage(), LOGGER_NORMAL);
 174                         info(t("User profile creation error"));
 175                         dba::delete('user', array('uid' => $newuid));
 176                         return;
 177                 }
 178         }
 179
 180         $errorcount = 0;
 181         foreach ($account['contact'] as &$contact) {
 182                 if ($contact['uid'] == $olduid && $contact['self'] == '1') {
 183                         foreach ($contact as $k => &$v) {
 184                                 $v = str_replace(array($oldbaseurl, $oldaddr), array($newbaseurl, $newaddr), $v);
 185                                 foreach (array("profile", "avatar", "micro") as $k) {
 186                                         $v = str_replace($oldbaseurl . "/photo/" . $k . "/" . $olduid . ".jpg", $newbaseurl . "/photo/" . $k . "/" . $newuid . ".jpg", $v);
 187                                 }
 188                         }
 189                 }
 190                 if ($contact['uid'] == $olduid && $contact['self'] == '0') {
 191                         // set contacts 'avatar-date' to NULL_DATE to let worker to update urls
 192                         $contact["avatar-date"] = NULL_DATE;
 193
 194                         switch ($contact['network']) {
 195                                 case NETWORK_DFRN:
 196                                 case NETWORK_DIASPORA:
 197                                         //  send relocate message (below)
 198                                         break;
 199                                 case NETWORK_ZOT:
 200                                         /// @TODO handle zot network
 201                                         break;
 202                                 case NETWORK_MAIL2:
 203                                         /// @TODO ?
 204                                         break;
 205                                 case NETWORK_FEED:
 206                                 case NETWORK_MAIL:
 207                                         // Nothing to do
 208                                         break;
 209                                 default:
 210                                         // archive other contacts
 211                                         $contact['archive'] = "1";
 212                         }
 213                 }
 214                 $contact['uid'] = $newuid;
 215                 $r = db_import_assoc('contact', $contact);
 216                 if ($r === false) {
 217                         logger("uimport:insert contact " . $contact['nick'] . "," . $contact['network'] . " : ERROR : " . dba::errorMessage(), LOGGER_NORMAL);
 218                         $errorcount++;
 219                 } else {
 220                         $contact['newid'] = last_insert_id();
 221                 }
 222         }
 223         if ($errorcount > 0) {
 224                 notice(sprintf(tt("%d contact not imported", "%d contacts not imported", $errorcount), $errorcount));
 225         }
 226
 227         foreach ($account['group'] as &$group) {
 228                 $group['uid'] = $newuid;
 229                 $r = db_import_assoc('group', $group);
 230                 if ($r === false) {
 231                         logger("uimport:insert group " . $group['name'] . " : ERROR : " . dba::errorMessage(), LOGGER_NORMAL);
 232                 } else {
 233                         $group['newid'] = last_insert_id();
 234                 }
 235         }
 236
 237         foreach ($account['group_member'] as &$group_member) {
 238                 $group_member['uid'] = $newuid;
 239
 240                 $import = 0;
 241                 foreach ($account['group'] as $group) {
 242                         if ($group['id'] == $group_member['gid'] && isset($group['newid'])) {
 243                                 $group_member['gid'] = $group['newid'];
 244                                 $import++;
 245                                 break;
 246                         }
 247                 }
 248                 foreach ($account['contact'] as $contact) {
 249                         if ($contact['id'] == $group_member['contact-id'] && isset($contact['newid'])) {
 250                                 $group_member['contact-id'] = $contact['newid'];
 251                                 $import++;
 252                                 break;
 253                         }
 254                 }
 255                 if ($import == 2) {
 256                         $r = db_import_assoc('group_member', $group_member);
 257                         if ($r === false) {
 258                                 logger("uimport:insert group member " . $group_member['id'] . " : ERROR : " . dba::errorMessage(), LOGGER_NORMAL);
 259                         }
 260                 }
 261         }
 262
 263         foreach ($account['photo'] as &$photo) {
 264                 $photo['uid'] = $newuid;
 265                 $photo['data'] = hex2bin($photo['data']);
 266
 267                 $p = new Photo($photo['data'], $photo['type']);
 268                 $r = $p->store(
 269                                 $photo['uid'], $photo['contact-id'], //0
 270                                 $photo['resource-id'], $photo['filename'], $photo['album'], $photo['scale'], $photo['profile'], //1
 271                                 $photo['allow_cid'], $photo['allow_gid'], $photo['deny_cid'], $photo['deny_gid']
 272                 );
 273
 274                 if ($r === false) {
 275                         logger("uimport:insert photo " . $photo['resource-id'] . "," . $photo['scale'] . " : ERROR : " . dba::errorMessage(), LOGGER_NORMAL);
 276                 }
 277         }
 278
 279         foreach ($account['pconfig'] as &$pconfig) {
 280                 $pconfig['uid'] = $newuid;
 281                 $r = db_import_assoc('pconfig', $pconfig);
 282                 if ($r === false) {
 283                         logger("uimport:insert pconfig " . $pconfig['id'] . " : ERROR : " . dba::errorMessage(), LOGGER_NORMAL);
 284                 }
 285         }
 286
 287         // send relocate messages
 288         Worker::add(PRIORITY_HIGH, 'Notifier', 'relocate', $newuid);
 289
 290         info(t("Done. You can now login with your username and password"));
 291         goaway(System::baseUrl() . "/login");
 292 }