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("insert_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 = $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         $field = implode(",", $field);
  86         $value = implode(",", $value);
  87         $query      = sprintf($rawQuery, $this->_table, $field, $value);
  88
  89         if (!$this->_makeQuery($query))
  90         {
  91             return false;
  92         }
  93         return true;
  94     }
  95
  96     /**
  97      * Selects a single row
  98      */
  99     public function select($fields, $where = null, $limit = null)
 100     {
 101         // Fix fields
 102         if (is_array($fields))
 103         {
 104             $fields = implode(", ", $fields);
 105         }
 106
 107         $raw = "SELECT %s FROM %s";
 108         $query = sprintf($raw, $fields, $this->_table);
 109
 110         if ($where)
 111         {
 112             $query .= sprintf(" WHERE %s", $where);
 113         }
 114
 115         if ($limit)
 116         {
 117             $query .= sprintf(" LIMIT %s", $limit);
 118         }
 119
 120         if (!$this->_makeQuery($query))
 121         {
 122             return false;
 123         }
 124
 125         return mysql_fetch_assoc($this->_query);
 126     }
 127
 128     /**
 129      * Update values
 130      * @param array $fields Fields
 131      * @param array $values Values
 132      * @param string $where Valid where-clause
 133      * @return bool
 134      */
 135     public function update($fields, $values, $where)
 136     {
 137         $raw = "UPDATE %s SET %s WHERE %s";
 138         $vals = array();
 139         $merged = "";
 140
 141         for ($i = 0; $i < count($fields); $i++)
 142         {
 143             $vals[] = $fields[$i]." = ".$values[$i];
 144         }
 145         $merged = implode(", ", $vals);
 146
 147         $query = sprintf($raw, $this->_table, $merged, $where);
 148
 149         if (!$this->_makeQuery($query))
 150         {
 151             return false;
 152         }
 153
 154         return true;
 155     }
 156
 157     /**
 158      * Deletes a row. See AOOSStorageDevice::deleteFromArray.
 159      * @param string $where
 160      * @return bool
 161      */
 162     public function delete($where, $limit = null)
 163     {
 164         $raw = "DELETE FROM %s WHERE %s";
 165         $query = sprintf($raw, $this->_table, $where);
 166
 167         if ($limit)
 168         {
 169             $query .= " LIMIT ".$limit;
 170         }
 171
 172         if (!$this->_makeQuery($query))
 173         {
 174             return false;
 175         }
 176
 177         return true;
 178     }
 179
 180     /**
 181      * Num rows
 182      * @return int
 183      */
 184     public function numRows($where = null)
 185     {
 186         $query = "SELECT COUNT(*)";
 187         if ($where)
 188         {
 189             $query .= "WHERE ".$where;
 190         }
 191         return $this->_makeQuery($query);
 192     }
 193
 194
 195     /**
 196      * Escapes string
 197      */
 198     private function _escape($array)
 199     {
 200         $array = $this->_strip($array);
 201         foreach ($array as &$a)
 202         {
 203             $a = mysql_real_escape_string($a);
 204         }
 205         return $array;
 206     }
 207
 208     /**
 209      * Strip slashes from array values
 210      */
 211     private function _strip($array)
 212     {
 213         foreach ($array as &$a)
 214         {
 215             $a = stripslashes($a);
 216         }
 217         return $array;
 218     }
 219 }
 220 ?>