From 1306aec37ab1bc8d0eaeb8beebd991ac7a48def5 Mon Sep 17 00:00:00 2001
From: "Dr. Patrick Urbanke" <patrick@getml.com>
Date: Tue, 15 Apr 2025 08:18:28 +0200
Subject: [PATCH] Added psql as a dependency

---
 CMakeLists.txt | 31 ++++++++++++++++++++++---------
 README.md      |  6 ++++++
 vcpkg.json     | 39 ++++++++++++++++++++++++++++++---------
 3 files changed, 58 insertions(+), 18 deletions(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 1eed59d..90c33b1 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -2,9 +2,10 @@ cmake_minimum_required(VERSION 3.23)
 
 option(SQLGEN_BUILD_SHARED "Build shared library" ${BUILD_SHARED_LIBS})
 
-option(SQLGEN_SQLITE3 "Enable sqlite3 support" ON) # enabled by default
+option(SQLGEN_POSTGRES "Enable PostgreSQL support" ON) # enabled by default
+
+option(SQLGEN_SQLITE3 "Enable SQLite3 support" ON) # enabled by default
 
-#option(SQLGEN_BUILD_BENCHMARKS "Build benchmarks" OFF)
 option(SQLGEN_BUILD_TESTS "Build tests" OFF)
 
 set(CMAKE_CXX_STANDARD_REQUIRED ON)
@@ -17,6 +18,18 @@ set(SQLGEN_USE_VCPKG_DEFAULT ON)
 option(SQLGEN_USE_VCPKG "Use VCPKG to download and build dependencies" ${SQLGEN_USE_VCPKG_DEFAULT})
 
 if (SQLGEN_USE_VCPKG)
+    if (SQLGEN_BUILD_TESTS)
+        list(APPEND VCPKG_MANIFEST_FEATURES "tests")
+    endif()
+    
+    if (SQLGEN_POSTGRES)
+        list(APPEND VCPKG_MANIFEST_FEATURES "postgres")
+    endif()
+    
+    if (SQLGEN_SQLITE3)
+        list(APPEND VCPKG_MANIFEST_FEATURES "sqlite3")
+    endif()
+    
     set(CMAKE_TOOLCHAIN_FILE ${CMAKE_CURRENT_SOURCE_DIR}/vcpkg/scripts/buildsystems/vcpkg.cmake CACHE STRING "Vcpkg toolchain file")
 endif ()
 
@@ -44,6 +57,13 @@ target_include_directories(
     $<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/include>
     $<INSTALL_INTERFACE:include>)
 
+if (SQLGEN_POSTGRES)
+    if (NOT TARGET PostgreSQL)
+        find_package(PostgreSQL REQUIRED)
+    endif()
+    target_link_libraries(sqlgen PUBLIC PostgreSQL::PostgreSQL)
+endif()
+
 if (SQLGEN_SQLITE3)
     list(APPEND SQLGEN_SOURCES src/sqlgen_sqlite.cpp)
     if (NOT TARGET unofficial-sqlite3)
@@ -66,13 +86,6 @@ if (SQLGEN_BUILD_TESTS)
     add_subdirectory(tests)
 endif ()
 
-if (SQLGEN_BUILD_BENCHMARKS)
-    if (NOT TARGET benchmark::benchmark)
-        find_package(benchmark CONFIG REQUIRED)
-    endif ()
-    add_subdirectory(benchmarks)
-endif ()
-
 if (PROJECT_IS_TOP_LEVEL)
     include(GNUInstallDirs)
     include(CMakePackageConfigHelpers)
diff --git a/README.md b/README.md
index d593da5..3735050 100644
--- a/README.md
+++ b/README.md
@@ -57,6 +57,12 @@ if (result) {
 
 ## Installation
 
+These three libraries are needed for PostgreSQL support:
+
+```
+sudo apt-get install autoconf bison flex
+```
+
 To install vcpkg:
 
 ```
diff --git a/vcpkg.json b/vcpkg.json
index 29196a9..5405719 100644
--- a/vcpkg.json
+++ b/vcpkg.json
@@ -1,16 +1,37 @@
 {
   "dependencies": [
-    {
-      "name": "gtest",
-      "version>=": "1.14.0"
-    },
     {
       "name": "reflectcpp",
       "version>=": "0.18.0"
-    },
-    {
-      "name": "sqlite3",
-      "version>=": "3.49.1"
     }
-  ]
+  ],
+  "features": {
+    "postgres": {
+      "description": "Enable PostgreSQL support",
+      "dependencies": [
+        {
+          "name": "libpq",
+          "version>=": "16.4"
+        }
+      ]
+    },
+    "sqlite3": {
+      "description": "Enable SQLite3 support",
+      "dependencies": [
+        {
+          "name": "sqlite3",
+          "version>=": "3.49.1"
+        }
+      ]
+    },
+    "tests": {
+      "description": "Build the tests",
+      "dependencies": [
+        {
+          "name": "gtest",
+          "version>=": "1.14.0"
+        }
+      ]
+    }
+  }
 }