1: <?php
  2: 
  3: /**
  4:  * This file is part of the Nette Framework (https://nette.org)
  5:  * Copyright (c) 2004 David Grudl (http://davidgrudl.com)
  6:  * @package Nette\Database\Drivers
  7:  */
  8: 
  9: 
 10: 
 11: /**
 12:  * Supplemental MySQL database driver.
 13:  *
 14:  * @author     David Grudl
 15:  * @package Nette\Database\Drivers
 16:  */
 17: class NMySqlDriver extends NObject implements ISupplementalDriver
 18: {
 19:     const ERROR_ACCESS_DENIED = 1045;
 20:     const ERROR_DUPLICATE_ENTRY = 1062;
 21:     const ERROR_DATA_TRUNCATED = 1265;
 22: 
 23:     /** @var NConnection */
 24:     private $connection;
 25: 
 26: 
 27:     /**
 28:      * Driver options:
 29:      *   - charset => character encoding to set (default is utf8)
 30:      *   - sqlmode => see http://dev.mysql.com/doc/refman/5.0/en/server-sql-mode.html
 31:      */
 32:     public function __construct(NConnection $connection, array $options)
 33:     {
 34:         $this->connection = $connection;
 35:         $charset = isset($options['charset']) ? $options['charset'] : 'utf8';
 36:         if ($charset) {
 37:             $connection->query("SET NAMES '$charset'");
 38:         }
 39:         if (isset($options['sqlmode'])) {
 40:             $connection->query("SET sql_mode='$options[sqlmode]'");
 41:         }
 42:         $connection->query("SET time_zone='" . date('P') . "'");
 43:     }
 44: 
 45: 
 46:     /********************* SQL ****************d*g**/
 47: 
 48: 
 49:     /**
 50:      * Delimites identifier for use in a SQL statement.
 51:      */
 52:     public function delimite($name)
 53:     {
 54:         // @see http://dev.mysql.com/doc/refman/5.0/en/identifiers.html
 55:         return '`' . str_replace('`', '``', $name) . '`';
 56:     }
 57: 
 58: 
 59:     /**
 60:      * Formats boolean for use in a SQL statement.
 61:      */
 62:     public function formatBool($value)
 63:     {
 64:         return $value ? '1' : '0';
 65:     }
 66: 
 67: 
 68:     /**
 69:      * Formats date-time for use in a SQL statement.
 70:      */
 71:     public function formatDateTime(DateTime $value)
 72:     {
 73:         return $value->format("'Y-m-d H:i:s'");
 74:     }
 75: 
 76: 
 77:     /**
 78:      * Encodes string for use in a LIKE statement.
 79:      */
 80:     public function formatLike($value, $pos)
 81:     {
 82:         $value = addcslashes(str_replace('\\', '\\\\', $value), "\x00\n\r\\'%_");
 83:         return ($pos <= 0 ? "'%" : "'") . $value . ($pos >= 0 ? "%'" : "'");
 84:     }
 85: 
 86: 
 87:     /**
 88:      * Injects LIMIT/OFFSET to the SQL query.
 89:      */
 90:     public function applyLimit(& $sql, $limit, $offset)
 91:     {
 92:         if ($limit >= 0 || $offset > 0) {
 93:             // see http://dev.mysql.com/doc/refman/5.0/en/select.html
 94:             $sql .= ' LIMIT ' . ($limit < 0 ? '18446744073709551615' : (int) $limit)
 95:                 . ($offset > 0 ? ' OFFSET ' . (int) $offset : '');
 96:         }
 97:     }
 98: 
 99: 
100:     /**
101:      * Normalizes result row.
102:      */
103:     public function normalizeRow($row, $statement)
104:     {
105:         return $row;
106:     }
107: 
108: 
109:     /********************* reflection ****************d*g**/
110: 
111: 
112:     /**
113:      * Returns list of tables.
114:      */
115:     public function getTables()
116:     {
117:         $tables = array();
118:         foreach ($this->connection->query('SHOW FULL TABLES') as $row) {
119:             $tables[] = array(
120:                 'name' => $row[0],
121:                 'view' => isset($row[1]) && $row[1] === 'VIEW',
122:             );
123:         }
124:         return $tables;
125:     }
126: 
127: 
128:     /**
129:      * Returns metadata for all columns in a table.
130:      */
131:     public function getColumns($table)
132:     {
133:         $columns = array();
134:         foreach ($this->connection->query('SHOW FULL COLUMNS FROM ' . $this->delimite($table)) as $row) {
135:             $type = explode('(', $row['Type']);
136:             $columns[] = array(
137:                 'name' => $row['Field'],
138:                 'table' => $table,
139:                 'nativetype' => strtoupper($type[0]),
140:                 'size' => isset($type[1]) ? (int) $type[1] : NULL,
141:                 'unsigned' => (bool) strstr($row['Type'], 'unsigned'),
142:                 'nullable' => $row['Null'] === 'YES',
143:                 'default' => $row['Default'],
144:                 'autoincrement' => $row['Extra'] === 'auto_increment',
145:                 'primary' => $row['Key'] === 'PRI',
146:                 'vendor' => (array) $row,
147:             );
148:         }
149:         return $columns;
150:     }
151: 
152: 
153:     /**
154:      * Returns metadata for all indexes in a table.
155:      */
156:     public function getIndexes($table)
157:     {
158:         $indexes = array();
159:         foreach ($this->connection->query('SHOW INDEX FROM ' . $this->delimite($table)) as $row) {
160:             $indexes[$row['Key_name']]['name'] = $row['Key_name'];
161:             $indexes[$row['Key_name']]['unique'] = !$row['Non_unique'];
162:             $indexes[$row['Key_name']]['primary'] = $row['Key_name'] === 'PRIMARY';
163:             $indexes[$row['Key_name']]['columns'][$row['Seq_in_index'] - 1] = $row['Column_name'];
164:         }
165:         return array_values($indexes);
166:     }
167: 
168: 
169:     /**
170:      * Returns metadata for all foreign keys in a table.
171:      */
172:     public function getForeignKeys($table)
173:     {
174:         $keys = array();
175:         $query = 'SELECT CONSTRAINT_NAME, COLUMN_NAME, REFERENCED_TABLE_NAME, REFERENCED_COLUMN_NAME FROM information_schema.KEY_COLUMN_USAGE '
176:             . 'WHERE TABLE_SCHEMA = DATABASE() AND REFERENCED_TABLE_NAME IS NOT NULL AND TABLE_NAME = ' . $this->connection->quote($table);
177: 
178:         foreach ($this->connection->query($query) as $id => $row) {
179:             $keys[$id]['name'] = $row['CONSTRAINT_NAME']; // foreign key name
180:             $keys[$id]['local'] = $row['COLUMN_NAME']; // local columns
181:             $keys[$id]['table'] = $row['REFERENCED_TABLE_NAME']; // referenced table
182:             $keys[$id]['foreign'] = $row['REFERENCED_COLUMN_NAME']; // referenced columns
183:         }
184: 
185:         return array_values($keys);
186:     }
187: 
188: 
189:     /**
190:      * @return bool
191:      */
192:     public function isSupported($item)
193:     {
194:         return $item === self::SUPPORT_COLUMNS_META || $item === self::SUPPORT_SELECT_UNGROUPED_COLUMNS;
195:     }
196: 
197: }
198: