allow building with the bundled sqlite source code

1 files changed, 29 insertions, 9 deletions

diff --git a/build.zig b/build.zig
index a3f0de5..be2a729 100644
--- a/build.zig
+++ b/build.zig
@@ -1,28 +1,48 @@
 const std = @import("std");
-const Builder = std.build.Builder;
 
-fn linkAll(obj: *std.build.LibExeObjStep) void {
-    obj.linkLibC();
-    obj.linkSystemLibrary("sqlite3");
+var sqlite3: ?*std.build.LibExeObjStep = null;
+
+fn linkSqlite(b: *std.build.LibExeObjStep) void {
+    b.linkLibC();
+
+    if (sqlite3) |lib| {
+        b.linkLibrary(lib);
+    } else {
+        b.linkSystemLibrary("sqlite3");
+    }
 }
 
-pub fn build(b: *Builder) void {
+pub fn build(b: *std.build.Builder) void {
     const target = b.standardTargetOptions(.{});
     const mode = b.standardReleaseOptions();
 
+    const in_memory = b.option(bool, "in_memory", "Should the tests run with sqlite in memory") orelse false;
+    const use_bundled = b.option(bool, "use_bundled", "Use the bundled sqlite3 source instead of linking the system library") orelse false;
+
+    // Build sqlite from source if asked
+    if (use_bundled) {
+        const lib = b.addStaticLibrary("sqlite", null);
+        lib.addCSourceFile("sqlite3.c", &[_][]const u8{"-std=c99"});
+        lib.linkLibC();
+        lib.setTarget(target);
+        lib.setBuildMode(mode);
+        sqlite3 = lib;
+    }
+
     const lib = b.addStaticLibrary("zig-sqlite", "sqlite.zig");
+    lib.addIncludeDir(".");
+    linkSqlite(lib);
     lib.setTarget(target);
     lib.setBuildMode(mode);
-    linkAll(lib);
     lib.install();
 
-    const in_memory = b.option(bool, "in_memory", "Should the tests run with sqlite in memory") orelse false;
-
     var main_tests = b.addTest("sqlite.zig");
+    main_tests.addIncludeDir(".");
+    linkSqlite(main_tests);
+    main_tests.setBuildMode(mode);
     main_tests.setTarget(target);
     main_tests.setBuildMode(mode);
     main_tests.addBuildOption(bool, "in_memory", in_memory);
-    linkAll(main_tests);
 
     const test_step = b.step("test", "Run library tests");
     test_step.dependOn(&main_tests.step);