From 32b73a0b1dfcd33e07e7fd2b2cf2fd7d42cf48a0 Mon Sep 17 00:00:00 2001
From: wei <>
Date: Thu, 12 Apr 2007 12:43:49 +0000
Subject: Add Sqlite driver.

---
 framework/Data/Common/Pgsql/TPgsqlMetaData.php | 47 ++++++++++++--------------
 1 file changed, 22 insertions(+), 25 deletions(-)

(limited to 'framework/Data/Common/Pgsql/TPgsqlMetaData.php')

diff --git a/framework/Data/Common/Pgsql/TPgsqlMetaData.php b/framework/Data/Common/Pgsql/TPgsqlMetaData.php
index b789192f..e964cfe7 100644
--- a/framework/Data/Common/Pgsql/TPgsqlMetaData.php
+++ b/framework/Data/Common/Pgsql/TPgsqlMetaData.php
@@ -119,12 +119,27 @@ EOD;
 	 */
 	protected function createNewTableInfo($schemaName,$tableName)
 	{
-		$info['SchemaName'] = $schemaName;
-		$info['TableName'] = $tableName;
+		$info['SchemaName'] = $this->assertIdentifier($schemaName);
+		$info['TableName'] = $this->assertIdentifier($tableName);
 		if($this->getIsView($schemaName,$tableName))
 			$info['IsView'] = true;
-		list($primary, $foreign, $unique) = $this->getConstraintKeys($schemaName, $tableName);
-		return new TPgsqlTableInfo($info,$primary,$foreign, $unique);
+		list($primary, $foreign) = $this->getConstraintKeys($schemaName, $tableName);
+		return new TPgsqlTableInfo($info,$primary,$foreign);
+	}
+
+	/**
+	 * @param string table name, schema name or column name.
+	 * @return string a valid identifier.
+	 * @throws TDbException when table name contains a double quote (").
+	 */
+	protected function assertIdentifier($name)
+	{
+		if(strpos($name, '"')!==false)
+		{
+			$ref = 'http://www.postgresql.org/docs/7.4/static/sql-syntax.html#SQL-SYNTAX-IDENTIFIERS';
+			throw new TDbException('dbcommon_invalid_identifier_name', $name, $ref);
+		}
+		return $name;
 	}
 
 	/**
@@ -163,8 +178,6 @@ EOD;
 			$info['IsPrimaryKey'] = true;
 		if($this->isForeignKeyColumn($columnId, $tableInfo))
 			$info['IsForeignKey'] = true;
-		if(in_array($columnId, $tableInfo->getUniqueKeys()))
-			$info['IsUnique'] = true;
 
 		if($col['atttypmod'] > 0)
 			$info['ColumnSize'] =  $col['atttypmod'] - 4;
@@ -222,10 +235,10 @@ EOD;
 	}
 
 	/**
-	 * Gets the primary, foreign key, and unique column details for the given table.
+	 * Gets the primary and foreign key column details for the given table.
 	 * @param string schema name
 	 * @param string table name.
-	 * @return array tuple ($primary, $foreign, $unique)
+	 * @return array tuple ($primary, $foreign)
 	 */
 	protected function getConstraintKeys($schemaName, $tableName)
 	{
@@ -245,7 +258,6 @@ EOD;
 		$command->bindValue(':schema', $schemaName);
 		$primary = array();
 		$foreign = array();
-		$unique = array();
 		foreach($command->query() as $row)
 		{
 			switch($row['contype'])
@@ -257,13 +269,9 @@ EOD;
 					if(($fkey = $this->getForeignKeys($row['consrc']))!==null)
 						$foreign[] = $fkey;
 					break;
-				case 'u':
-					if(($ukey = $this->getUniqueKey($row['consrc']))!==null)
-						$unique[] = $ukey;
-					break;
 			}
 		}
-		return array($primary,$foreign,$unique);
+		return array($primary,$foreign);
 	}
 
 	/**
@@ -279,17 +287,6 @@ EOD;
 		return array();
 	}
 
-	/**
-	 * @param string pgsql unique constraint definition
-	 * @return string column id if found, null otherwise.
-	 */
-	protected function getUniqueKey($src)
-	{
-		$matches=array();
-		if(preg_match('/UNIQUE\s+\(([^\)]+)\)/i', $src, $matches))
-			return $matches[1];
-	}
-
 	/**
 	 * Gets foreign relationship constraint keys and table name
 	 * @param string pgsql foreign key definition
-- 
cgit v1.2.3