<?php
# ***** BEGIN LICENSE BLOCK *****
# This file is part of Clearbricks.
# Copyright (c) 2003-2010 Olivier Meunier & Association Dotclear
# All rights reserved.
#
# Clearbricks is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
# 
# Clearbricks is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
# 
# You should have received a copy of the GNU General Public License
# along with Clearbricks; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
#
# ***** END LICENSE BLOCK *****

if (class_exists('dbSchema'))
{
	class sqliteSchema extends dbSchema implements i_dbSchema
	{
		private $table_hist = array();
		
		private $table_stack = array();	// Stack for tables creation
		private $x_stack = array();		// Execution stack
		
		public function dbt2udt($type,&$len,&$default)
		{
			$type = parent::dbt2udt($type,$len,$default);
			
			switch ($type)
			{
				case 'float':
					return 'real';
				case 'double':
					return 'float';
				case 'timestamp':
					# DATETIME real type is TIMESTAMP
					if ($default == "'1970-01-01 00:00:00'") {
						# Bad hack
						$default = 'now()';
					}
					return 'timestamp';
				case 'integer':
				case 'mediumint':
				case 'bigint':
				case 'tinyint':
				case 'smallint':
				case 'numeric':
					return 'integer';
				case 'tinytext':
				case 'longtext':
					return 'text';
			}
			
			return $type;
		}
		
		public function udt2dbt($type,&$len,&$default)
		{
			$type = parent::udt2dbt($type,$len,$default);
			
			switch ($type)
			{
				case 'integer':
				case 'smallint':
				case 'bigint':
					return 'integer';
				case 'real':
				case 'float:':
					return 'real';
				case 'date':
				case 'time':
					return 'timestamp';
				case 'timestamp':
					if ($default == 'now()') {
						# SQLite does not support now() default value...
						$default = "'1970-01-01 00:00:00'";
					}
					return $type;
			}
			
			return $type;
		}
		
		public function flushStack()
		{
			foreach ($this->table_stack as $table => $def)
			{
				$sql = 'CREATE TABLE '.$table." (\n".implode(",\n",$def)."\n)\n ";
				$this->con->execute($sql);
			}
			
			foreach ($this->x_stack as $x)
			{
				$this->con->execute($x);
			}
		}
		
		public function db_get_tables()
		{
			$res = array();
			$sql = "SELECT * FROM sqlite_master WHERE type = 'table'";
			$rs = $this->con->select($sql);
			
			$res = array();
			while ($rs->fetch()) {
				$res[] = $rs->tbl_name;
			}
			
			return $res;
		}
		
		public function db_get_columns($table)
		{
			$sql = 'PRAGMA table_info('.$this->con->escapeSystem($table).')';
			$rs = $this->con->select($sql);
			
			$res = array();
			while ($rs->fetch())
			{
				$field = trim($rs->name);
				$type = trim($rs->type);
				$null = trim($rs->notnull) == 0;
				$default = trim($rs->dflt_value);
				
				$len = null;
				if (preg_match('/^(.+?)\(([\d,]+)\)$/si',$type,$m)) {
					$type = $m[1];
					$len = (integer) $m[2];
				}
				
				$res[$field] = array(
					'type' => $type,
					'len' => $len,
					'null' => $null,
					'default' => $default
				);
			}
			return $res;
		}
		
		public function db_get_keys($table)
		{
			$t = array();
			$res = array();
			
			# Get primary keys first
			$sql = "SELECT sql FROM sqlite_master WHERE type='table' AND name='".$this->con->escape($table)."'";
			$rs = $this->con->select($sql);
			
			if ($rs->isEmpty()) {
				return array();
			}
			
			# Get primary keys
			$n = preg_match_all('/^\s*CONSTRAINT\s+([^,]+?)\s+PRIMARY\s+KEY\s+\((.+?)\)/msi',$rs->sql,$match);
			if ($n > 0)
			{
				foreach ($match[1] as $i => $name)
				{
					$cols = preg_split('/\s*,\s*/',$match[2][$i]);
					$res[] = array(
						'name' => $name,
						'primary' => true,
						'unique' => false,
						'cols' => $cols
					);
				}
			}
			
			# Get unique keys
			$n = preg_match_all('/^\s*CONSTRAINT\s+([^,]+?)\s+UNIQUE\s+\((.+?)\)/msi',$rs->sql,$match);
			if ($n > 0)
			{
				foreach ($match[1] as $i => $name)
				{
					$cols = preg_split('/\s*,\s*/',$match[2][$i]);
					$res[] = array(
						'name' => $name,
						'primary' => false,
						'unique' => true,
						'cols' => $cols
					);
				}
			}
			
			return $res;
		}
		
		public function db_get_indexes($table)
		{
			$sql = 'PRAGMA index_list('.$this->con->escapeSystem($table).')';
			$rs = $this->con->select($sql);
			
			$res = array();
			while ($rs->fetch())
			{
				if (preg_match('/^sqlite_/',$rs->name)) {
					continue;
				}
				
				$idx = $this->con->select('PRAGMA index_info('.$this->con->escapeSystem($rs->name).')');
				$cols = array();
				while ($idx->fetch()) {
					$cols[] = $idx->name;
				}
				
				$res[] = array(
					'name' => $rs->name,
					'type' => 'btree',
					'cols' => $cols
				);
			}
			
			return $res;
		}
		
		public function db_get_references($table)
		{
			$sql = 'SELECT * FROM sqlite_master WHERE type=\'trigger\' AND tbl_name = \'%1$s\' AND name LIKE \'%2$s_%%\' ';
			$res = array();
			
			# Find constraints on table
			$bir = $this->con->select(sprintf($sql,$this->con->escape($table),'bir'));
			$bur = $this->con->select(sprintf($sql,$this->con->escape($table),'bur'));
			
			if ($bir->isEmpty() || $bur->isempty()) {
				return $res;
			}
			
			while ($bir->fetch())
			{
				# Find child column and parent table and column
				if (!preg_match('/FROM\s+(.+?)\s+WHERE\s+(.+?)\s+=\s+NEW\.(.+?)\s*?\) IS\s+NULL/msi',$bir->sql,$m)) {
					continue;
				}
				
				$c_col = $m[3];
				$p_table = $m[1];
				$p_col = $m[2];
				
				# Find on update
				$on_update = 'restrict';
				$aur = $this->con->select(sprintf($sql,$this->con->escape($p_table),'aur'));
				while ($aur->fetch())
				{
					if (!preg_match('/AFTER\s+UPDATE/msi',$aur->sql)) {
						continue;
					}
					
					if (preg_match('/UPDATE\s+'.$table.'\s+SET\s+'.$c_col.'\s*=\s*NEW.'.$p_col.
						'\s+WHERE\s+'.$c_col.'\s*=\s*OLD\.'.$p_col.'/msi',$aur->sql)) {
						$on_update = 'cascade';
						break;
					}
					
					if (preg_match('/UPDATE\s+'.$table.'\s+SET\s+'.$c_col.'\s*=\s*NULL'.
						'\s+WHERE\s+'.$c_col.'\s*=\s*OLD\.'.$p_col.'/msi',$aur->sql)) {
						$on_update = 'set null';
						break;
					}
				}
				
				# Find on delete
				$on_delete = 'restrict';
				$bdr = $this->con->select(sprintf($sql,$this->con->escape($p_table),'bdr'));
				while ($bdr->fetch())
				{
					if (!preg_match('/BEFORE\s+DELETE/msi',$bdr->sql)) {
						continue;
					}
					
					if (preg_match('/DELETE\s+FROM\s+'.$table.'\s+WHERE\s+'.$c_col.'\s*=\s*OLD\.'.$p_col.'/msi',$bdr->sql)) {
						$on_delete = 'cascade';
						break;
					}
					
					if (preg_match('/UPDATE\s+'.$table.'\s+SET\s+'.$c_col.'\s*=\s*NULL'.
						'\s+WHERE\s+'.$c_col.'\s*=\s*OLD\.'.$p_col.'/msi',$bdr->sql)) {
						$on_update = 'set null';
						break;
					}
				}
				
				$res[] = array(
					'name' => substr($bir->name,4),
					'c_cols' => array($c_col),
					'p_table' => $p_table,
					'p_cols' => array($p_col),
					'update' => $on_update,
					'delete' => $on_delete
				);
			}
			
			return $res;
		}
		
		public function db_create_table($name,$fields)
		{
			$a = array();
			
			foreach ($fields as $n => $f)
			{
				$type = $f['type'];
				$len = (integer) $f['len'];
				$default = $f['default'];
				$null = $f['null'];
				
				$type = $this->udt2dbt($type,$len,$default);
				$len = $len > 0 ? '('.$len.')' : '';
				$null = $null ? 'NULL' : 'NOT NULL';
				
				if ($default === null) {
					$default = 'DEFAULT NULL';
				} elseif ($default !== false) {
					$default = 'DEFAULT '.$default.' ';
				} else {
					$default = '';
				}
				
				$a[] = $n.' '.$type.$len.' '.$null.' '.$default;
			}
			
			$this->table_stack[$name][] = implode(",\n",$a);
			$this->table_hist[$name] = $fields;
		}
		
		public function db_create_field($table,$name,$type,$len,$null,$default)
		{
			$type = $this->udt2dbt($type,$len,$default);
			$len = $len > 0 ? '('.$len.')' : '';
			$null = $null ? 'NULL' : 'NOT NULL';
			
			if ($default === null) {
				$default = 'DEFAULT NULL';
			} elseif ($default !== false) {
				$default = 'DEFAULT '.$default.' ';
			} else {
				$default = '';
			}
			
			$sql =
			'ALTER TABLE '.$this->con->escapeSystem($table).' '.
			'ADD COLUMN '.$this->con->escapeSystem($name).' '.
			$type.$len.' '.$null.' '.$default;
			
			$this->con->execute($sql);
		}
		
		public function db_create_primary($table,$name,$cols)
		{
			$this->table_stack[$table][] = 'CONSTRAINT '.$name.' PRIMARY KEY ('.implode(',',$cols).') ';
		}
		
		public function db_create_unique($table,$name,$cols)
		{
			$this->table_stack[$table][] = 'CONSTRAINT '.$name.' UNIQUE ('.implode(',',$cols).') ';
		}
		
		public function db_create_index($table,$name,$type,$cols)
		{
			$this->x_stack[] = 'CREATE INDEX '.$name.' ON '.$table.' ('.implode(',',$cols).') ';
		}
		
		public function db_create_reference($name,$c_table,$c_cols,$p_table,$p_cols,$update,$delete)
		{
			if (!isset($this->table_hist[$c_table])) {
				return;
			}
			
			if (count($c_cols) > 1 || count($p_cols) > 1) {
				throw new Exception('SQLite UDBS does not support multiple columns foreign keys');
			}
			
			$c_col = $c_cols[0];
			$p_col = $p_cols[0];
			
			$update = strtolower($update);
			$delete = strtolower($delete);
			
			$cnull = $this->table_hist[$c_table][$c_col]['null'];
			
			# Create constraint
			$this->x_stack[] =
			'CREATE TRIGGER bir_'.$name."\n".
			'BEFORE INSERT ON '.$c_table."\n".
			"FOR EACH ROW BEGIN\n".
			'  SELECT RAISE(ROLLBACK,\'insert on table "'.$c_table.'" violates foreign key constraint "'.$name.'"\')'."\n".
			'  WHERE '.
			($cnull ? 'NEW.'.$c_col." IS NOT NULL\n  AND " : '').
			'(SELECT '.$p_col.' FROM '.$p_table.' WHERE '.$p_col.' = NEW.'.$c_col.") IS NULL;\n".
			"END;\n";
			
			# Update constraint
			$this->x_stack[] =
			'CREATE TRIGGER bur_'.$name."\n".
			'BEFORE UPDATE ON '.$c_table."\n".
			"FOR EACH ROW BEGIN\n".
			'  SELECT RAISE(ROLLBACK,\'update on table "'.$c_table.'" violates foreign key constraint "'.$name.'"\')'."\n".
			'  WHERE '.
			($cnull ? 'NEW.'.$c_col." IS NOT NULL\n  AND " : '').
			'(SELECT '.$p_col.' FROM '.$p_table.' WHERE '.$p_col.' = NEW.'.$c_col.") IS NULL;\n".
			"END;\n";
			
			# ON UPDATE
			if ($update == 'cascade')
			{
				$this->x_stack[] =
				'CREATE TRIGGER aur_'.$name."\n".
				'AFTER UPDATE ON '.$p_table."\n".
				"FOR EACH ROW BEGIN\n".
				'  UPDATE '.$c_table.' SET '.$c_col.' = NEW.'.$p_col.' WHERE '.$c_col.' = OLD.'.$p_col.";\n".
				"END;\n";
			}
			elseif ($update == 'set null')
			{
				$this->x_stack[] =
				'CREATE TRIGGER aur_'.$name."\n".
				'AFTER UPDATE ON '.$p_table."\n".
				"FOR EACH ROW BEGIN\n".
				'  UPDATE '.$c_table.' SET '.$c_col.' = NULL WHERE '.$c_col.' = OLD.'.$p_col.";\n".
				"END;\n";
			}
			else # default on restrict
			{
				$this->x_stack[] =
				'CREATE TRIGGER burp_'.$name."\n".
				'BEFORE UPDATE ON '.$p_table."\n".
				"FOR EACH ROW BEGIN\n".
				'  SELECT RAISE (ROLLBACK,\'update on table "'.$p_table.'" violates foreign key constraint "'.$name.'"\')'."\n".
				'  WHERE (SELECT '.$c_col.' FROM '.$c_table.' WHERE '.$c_col.' = OLD.'.$p_col.") IS NOT NULL;\n".
				"END;\n";
			}
			
			# ON DELETE
			if ($delete == 'cascade')
			{
				$this->x_stack[] =
				'CREATE TRIGGER bdr_'.$name."\n".
				'BEFORE DELETE ON '.$p_table."\n".
				"FOR EACH ROW BEGIN\n".
				'  DELETE FROM '.$c_table.' WHERE '.$c_col.' = OLD.'.$p_col.";\n".
				"END;\n";
			}
			elseif ($delete == 'set null')
			{
				$this->x_stack[] =
				'CREATE TRIGGER bdr_'.$name."\n".
				'BEFORE DELETE ON '.$p_table."\n".
				"FOR EACH ROW BEGIN\n".
				'  UPDATE '.$c_table.' SET '.$c_col.' = NULL WHERE '.$c_col.' = OLD.'.$p_col.";\n".
				"END;\n";
			}
			else
			{
				$this->x_stack[] =
				'CREATE TRIGGER bdr_'.$name."\n".
				'BEFORE DELETE ON '.$p_table."\n".
				"FOR EACH ROW BEGIN\n".
				'  SELECT RAISE (ROLLBACK,\'delete on table "'.$p_table.'" violates foreign key constraint "'.$name.'"\')'."\n".
				'  WHERE (SELECT '.$c_col.' FROM '.$c_table.' WHERE '.$c_col.' = OLD.'.$p_col.") IS NOT NULL;\n".
				"END;\n";
			}
		}
		
		public function db_alter_field($table,$name,$type,$len,$null,$default)
		{
			$type = $this->udt2dbt($type,$len,$default);
			if ($type != 'integer' && $type != 'text' && $type != 'timestamp') {
				throw new Exception('SQLite fields cannot be changed.');
			}
		}
		
		public function db_alter_primary($table,$name,$newname,$cols)
		{
			throw new Exception('SQLite primary key cannot be changed.');
		}
		
		public function db_alter_unique($table,$name,$newname,$cols)
		{
			throw new Exception('SQLite unique index cannot be changed.');
		}
		
		public function db_alter_index($table,$name,$newname,$type,$cols)
		{
			$this->con->execute('DROP INDEX IF EXISTS '.$name);
			$this->con->execute('CREATE INDEX '.$newname.' ON '.$table.' ('.implode(',',$cols).') ');
		}
		
		public function db_alter_reference($name,$newname,$c_table,$c_cols,$p_table,$p_cols,$update,$delete)
		{
			$this->con->execute('DROP TRIGGER IF EXISTS bur_'.$name);
			$this->con->execute('DROP TRIGGER IF EXISTS burp_'.$name);
			$this->con->execute('DROP TRIGGER IF EXISTS bir_'.$name);
			$this->con->execute('DROP TRIGGER IF EXISTS aur_'.$name);
			$this->con->execute('DROP TRIGGER IF EXISTS bdr_'.$name);
			
			$this->table_hist[$c_table] = $this->db_get_columns($c_table);
			$this->db_create_reference($newname,$c_table,$c_cols,$p_table,$p_cols,$update,$delete);
		}
		
		public function db_drop_unique($table,$name)
		{
			throw new Exception('SQLite unique index cannot be removed.');
		}
	}
}
?>