@ -1,56 +1,56 @@
< ? php
/**
* This file defines CreateQuery class .
*
*
* @ package SQLite Integration
* @ author Kojima Toshiyasu
*/
/**
* This class provides a function to rewrite CREATE query .
*
*
*/
class CreateQuery {
/**
* The query string to be rewritten in this class .
*
*
* @ var string
* @ access private
*/
private $_query = '' ;
/**
* The array to contain CREATE INDEX queries .
*
*
* @ var array of strings
* @ access private
*/
private $index_queries = array ();
/**
* The array to contain error messages .
*
*
* @ var array of string
* @ access private
*/
private $_errors = array ();
/**
* Variable to have the table name to be executed .
*
*
* @ var string
* @ access private
*/
private $table_name = '' ;
/**
* Variable to check if the query has the primary key .
*
*
* @ var boolean
* @ access private
*/
private $has_primary_key = false ;
/**
* Function to rewrite query .
*
*
* @ param string $query the query being processed
* @ return string | array the processed ( rewritten ) query
*/
@ -85,50 +85,50 @@ class CreateQuery{
$this -> rewrite_set ();
$this -> rewrite_key ();
$this -> add_if_not_exists ();
return $this -> post_process ();
}
/**
* Method to get table name from the query string .
*
*
* 'IF NOT EXISTS' clause is removed for the easy regular expression usage .
* It will be added at the end of the process .
*
*
* @ access private
*/
private function get_table_name (){
// $pattern = '/^\\s*CREATE\\s*(TEMP|TEMPORARY)?\\s*TABLE\\s*(IF NOT EXISTS)?\\s*([^\(]*)/imsx';
$pattern = '/^\\s*CREATE\\s*(?:TEMP|TEMPORARY)?\\s*TABLE\\s*(?:IF\\s*NOT\\s*EXISTS)?\\s*([^\(]*)/imsx' ;
$pattern = '/^\\s*CREATE\\s*(?:TEMP|TEMPORARY)?\\s*TABLE\\s*(?:IF\\s*NOT\\s*EXISTS)?\\s*([^\(]*)/imsx' ;
if ( preg_match ( $pattern , $this -> _query , $matches )) {
$this -> table_name = trim ( $matches [ 1 ]);
}
$this -> table_name = trim ( $matches [ 1 ]);
}
}
/**
* Method to change the MySQL field types to SQLite compatible types .
*
*
* Order of the key value is important . Don ' t change it .
*
*
* @ access private
*/
private function rewrite_field_types (){
$array_types = array (
'bit' => 'integer' , 'bool' => 'integer' ,
'boolean' => 'integer' , 'tinyint' => 'integer' ,
'smallint' => 'integer' , 'mediumint' => 'integer' ,
'int' => 'integer' , 'integer' => 'integer' ,
'bigint' => 'integer' , 'float' => 'real' ,
'double' => 'real' , 'decimal' => 'real' ,
'dec' => 'real' , 'numeric' => 'real' ,
'fixed' => 'real' , 'date' => 'text' ,
'datetime' => 'text' , 'timestamp' => 'text' ,
'time' => 'text' , 'year' => 'text' ,
'char' => 'text' , 'varchar' => 'text' ,
'binary' => 'integer' , 'varbinary' => 'blob' ,
'tinyblob' => 'blob' , 'tinytext' => 'text' ,
'blob' => 'blob' , 'text' => 'text' ,
'mediumblob' => 'blob' , 'mediumtext' => 'text' ,
'longblob' => 'blob' , 'longtext' => 'text'
);
'bit' => 'integer' , 'bool' => 'integer' ,
'boolean' => 'integer' , 'tinyint' => 'integer' ,
'smallint' => 'integer' , 'mediumint' => 'integer' ,
'int' => 'integer' , 'integer' => 'integer' ,
'bigint' => 'integer' , 'float' => 'real' ,
'double' => 'real' , 'decimal' => 'real' ,
'dec' => 'real' , 'numeric' => 'real' ,
'fixed' => 'real' , 'date' => 'text' ,
'datetime' => 'text' , 'timestamp' => 'text' ,
'time' => 'text' , 'year' => 'text' ,
'char' => 'text' , 'varchar' => 'text' ,
'binary' => 'integer' , 'varbinary' => 'blob' ,
'tinyblob' => 'blob' , 'tinytext' => 'text' ,
'blob' => 'blob' , 'text' => 'text' ,
'mediumblob' => 'blob' , 'mediumtext' => 'text' ,
'longblob' => 'blob' , 'longtext' => 'text'
);
foreach ( $array_types as $o => $r ){
$pattern = " / \\ b(?<!`) $o\\b\\s *( \ ([^ \ )]* \ )*)? \\ s*/ims " ;
if ( preg_match ( " /^ \\ s*.*? \\ s* \ (.*? $o .*? \ )/im " , $this -> _query )) {
@ -140,43 +140,43 @@ class CreateQuery{
}
/**
* Method for stripping the comments from the SQL statement .
*
*
* @ access private
*/
*/
private function rewrite_comments (){
$this -> _query = preg_replace ( " /# --------------------------------------------------------/ " , " -- ****************************************************** " , $this -> _query );
$this -> _query = preg_replace ( " /#/ " , " -- " , $this -> _query );
}
/**
* Method for stripping the engine and other stuffs .
*
*
* TYPE , ENGINE and AUTO_INCREMENT are removed here .
* @ access private
*/
*/
private function rewrite_engine_info (){
$this -> _query = preg_replace ( " / \\ s*(TYPE|ENGINE) \\ s*= \\ s*.*(?<!;)/ims " , '' , $this -> _query );
$this -> _query = preg_replace ( " / AUTO_INCREMENT \\ s*= \\ s*[0-9]*/ims " , '' , $this -> _query );
}
/**
* Method for stripping unsigned .
*
*
* SQLite doesn ' t have unsigned int data type . So UNSIGNED INT ( EGER ) is converted
* to INTEGER here .
*
*
* @ access private
*/
*/
private function rewrite_unsigned (){
$this -> _query = preg_replace ( '/\\bunsigned\\b/ims' , ' ' , $this -> _query );
}
/**
* Method for rewriting primary key auto_increment .
*
*
* If the field type is 'INTEGER PRIMARY KEY' , it is automatically autoincremented
* by SQLite . There ' s a little difference between PRIMARY KEY and AUTOINCREMENT , so
* we may well convert to PRIMARY KEY only .
*
*
* @ access private
*/
*/
private function rewrite_autoincrement (){
$this -> _query = preg_replace ( '/\\bauto_increment\\s*primary\\s*key\\s*(,)?/ims' , ' PRIMARY KEY AUTOINCREMENT \\1' , $this -> _query , - 1 , $count );
$this -> _query = preg_replace ( '/\\bauto_increment\\b\\s*(,)?/ims' , ' PRIMARY KEY AUTOINCREMENT $1' , $this -> _query , - 1 , $count );
@ -186,72 +186,72 @@ class CreateQuery{
}
/**
* Method for rewriting primary key .
*
*
* @ access private
*/
*/
private function rewrite_primary_key (){
if ( $this -> has_primary_key ) {
$this -> _query = preg_replace ( '/\\s*primary key\\s*.*?\([^\)]*\)\\s*(,|)/i' , ' ' , $this -> _query );
} else {
// If primary key has an index name, we remove that name.
$this -> _query = preg_replace ( '/\\bprimary\\s*key\\s*.*?\\s*(\(.*?\))/im' , 'PRIMARY KEY \\1' , $this -> _query );
// If primary key has an index name, we remove that name.
$this -> _query = preg_replace ( '/\\bprimary\\s*key\\s*.*?\\s*(\(.*?\))/im' , 'PRIMARY KEY \\1' , $this -> _query );
}
}
/**
* Method for rewriting foreign key .
*
* @ access private
*/
private function rewrite_foreign_key () {
$pattern = '/\\s*foreign\\s*key\\s*(|.*?)\([^\)]+?\)\\s*references\\s*.*/i' ;
if ( preg_match_all ( $pattern , $this -> _query , $match )) {
if ( isset ( $match [ 1 ])) {
$this -> _query = str_ireplace ( $match [ 1 ], '' , $this -> _query );
}
}
}
/**
* Method for rewriting foreign key .
*
* @ access private
*/
private function rewrite_foreign_key () {
$pattern = '/\\s*foreign\\s*key\\s*(|.*?)\([^\)]+?\)\\s*references\\s*.*/i' ;
if ( preg_match_all ( $pattern , $this -> _query , $match )) {
if ( isset ( $match [ 1 ])) {
$this -> _query = str_ireplace ( $match [ 1 ], '' , $this -> _query );
}
}
}
/**
* Method for rewriting unique key .
*
*
* @ access private
*/
*/
private function rewrite_unique_key (){
$this -> _query = preg_replace_callback ( '/\\bunique key\\b([^\(]*)(\(.*\))/im' , array ( $this , '_rewrite_unique_key' ), $this -> _query );
}
/**
* Callback method for rewrite_unique_key .
*
*
* @ param array $matches an array of matches from the Regex
* @ access private
*/
private function _rewrite_unique_key ( $matches ){
$index_name = trim ( $matches [ 1 ]);
$col_name = trim ( $matches [ 2 ]);
$tbl_name = $this -> table_name ;
if ( preg_match ( '/\(\\d+?\)/' , $col_name )) {
$col_name = preg_replace ( '/\(\\d+?\)/' , '' , $col_name );
}
$_wpdb = new PDODB ();
$results = $_wpdb -> get_results ( " SELECT name FROM sqlite_master WHERE type='index' " );
$_wpdb = null ;
if ( $results ) {
foreach ( $results as $result ) {
if ( $result -> name == $index_name ) {
$r = rand ( 0 , 50 );
$index_name = $index_name . " _ $r " ;
break ;
}
}
}
$index_name = trim ( $matches [ 1 ]);
$col_name = trim ( $matches [ 2 ]);
$tbl_name = $this -> table_name ;
if ( preg_match ( '/\(\\d+?\)/' , $col_name )) {
$col_name = preg_replace ( '/\(\\d+?\)/' , '' , $col_name );
}
$_wpdb = new PDODB ();
$results = $_wpdb -> get_results ( " SELECT name FROM sqlite_master WHERE type='index' " );
$_wpdb = null ;
if ( $results ) {
foreach ( $results as $result ) {
if ( $result -> name == $index_name ) {
$r = rand ( 0 , 50 );
$index_name = $index_name . " _ $r " ;
break ;
}
}
}
$index_name = str_replace ( ' ' , '' , $index_name );
$this -> index_queries [] = " CREATE UNIQUE INDEX $index_name ON " . $tbl_name . $col_name ;
return '' ;
}
/**
* Method for handling ENUM fields .
*
*
* SQLite doesn ' t support enum , so we change it to check constraint .
*
*
* @ access private
*/
private function rewrite_enum (){
@ -260,7 +260,7 @@ class CreateQuery{
}
/**
* Call back method for rewrite_enum () and rewrite_set () .
*
*
* @ access private
*/
private function _rewrite_enum ( $matches ){
@ -269,9 +269,9 @@ class CreateQuery{
}
/**
* Method for rewriting usage of set .
*
*
* It is similar but not identical to enum . SQLite does not support either .
*
*
* @ access private
*/
private function rewrite_set (){
@ -280,10 +280,10 @@ class CreateQuery{
}
/**
* Method for rewriting usage of key to create an index .
*
*
* SQLite cannot create non - unique indices as part of the create query ,
* so we need to create an index by hand and append it to the create query .
*
*
* @ access private
*/
private function rewrite_key (){
@ -291,72 +291,72 @@ class CreateQuery{
}
/**
* Callback method for rewrite_key .
*
*
* @ param array $matches an array of matches from the Regex
* @ access private
*/
*/
private function _rewrite_key ( $matches ){
$index_name = trim ( $matches [ 2 ]);
$col_name = trim ( $matches [ 3 ]);
if ( preg_match ( '/\([0-9]+?\)/' , $col_name , $match )) {
$col_name = preg_replace_callback ( '/\([0-9]+?\)/' , array ( $this , '_remove_length' ), $col_name );
}
$tbl_name = $this -> table_name ;
$_wpdb = new PDODB ();
$results = $_wpdb -> get_results ( " SELECT name FROM sqlite_master WHERE type='index' " );
$_wpdb = null ;
if ( $results ) {
foreach ( $results as $result ) {
if ( $result -> name == $index_name ) {
$r = rand ( 0 , 50 );
$index_name = $index_name . " _ $r " ;
break ;
}
}
}
$index_name = trim ( $matches [ 2 ]);
$col_name = trim ( $matches [ 3 ]);
if ( preg_match ( '/\([0-9]+?\)/' , $col_name , $match )) {
$col_name = preg_replace_callback ( '/\([0-9]+?\)/' , array ( $this , '_remove_length' ), $col_name );
}
$tbl_name = $this -> table_name ;
$_wpdb = new PDODB ();
$results = $_wpdb -> get_results ( " SELECT name FROM sqlite_master WHERE type='index' " );
$_wpdb = null ;
if ( $results ) {
foreach ( $results as $result ) {
if ( $result -> name == $index_name ) {
$r = rand ( 0 , 50 );
$index_name = $index_name . " _ $r " ;
break ;
}
}
}
$this -> index_queries [] = 'CREATE INDEX ' . $index_name . ' ON ' . $tbl_name . $col_name ;
return '' ;
}
/**
* Call back method to remove unnecessary string .
*
*
* This method is deprecated .
*
*
* @ param string $match
* @ return string whose length is zero
* @ access private
*/
private function _remove_length ( $match ) {
return '' ;
return '' ;
}
/**
* Method to assemble the main query and index queries into an array .
*
*
* It return the array of the queries to be executed separately .
*
*
* @ return array
* @ access private
*/
private function post_process (){
private function post_process ()
$mainquery = $this -> _query ;
do {
$count = 0 ;
$mainquery = preg_replace ( '/,\\s*\)/imsx' , ')' , $mainquery , - 1 , $count );
} while ( $count > 0 );
do {
$count = 0 ;
$mainquery = preg_replace ( '/\(\\s*?,/imsx' , '(' , $mainquery , - 1 , $count );
$count = 0 ;
$mainquery = preg_replace ( '/\(\\s*?,/imsx' , '(' , $mainquery , - 1 , $count );
} while ( $count > 0 );
$return_val [] = $mainquery ;
$return_val = array_merge ( $return_val , $this -> index_queries );
return $return_val ;
return $return_val ;
}
/**
* Method to add IF NOT EXISTS to query string .
*
*
* This adds IF NOT EXISTS to every query string , which prevent the exception
* from being thrown .
*
*
* @ access private
*/
private function add_if_not_exists (){
@ -364,33 +364,33 @@ class CreateQuery{
$this -> _query = preg_replace ( $pattern_table , 'CREATE $1 TABLE IF NOT EXISTS ' , $this -> _query );
$pattern_index = '/^\\s*CREATE\\s*(UNIQUE)?\\s*INDEX\\s*(IF NOT EXISTS)?\\s*/ims' ;
for ( $i = 0 ; $i < count ( $this -> index_queries ); $i ++ ) {
$this -> index_queries [ $i ] = preg_replace ( $pattern_index , 'CREATE $1 INDEX IF NOT EXISTS ' , $this -> index_queries [ $i ]);
$this -> index_queries [ $i ] = preg_replace ( $pattern_index , 'CREATE $1 INDEX IF NOT EXISTS ' , $this -> index_queries [ $i ]);
}
}
/**
* Method to strip back quotes .
*
*
* @ access private
*/
*/
private function strip_backticks (){
$this -> _query = str_replace ( '`' , '' , $this -> _query );
foreach ( $this -> index_queries as & $query ) {
$query = str_replace ( '`' , '' , $query );
}
foreach ( $this -> index_queries as & $query ) {
$query = str_replace ( '`' , '' , $query );
}
}
/**
* Method to remove the character set information from within mysql queries .
*
*
* This removes DEFAULT CHAR ( ACTER ) SET and COLLATE , which is meaningless for
* SQLite .
*
*
* @ access private
*/
*/
private function rewrite_character_set (){
$pattern_charset = '/\\b(default\\s*character\\s*set|default\\s*charset|character\\s*set)\\s*(?<!\()[^ ]*/im' ;
$pattern_collate1 = '/\\s*collate\\s*[^ ]*(?=,)/im' ;
$pattern_collate2 = '/\\s*collate\\s*[^ ]*(?<!;)/im' ;
$patterns = array ( $pattern_charset , $pattern_collate1 , $pattern_collate2 );
$pattern_collate2 = '/\\s*collate\\s*[^ ]*(?<!;)/im' ;
$patterns = array ( $pattern_charset , $pattern_collate1 , $pattern_collate2 );
$this -> _query = preg_replace ( $patterns , '' , $this -> _query );
}
}
kjmtsh
8 years ago