Added FK option

Author: wickwirew

Sources/PureSQL/ConnectionPool.swift

@@ -22,6 +22,8 @@ public actor ConnectionPool: Sendable {
     private var count: Int = 1
     /// The maximum number of connections we can create
     private let limit: Int
+    /// Whether to enforce foreign key constraints on each connection.
+    private let foreignKeys: Bool
     /// Any connections available for use
     private var availableConnections: [RawConnection]
     /// Any caller waiting for a connection
@@ -37,21 +39,27 @@ public actor ConnectionPool: Sendable {
         path: String,
         limit: Int,
         migrations: [String],
-        runMigrations: Bool = true
+        runMigrations: Bool = true,
+        foreignKeys: Bool = true
     ) throws {
         guard limit > 0 else {
             throw SQLError.poolCannotHaveZeroConnections
         }
 
         self.path = path
         self.limit = limit
+        self.foreignKeys = foreignKeys
 
         let connection = try SQLiteConnection(path: path)
         self.observer.installHooks(into: connection)
 
         // Turn on WAL mode
         try connection.execute(sql: "PRAGMA journal_mode=WAL;")
 
+        if foreignKeys {
+            try connection.execute(sql: "PRAGMA foreign_keys=ON;")
+        }
+        
         if runMigrations {
             try MigrationRunner.execute(migrations: migrations, connection: connection)
         }
@@ -115,6 +123,12 @@ public actor ConnectionPool: Sendable {
         count += 1
         let connection = try SQLiteConnection(path: path)
         observer.installHooks(into: connection)
+
+        // Foreign key enforcement is per connection
+        if foreignKeys {
+            try connection.execute(sql: "PRAGMA foreign_keys=ON;")
+        }
+
         return connection
     }
 

Sources/PureSQL/Database.swift

@@ -49,14 +49,16 @@ public extension Database {
                 path: path,
                 limit: config.maxConnectionCount,
                 migrations: Self.sanitizedMigrations,
-                runMigrations: config.autoMigrate
+                runMigrations: config.autoMigrate,
+                foreignKeys: config.foreignKeys
             )
         } else {
             try ConnectionPool(
                 path: ":memory:",
                 limit: 1,
                 migrations: Self.sanitizedMigrations,
-                runMigrations: config.autoMigrate
+                runMigrations: config.autoMigrate,
+                foreignKeys: config.foreignKeys
             )
         }
 

Sources/PureSQL/DatabaseConfig.swift

@@ -18,14 +18,18 @@ public struct DatabaseConfig {
     /// If `true` the migrations will run when the connection is opened.
     /// Default is `true`.
     public var autoMigrate: Bool
+    /// If `true`, `PRAGMA foreign_keys` is enabled on every connection.
+    public var foreignKeys: Bool
 
     public init(
         path: String?,
         maxConnectionCount: Int = 5,
-        autoMigrate: Bool = true
+        autoMigrate: Bool = true,
+        foreignKeys: Bool = true
     ) {
         self.path = path
         self.maxConnectionCount = maxConnectionCount
         self.autoMigrate = autoMigrate
+        self.foreignKeys = foreignKeys
     }
 }