Add Swift Package support

Author: Gioee

.gitignore

@@ -7,4 +7,5 @@
 /dist
 *.sqlite
 *.a
-.vscode
+.vscode
+.build

Package.swift

@@ -0,0 +1,33 @@
+// swift-tools-version: 6.1
+// The swift-tools-version declares the minimum version of Swift required to build this package.
+
+import PackageDescription
+
+let package = Package(
+    name: "vector",
+    platforms: [.macOS(.v11), .iOS(.v11)],
+    products: [
+        // Products can be used to vend plugins, making them visible to other packages.
+        .plugin(
+            name: "vectorPlugin",
+            targets: ["vectorPlugin"]),
+        .library(
+            name: "vector",
+            targets: ["vector"])
+    ],
+    targets: [
+        // Build tool plugin that invokes the Makefile
+        .plugin(
+            name: "vectorPlugin",
+            capability: .buildTool(),
+            path: "packages/swift/plugin"
+        ),
+        // vector library target
+        .target(
+            name: "vector",
+            dependencies: [],
+            path: "packages/swift/extension",
+            plugins: ["vectorPlugin"]
+        ),
+    ]
+)

README.md

@@ -44,6 +44,29 @@ Download the appropriate pre-built binary for your platform from the official [R
 
 You can download the WebAssembly (WASM) version of SQLite with the SQLite Vector extension enabled from: https://www.npmjs.com/package/@sqliteai/sqlite-wasm
 
+### Swift Package
+
+You can [add this repository as a package dependency to your Swift project](https://developer.apple.com/documentation/xcode/adding-package-dependencies-to-your-app#Add-a-package-dependency). After adding the package, you'll need to set up SQLite with extension loading by following steps 4 and 5 of [this guide](https://github.com/sqliteai/sqlite-extensions-guide/blob/main/platforms/ios.md#4-set-up-sqlite-with-extension-loading).
+
+Here's an example of how to use the package:
+```swift
+import vector
+
+...
+
+var db: OpaquePointer?
+sqlite3_open(":memory:", &db)
+sqlite3_enable_load_extension(db, 1)
+var errMsg: UnsafeMutablePointer<Int8>? = nil
+sqlite3_load_extension(db, vector.path, nil, &errMsg)
+var stmt: OpaquePointer?
+sqlite3_prepare_v2(db, "SELECT vector_version()", -1, &stmt, nil)
+defer { sqlite3_finalize(stmt) }
+sqlite3_step(stmt)
+log("vector_version(): \(String(cString: sqlite3_column_text(stmt, 0)))")
+sqlite3_close(db)
+```
+
 ### Loading the Extension
 
 ```sql

packages/swift/extension/vector.swift

@@ -0,0 +1,19 @@
+// vector.swift
+// This file serves as a placeholder for the vector target.
+// The actual SQLite extension is built using the Makefile through the build plugin.
+
+import Foundation
+
+/// Placeholder structure for vector
+public struct vector {
+    /// Returns the path to the built vector dylib inside the XCFramework
+    public static var path: String {
+        #if os(macOS)
+        return "vector.xcframework/macos-arm64_x86_64/vector.framework/vector"
+        #elseif targetEnvironment(simulator)
+        return "vector.xcframework/ios-arm64_x86_64-simulator/vector.framework/vector"
+        #else
+        return "vector.xcframework/ios-arm64/vector.framework/vector"
+        #endif
+    }
+}

packages/swift/plugin/vector.swift

@@ -0,0 +1,60 @@
+import PackagePlugin
+import Foundation
+
+@main
+struct vector: BuildToolPlugin {
+    /// Entry point for creating build commands for targets in Swift packages.
+    func createBuildCommands(context: PluginContext, target: Target) async throws -> [Command] {
+        let packageDirectory = context.package.directoryURL
+        let outputDirectory = context.pluginWorkDirectoryURL
+        return createvectorBuildCommands(packageDirectory: packageDirectory, outputDirectory: outputDirectory)
+    }
+}
+
+#if canImport(XcodeProjectPlugin)
+import XcodeProjectPlugin
+
+extension vector: XcodeBuildToolPlugin {
+    // Entry point for creating build commands for targets in Xcode projects.
+    func createBuildCommands(context: XcodePluginContext, target: XcodeTarget) throws -> [Command] {
+        let outputDirectory = context.pluginWorkDirectoryURL
+        return createvectorBuildCommands(packageDirectory: nil, outputDirectory: outputDirectory)
+    }
+}
+
+#endif
+
+/// Shared function to create vector build commands
+func createvectorBuildCommands(packageDirectory: URL?, outputDirectory: URL) -> [Command] {
+
+    // For Xcode projects, use current directory; for Swift packages, use provided packageDirectory
+    let workingDirectory = packageDirectory?.path ?? "$(pwd)"
+    let packageDirInfo = packageDirectory != nil ? "Package directory: \(packageDirectory!.path)" : "Working directory: $(pwd)"
+
+    return [
+        .prebuildCommand(
+            displayName: "Building vector XCFramework",
+            executable: URL(fileURLWithPath: "/bin/bash"),
+            arguments: [
+                "-c",
+                """
+                set -e
+                echo "Starting vector XCFramework prebuild..."
+                echo "\(packageDirInfo)"
+                
+                # Clean and create output directory
+                rm -rf "\(outputDirectory.path)"
+                mkdir -p "\(outputDirectory.path)"
+                
+                # Build directly from source directory with custom output paths
+                cd "\(workingDirectory)" && \
+                echo "Building XCFramework with native network..." && \
+                make xcframework NATIVE_NETWORK=ON DIST_DIR="\(outputDirectory.path)" BUILD_RELEASE="\(outputDirectory.path)/build/release" BUILD_TEST="\(outputDirectory.path)/build/test" && \
+                rm -rf "\(outputDirectory.path)/build" && \
+                echo "XCFramework build completed successfully!"
+                """
+            ],
+            outputFilesDirectory: outputDirectory
+        )
+    ]
+}