From 1327b9d82ff5ebed7bac0f6ca6b6ee7d7bea4d01 Mon Sep 17 00:00:00 2001
From: Matthew Bucci <buccim2@gene.com>
Date: Fri, 18 Oct 2024 23:31:21 -0700
Subject: [PATCH] add end to list of reserved keywords

---
 pg4wp/rewriters/CreateTableSQLRewriter.php |  2 +-
 tests/rewriteTest.php                      | 61 ++++++++++++++++++++++
 2 files changed, 62 insertions(+), 1 deletion(-)

diff --git a/pg4wp/rewriters/CreateTableSQLRewriter.php b/pg4wp/rewriters/CreateTableSQLRewriter.php
index a919124..a19af1a 100644
--- a/pg4wp/rewriters/CreateTableSQLRewriter.php
+++ b/pg4wp/rewriters/CreateTableSQLRewriter.php
@@ -150,7 +150,7 @@ class CreateTableSQLRewriter extends AbstractSQLRewriter
             $columnsAndKeys = $matches[3];
             $suffix = ')' . $matches[4];
     
-            $regex = '/(?:^|\s*,\s*)(\b(?:timestamp|date|time|default)\b)\s*(?=\s+\w+)/i'; 
+            $regex = '/(?:^|\s*,\s*)(\b(?:timestamp|date|time|default|end)\b)\s*(?=\s+\w+)/i'; 
 
             // Callback function to add quotes around protected column names
             $callback = function($matches) {
diff --git a/tests/rewriteTest.php b/tests/rewriteTest.php
index 7555682..ab40ca2 100644
--- a/tests/rewriteTest.php
+++ b/tests/rewriteTest.php
@@ -785,6 +785,67 @@ final class rewriteTest extends TestCase
         $this->assertSame(trim($expected), trim($postgresql));
     }
 
+    public function test_it_quotes_reserved_columns() 
+    {
+        $sql = <<<SQL
+            CREATE TABLE wp_aioseo_notifications (
+                id bigint(20) unsigned NOT NULL AUTO_INCREMENT,
+                slug varchar(13) NOT NULL,
+                title text NOT NULL,
+                content longtext NOT NULL,
+                type varchar(64) NOT NULL,
+                level text NOT NULL,
+                notification_id bigint(20) unsigned DEFAULT NULL,
+                notification_name varchar(255) DEFAULT NULL,
+                start datetime DEFAULT NULL,
+                end datetime DEFAULT NULL,
+                button1_label varchar(255) DEFAULT NULL,
+                button1_action varchar(255) DEFAULT NULL,
+                button2_label varchar(255) DEFAULT NULL,
+                button2_action varchar(255) DEFAULT NULL,
+                dismissed tinyint(1) NOT NULL DEFAULT 0,
+                created datetime NOT NULL,
+                updated datetime NOT NULL,
+                PRIMARY KEY (id),
+                UNIQUE KEY ndx_aioseo_notifications_slug (slug),
+                KEY ndx_aioseo_notifications_dates (start, end),
+                KEY ndx_aioseo_notifications_type (type),
+                KEY ndx_aioseo_notifications_dismissed (dismissed)
+            ) DEFAULT CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_520_ci;
+        SQL;
+
+        $expected = <<<SQL
+            CREATE TABLE IF NOT EXISTS wp_aioseo_notifications (
+                id bigserial,
+                slug varchar(13) NOT NULL,
+                title text NOT NULL,
+                content text NOT NULL,
+                type varchar(64) NOT NULL,
+                level text NOT NULL,
+                notification_id bigint  DEFAULT NULL,
+                notification_name varchar(255) DEFAULT NULL,
+                start timestamp DEFAULT NULL,
+                "end" timestamp DEFAULT NULL,
+                button1_label varchar(255) DEFAULT NULL,
+                button1_action varchar(255) DEFAULT NULL,
+                button2_label varchar(255) DEFAULT NULL,
+                button2_action varchar(255) DEFAULT NULL,
+                dismissed smallint NOT NULL DEFAULT 0,
+                created timestamp NOT NULL,
+                updated timestamp NOT NULL,
+                PRIMARY KEY (id)
+            );
+        CREATE UNIQUE INDEX IF NOT EXISTS wp_aioseo_notifications_ndx_aioseo_notifications_slug ON wp_aioseo_notifications (slug);
+        CREATE INDEX IF NOT EXISTS wp_aioseo_notifications_ndx_aioseo_notifications_dates ON wp_aioseo_notifications (start, end);
+        CREATE INDEX IF NOT EXISTS wp_aioseo_notifications_ndx_aioseo_notifications_type ON wp_aioseo_notifications (type);
+        CREATE INDEX IF NOT EXISTS wp_aioseo_notifications_ndx_aioseo_notifications_dismissed ON wp_aioseo_notifications (dismissed);
+        SQL;
+
+        $postgresql = pg4wp_rewrite($sql);
+        $this->assertSame(trim($expected), trim($postgresql));
+    }
+   
+
     protected function setUp(): void
     {
         global $wpdb;