classes/opml.php

   1 <?php
   2 class Opml extends Handler_Protected {
   3
   4         function csrf_ignore($method) {
   5                 $csrf_ignored = array("export", "import");
   6
   7                 return array_search($method, $csrf_ignored) !== false;
   8         }
   9
  10         function export() {
  11                 $output_name = $_REQUEST["filename"];
  12                 if (!$output_name) $output_name = "TinyTinyRSS.opml";
  13
  14                 $show_settings = $_REQUEST["settings"];
  15
  16                 $owner_uid = $_SESSION["uid"];
  17                 return $this->opml_export($output_name, $owner_uid, false, ($show_settings == 1));
  18         }
  19
  20         function import() {
  21                 $owner_uid = $_SESSION["uid"];
  22
  23                 header('Content-Type: text/html; charset=utf-8');
  24
  25                 print "<html>
  26                         <head>
  27                                 <link rel=\"stylesheet\" href=\"utility.css\" type=\"text/css\">
  28                                 <title>".__("OPML Utility")."</title>
  29                                 <meta http-equiv=\"Content-Type\" content=\"text/html; charset=utf-8\"/>
  30                         </head>
  31                         <body>
  32                         <div class=\"floatingLogo\"><img src=\"images/logo_wide.png\"></div>
  33                         <h1>".__('OPML Utility')."</h1>";
  34
  35                 add_feed_category($this->link, "Imported feeds");
  36
  37                 $this->opml_notice(__("Importing OPML..."));
  38                 $this->opml_import($owner_uid);
  39
  40                 print "<br><form method=\"GET\" action=\"prefs.php\">
  41                         <input type=\"submit\" value=\"".__("Return to preferences")."\">
  42                         </form>";
  43
  44                 print "</body></html>";
  45
  46
  47         }
  48
  49         // Export
  50
  51         private function opml_export_category($owner_uid, $cat_id, $hide_private_feeds=false) {
  52
  53                 if ($cat_id) {
  54                         $cat_qpart = "parent_cat = '$cat_id'";
  55                         $feed_cat_qpart = "cat_id = '$cat_id'";
  56                 } else {
  57                         $cat_qpart = "parent_cat IS NULL";
  58                         $feed_cat_qpart = "cat_id IS NULL";
  59                 }
  60
  61                 if ($hide_private_feeds)
  62                         $hide_qpart = "(private IS false AND auth_login = '' AND auth_pass = '')";
  63                 else
  64                         $hide_qpart = "true";
  65
  66                 $out = "";
  67
  68                 if ($cat_id) {
  69                         $result = db_query($this->link, "SELECT title FROM ttrss_feed_categories WHERE id = '$cat_id'
  70                                 AND owner_uid = '$owner_uid'");
  71                         $cat_title = db_fetch_result($result, 0, "title");
  72                 }
  73
  74                 if ($cat_title) $out .= "<outline title=\"$cat_title\">\n";
  75
  76                 $result = db_query($this->link, "SELECT id,title
  77                         FROM ttrss_feed_categories WHERE
  78                         $cat_qpart AND owner_uid = '$owner_uid' ORDER BY order_id, title");
  79
  80                 while ($line = db_fetch_assoc($result)) {
  81                         $title = htmlspecialchars($line["title"]);
  82                         $out .= $this->opml_export_category($owner_uid, $line["id"], $hide_private_feeds);
  83                 }
  84
  85                 $feeds_result = db_query($this->link, "select title, feed_url, site_url
  86                                 from ttrss_feeds where $feed_cat_qpart AND owner_uid = '$owner_uid' AND $hide_qpart
  87                                 order by order_id, title");
  88
  89                 while ($fline = db_fetch_assoc($feeds_result)) {
  90                         $title = htmlspecialchars($fline["title"]);
  91                         $url = htmlspecialchars($fline["feed_url"]);
  92                         $site_url = htmlspecialchars($fline["site_url"]);
  93
  94                         if ($site_url) {
  95                                 $html_url_qpart = "htmlUrl=\"$site_url\"";
  96                         } else {
  97                                 $html_url_qpart = "";
  98                         }
  99
 100                         $out .= "<outline text=\"$title\" xmlUrl=\"$url\" $html_url_qpart/>\n";
 101                 }
 102
 103                 if ($cat_title) $out .= "</outline>\n";
 104
 105                 return $out;
 106         }
 107
 108         function opml_export($name, $owner_uid, $hide_private_feeds=false, $include_settings=true) {
 109                 if (!$owner_uid) return;
 110
 111                 if (!isset($_REQUEST["debug"])) {
 112                         header("Content-type: application/xml+opml");
 113                         header("Content-Disposition: attachment; filename=" . $name );
 114                 } else {
 115                         header("Content-type: text/xml");
 116                 }
 117
 118                 $out = "<?xml version=\"1.0\" encoding=\"utf-8\"?".">";
 119
 120                 $out .= "<opml version=\"1.0\">";
 121                 $out .= "<head>
 122                         <dateCreated>" . date("r", time()) . "</dateCreated>
 123                         <title>Tiny Tiny RSS Feed Export</title>
 124                 </head>";
 125                 $out .= "<body>";
 126
 127                 $out .= $this->opml_export_category($owner_uid, false, $hide_private_feeds);
 128
 129                 # export tt-rss settings
 130
 131                 if ($include_settings) {
 132                         $out .= "<outline title=\"tt-rss-prefs\" schema-version=\"".SCHEMA_VERSION."\">";
 133
 134                         $result = db_query($this->link, "SELECT pref_name, value FROM ttrss_user_prefs WHERE
 135                            profile IS NULL AND owner_uid = " . $_SESSION["uid"] . " ORDER BY pref_name");
 136
 137                         while ($line = db_fetch_assoc($result)) {
 138                                 $name = $line["pref_name"];
 139                                 $value = htmlspecialchars($line["value"]);
 140
 141                                 $out .= "<outline pref-name=\"$name\" value=\"$value\"/>";
 142                         }
 143
 144                         $out .= "</outline>";
 145
 146                         $out .= "<outline title=\"tt-rss-labels\" schema-version=\"".SCHEMA_VERSION."\">";
 147
 148                         $result = db_query($this->link, "SELECT * FROM ttrss_labels2 WHERE
 149                                 owner_uid = " . $_SESSION['uid']);
 150
 151                         while ($line = db_fetch_assoc($result)) {
 152                                 $name = htmlspecialchars($line['caption']);
 153                                 $fg_color = htmlspecialchars($line['fg_color']);
 154                                 $bg_color = htmlspecialchars($line['bg_color']);
 155
 156                                 $out .= "<outline label-name=\"$name\" label-fg-color=\"$fg_color\" label-bg-color=\"$bg_color\"/>";
 157
 158                         }
 159
 160                         $out .= "</outline>";
 161
 162                         $out .= "<outline title=\"tt-rss-filters\" schema-version=\"".SCHEMA_VERSION."\">";
 163
 164                         $result = db_query($this->link, "SELECT * FROM ttrss_filters2
 165                                 WHERE owner_uid = ".$_SESSION["uid"]." ORDER BY id");
 166
 167                         while ($line = db_fetch_assoc($result)) {
 168                                 foreach (array('enabled', 'match_any_rule') as $b) {
 169                                         $line[$b] = sql_bool_to_bool($line[$b]);
 170                                 }
 171
 172                                 $line["rules"] = array();
 173                                 $line["actions"] = array();
 174
 175                                 $tmp_result = db_query($this->link, "SELECT * FROM ttrss_filters2_rules
 176                                         WHERE filter_id = ".$line["id"]);
 177
 178                                 while ($tmp_line = db_fetch_assoc($tmp_result)) {
 179                                         unset($tmp_line["id"]);
 180                                         unset($tmp_line["filter_id"]);
 181
 182                                         $cat_filter = sql_bool_to_bool($tmp_line["cat_filter"]);
 183
 184                                         if ($cat_filter && $tmp_line["cat_id"] || $tmp_line["feed_id"]) {
 185                                                 $tmp_line["feed"] = getFeedTitle($this->link,
 186                                                         $cat_filter ? $tmp_line["cat_id"] : $tmp_line["feed_id"],
 187                                                         $cat_filter);
 188                                         } else {
 189                                                 $tmp_line["feed"] = "";
 190                                         }
 191
 192                                         $tmp_line["cat_filter"] = sql_bool_to_bool($tmp_line["cat_filter"]);
 193
 194                                         unset($tmp_line["feed_id"]);
 195                                         unset($tmp_line["cat_id"]);
 196
 197                                         array_push($line["rules"], $tmp_line);
 198                                 }
 199
 200                                 $tmp_result = db_query($this->link, "SELECT * FROM ttrss_filters2_actions
 201                                         WHERE filter_id = ".$line["id"]);
 202
 203                                 while ($tmp_line = db_fetch_assoc($tmp_result)) {
 204                                         unset($tmp_line["id"]);
 205                                         unset($tmp_line["filter_id"]);
 206
 207                                         array_push($line["actions"], $tmp_line);
 208                                 }
 209
 210                                 unset($line["id"]);
 211                                 unset($line["owner_uid"]);
 212                                 $filter = json_encode($line);
 213
 214                                 $out .= "<outline filter-type=\"2\">$filter</outline>";
 215
 216                         }
 217
 218
 219                         $out .= "</outline>";
 220                 }
 221
 222                 $out .= "</body></opml>";
 223
 224                 // Format output.
 225                 $doc = new DOMDocument();
 226                 $doc->formatOutput = true;
 227                 $doc->preserveWhiteSpace = false;
 228                 $doc->loadXML($out);
 229
 230                 $xpath = new DOMXpath($doc);
 231                 $outlines = $xpath->query("//outline[@title]");
 232
 233                 // cleanup empty categories
 234                 foreach ($outlines as $node) {
 235                         if ($node->getElementsByTagName('outline')->length == 0)
 236                                 $node->parentNode->removeChild($node);
 237                 }
 238
 239                 $res = $doc->saveXML();
 240
 241                 // saveXML uses a two-space indent.  Change to tabs.
 242                 $res = preg_replace_callback('/^(?:  )+/mu',
 243                         create_function(
 244                                 '$matches',
 245                                 'return str_repeat("\t", intval(strlen($matches[0])/2));'),
 246                         $res);
 247
 248                 print $res;
 249         }
 250
 251         // Import
 252
 253         private function opml_import_feed($doc, $node, $cat_id, $owner_uid) {
 254                 $attrs = $node->attributes;
 255
 256                 $feed_title = db_escape_string($attrs->getNamedItem('text')->nodeValue);
 257                 if (!$feed_title) $feed_title = db_escape_string($attrs->getNamedItem('title')->nodeValue);
 258
 259                 $feed_url = db_escape_string($attrs->getNamedItem('xmlUrl')->nodeValue);
 260                 if (!$feed_url) $feed_url = db_escape_string($attrs->getNamedItem('xmlURL')->nodeValue);
 261
 262                 $site_url = db_escape_string($attrs->getNamedItem('htmlUrl')->nodeValue);
 263
 264                 if ($feed_url && $feed_title) {
 265                         $result = db_query($this->link, "SELECT id FROM ttrss_feeds WHERE
 266                                 feed_url = '$feed_url' AND owner_uid = '$owner_uid'");
 267
 268                         if (db_num_rows($result) == 0) {
 269                                 #$this->opml_notice("[FEED] [$feed_title/$feed_url] dst_CAT=$cat_id");
 270                                 $this->opml_notice(T_sprintf("Adding feed: %s", $feed_title));
 271
 272                                 $query = "INSERT INTO ttrss_feeds
 273                                         (title, feed_url, owner_uid, cat_id, site_url, order_id) VALUES
 274                                         ('$feed_title', '$feed_url', '$owner_uid',
 275                                         '$cat_id', '$site_url', 0)";
 276                                 db_query($this->link, $query);
 277
 278                         } else {
 279                                 $this->opml_notice(T_sprintf("Duplicate feed: %s", $feed_title));
 280                         }
 281                 }
 282         }
 283
 284         private function opml_import_label($doc, $node, $owner_uid) {
 285                 $attrs = $node->attributes;
 286                 $label_name = db_escape_string($attrs->getNamedItem('label-name')->nodeValue);
 287
 288                 if ($label_name) {
 289                         $fg_color = db_escape_string($attrs->getNamedItem('label-fg-color')->nodeValue);
 290                         $bg_color = db_escape_string($attrs->getNamedItem('label-bg-color')->nodeValue);
 291
 292                         if (!label_find_id($this->link, $label_name, $_SESSION['uid'])) {
 293                                 $this->opml_notice(T_sprintf("Adding label %s", htmlspecialchars($label_name)));
 294                                 label_create($this->link, $label_name, $fg_color, $bg_color);
 295                         } else {
 296                                 $this->opml_notice(T_sprintf("Duplicate label: %s", htmlspecialchars($label_name)));
 297                         }
 298                 }
 299         }
 300
 301         private function opml_import_preference($doc, $node, $owner_uid) {
 302                 $attrs = $node->attributes;
 303                 $pref_name = db_escape_string($attrs->getNamedItem('pref-name')->nodeValue);
 304
 305                 if ($pref_name) {
 306                         $pref_value = db_escape_string($attrs->getNamedItem('value')->nodeValue);
 307
 308                         $this->opml_notice(T_sprintf("Setting preference key %s to %s",
 309                                 $pref_name, $pref_value));
 310
 311                         set_pref($this->link, $pref_name, $pref_value);
 312                 }
 313         }
 314
 315         /* private function opml_import_filter($doc, $node, $owner_uid) {
 316                 $attrs = $node->attributes;
 317
 318                 $filter_name = db_escape_string($attrs->getNamedItem('filter-name')->nodeValue);
 319
 320                 if ($filter_name) {
 321
 322                 $filter = json_decode($node->nodeValue, true);
 323
 324                         if ($filter) {
 325                                 $reg_exp = db_escape_string($filter['reg_exp']);
 326                                 $filter_type = (int)$filter['filter_type'];
 327                                 $action_id = (int)$filter['action_id'];
 328
 329                                 $result = db_query($this->link, "SELECT id FROM ttrss_filters WHERE
 330                                         reg_exp = '$reg_exp' AND
 331                                         filter_type = '$filter_type' AND
 332                                         action_id = '$action_id' AND
 333                                         owner_uid = " .$_SESSION['uid']);
 334
 335                                 if (db_num_rows($result) == 0) {
 336                                         $enabled = bool_to_sql_bool($filter['enabled']);
 337                                         $action_param = db_escape_string($filter['action_param']);
 338                                         $inverse = bool_to_sql_bool($filter['inverse']);
 339                                         $filter_param = db_escape_string($filter['filter_param']);
 340                                         $cat_filter = bool_to_sql_bool($filter['cat_filter']);
 341
 342                                         $feed_url = db_escape_string($filter['feed_url']);
 343                                         $cat_title = db_escape_string($filter['cat_title']);
 344
 345                                         $result = db_query($this->link, "SELECT id FROM ttrss_feeds WHERE
 346                                                 feed_url = '$feed_url' AND owner_uid = $owner_uid");
 347
 348                                         if (db_num_rows($result) != 0) {
 349                                                 $feed_id = db_fetch_result($result, 0, "id");
 350                                         } else {
 351                                                 $feed_id = "NULL";
 352                                         }
 353
 354                                         $result = db_query($this->link, "SELECT id FROM ttrss_feed_categories WHERE
 355                                                 title = '$cat_title' AND  owner_uid = $owner_uid");
 356
 357                                         if (db_num_rows($result) != 0) {
 358                                                 $cat_id = db_fetch_result($result, 0, "id");
 359                                         } else {
 360                                                 $cat_id = "NULL";
 361                                         }
 362
 363                                         $this->opml_notice(T_sprintf("Adding filter %s", htmlspecialchars($reg_exp)));
 364
 365                                         $query = "INSERT INTO ttrss_filters (filter_type, action_id,
 366                                                         enabled, inverse, action_param, filter_param,
 367                                                         cat_filter, feed_id,
 368                                                         cat_id, reg_exp,
 369                                                         owner_uid)
 370                                                 VALUES ($filter_type, $action_id,
 371                                                         $enabled, $inverse, '$action_param', '$filter_param',
 372                                                         $cat_filter, $feed_id,
 373                                                         $cat_id, '$reg_exp', ".
 374                                                         $_SESSION['uid'].")";
 375
 376                                         db_query($this->link, $query);
 377
 378                                 } else {
 379                                         $this->opml_notice(T_sprintf("Duplicate filter %s", htmlspecialchars($reg_exp)));
 380                                 }
 381                         }
 382                 }
 383         } */
 384
 385         private function opml_import_category($doc, $root_node, $owner_uid, $parent_id) {
 386                 $body = $doc->getElementsByTagName('body');
 387
 388                 $default_cat_id = (int) get_feed_category($this->link, 'Imported feeds', false);
 389
 390                 if ($root_node) {
 391                         $cat_title = db_escape_string($root_node->attributes->getNamedItem('title')->nodeValue);
 392
 393                         if (!in_array($cat_title, array("tt-rss-filters", "tt-rss-labels", "tt-rss-prefs"))) {
 394                                 $cat_id = get_feed_category($this->link, $cat_title, $parent_id);
 395                                 db_query($this->link, "BEGIN");
 396                                 if ($cat_id === false) {
 397                                         add_feed_category($this->link, $cat_title, $parent_id);
 398                                         $cat_id = get_feed_category($this->link, $cat_title, $parent_id);
 399                                 }
 400                                 db_query($this->link, "COMMIT");
 401                         } else {
 402                                 $cat_id = 0;
 403                         }
 404
 405                         $outlines = $root_node->childNodes;
 406
 407                 } else {
 408                         $xpath = new DOMXpath($doc);
 409                         $outlines = $xpath->query("//opml/body/outline");
 410
 411                         $cat_id = 0;
 412                 }
 413
 414                 #$this->opml_notice("[CAT] $cat_title id: $cat_id P_id: $parent_id");
 415                 $this->opml_notice(T_sprintf("Processing category: %s", $cat_title ? $cat_title : __("Uncategorized")));
 416
 417                 foreach ($outlines as $node) {
 418                         if ($node->hasAttributes() && strtolower($node->tagName) == "outline") {
 419                                 $attrs = $node->attributes;
 420                                 $node_cat_title = db_escape_string($attrs->getNamedItem('title')->nodeValue);
 421                                 $node_feed_url = db_escape_string($attrs->getNamedItem('xmlUrl')->nodeValue);
 422
 423                                 if ($node_cat_title && !$node_feed_url) {
 424                                         $this->opml_import_category($doc, $node, $owner_uid, $cat_id);
 425                                 } else {
 426
 427                                         if (!$cat_id) {
 428                                                 $dst_cat_id = $default_cat_id;
 429                                         } else {
 430                                                 $dst_cat_id = $cat_id;
 431                                         }
 432
 433                                         switch ($cat_title) {
 434                                         case "tt-rss-prefs":
 435                                                 $this->opml_import_preference($doc, $node, $owner_uid);
 436                                                 break;
 437                                         case "tt-rss-labels":
 438                                                 $this->opml_import_label($doc, $node, $owner_uid);
 439                                                 break;
 440                                         case "tt-rss-filters":
 441                                                 //$this->opml_import_filter($doc, $node, $owner_uid);
 442                                                 break;
 443                                         default:
 444                                                 $this->opml_import_feed($doc, $node, $dst_cat_id, $owner_uid);
 445                                         }
 446                                 }
 447                         }
 448                 }
 449         }
 450
 451         function opml_import($owner_uid) {
 452                 if (!$owner_uid) return;
 453
 454                 $debug = isset($_REQUEST["debug"]);
 455                 $doc = false;
 456
 457                 #if ($debug) $doc = DOMDocument::load("/tmp/test.opml");
 458
 459                 if (is_file($_FILES['opml_file']['tmp_name'])) {
 460                         $doc = DOMDocument::load($_FILES['opml_file']['tmp_name']);
 461                 } else if (!$doc) {
 462                         print_error(__('Error: please upload OPML file.'));
 463                         return;
 464                 }
 465
 466                 if ($doc) {
 467                         $this->opml_import_category($doc, false, $owner_uid, false);
 468                 } else {
 469                         print_error(__('Error while parsing document.'));
 470                 }
 471         }
 472
 473         private function opml_notice($msg) {
 474                 print "$msg<br/>";
 475         }
 476
 477 }
 478 ?>