From 2d5f178426f365c50ead603ca1e4666252df54f7 Mon Sep 17 00:00:00 2001
From: Matthew Bucci <buccim2@gene.com>
Date: Mon, 26 Feb 2024 23:44:17 -0800
Subject: [PATCH] improve primary key detection to use first column found if no
 primary key found

---
 pg4wp/driver_pgsql.php | 26 +++++++++++++++++++-------
 1 file changed, 19 insertions(+), 7 deletions(-)

diff --git a/pg4wp/driver_pgsql.php b/pg4wp/driver_pgsql.php
index d81fc58..d6a8469 100644
--- a/pg4wp/driver_pgsql.php
+++ b/pg4wp/driver_pgsql.php
@@ -469,18 +469,30 @@ function wpsqli_rollback(&$connection, $flags = 0, $name = null)
 function get_primary_key_for_table(&$connection, $table)
 {
     $query = <<<SQL
-    SELECT a.attname
-        FROM   pg_index i
-        JOIN   pg_attribute a ON a.attrelid = i.indrelid
-                            AND a.attnum = ANY(i.indkey)
+    SELECT a.attname, i.indisprimary 
+        FROM   pg_index i 
+        JOIN   pg_attribute a ON a.attrelid = i.indrelid AND a.attnum = ANY(i.indkey)
         WHERE  i.indrelid = '$table'::regclass
-        AND    i.indisprimary
     SQL;
 
     $result = pg_query($connection, $query);
+    if (!$result) {
+        return null;
+    }
 
-    $row = pg_fetch_row($result);
-    return $row[0];
+    $firstRow = null;
+    while ($row = pg_fetch_row($result)) {
+        if ($firstRow === null) {
+            $firstRow = $row; // Save the first row in case no match is found
+        }
+        
+        if ($row[1] == true) {
+            return $row[0]; // Return the first row where $row[1] == true
+        }
+    }
+
+    // If no row where $row[1] == true was found, return the first row encountered
+    return $firstRow ? $firstRow[0] : null;
 }
 
 /**