fix extension

apache · Jun 19, 2024 · 1665b36 · 1665b36
1 parent 0e800ec
commit 1665b36
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/c/driver/sqlite/CMakeLists.txt b/c/driver/sqlite/CMakeLists.txt
@@ -23,10 +23,12 @@ else()
   find_package(unofficial-sqlite3 CONFIG REQUIRED)
   set(SQLite3_LINK_LIBRARIES unofficial::sqlite3::sqlite3)
   set(SQLite3_INCLUDE_DIRS)
+  set(ADBC_SQLITE_WITH_LOAD_EXTENSION ON)
 endif()
 
 # Check for sqlite3_load_extension() in sqlite3.h
-if(EXISTS "${SQLite3_INCLUDE_DIRS}/sqlite3.h")
+if(NOT ADBC_SQLITE_WITH_LOAD_EXTENSION AND EXISTS "${SQLite3_INCLUDE_DIRS}/sqlite3.h")
+  file(READ "${SQLite3_INCLUDE_DIRS}/sqlite3.h" ADBC_SQLITE_H_CONTENT)
   string(FIND "${ADBC_SQLITE_H_CONTENT}" "sqlite3_load_extension"
               ADBC_SQLITE3_LOAD_EXTENSION_POS)
 
@@ -35,6 +37,11 @@ if(EXISTS "${SQLite3_INCLUDE_DIRS}/sqlite3.h")
   else()
     set(ADBC_SQLITE_WITH_LOAD_EXTENSION ON)
   endif()
+  message(STATUS "Inferred SQLite3 extension support from ${SQLite3_INCLUDE_DIRS}/sqlite3.h (${ADBC_SQLITE_WITH_LOAD_EXTENSION})"
+  )
+else()
+  message(STATUS "Can't detect SQLite3 extension support (defaulting to ON)")
+  set(ADBC_SQLITE_WITH_LOAD_EXTENSION ON)
 endif()
 
 if(NOT ADBC_SQLITE_WITH_LOAD_EXTENSION)