- DibiSQLiteDriver: added ability to convert strings to requested character encoding

- added experimental Microsoft SQL Server 2005 Driver for PHP

dibi/drivers/mssql2005.php

@@ -0,0 +1,408 @@
+<?php
+
+/**
+ * dibi - tiny'n'smart database abstraction layer
+ * ----------------------------------------------
+ *
+ * Copyright (c) 2005, 2009 David Grudl (http://davidgrudl.com)
+ *
+ * This source file is subject to the "dibi license" that is bundled
+ * with this package in the file license.txt.
+ *
+ * For more information please see http://dibiphp.com
+ *
+ * @copyright  Copyright (c) 2005, 2009 David Grudl
+ * @license    http://dibiphp.com/license  dibi license
+ * @link       http://dibiphp.com
+ * @package    dibi
+ * @version    $Id$
+ */
+
+
+/**
+ * The dibi driver for MS SQL Driver 2005 database.
+ *
+ * Connection options:
+ *   - 'host' - the MS SQL server host name. It can also include a port number (hostname:port)
+ *   - 'options' - connection info array {@link http://msdn.microsoft.com/en-us/library/cc296161(SQL.90).aspx}
+ *   - 'lazy' - if TRUE, connection will be established only when required
+ *   - 'charset' - character encoding to set (default is UTF-8)
+ *
+ * @author     David Grudl
+ * @copyright  Copyright (c) 2005, 2009 David Grudl
+ * @package    dibi
+ */
+class DibiMsSql2005Driver extends DibiObject implements IDibiDriver
+{
+	/** @var resource  Connection resource */
+	private $connection;
+
+	/** @var resource  Resultset resource */
+	private $resultSet;
+
+	/** @var string  character encoding */
+	private $charset;
+
+
+
+	/**
+	 * @throws DibiException
+	 */
+	public function __construct()
+	{
+		if (!extension_loaded('sqlsrv')) {
+			throw new DibiDriverException("PHP extension 'sqlsrv' is not loaded.");
+		}
+	}
+
+
+
+	/**
+	 * Connects to a database.
+	 * @return void
+	 * @throws DibiException
+	 */
+	public function connect(array &$config)
+	{
+		DibiConnection::alias($config, 'host', 'hostname');
+
+		if (isset($config['options'])) {
+			$this->connection = sqlsrv_connect($config['host'], $config['options']);
+		} else {
+			$this->connection = sqlsrv_connect($config['host']);
+		}
+
+		if (!is_resource($this->connection)) {
+			$info = sqlsrv_errors();
+			throw new DibiDriverException($info[0]['message'], $info[0]['code']);
+		}
+
+		$this->charset = empty($config['charset']) ? 'UTF-8' : $config['charset'];
+	}
+
+
+
+	/**
+	 * Disconnects from a database.
+	 * @return void
+	 */
+	public function disconnect()
+	{
+		sqlsrv_close($this->connection);
+	}
+
+
+
+	/**
+	 * Executes the SQL query.
+	 * @param  string      SQL statement.
+	 * @return IDibiDriver|NULL
+	 * @throws DibiDriverException
+	 */
+	public function query($sql)
+	{
+		$sql = iconv($this->charset, 'UTF-16LE', $sql);
+		$this->resultSet = sqlsrv_query($this->connection, $sql);
+
+		if ($this->resultSet === FALSE) {
+			$info = sqlsrv_errors();
+			throw new DibiDriverException($info[0]['message'], $info[0]['code'], $sql);
+		}
+
+		return is_resource($this->resultSet) ? clone $this : NULL;
+	}
+
+
+
+	/**
+	 * Gets the number of affected rows by the last INSERT, UPDATE or DELETE query.
+	 * @return int|FALSE  number of rows or FALSE on error
+	 */
+	public function affectedRows()
+	{
+		return sqlsrv_rows_affected($this->resultSet);
+	}
+
+
+
+	/**
+	 * Retrieves the ID generated for an AUTO_INCREMENT column by the previous INSERT query.
+	 * @return int|FALSE  int on success or FALSE on failure
+	 */
+	public function insertId($sequence)
+	{
+		$res = sqlsrv_query($this->connection, 'SELECT @@IDENTITY');
+		if (is_resource($res)) {
+			$row = sqlsrv_fetch_array($res, SQLSRV_FETCH_NUMERIC);
+			return $row[0];
+		}
+		return FALSE;
+	}
+
+
+
+	/**
+	 * Begins a transaction (if supported).
+	 * @param  string  optinal savepoint name
+	 * @return void
+	 * @throws DibiDriverException
+	 */
+	public function begin($savepoint = NULL)
+	{
+		$this->query('BEGIN TRANSACTION');
+	}
+
+
+
+	/**
+	 * Commits statements in a transaction.
+	 * @param  string  optinal savepoint name
+	 * @return void
+	 * @throws DibiDriverException
+	 */
+	public function commit($savepoint = NULL)
+	{
+		$this->query('COMMIT');
+	}
+
+
+
+	/**
+	 * Rollback changes in a transaction.
+	 * @param  string  optinal savepoint name
+	 * @return void
+	 * @throws DibiDriverException
+	 */
+	public function rollback($savepoint = NULL)
+	{
+		$this->query('ROLLBACK');
+	}
+
+
+
+	/**
+	 * Returns the connection resource.
+	 * @return mixed
+	 */
+	public function getResource()
+	{
+		return $this->connection;
+	}
+
+
+
+	/********************* SQL ****************d*g**/
+
+
+
+	/**
+	 * Encodes data for use in an SQL statement.
+	 * @param  string    value
+	 * @param  string    type (dibi::FIELD_TEXT, dibi::FIELD_BOOL, ...)
+	 * @return string    encoded value
+	 * @throws InvalidArgumentException
+	 */
+	public function escape($value, $type)
+	{
+		switch ($type) {
+		case dibi::FIELD_TEXT:
+		case dibi::FIELD_BINARY:
+			return "'" . str_replace("'", "''", $value) . "'";
+
+		case dibi::IDENTIFIER:
+			// @see http://msdn.microsoft.com/en-us/library/ms176027.aspx
+			$value = str_replace(array('[', ']'), array('[[', ']]'), $value);
+			return '[' . str_replace('.', '].[', $value) . ']';
+
+		case dibi::FIELD_BOOL:
+			return $value ? -1 : 0;
+
+		case dibi::FIELD_DATE:
+			return date("'Y-m-d'", $value);
+
+		case dibi::FIELD_DATETIME:
+			return date("'Y-m-d H:i:s'", $value);
+
+		default:
+			throw new InvalidArgumentException('Unsupported type.');
+		}
+	}
+
+
+
+	/**
+	 * Decodes data from result set.
+	 * @param  string    value
+	 * @param  string    type (dibi::FIELD_BINARY)
+	 * @return string    decoded value
+	 * @throws InvalidArgumentException
+	 */
+	public function unescape($value, $type)
+	{
+		throw new InvalidArgumentException('Unsupported type.');
+	}
+
+
+
+	/**
+	 * Injects LIMIT/OFFSET to the SQL query.
+	 * @param  string &$sql  The SQL query that will be modified.
+	 * @param  int $limit
+	 * @param  int $offset
+	 * @return void
+	 */
+	public function applyLimit(&$sql, $limit, $offset)
+	{
+		// offset support is missing
+		if ($limit >= 0) {
+			$sql = 'SELECT TOP ' . (int) $limit . ' * FROM (' . $sql . ')';
+		}
+
+		if ($offset) {
+			throw new NotImplementedException('Offset is not implemented.');
+		}
+	}
+
+
+
+	/********************* result set ****************d*g**/
+
+
+
+	/**
+	 * Returns the number of rows in a result set.
+	 * @return int
+	 */
+	public function rowCount()
+	{
+		throw new DibiDriverException('Row count is not available for unbuffered queries.');
+	}
+
+
+
+	/**
+	 * Fetches the row at current position and moves the internal cursor to the next position.
+	 * @param  bool     TRUE for associative array, FALSE for numeric
+	 * @return array    array on success, nonarray if no next record
+	 * @internal
+	 */
+	public function fetch($assoc)
+	{
+		$row = sqlsrv_fetch_array($this->resultSet, $assoc ? SQLSRV_FETCH_ASSOC : SQLSRV_FETCH_NUMERIC);
+		foreach ($row as $k => $v) {
+			if (is_string($v)) {
+				$row[$k] = iconv('UTF-16LE', $this->charset, $v);
+			}
+		}
+		return $row;
+	}
+
+
+
+	/**
+	 * Moves cursor position without fetching row.
+	 * @param  int      the 0-based cursor pos to seek to
+	 * @return boolean  TRUE on success, FALSE if unable to seek to specified record
+	 * @throws DibiException
+	 */
+	public function seek($row)
+	{
+		throw new DibiDriverException('Cannot seek an unbuffered result set.');
+	}
+
+
+
+	/**
+	 * Frees the resources allocated for this result set.
+	 * @return void
+	 */
+	public function free()
+	{
+		sqlsrv_free_stmt($this->resultSet);
+		$this->resultSet = NULL;
+	}
+
+
+
+	/**
+	 * Returns metadata for all columns in a result set.
+	 * @return array
+	 */
+	public function getColumnsMeta()
+	{
+		$count = sqlsrv_num_fields($this->resultSet);
+		$res = array();
+		for ($i = 0; $i < $count; $i++) {
+			$row = (array) sqlsrv_field_metadata($this->resultSet, $i);
+			$res[] = array(
+				'name' => $row['Name'],
+				'fullname' => $row['Name'],
+				'nativetype' => $row['Type'],
+			);
+		}
+		return $res;
+	}
+
+
+
+	/**
+	 * Returns the result set resource.
+	 * @return mixed
+	 */
+	public function getResultResource()
+	{
+		return $this->resultSet;
+	}
+
+
+
+	/********************* reflection ****************d*g**/
+
+
+
+	/**
+	 * Returns list of tables.
+	 * @return array
+	 */
+	public function getTables()
+	{
+		throw new NotImplementedException;
+	}
+
+
+
+	/**
+	 * Returns metadata for all columns in a table.
+	 * @param  string
+	 * @return array
+	 */
+	public function getColumns($table)
+	{
+		throw new NotImplementedException;
+	}
+
+
+
+	/**
+	 * Returns metadata for all indexes in a table.
+	 * @param  string
+	 * @return array
+	 */
+	public function getIndexes($table)
+	{
+		throw new NotImplementedException;
+	}
+
+
+
+	/**
+	 * Returns metadata for all foreign keys in a table.
+	 * @param  string
+	 * @return array
+	 */
+	public function getForeignKeys($table)
+	{
+		throw new NotImplementedException;
+	}
+
+}

dibi/drivers/sqlite.php

@@ -29,6 +29,8 @@
  *   - 'lazy' - if TRUE, connection will be established only when required
  *   - 'formatDate' - how to format date in SQL (@see date)
  *   - 'formatDateTime' - how to format datetime in SQL (@see date)
+ *   - 'dbcharset' - database character encoding (will be converted to 'charset')
+ *   - 'charset' - character encoding to set (default is UTF-8)
  *
  * @author     David Grudl
  * @copyright  Copyright (c) 2005, 2009 David Grudl
@@ -36,22 +38,21 @@
  */
 class DibiSqliteDriver extends DibiObject implements IDibiDriver
 {
-
 	/** @var resource  Connection resource */
 	private $connection;
 
-
 	/** @var resource  Resultset resource */
 	private $resultSet;
 
-
 	/** @var bool  Is buffered (seekable and countable)? */
 	private $buffered;
 
-
 	/** @var string  Date and datetime format */
 	private $fmtDate, $fmtDateTime;
 
+	/** @var string  character encoding */
+	private $dbcharset, $charset;
+
 
 
 	/**
@@ -89,6 +90,12 @@ class DibiSqliteDriver extends DibiObject implements IDibiDriver
 		}
 
 		$this->buffered = empty($config['unbuffered']);
+
+		$this->dbcharset = empty($config['dbcharset']) ? 'UTF-8' : $config['dbcharset'];
+		$this->charset = empty($config['charset']) ? 'UTF-8' : $config['charset'];
+		if (strcasecmp($this->dbcharset, $this->charset) === 0) {
+			$this->dbcharset = $this->charset = NULL;
+		}
 	}
 
 
@@ -112,6 +119,10 @@ class DibiSqliteDriver extends DibiObject implements IDibiDriver
 	 */
 	public function query($sql)
 	{
+		if ($this->dbcharset !== NULL) {
+			$sql = iconv($this->charset, $this->dbcharset . '//IGNORE', $sql);
+		}
+
 		DibiDriverException::tryError();
 		if ($this->buffered) {
 			$this->resultSet = sqlite_query($this->connection, $sql);
@@ -292,9 +303,13 @@ class DibiSqliteDriver extends DibiObject implements IDibiDriver
 	public function fetch($assoc)
 	{
 		$row = sqlite_fetch_array($this->resultSet, $assoc ? SQLITE_ASSOC : SQLITE_NUM);
-		if ($assoc && $row) {
+		$charset = $this->charset === NULL ? NULL : $this->charset . '//TRANSLIT';
+		if ($row && ($assoc || $charset)) {
 			$tmp = array();
 			foreach ($row as $k => $v) {
+				if ($charset !== NULL && is_string($v)) {
+					$v = iconv($this->dbcharset, $charset, $v);
+				}
 				$tmp[str_replace(array('[', ']'), '', $k)] = $v;
 			}
 			return $tmp;
@@ -381,7 +396,10 @@ class DibiSqliteDriver extends DibiObject implements IDibiDriver
 			SELECT name, type = 'view' as view FROM sqlite_temp_master WHERE type IN ('table', 'view')
 			ORDER BY name
 		");
-		$res = sqlite_fetch_all($this->resultSet, SQLITE_ASSOC);
+		$res = array();
+		while ($row = $this->fetch(TRUE)) {
+			$res[] = $row;
+		}
 		$this->free();
 		return $res;
 	}