/system/libraries/drivers/Database/Pdosqlite.php
PHP | 473 lines | 411 code | 35 blank | 27 comment | 22 complexity | a7b968ed6a966c6e18940ff1272b209b MD5 | raw file
Possible License(s): LGPL-2.1
- <?php defined('SYSPATH') OR die('No direct access allowed.');
- /*
- * Class: Database_PdoSqlite_Driver
- * Provides specific database items for Sqlite.
- *
- * Connection string should be, eg: "pdosqlite://path/to/database.db"
- *
- * Version 1.0 alpha
- * author - Doutu, updated by gregmac
- * copyright - (c) BSD
- * license - <no>
- */
- class Database_Pdosqlite_Driver extends Database_Driver {
- // Database connection link
- protected $link;
- protected $db_config;
- /*
- * Constructor: __construct
- * Sets up the config for the class.
- *
- * Parameters:
- * config - database configuration
- *
- */
- public function __construct($config)
- {
- $this->db_config = $config;
- Kohana::log('debug', 'PDO:Sqlite Database Driver Initialized');
- }
- public function connect()
- {
- // Import the connect variables
- extract($this->db_config['connection']);
- try
- {
- $this->link = new PDO('sqlite:'.$socket.$database, $user, $pass,
- array(PDO::ATTR_PERSISTENT => $this->db_config['persistent']));
- $this->link->setAttribute(PDO::ATTR_CASE, PDO::CASE_NATURAL);
- $this->link->query('PRAGMA count_changes=1;');
- if ($charset = $this->db_config['character_set'])
- {
- $this->set_charset($charset);
- }
- }
- catch (PDOException $e)
- {
- throw new Kohana_Database_Exception('database.error', $e->getMessage());
- }
- // Clear password after successful connect
- $this->db_config['connection']['pass'] = NULL;
- return $this->link;
- }
- public function query($sql)
- {
- try
- {
- $sth = $this->link->prepare($sql);
- }
- catch (PDOException $e)
- {
- throw new Kohana_Database_Exception('database.error', $e->getMessage());
- }
- return new Pdosqlite_Result($sth, $this->link, $this->db_config['object'], $sql);
- }
- public function set_charset($charset)
- {
- $this->link->query('PRAGMA encoding = '.$this->escape_str($charset));
- }
- public function escape_table($table)
- {
- if ( ! $this->db_config['escape'])
- return $table;
- return '`'.str_replace('.', '`.`', $table).'`';
- }
- public function escape_column($column)
- {
- if ( ! $this->db_config['escape'])
- return $column;
- if (strtolower($column) == 'count(*)' OR $column == '*')
- return $column;
- // This matches any modifiers we support to SELECT.
- if ( ! preg_match('/\b(?:rand|all|distinct(?:row)?|high_priority|sql_(?:small_result|b(?:ig_result|uffer_result)|no_cache|ca(?:che|lc_found_rows)))\s/i', $column))
- {
- if (stripos($column, ' AS ') !== FALSE)
- {
- // Force 'AS' to uppercase
- $column = str_ireplace(' AS ', ' AS ', $column);
- // Runs escape_column on both sides of an AS statement
- $column = array_map(array($this, __FUNCTION__), explode(' AS ', $column));
- // Re-create the AS statement
- return implode(' AS ', $column);
- }
- return preg_replace('/[^.*]+/', '`$0`', $column);
- }
- $parts = explode(' ', $column);
- $column = '';
- for ($i = 0, $c = count($parts); $i < $c; $i++)
- {
- // The column is always last
- if ($i == ($c - 1))
- {
- $column .= preg_replace('/[^.*]+/', '`$0`', $parts[$i]);
- }
- else // otherwise, it's a modifier
- {
- $column .= $parts[$i].' ';
- }
- }
- return $column;
- }
- public function limit($limit, $offset = 0)
- {
- return 'LIMIT '.$offset.', '.$limit;
- }
- public function compile_select($database)
- {
- $sql = ($database['distinct'] == TRUE) ? 'SELECT DISTINCT ' : 'SELECT ';
- $sql .= (count($database['select']) > 0) ? implode(', ', $database['select']) : '*';
- if (count($database['from']) > 0)
- {
- $sql .= "\nFROM ";
- $sql .= implode(', ', $database['from']);
- }
- if (count($database['join']) > 0)
- {
- foreach($database['join'] AS $join)
- {
- $sql .= "\n".$join['type'].'JOIN '.implode(', ', $join['tables']).' ON '.$join['conditions'];
- }
- }
- if (count($database['where']) > 0)
- {
- $sql .= "\nWHERE ";
- }
- $sql .= implode("\n", $database['where']);
- if (count($database['groupby']) > 0)
- {
- $sql .= "\nGROUP BY ";
- $sql .= implode(', ', $database['groupby']);
- }
- if (count($database['having']) > 0)
- {
- $sql .= "\nHAVING ";
- $sql .= implode("\n", $database['having']);
- }
- if (count($database['orderby']) > 0)
- {
- $sql .= "\nORDER BY ";
- $sql .= implode(', ', $database['orderby']);
- }
- if (is_numeric($database['limit']))
- {
- $sql .= "\n";
- $sql .= $this->limit($database['limit'], $database['offset']);
- }
- return $sql;
- }
- public function escape_str($str)
- {
- if ( ! $this->db_config['escape'])
- return $str;
- if (function_exists('sqlite_escape_string'))
- {
- $res = sqlite_escape_string($str);
- }
- else
- {
- $res = str_replace("'", "''", $str);
- }
- return $res;
- }
- public function list_tables(Database $db)
- {
- $sql = "SELECT `name` FROM `sqlite_master` WHERE `type`='table' ORDER BY `name`;";
- try
- {
- $result = $db->query($sql)->result(FALSE, PDO::FETCH_ASSOC);
- $tables = array();
- foreach ($result as $row)
- {
- $tables[] = current($row);
- }
- }
- catch (PDOException $e)
- {
- throw new Kohana_Database_Exception('database.error', $e->getMessage());
- }
- return $tables;
- }
- public function show_error()
- {
- $err = $this->link->errorInfo();
- return isset($err[2]) ? $err[2] : 'Unknown error!';
- }
- public function list_fields($table, $query = FALSE)
- {
- static $tables;
- if (is_object($query))
- {
- if (empty($tables[$table]))
- {
- $tables[$table] = array();
- foreach ($query->result() as $row)
- {
- $tables[$table][] = $row->name;
- }
- }
- return $tables[$table];
- }
- else
- {
- $result = $this->link->query( 'PRAGMA table_info('.$this->escape_table($table).')' );
- foreach ($result as $row)
- {
- $tables[$table][$row['name']] = $this->sql_type($row['type']);
- }
- return $tables[$table];
- }
- }
- public function field_data($table)
- {
- Kohana::log('error', 'This method is under developing');
- }
- /**
- * Version number query string
- *
- * @access public
- * @return string
- */
- function version()
- {
- return $this->link->getAttribute(constant("PDO::ATTR_SERVER_VERSION"));
- }
- } // End Database_PdoSqlite_Driver Class
- /*
- * PDO-sqlite Result
- */
- class Pdosqlite_Result extends Database_Result {
- // Data fetching types
- protected $fetch_type = PDO::FETCH_OBJ;
- protected $return_type = PDO::FETCH_ASSOC;
- /**
- * Sets up the result variables.
- *
- * @param resource query result
- * @param resource database link
- * @param boolean return objects or arrays
- * @param string SQL query that was run
- */
- public function __construct($result, $link, $object = TRUE, $sql)
- {
- if (is_object($result) OR $result = $link->prepare($sql))
- {
- // run the query
- try
- {
- $result->execute();
- }
- catch (PDOException $e)
- {
- throw new Kohana_Database_Exception('database.error', $e->getMessage());
- }
- if (preg_match('/^SELECT|PRAGMA|EXPLAIN/i', $sql))
- {
- $this->result = $result;
- $this->current_row = 0;
- $this->total_rows = $this->sqlite_row_count();
- $this->fetch_type = ($object === TRUE) ? PDO::FETCH_OBJ : PDO::FETCH_ASSOC;
- }
- elseif (preg_match('/^DELETE|INSERT|UPDATE/i', $sql))
- {
- $this->insert_id = $link->lastInsertId();
- }
- }
- else
- {
- // SQL error
- throw new Kohana_Database_Exception('database.error', $link->errorInfo().' - '.$sql);
- }
- // Set result type
- $this->result($object);
- // Store the SQL
- $this->sql = $sql;
- }
- private function sqlite_row_count()
- {
- $count = 0;
- while ($this->result->fetch())
- {
- $count++;
- }
- // The query must be re-fetched now.
- $this->result->execute();
- return $count;
- }
- /*
- * Destructor: __destruct
- * Magic __destruct function, frees the result.
- */
- public function __destruct()
- {
- if (is_object($this->result))
- {
- $this->result->closeCursor();
- $this->result = NULL;
- }
- }
- public function result($object = TRUE, $type = PDO::FETCH_BOTH)
- {
- $this->fetch_type = (bool) $object ? PDO::FETCH_OBJ : PDO::FETCH_BOTH;
- if ($this->fetch_type == PDO::FETCH_OBJ)
- {
- $this->return_type = (is_string($type) AND Kohana::auto_load($type)) ? $type : 'stdClass';
- }
- else
- {
- $this->return_type = $type;
- }
- return $this;
- }
- public function as_array($object = NULL, $type = PDO::FETCH_ASSOC)
- {
- return $this->result_array($object, $type);
- }
- public function result_array($object = NULL, $type = PDO::FETCH_ASSOC)
- {
- $rows = array();
- if (is_string($object))
- {
- $fetch = $object;
- }
- elseif (is_bool($object))
- {
- if ($object === TRUE)
- {
- $fetch = PDO::FETCH_OBJ;
- // NOTE - The class set by $type must be defined before fetching the result,
- // autoloading is disabled to save a lot of stupid overhead.
- $type = (is_string($type) AND Kohana::auto_load($type)) ? $type : 'stdClass';
- }
- else
- {
- $fetch = PDO::FETCH_OBJ;
- }
- }
- else
- {
- // Use the default config values
- $fetch = $this->fetch_type;
- if ($fetch == PDO::FETCH_OBJ)
- {
- $type = (is_string($type) AND Kohana::auto_load($type)) ? $type : 'stdClass';
- }
- }
- try
- {
- while ($row = $this->result->fetch($fetch))
- {
- $rows[] = $row;
- }
- }
- catch(PDOException $e)
- {
- throw new Kohana_Database_Exception('database.error', $e->getMessage());
- return FALSE;
- }
- return $rows;
- }
- public function list_fields()
- {
- $field_names = array();
- for ($i = 0, $max = $this->result->columnCount(); $i < $max; $i++)
- {
- $info = $this->result->getColumnMeta($i);
- $field_names[] = $info['name'];
- }
- return $field_names;
- }
- public function seek($offset)
- {
- // To request a scrollable cursor for your PDOStatement object, you must
- // set the PDO::ATTR_CURSOR attribute to PDO::CURSOR_SCROLL when you
- // prepare the statement.
- Kohana::log('error', get_class($this).' does not support scrollable cursors, '.__FUNCTION__.' call ignored');
- return FALSE;
- }
- public function offsetGet($offset)
- {
- try
- {
- return $this->result->fetch($this->fetch_type, PDO::FETCH_ORI_ABS, $offset);
- }
- catch(PDOException $e)
- {
- throw new Kohana_Database_Exception('database.error', $e->getMessage());
- }
- }
- public function rewind()
- {
- // Same problem that seek() has, see above.
- return $this->seek(0);
- }
- } // End PdoSqlite_Result Class