set csrf and access_level in single user mode session

[tt-rss.git] / update.php

diff --git a/update.php b/update.php
index 92e2ce6ef085c7090180a6286e0b53ae10ebaa6e..68add9f0ccfc34563c19db262373f868dd53abb1 100755 (executable)
--- a/update.php
+++ b/update.php
@@ -1,10 +1,14 @@
 #!/usr/bin/php
 <?php
+       set_include_path(get_include_path() . PATH_SEPARATOR .
+               dirname(__FILE__) . "/include");
+
        define('DISABLE_SESSIONS', true);
 
        chdir(dirname(__FILE__));
 
        require_once "functions.php";
+       require_once "rssfuncs.php";
        require_once "sanity_check.php";
        require_once "config.php";
        require_once "db.php";
@@ -18,12 +22,13 @@
        if (!$op || $op == "-help") {
                print "Tiny Tiny RSS data update script.\n\n";
                print "Options:\n";
-               print "  -feeds         - update feeds\n";
-               print "  -feedbrowser   - update feedbrowser\n";
-               print "  -daemon        - start single-process update daemon\n";
-               print "  -cleanup-tags  - perform tags table maintenance\n";
-               print "  -get-feeds     - receive popular feeds from linked instances\n";
-               print "  -help          - show this help\n";
+               print "  -feeds              - update feeds\n";
+               print "  -feedbrowser        - update feedbrowser\n";
+               print "  -daemon             - start single-process update daemon\n";
+               print "  -cleanup-tags       - perform tags table maintenance\n";
+               print "  -get-feeds          - receive popular feeds from linked instances\n";
+               print "  -import USER FILE   - import articles from XML\n";
+               print "  -help               - show this help\n";
                return;
        }
 
@@ -45,14 +50,6 @@
        // Create a database connection.
        $link = db_connect(DB_HOST, DB_USER, DB_PASS, DB_NAME);
 
-       if (!$link) {
-               if (DB_TYPE == "mysql") {
-                       print mysql_error();
-               }
-               // PG seems to display its own errors just fine by default.
-               return;
-       }
-
        init_connection($link);
 
        if ($op == "-feeds") {
@@ -68,6 +65,8 @@
 
                $rc = cleanup_tags($link, 14, 50000);
                _debug("Cleaned $rc cached tags.");
+
+               get_linked_feeds($link);
        }
 
        if ($op == "-feedbrowser") {
@@ -102,6 +101,8 @@
                        $rc = cleanup_tags($link, 14, 50000);
 
                        _debug("Cleaned $rc cached tags.");
+
+                       get_linked_feeds($link);
                }
 
        }
@@ -115,7 +116,40 @@
                get_linked_feeds($link);
        }
 
+       if ($op == "-import") {
+               $username = $argv[2];
+               $filename = $argv[3];
+
+               if (!$username) {
+                       print "error: please specify username.\n";
+                       return;
+               }
+
+               if (!is_file($filename)) {
+                       print "error: input filename ($filename) doesn't exist.\n";
+                       return;
+               }
+
+               print "importing $filename for user $username...\n";
+
+               $result = db_query($link, "SELECT id FROM ttrss_users WHERE login = '$username'");
+
+               if (db_num_rows($result) == 0) {
+                       print "error: could not find user $username.\n";
+                       return;
+               }
+
+               $owner_uid = db_fetch_result($result, 0, "id");
+
+               perform_data_import($link, $filename, $owner_uid);
+
+       }
+
        db_close($link);
 
+       if ($lock_handle != false) {
+               fclose($lock_handle);
+       }
+
        unlink(LOCK_DIRECTORY . "/$lock_filename");
 ?>