include/uimport.php

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