inc/poche/Database.class.php

   1 <?php
   2 /**
   3  * poche, a read it later open source system
   4  *
   5  * @category   poche
   6  * @author     Nicolas Lœuillet <support@inthepoche.com>
   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 WHERE id=?";
  41         $query = $this->executeQuery($sql, array('1'));
  42         $hasAdmin = $query->fetchAll();
  43
  44         if (count($hasAdmin) == 0)
  45             return FALSE;
  46
  47         return TRUE;
  48     }
  49
  50     public function install($login, $password) {
  51         $sql = 'INSERT INTO users ( username, password ) VALUES (?, ?)';
  52         $params = array($login, $password);
  53         $query = $this->executeQuery($sql, $params);
  54
  55         return TRUE;
  56     }
  57
  58     private function getConfigUser($id) {
  59         $sql = "SELECT * FROM users_config WHERE user_id = ?";
  60         $query = $this->executeQuery($sql, array($id));
  61         $result = $query->fetchAll();
  62         $user_config = array();
  63
  64         foreach ($result as $key => $value) {
  65             $user_config[$value['name']] = $value['value'];
  66         }
  67
  68         return $user_config;
  69     }
  70
  71     public function login($username, $password) {
  72         $sql = "SELECT * FROM users WHERE username=? AND password=?";
  73         $query = $this->executeQuery($sql, array($username, $password));
  74         $login = $query->fetchAll();
  75
  76         $user = array();
  77         if (isset($login[0])) {
  78             $user['id'] = $login[0]['id'];
  79             $user['username'] = $login[0]['username'];
  80             $user['password'] = $login[0]['password'];
  81             $user['name'] = $login[0]['name'];
  82             $user['email'] = $login[0]['email'];
  83             $user['config'] = $this->getConfigUser($login[0]['id']);
  84         }
  85
  86         return $user;
  87     }
  88
  89     public function updatePassword($id, $password)
  90     {
  91         $sql_update = "UPDATE users SET password=? WHERE id=?";
  92         $params_update = array($password, $id);
  93         $query = $this->executeQuery($sql_update, $params_update);
  94     }
  95
  96     private function executeQuery($sql, $params) {
  97         try
  98         {
  99             $query = $this->getHandle()->prepare($sql);
 100             $query->execute($params);
 101             return $query;
 102         }
 103         catch (Exception $e)
 104         {
 105             Tools::logm('execute query error : '.$e->getMessage());
 106             return FALSE;
 107         }
 108     }
 109
 110     public function retrieveAll($user_id) {
 111         $sql        = "SELECT * FROM entries WHERE user_id=? ORDER BY id";
 112         $query      = $this->executeQuery($sql, array($user_id));
 113         $entries    = $query->fetchAll();
 114
 115         return $entries;
 116     }
 117
 118     public function retrieveOneById($id, $user_id) {
 119         $entry  = NULL;
 120         $sql    = "SELECT * FROM entries WHERE id=? AND user_id=?";
 121         $params = array(intval($id), $user_id);
 122         $query  = $this->executeQuery($sql, $params);
 123         $entry  = $query->fetchAll();
 124
 125         return $entry[0];
 126     }
 127
 128     public function getEntriesByView($view, $user_id, $limit = '') {
 129         switch ($_SESSION['sort'])
 130         {
 131             case 'ia':
 132                 $order = 'ORDER BY id';
 133                 break;
 134             case 'id':
 135                 $order = 'ORDER BY id DESC';
 136                 break;
 137             case 'ta':
 138                 $order = 'ORDER BY lower(title)';
 139                 break;
 140             case 'td':
 141                 $order = 'ORDER BY lower(title) DESC';
 142                 break;
 143             default:
 144                 $order = 'ORDER BY id';
 145                 break;
 146         }
 147
 148         switch ($view)
 149         {
 150             case 'archive':
 151                 $sql    = "SELECT * FROM entries WHERE user_id=? AND is_read=? " . $order;
 152                 $params = array($user_id, 1);
 153                 break;
 154             case 'fav' :
 155                 $sql    = "SELECT * FROM entries WHERE user_id=? AND is_fav=? " . $order;
 156                 $params = array($user_id, 1);
 157                 break;
 158             default:
 159                 $sql    = "SELECT * FROM entries WHERE user_id=? AND is_read=? " . $order;
 160                 $params = array($user_id, 0);
 161                 break;
 162         }
 163
 164         $sql .= ' ' . $limit;
 165
 166         $query = $this->executeQuery($sql, $params);
 167         $entries = $query->fetchAll();
 168
 169         return $entries;
 170     }
 171
 172     public function add($url, $title, $content, $user_id) {
 173         $sql_action = 'INSERT INTO entries ( url, title, content, user_id ) VALUES (?, ?, ?, ?)';
 174         $params_action = array($url, $title, $content, $user_id);
 175         $query = $this->executeQuery($sql_action, $params_action);
 176         return $query;
 177     }
 178
 179     public function deleteById($id, $user_id) {
 180         $sql_action     = "DELETE FROM entries WHERE id=? AND user_id=?";
 181         $params_action  = array($id, $user_id);
 182         $query          = $this->executeQuery($sql_action, $params_action);
 183         return $query;
 184     }
 185
 186     public function favoriteById($id, $user_id) {
 187         $sql_action     = "UPDATE entries SET is_fav=NOT is_fav WHERE id=? AND user_id=?";
 188         $params_action  = array($id, $user_id);
 189         $query          = $this->executeQuery($sql_action, $params_action);
 190     }
 191
 192     public function archiveById($id, $user_id) {
 193         $sql_action     = "UPDATE entries SET is_read=NOT is_read WHERE id=? AND user_id=?";
 194         $params_action  = array($id, $user_id);
 195         $query          = $this->executeQuery($sql_action, $params_action);
 196     }
 197
 198     public function getLastId() {
 199         return $this->getHandle()->lastInsertId();
 200     }
 201 }