From 99fd3379d751636b4a54ac7ba3356c121eb4f0fd Mon Sep 17 00:00:00 2001
From: Sven Jacobs <github@svenjacobs.com>
Date: Sun, 18 Sep 2022 15:24:33 +0200
Subject: [PATCH] Add callback to hook into DatabaseConfiguration creation of
 NativeSqliteDriver

---
 .../app/cash/sqldelight/driver/native/NativeSqlDatabase.kt  | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/drivers/native-driver/src/nativeMain/kotlin/app/cash/sqldelight/driver/native/NativeSqlDatabase.kt b/drivers/native-driver/src/nativeMain/kotlin/app/cash/sqldelight/driver/native/NativeSqlDatabase.kt
index 797b6931643..f171a369598 100644
--- a/drivers/native-driver/src/nativeMain/kotlin/app/cash/sqldelight/driver/native/NativeSqlDatabase.kt
+++ b/drivers/native-driver/src/nativeMain/kotlin/app/cash/sqldelight/driver/native/NativeSqlDatabase.kt
@@ -112,10 +112,14 @@ class NativeSqliteDriver(
     maxReaderConnections = maxReaderConnections,
   )
 
+  /**
+   * @param onConfiguration Callback to hook into [DatabaseConfiguration] creation.
+   */
   constructor(
     schema: SqlSchema,
     name: String,
     maxReaderConnections: Int = 1,
+    onConfiguration: (DatabaseConfiguration) -> DatabaseConfiguration = { it },
   ) : this(
     configuration = DatabaseConfiguration(
       name = name,
@@ -126,7 +130,7 @@ class NativeSqliteDriver(
       upgrade = { connection, oldVersion, newVersion ->
         wrapConnection(connection) { schema.migrate(it, oldVersion, newVersion) }
       },
-    ),
+    ).let(onConfiguration),
     maxReaderConnections = maxReaderConnections,
   )