inc/poche/Database.class.php

   1 <?php
   2 /**
   3  * wallabag, self hostable application allowing you to not miss any content anymore
   4  *
   5  * @category   wallabag
   6  * @author     Nicolas Lœuillet <nicolas@loeuillet.org>
   7  * @copyright  2013
   8  * @license    http://www.wtfpl.net/ see COPYING file
   9  */
  10
  11 class Database {
  12     var $handle;
  13
  14     function __construct()
  15     {
  16         switch (STORAGE) {
  17             case 'sqlite':
  18                 $db_path = 'sqlite:' . STORAGE_SQLITE;
  19                 $this->handle = new PDO($db_path);
  20                 break;
  21             case 'mysql':
  22                 $db_path = 'mysql:host=' . STORAGE_SERVER . ';dbname=' . STORAGE_DB;
  23                 $this->handle = new PDO($db_path, STORAGE_USER, STORAGE_PASSWORD);
  24                 break;
  25             case 'postgres':
  26                 $db_path = 'pgsql:host=' . STORAGE_SERVER . ';dbname=' . STORAGE_DB;
  27                 $this->handle = new PDO($db_path, STORAGE_USER, STORAGE_PASSWORD);
  28                 break;
  29         }
  30
  31         $this->handle->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
  32         Tools::logm('storage type ' . STORAGE);
  33     }
  34
  35     private function getHandle() {
  36         return $this->handle;
  37     }
  38
  39     public function isInstalled() {
  40         $sql = "SELECT username FROM users";
  41         $query = $this->executeQuery($sql, array());
  42         if ($query == false) {
  43             die(STORAGE . ' database looks empty. You have to create it (you can find database structure in install folder).');
  44         }
  45         $hasAdmin = count($query->fetchAll());
  46
  47         if ($hasAdmin == 0)
  48             return false;
  49
  50         return true;
  51     }
  52
  53     public function checkTags() {
  54
  55         if (STORAGE == 'sqlite') {
  56             $sql = '
  57                 CREATE TABLE IF NOT EXISTS tags (
  58                     id INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL UNIQUE,
  59                     value TEXT
  60                 )';
  61         }
  62         elseif(STORAGE == 'mysql') {
  63             $sql = '
  64                 CREATE TABLE IF NOT EXISTS `tags` (
  65                   `id` int(11) NOT NULL AUTO_INCREMENT,
  66                   `value` varchar(255) NOT NULL,
  67                   PRIMARY KEY (`id`)
  68                 ) ENGINE=InnoDB DEFAULT CHARSET=utf8;
  69             ';
  70         }
  71         else {
  72             $sql = '
  73                 CREATE TABLE tags (
  74                   id bigserial primary key,
  75                   value varchar(255) NOT NULL
  76                 );
  77             ';
  78         }
  79
  80         $query = $this->executeQuery($sql, array());
  81
  82         if (STORAGE == 'sqlite') {
  83             $sql = '
  84                 CREATE TABLE IF NOT EXISTS tags_entries (
  85                     id INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL UNIQUE,
  86                     entry_id INTEGER,
  87                     tag_id INTEGER,
  88                     FOREIGN KEY(entry_id) REFERENCES entries(id) ON DELETE CASCADE,
  89                     FOREIGN KEY(tag_id) REFERENCES tags(id) ON DELETE CASCADE
  90                 )';
  91         }
  92         elseif(STORAGE == 'mysql') {
  93             $sql = '
  94                 CREATE TABLE IF NOT EXISTS `tags_entries` (
  95                   `id` int(11) NOT NULL AUTO_INCREMENT,
  96                   `entry_id` int(11) NOT NULL,
  97                   `tag_id` int(11) NOT NULL,
  98                   FOREIGN KEY(entry_id) REFERENCES entries(id) ON DELETE CASCADE,
  99                   FOREIGN KEY(tag_id) REFERENCES tags(id) ON DELETE CASCADE,
 100                   PRIMARY KEY (`id`)
 101                 ) ENGINE=InnoDB  DEFAULT CHARSET=utf8;
 102             ';
 103         }
 104         else {
 105             $sql = '
 106                 CREATE TABLE tags_entries (
 107                   id bigserial primary key,
 108                   entry_id integer NOT NULL,
 109                   tag_id integer NOT NULL
 110                 )
 111             ';
 112         }
 113
 114         $query = $this->executeQuery($sql, array());
 115     }
 116
 117     public function install($login, $password) {
 118         $sql = 'INSERT INTO users ( username, password, name, email) VALUES (?, ?, ?, ?)';
 119         $params = array($login, $password, $login, ' ');
 120         $query = $this->executeQuery($sql, $params);
 121
 122         $sequence = '';
 123         if (STORAGE == 'postgres') {
 124             $sequence = 'users_id_seq';
 125         }
 126
 127         $id_user = intval($this->getLastId($sequence));
 128
 129         $sql = 'INSERT INTO users_config ( user_id, name, value ) VALUES (?, ?, ?)';
 130         $params = array($id_user, 'pager', PAGINATION);
 131         $query = $this->executeQuery($sql, $params);
 132
 133         $sql = 'INSERT INTO users_config ( user_id, name, value ) VALUES (?, ?, ?)';
 134         $params = array($id_user, 'language', LANG);
 135         $query = $this->executeQuery($sql, $params);
 136
 137         $sql = 'INSERT INTO users_config ( user_id, name, value ) VALUES (?, ?, ?)';
 138         $params = array($id_user, 'theme', DEFAULT_THEME);
 139         $query = $this->executeQuery($sql, $params);
 140
 141         return TRUE;
 142     }
 143
 144     public function getConfigUser($id) {
 145         $sql = "SELECT * FROM users_config WHERE user_id = ?";
 146         $query = $this->executeQuery($sql, array($id));
 147         $result = $query->fetchAll();
 148         $user_config = array();
 149
 150         foreach ($result as $key => $value) {
 151             $user_config[$value['name']] = $value['value'];
 152         }
 153
 154         return $user_config;
 155     }
 156
 157     public function userExists($username) {
 158         $sql = "SELECT * FROM users WHERE username=?";
 159         $query = $this->executeQuery($sql, array($username));
 160         $login = $query->fetchAll();
 161         if (isset($login[0])) {
 162             return true;
 163         } else {
 164             return false;
 165         }
 166     }
 167
 168     public function login($username, $password, $isauthenticated=false) {
 169         if ($isauthenticated) {
 170           $sql = "SELECT * FROM users WHERE username=?";
 171           $query = $this->executeQuery($sql, array($username));
 172         } else {
 173           $sql = "SELECT * FROM users WHERE username=? AND password=?";
 174           $query = $this->executeQuery($sql, array($username, $password));
 175         }
 176         $login = $query->fetchAll();
 177
 178         $user = array();
 179         if (isset($login[0])) {
 180             $user['id'] = $login[0]['id'];
 181             $user['username'] = $login[0]['username'];
 182             $user['password'] = $login[0]['password'];
 183             $user['name'] = $login[0]['name'];
 184             $user['email'] = $login[0]['email'];
 185             $user['config'] = $this->getConfigUser($login[0]['id']);
 186         }
 187
 188         return $user;
 189     }
 190
 191     public function updatePassword($userId, $password)
 192     {
 193         $sql_update = "UPDATE users SET password=? WHERE id=?";
 194         $params_update = array($password, $userId);
 195         $query = $this->executeQuery($sql_update, $params_update);
 196     }
 197
 198     public function updateUserConfig($userId, $key, $value) {
 199         $config = $this->getConfigUser($userId);
 200
 201         if (! isset($config[$key])) {
 202             $sql = "INSERT INTO users_config (value, user_id, name) VALUES (?, ?, ?)";
 203         }
 204         else {
 205             $sql = "UPDATE users_config SET value=? WHERE user_id=? AND name=?";
 206         }
 207
 208         $params = array($value, $userId, $key);
 209         $query = $this->executeQuery($sql, $params);
 210     }
 211
 212     private function executeQuery($sql, $params) {
 213         try
 214         {
 215             $query = $this->getHandle()->prepare($sql);
 216             $query->execute($params);
 217             return $query;
 218         }
 219         catch (Exception $e)
 220         {
 221             Tools::logm('execute query error : '.$e->getMessage());
 222             return FALSE;
 223         }
 224     }
 225
 226     public function retrieveAll($user_id) {
 227         $sql        = "SELECT * FROM entries WHERE user_id=? ORDER BY id";
 228         $query      = $this->executeQuery($sql, array($user_id));
 229         $entries    = $query->fetchAll();
 230
 231         return $entries;
 232     }
 233
 234     public function retrieveOneById($id, $user_id) {
 235         $entry  = NULL;
 236         $sql    = "SELECT * FROM entries WHERE id=? AND user_id=?";
 237         $params = array(intval($id), $user_id);
 238         $query  = $this->executeQuery($sql, $params);
 239         $entry  = $query->fetchAll();
 240
 241         return isset($entry[0]) ? $entry[0] : null;
 242     }
 243
 244     public function retrieveOneByURL($url, $user_id) {
 245         $entry  = NULL;
 246         $sql    = "SELECT * FROM entries WHERE url=? AND user_id=?";
 247         $params = array($url, $user_id);
 248         $query  = $this->executeQuery($sql, $params);
 249         $entry  = $query->fetchAll();
 250
 251         return isset($entry[0]) ? $entry[0] : null;
 252     }
 253
 254     public function reassignTags($old_entry_id, $new_entry_id) {
 255         $sql    = "UPDATE tags_entries SET entry_id=? WHERE entry_id=?";
 256         $params = array($new_entry_id, $old_entry_id);
 257         $query  = $this->executeQuery($sql, $params);
 258     }
 259
 260     public function getEntriesByView($view, $user_id, $limit = '') {
 261         switch ($_SESSION['sort'])
 262         {
 263             case 'ia':
 264                 $order = 'ORDER BY id';
 265                 break;
 266             case 'id':
 267                 $order = 'ORDER BY id DESC';
 268                 break;
 269             case 'ta':
 270                 $order = 'ORDER BY lower(title)';
 271                 break;
 272             case 'td':
 273                 $order = 'ORDER BY lower(title) DESC';
 274                 break;
 275             default:
 276                 $order = 'ORDER BY id';
 277                 break;
 278         }
 279
 280         switch ($view)
 281         {
 282             case 'archive':
 283                 $sql    = "SELECT * FROM entries WHERE user_id=? AND is_read=? " . $order;
 284                 $params = array($user_id, 1);
 285                 break;
 286             case 'fav' :
 287                 $sql    = "SELECT * FROM entries WHERE user_id=? AND is_fav=? " . $order;
 288                 $params = array($user_id, 1);
 289                 break;
 290             default:
 291                 $sql    = "SELECT * FROM entries WHERE user_id=? AND is_read=? " . $order;
 292                 $params = array($user_id, 0);
 293                 break;
 294         }
 295
 296         $sql .= ' ' . $limit;
 297
 298         $query = $this->executeQuery($sql, $params);
 299         $entries = $query->fetchAll();
 300
 301         return $entries;
 302     }
 303
 304     public function updateContent($id, $content, $user_id) {
 305         $sql_action = 'UPDATE entries SET content = ? WHERE id=? AND user_id=?';
 306         $params_action = array($content, $id, $user_id);
 307         $query = $this->executeQuery($sql_action, $params_action);
 308         return $query;
 309     }
 310
 311     public function add($url, $title, $content, $user_id) {
 312         $sql_action = 'INSERT INTO entries ( url, title, content, user_id ) VALUES (?, ?, ?, ?)';
 313         $params_action = array($url, $title, $content, $user_id);
 314         $query = $this->executeQuery($sql_action, $params_action);
 315         return $query;
 316     }
 317
 318     public function deleteById($id, $user_id) {
 319         $sql_action     = "DELETE FROM entries WHERE id=? AND user_id=?";
 320         $params_action  = array($id, $user_id);
 321         $query          = $this->executeQuery($sql_action, $params_action);
 322         return $query;
 323     }
 324
 325     public function favoriteById($id, $user_id) {
 326         $sql_action     = "UPDATE entries SET is_fav=NOT is_fav WHERE id=? AND user_id=?";
 327         $params_action  = array($id, $user_id);
 328         $query          = $this->executeQuery($sql_action, $params_action);
 329     }
 330
 331     public function archiveById($id, $user_id) {
 332         $sql_action     = "UPDATE entries SET is_read=NOT is_read WHERE id=? AND user_id=?";
 333         $params_action  = array($id, $user_id);
 334         $query          = $this->executeQuery($sql_action, $params_action);
 335     }
 336
 337     public function archiveAll($user_id) {
 338         $sql_action     = "UPDATE entries SET is_read=? WHERE user_id=? AND is_read=?";
 339         $params_action  = array($user_id, 1, 0);
 340         $query          = $this->executeQuery($sql_action, $params_action);
 341     }
 342
 343     public function getLastId($column = '') {
 344         return $this->getHandle()->lastInsertId($column);
 345     }
 346
 347     public function retrieveAllTags($user_id) {
 348         $sql = "SELECT DISTINCT tags.* FROM tags
 349           LEFT JOIN tags_entries ON tags_entries.tag_id=tags.id
 350           LEFT JOIN entries ON tags_entries.entry_id=entries.id
 351           WHERE entries.user_id=?";
 352         $query = $this->executeQuery($sql, array($user_id));
 353         $tags = $query->fetchAll();
 354
 355         return $tags;
 356     }
 357
 358     public function retrieveTag($id, $user_id) {
 359         $tag  = NULL;
 360         $sql    = "SELECT DISTINCT tags.* FROM tags
 361           LEFT JOIN tags_entries ON tags_entries.tag_id=tags.id
 362           LEFT JOIN entries ON tags_entries.entry_id=entries.id
 363           WHERE tags.id=? AND entries.user_id=?";
 364         $params = array(intval($id), $user_id);
 365         $query  = $this->executeQuery($sql, $params);
 366         $tag  = $query->fetchAll();
 367
 368         return isset($tag[0]) ? $tag[0] : null;
 369     }
 370
 371     public function retrieveEntriesByTag($tag_id, $user_id) {
 372         $sql =
 373             "SELECT entries.* FROM entries
 374             LEFT JOIN tags_entries ON tags_entries.entry_id=entries.id
 375             WHERE tags_entries.tag_id = ? AND entries.user_id=?";
 376         $query = $this->executeQuery($sql, array($tag_id, $user_id));
 377         $entries = $query->fetchAll();
 378
 379         return $entries;
 380     }
 381
 382     public function retrieveTagsByEntry($entry_id) {
 383         $sql =
 384             "SELECT tags.* FROM tags
 385             LEFT JOIN tags_entries ON tags_entries.tag_id=tags.id
 386             WHERE tags_entries.entry_id = ?";
 387         $query = $this->executeQuery($sql, array($entry_id));
 388         $tags = $query->fetchAll();
 389
 390         return $tags;
 391     }
 392
 393     public function removeTagForEntry($entry_id, $tag_id) {
 394         $sql_action     = "DELETE FROM tags_entries WHERE tag_id=? AND entry_id=?";
 395         $params_action  = array($tag_id, $entry_id);
 396         $query          = $this->executeQuery($sql_action, $params_action);
 397         return $query;
 398     }
 399
 400     public function retrieveTagByValue($value) {
 401         $tag  = NULL;
 402         $sql    = "SELECT * FROM tags WHERE value=?";
 403         $params = array($value);
 404         $query  = $this->executeQuery($sql, $params);
 405         $tag  = $query->fetchAll();
 406
 407         return isset($tag[0]) ? $tag[0] : null;
 408     }
 409
 410     public function createTag($value) {
 411         $sql_action = 'INSERT INTO tags ( value ) VALUES (?)';
 412         $params_action = array($value);
 413         $query = $this->executeQuery($sql_action, $params_action);
 414         return $query;
 415     }
 416
 417     public function setTagToEntry($tag_id, $entry_id) {
 418         $sql_action = 'INSERT INTO tags_entries ( tag_id, entry_id ) VALUES (?, ?)';
 419         $params_action = array($tag_id, $entry_id);
 420         $query = $this->executeQuery($sql_action, $params_action);
 421         return $query;
 422     }
 423 }