op-sqlite.podspec

require "json"

log_message = lambda do |message|
  puts "\e[34m#{message}\e[0m"
end

# In the sample app the dir is not inside of node_modules
is_user_app = __dir__.include?("node_modules")
package = JSON.parse(File.read(File.join(__dir__, "package.json")))
app_package = nil
package_json_path = nil

# When installed on user node_modules lives inside node_modules/@op-engineering/op-sqlite
# Find the users package.json by searching up through parent directories
if is_user_app
  # Prefer the app, the same package.json Android reads. It is the folder above
  # the Podfile; the search further down can't find it under pnpm or when
  # hoisted, where it ends at the workspace root. `pod install` finds the Podfile
  # before evaluating any podspec; when none was found, podfile_path is nil.
  podfile_path = defined?(Pod::Config) ? Pod::Config.instance.podfile_path : nil
  if podfile_path
    package_path = File.join(File.dirname(File.dirname(podfile_path.to_s)), "package.json")
    candidate = begin
      JSON.parse(File.read(package_path))
    rescue Errno::ENOENT, JSON::ParserError
      nil
    end
    package_json_path = package_path if candidate.is_a?(Hash) && candidate.key?("op-sqlite")
  end

  if package_json_path.nil?
    current_dir = File.expand_path(__dir__)
    # Move one level up to the parent directory
    current_dir = File.dirname(current_dir)

    # Find the package.json by searching up through parent directories
    loop do
      package_path = File.join(current_dir, "package.json")
      if File.exist?(package_path)
        package_json_path = package_path
        break
      end

      parent_dir = File.dirname(current_dir)
      break if parent_dir == current_dir  # reached filesystem root
      current_dir = parent_dir
    end

    raise "package.json not found" if package_json_path.nil?
  end
else
  package_json_path = File.join(__dir__, "example", "package.json")
end

app_package = JSON.parse(File.read(package_json_path))

op_sqlite_config = app_package["op-sqlite"]
use_sqlcipher = false
use_libsql = false
use_turso = false
performance_mode = false
phone_version = false
sqlite_flags = ""
fts5 = false
rtree = false
use_sqlite_vec = false
tokenizers = []

if(op_sqlite_config != nil)
  use_sqlcipher = op_sqlite_config["sqlcipher"] == true
  use_libsql = op_sqlite_config["libsql"] == true
  use_turso = op_sqlite_config["turso"] == true
  performance_mode = op_sqlite_config["performanceMode"] || false
  phone_version = op_sqlite_config["iosSqlite"] == true
  sqlite_flags = op_sqlite_config["sqliteFlags"] || ""
  fts5 = op_sqlite_config["fts5"] == true
  rtree = op_sqlite_config["rtree"] == true
  use_sqlite_vec = op_sqlite_config["sqliteVec"] == true
  tokenizers = op_sqlite_config["tokenizers"] || []
end

if phone_version then
  if use_sqlcipher then
    raise "SQLCipher is not supported with phone version. It cannot load extensions."
  end

  if rtree then
    raise "RTree is not supported with phone version. It cannot load extensions."
  end

  if use_sqlite_vec then
    raise "sqlite-vec is not supported with phone version. It cannot load extensions."
  end
end

if use_libsql and use_sqlite_vec then
    raise "You cannot use sqlite-vec with libsql. libsql already has vector search included."
end

if use_turso and use_sqlite_vec then
  raise "You cannot use sqlite-vec with turso backend."
end

if use_turso and use_libsql then
  raise "You cannot enable both libsql and turso backend."
end

Pod::Spec.new do |s|
  s.name         = "op-sqlite"
  s.version      = package["version"]
  s.summary      = package["description"]
  s.homepage     = package["homepage"]
  s.license      = package["license"]
  s.authors      = package["author"]

  s.platforms    = { :ios => min_ios_version_supported, :tvos => "13.0", :osx => "10.15", :visionos => "1.0", :watchos => "9.0" }
  s.source       = { :git => "https://github.com/op-engineering/op-sqlite.git", :tag => "#{s.version}" }

  log_message.call("[OP-SQLITE] Configuration found at #{package_json_path}")

  install_modules_dependencies(s)

  # Base source files
  source_files = Dir.glob("ios/**/*.{h,hpp,m,mm}") + Dir.glob("cpp/**/*.{hpp,h,cpp,c}")

  # Backend bridges are selected explicitly by flags and should not be compiled by default.
  source_files.reject! { |path| path == "cpp/turso/OPTursoBridge.cpp" } unless use_turso

  # Strictly blocks all headers from being public
  s.public_header_files = ["ios/OPSQLite.h", "cpp/sqlite3.h", "cpp/OPBridge.hpp"]

  # `c_sources` lives next to the package.json holding the config. A path relative
  # to this podspec would land inside the pnpm store.
  c_sources_dir = File.join(File.dirname(package_json_path), "c_sources")

  if tokenizers.any?
    if use_turso then
      raise "Tokenizers are not supported with turso backend. Please disable tokenizers or do not enable turso."
    end

    # Generated by react-native.config.js. `use_native_modules!` loads it before any
    # podspec is evaluated, but a Podfile without it won't have.
    tokenizers_header = File.join(c_sources_dir, "tokenizers.h")
    unless File.exist?(tokenizers_header)
      raise "[OP-SQLITE] Tokenizers #{tokenizers} are enabled, but #{tokenizers_header} is missing. Run `npx react-native config` from your app's folder to create it."
    end

    FileUtils.cp_r(c_sources_dir, __dir__)
    # # Add all .h and .c files from the `c_sources` directory
    source_files += Dir.glob(File.join("c_sources", "**/*.{h,cpp}"))
  end

  # Assign the collected source files to `s.source_files`
  s.source_files = source_files

  xcconfig = {
    :GCC_PREPROCESSOR_DEFINITIONS => "",
    :CLANG_CXX_LANGUAGE_STANDARD => "c++23",
    :GCC_OPTIMIZATION_LEVEL => "2",
  }

  exclude_files = []

  if use_sqlcipher then
    log_message.call("[OP-SQLITE] using SQLCipher")
    exclude_files += ["cpp/sqlite3.c", "cpp/sqlite3.h", "cpp/libsql/OPLibsqlBridge.cpp", "cpp/libsql/OPLibsqlBridge.hpp", "ios/libsql.xcframework/**/*", "ios/libsql_experimental.xcframework/**/*", "ios/turso_sdk_kit.xcframework/**/*"]
    xcconfig[:GCC_PREPROCESSOR_DEFINITIONS] += " OP_SQLITE_USE_SQLCIPHER=1 HAVE_FULLFSYNC=1 SQLITE_HAS_CODEC SQLITE_TEMP_STORE=3 SQLITE_EXTRA_INIT=sqlcipher_extra_init SQLITE_EXTRA_SHUTDOWN=sqlcipher_extra_shutdown"
    s.dependency "OpenSSL-Universal"
  elsif use_turso then
    log_message.call("[OP-SQLITE] using Turso SDK kit")
    exclude_files += ["cpp/sqlite3.c", "cpp/sqlite3.h", "cpp/OPBridge.hpp", "cpp/OPBridge.cpp", "cpp/sqlcipher/sqlite3.c", "cpp/sqlcipher/sqlite3.h", "cpp/libsql/OPLibsqlBridge.hpp", "cpp/libsql/OPLibsqlBridge.cpp", "ios/libsql_experimental.xcframework/**/*"]
  elsif use_libsql then
    log_message.call("[OP-SQLITE] ⚠️ Using libsql. If you have libsql questions please ask in the Turso Discord server.")
    exclude_files += ["cpp/sqlite3.c", "cpp/sqlite3.h", "cpp/sqlcipher/sqlite3.c", "cpp/sqlcipher/sqlite3.h", "cpp/OPBridge.hpp", "cpp/OPBridge.cpp", "ios/turso_sdk_kit.xcframework/**/*"]
  else
    log_message.call("[OP-SQLITE] using pure SQLite")
    exclude_files += ["cpp/sqlcipher/sqlite3.c", "cpp/sqlcipher/sqlite3.h", "cpp/libsql/OPLibsqlBridge.hpp", "cpp/libsql/OPLibsqlBridge.cpp", "cpp/libsql/libsql.h", "ios/libsql_experimental.xcframework/**/*", "ios/turso_sdk_kit.xcframework/**/*"]
  end

   # Exclude xcframeworks that aren't being used
  if !use_sqlite_vec then
    exclude_files += ["ios/sqlitevec.xcframework/**/*"]
  end

  other_cflags = '$(inherited) -DSQLITE_DBCONFIG_ENABLE_LOAD_EXTENSION=1 -DHAVE_USLEEP=1 -DSQLITE_ENABLE_LOCKING_STYLE=0'
  optimizedCflags = ' -DSQLITE_DQS=0 -DSQLITE_DEFAULT_MEMSTATUS=0 -DSQLITE_DEFAULT_WAL_SYNCHRONOUS=1 -DSQLITE_LIKE_DOESNT_MATCH_BLOBS=1 -DSQLITE_MAX_EXPR_DEPTH=0 -DSQLITE_OMIT_DEPRECATED=1 -DSQLITE_OMIT_PROGRESS_CALLBACK=1 -DSQLITE_OMIT_SHARED_CACHE=1 -DSQLITE_USE_ALLOCA=1 -DSQLITE_STRICT_SUBTYPE=1 -DSQLITE_THREADSAFE=2'
  frameworks = []

  if fts5 then
    xcconfig[:GCC_PREPROCESSOR_DEFINITIONS] += " SQLITE_ENABLE_FTS5=1"
  end

  if rtree then
    xcconfig[:GCC_PREPROCESSOR_DEFINITIONS] += " SQLITE_ENABLE_RTREE=1"
  end

  if phone_version then
    log_message.call("[OP-SQLITE] using iOS embedded SQLite 📱")
    xcconfig[:GCC_PREPROCESSOR_DEFINITIONS] += " OP_SQLITE_USE_PHONE_VERSION=1"
    exclude_files += ["cpp/sqlite3.c", "cpp/sqlite3.h"]
    s.library = "sqlite3"
  end

  if performance_mode then
    log_message.call("[OP-SQLITE] Performance mode enabled")
    other_cflags += optimizedCflags
  end

  if use_sqlite_vec then
    log_message.call("[OP-SQLITE] using Sqlite Vec ↗️")
    xcconfig[:GCC_PREPROCESSOR_DEFINITIONS] += " OP_SQLITE_USE_SQLITE_VEC=1"
    frameworks.push("ios/sqlitevec.xcframework")
  end

  if use_libsql then
    xcconfig[:GCC_PREPROCESSOR_DEFINITIONS] += " OP_SQLITE_USE_LIBSQL=1"
    frameworks = ["ios/libsql_experimental.xcframework"]
  end

  if use_turso then
    xcconfig[:GCC_PREPROCESSOR_DEFINITIONS] += " OP_SQLITE_USE_TURSO=1"
    frameworks = ["ios/turso_sdk_kit.xcframework"]
  end

  if sqlite_flags != "" then
    log_message.call("[OP-SQLITE] Detected custom SQLite flags: #{sqlite_flags}")
    other_cflags += " #{sqlite_flags}"
  end

  if tokenizers.any? then
    log_message.call("[OP_SQLITE] Tokenizers enabled: #{tokenizers}")
    if is_user_app then
      other_cflags += " -DTOKENIZERS_HEADER_PATH=\\\"../c_sources/tokenizers.h\\\""
    else
      other_cflags += " -DTOKENIZERS_HEADER_PATH=\\\"../example/c_sources/tokenizers.h\\\""
    end
  end

  xcconfig[:OTHER_CFLAGS] = other_cflags
  s.pod_target_xcconfig = xcconfig
  s.vendored_frameworks = frameworks
  s.exclude_files = exclude_files
end