From 97270a389ad78f83190d8ff4cb595a696e6559a0 Mon Sep 17 00:00:00 2001
From: Gevik Babakhani <gevik@ext4yii.com>
Date: Fri, 31 May 2013 14:25:56 +0200
Subject: [PATCH] Savepoint, implementing fkeys.

---
 framework/yii/db/pgsql/Schema.php | 76 ++++++++++++++++++++++++++++++++-------
 tests/unit/data/postgres.sql      |  5 ++-
 2 files changed, 67 insertions(+), 14 deletions(-)

diff --git a/framework/yii/db/pgsql/Schema.php b/framework/yii/db/pgsql/Schema.php
index f7329bd..f42c132 100644
--- a/framework/yii/db/pgsql/Schema.php
+++ b/framework/yii/db/pgsql/Schema.php
@@ -136,12 +136,30 @@ class Schema extends \yii\db\Schema {
 		$table = new TableSchema();
 		$this->resolveTableNames($table, $name);
 		if ($this->findColumns($table)) {
-			$this->findForeignKeys($table);
+			$this->findConstraints($table);
 			return $table;
 		}
 	}
 
 	/**
+	 * Collects the foreign key column details for the given table.
+	 * @param TableSchema $table the table metadata
+	 */
+	protected function findConstraints($table) {
+		
+		try {
+			$constraints = $this->db->createCommand($sql)->queryAll();
+		} catch (\Exception $e) {
+			return false;
+		}
+		foreach ($constraints as $constraint) {
+			$column = $this->loadColumnSchema($column);
+			$table->columns[$column->name] = $column;
+		}
+		return true;
+	}
+
+	/**
 	 * Collects the metadata of table columns.
 	 * @param TableSchema $table the table metadata
 	 * @return boolean whether the table exists in the database
@@ -163,13 +181,41 @@ SELECT
         a.attnotnull = false AS is_nullable,	
         CAST(pg_get_expr(ad.adbin, ad.adrelid) AS varchar) AS column_default,
         coalesce(pg_get_expr(ad.adbin, ad.adrelid) ~ 'nextval',false) AS is_autoinc,
-        array_to_string((select array_agg(enumlabel) from pg_enum where enumtypid=a.atttypid)::varchar[],',') as enum_values
+        array_to_string((select array_agg(enumlabel) from pg_enum where enumtypid=a.atttypid)::varchar[],',') as enum_values,
+	CASE atttypid
+		 WHEN 21 /*int2*/ THEN 16
+		 WHEN 23 /*int4*/ THEN 32
+		 WHEN 20 /*int8*/ THEN 64
+		 WHEN 1700 /*numeric*/ THEN
+		      CASE WHEN atttypmod = -1
+			   THEN null
+			   ELSE ((atttypmod - 4) >> 16) & 65535
+			   END
+		 WHEN 700 /*float4*/ THEN 24 /*FLT_MANT_DIG*/
+		 WHEN 701 /*float8*/ THEN 53 /*DBL_MANT_DIG*/
+		 ELSE null
+	  END   AS numeric_precision,
+	  CASE 
+	    WHEN atttypid IN (21, 23, 20) THEN 0
+	    WHEN atttypid IN (1700) THEN            
+		CASE 
+		    WHEN atttypmod = -1 THEN null       
+		    ELSE (atttypmod - 4) & 65535
+		END
+	       ELSE null
+	  END AS numeric_scale,
+	CAST(
+             information_schema._pg_char_max_length(information_schema._pg_truetypid(a, t), information_schema._pg_truetypmod(a, t))
+             AS numeric
+	) AS size,
+	a.attnum = any (ct.conkey) as is_pkey			
 FROM
 	pg_class c
 	LEFT JOIN pg_attribute a ON a.attrelid = c.oid
 	LEFT JOIN pg_attrdef ad ON a.attrelid = ad.adrelid AND a.attnum = ad.adnum
 	LEFT JOIN pg_type t ON a.atttypid = t.oid
-	LEFT JOIN pg_namespace d ON d.oid = c.relnamespace         
+	LEFT JOIN pg_namespace d ON d.oid = c.relnamespace
+	LEFT join pg_constraint ct on ct.conrelid=c.oid and ct.contype='p'
 WHERE
 	a.attnum > 0
 	and c.relname = {$tableName}
@@ -186,10 +232,9 @@ SQL;
 		}
 		foreach ($columns as $column) {
 			$column = $this->loadColumnSchema($column);
-			if ($column->name == 'numbers')
-				print_r($column);
+			$table->columns[$column->name] = $column;
 		}
-		die();
+		return true;
 	}
 
 	/**
@@ -205,14 +250,19 @@ SQL;
 		$column->dbType = $info['data_type'];
 		$column->defaultValue = $info['column_default'];
 		$column->enumValues = explode(',', str_replace(array("''"), array("'"), $info['enum_values']));
-//$column->isPrimaryKey
+		$column->unsigned = false; // has no meanining in PG
+		$column->isPrimaryKey = $info['is_pkey'];
 		$column->name = $info['column_name'];
-		//$column->phpType
-//$column->precision
-//$column->scale
-//$column->size;
-//$column->type
-//$column->unsigned
+		$column->precision = $info['numeric_precision'];
+		$column->scale = $info['numeric_scale'];
+		$column->size = $info['size'];
+
+		if (isset($this->typeMap[$column->dbType])) {
+			$column->type = $this->typeMap[$column->dbType];
+		} else {
+			$column->type = self::TYPE_STRING;
+		}
+		$column->phpType = $this->getColumnPhpType($column);
 		return $column;
 	}
 
diff --git a/tests/unit/data/postgres.sql b/tests/unit/data/postgres.sql
index f158e58..6a02691 100644
--- a/tests/unit/data/postgres.sql
+++ b/tests/unit/data/postgres.sql
@@ -22,11 +22,14 @@ CREATE TABLE tbl_customer (
   id serial not null primary key,
   email varchar(128) NOT NULL,
   name varchar(128) NOT NULL,
+  age numeric(3),
+  zipcode varchar(6),
   address text,
   status integer DEFAULT 0,
   fullname fullname,
   mood mood,
-  numbers integer[]
+  numbers integer[],
+  amount numeric(6,4)
 );
 
 comment on column public.tbl_customer.email is 'someone@example.com';