lib/AOOSStorageDeviceMySQL.php

   1 <?php
   2 require_once("AOOSException.php");
   3 require_once("AOOSModule.php");
   4 require_once("AOOSInterfaces.php");
   5
   6 /**
   7  * The mysql-part of AOOSStorageDevice
   8  * @author Sebastian Skejø
   9  */
  10 class AOOSStorageDeviceMySQL extends AOOSModule implements AOOSStorageInterface
  11 {
  12     private $_connection    = null;
  13     private $_username      = null;
  14     private $_password      = null;
  15     private $_database      = null;
  16     private $_host          = null;
  17     private $_table         = null;
  18     private $_query         = null;
  19     private $_queryText     = null;
  20
  21     public function __construct($core)
  22     {
  23         parent::__construct($core);
  24
  25         try {
  26             $this->_username    = $core->getSetting("DBUsername");
  27             $this->_password    = $core->getSetting("DBPassword");
  28             $this->_database    = $core->getSetting("DBDatabase");
  29             $this->_host        = $core->getSetting("DBHost");
  30         }
  31         catch (AOOSException $e)
  32         {
  33         }
  34
  35         if (!$this->_connection = mysql_connect($this->_host, $this->_username, $this->_password))
  36             throw new AOOSException($this->core(), $this->tr("mysql_connect_fail"), $this->tr("error").": ".mysql_error());
  37         if (!mysql_select_db($this->_database))
  38             throw new AOOSException($this->core(), $this->tr("mysql_select_db_fail"), $this->tr("error").": ".mysql_error());
  39     }
  40
  41     public function __destruct()
  42     {
  43         if (!mysql_close($this->_connection))
  44             throw new AOOSException($this->core(), $this->tr("mysql_close_fail"), $this->tr("error").": ".mysql_error(), true, 1);
  45     }
  46
  47     private function _makeQuery($query)
  48     {
  49         if ($this->_queryText != $query)
  50         {
  51             $this->_queryText = $query;
  52             if (!($this->_query = mysql_query($this->_queryText)))
  53             {
  54                 throw new AOOSException($this->core(), $this->tr("query_fail_mysql"), $this->tr("error").": ".mysql_error()."<br />Query: ".$query);
  55                 return false;
  56             }
  57         }
  58
  59         return true;
  60     }
  61
  62     public function setTable($table)
  63     {
  64         $this->_table = $this->core()->getSetting("DBPrefix").$table;
  65     }
  66
  67     public function setQuery($query)
  68     {
  69         return $this->_makeQuery($query);
  70     }
  71
  72     /** Inserts a single value
  73      * @param string $field Field name
  74      * @param string $value Value
  75      * @return bool
  76      */
  77     public function insert($field, $value, $escape = null)
  78     {
  79         $rawQuery   = "INSERT INTO %s (%s) VALUES (%s)";
  80         if ($escape)
  81         {
  82             $field      = $this->_escape($field);
  83             $value      = $this->_escape($value);
  84         }
  85         $this->_quote($value);
  86         $field = implode(",", $field);
  87         $value = implode(",", $value);
  88         $query      = sprintf($rawQuery, $this->_table, $field, $value);
  89
  90         if (!$this->_makeQuery($query))
  91         {
  92             return false;
  93         }
  94         return true;
  95     }
  96
  97     /**
  98      * Selects a single row
  99      */
 100     public function select($fields, $where = null, $limit = null)
 101     {
 102         // Fix fields
 103         if (is_array($fields))
 104         {
 105             $fields = implode(", ", $fields);
 106         }
 107
 108         $raw = "SELECT %s FROM %s";
 109         $query = sprintf($raw, $fields, $this->_table);
 110
 111         if ($where)
 112         {
 113             $query .= sprintf(" WHERE %s", $where);
 114         }
 115
 116         if ($limit)
 117         {
 118             $query .= sprintf(" LIMIT %s", $limit);
 119         }
 120
 121         if (!$this->_makeQuery($query))
 122         {
 123             return false;
 124         }
 125
 126         return mysql_fetch_assoc($this->_query);
 127     }
 128
 129     /**
 130      * Update values
 131      * @param array $fields Fields
 132      * @param array $values Values
 133      * @param string $where Valid where-clause
 134      * @return bool
 135      */
 136     public function update($fields, $values, $where)
 137     {
 138         $raw = "UPDATE %s SET %s WHERE %s";
 139         $vals = array();
 140         $merged = "";
 141
 142         for ($i = 0; $i < count($fields); $i++)
 143         {
 144             $vals[] = $fields[$i]." = ".$values[$i];
 145         }
 146         $merged = implode(", ", $vals);
 147
 148         $query = sprintf($raw, $this->_table, $merged, $where);
 149
 150         if (!$this->_makeQuery($query))
 151         {
 152             return false;
 153         }
 154
 155         return true;
 156     }
 157
 158     /**
 159      * Deletes a row. See AOOSStorageDevice::deleteFromArray.
 160      * @param string $where
 161      * @return bool
 162      */
 163     public function delete($where, $limit = null)
 164     {
 165         $raw = "DELETE FROM %s WHERE %s";
 166         $query = sprintf($raw, $this->_table, $where);
 167
 168         if ($limit)
 169         {
 170             $query .= " LIMIT ".$limit;
 171         }
 172
 173         if (!$this->_makeQuery($query))
 174         {
 175             return false;
 176         }
 177
 178         return true;
 179     }
 180
 181     /**
 182      * Num rows
 183      * @return int
 184      */
 185     public function numRows($where = null)
 186     {
 187         $query = sprintf("SELECT COUNT(*) FROM %s ", $this->_table);
 188         if ($where)
 189         {
 190             $query .= "WHERE ".$where;
 191         }
 192         $this->_makeQuery($query);
 193         $result = mysql_fetch_array($this->_query);
 194         return $result[0];
 195     }
 196
 197
 198     /**
 199      * Escapes string
 200      */
 201     private function _escape($array)
 202     {
 203         $array = $this->_strip($array);
 204         foreach ($array as &$a)
 205         {
 206             $a = mysql_real_escape_string($a);
 207         }
 208         return $array;
 209     }
 210
 211     /**
 212      * Strip slashes from array values
 213      */
 214     private function _strip($array)
 215     {
 216         foreach ($array as &$a)
 217         {
 218             $a = stripslashes($a);
 219         }
 220         return $array;
 221     }
 222
 223     private function _quote(&$array)
 224     {
 225         foreach ($array as &$a)
 226         {
 227             $s = substr($a, 0,1);
 228             if ($s != "\"" && $s != "'")
 229             {
 230                 $a = "'".$a;
 231             }
 232             $e = substr($a, strlen($a)-1, strlen($a));
 233             if ($e != "\"" && $e != "'")
 234             {
 235                 $a = $a."'";
 236             }
 237         }
 238         return $array;
 239     }
 240 }
 241 ?>