99 lines
2.8 KiB
Zig
99 lines
2.8 KiB
Zig
const std = @import("std");
|
|
const c = @cImport({
|
|
@cInclude("sqlite3.h");
|
|
});
|
|
|
|
pub const Sqlite = struct {
|
|
db: *c.sqlite3,
|
|
|
|
pub fn open(path: [:0]const u8) !Sqlite {
|
|
var db: ?*c.sqlite3 = undefined;
|
|
const err = c.sqlite3_open_v2(path, &db, c.SQLITE_OPEN_READWRITE | c.SQLITE_OPEN_CREATE, null);
|
|
if (err != c.SQLITE_OK) return error.UnknownError;
|
|
|
|
return Sqlite{
|
|
.db = db.?,
|
|
};
|
|
}
|
|
|
|
pub fn close(self: *Sqlite) void {
|
|
return c.sqlite3_close_v2(self.db);
|
|
}
|
|
|
|
pub fn prepare(self: *Sqlite, sql: []const u8) !PreparedStmt {
|
|
var stmt: [*c]c.sqlite3_stmt = undefined;
|
|
const err = c.sqlite3_prepare_v2(self.db, sql.ptr, sql.len, &stmt, null);
|
|
if (err != c.SQLITE_OK) return error.UnknownError;
|
|
|
|
return PreparedStmt{ .stmt = stmt };
|
|
}
|
|
};
|
|
|
|
pub const Row = struct {
|
|
stmt: *c.sqlite3_stmt,
|
|
|
|
pub fn getI64(self: *Row, idx: u15) !i64 {
|
|
return @intCast(i64, c.sqlite3_column_int64(self.stmt, idx));
|
|
}
|
|
|
|
pub fn getText(self: *Row, idx: u15, buf: []u8) ![]u8 {
|
|
const ptr = c.sqlite3_column_text(self.stmt, idx);
|
|
|
|
const size = c.sqlite3_column_bytes(self.stmt, idx);
|
|
if (size > buf.len) return error.StreamTooLong;
|
|
for (ptr[0..size]) |ch, i| buf[i] = ch;
|
|
|
|
return buf[0..size];
|
|
}
|
|
|
|
pub fn getTextAlloc(self: *Row, idx: u15, alloc: std.mem.Allocator) ![]u8 {
|
|
const size = c.sqlite3_column_bytes(self.stmt, idx);
|
|
var buf = try alloc.alloc(u8, size);
|
|
errdefer alloc.free(buf);
|
|
|
|
return self.getText(idx, buf);
|
|
}
|
|
};
|
|
|
|
pub const PreparedStmt = struct {
|
|
stmt: *c.sqlite3_stmt,
|
|
|
|
pub fn bindNull(self: *PreparedStmt, idx: u15) !void {
|
|
return switch (c.sqlite3_bind_null(self.stmt, idx)) {
|
|
.SQLITE_OK => void,
|
|
else => error.UnknownError,
|
|
};
|
|
}
|
|
|
|
pub fn bindText(self: *PreparedStmt, idx: u15, str: []const u8) !void {
|
|
return switch (c.sqlite3_bind_text(self.stmt, idx, str.ptr, str.len, c.SQLITE_TRANSIENT)) {
|
|
.SQLITE_OK => void,
|
|
else => error.UnknownError,
|
|
};
|
|
}
|
|
|
|
pub fn bindI64(self: *PreparedStmt, idx: u15, val: i64) !void {
|
|
return switch (c.sqlite3_bind_int64(self.stmt, idx, val)) {
|
|
.SQLITE_OK => void,
|
|
else => error.UnknownError,
|
|
};
|
|
}
|
|
|
|
pub fn step(self: *PreparedStmt) !?Row {
|
|
return switch (c.sqlite3_step(self.stmt)) {
|
|
.SQLITE_ROW => Row{ .stmt = self.stmt },
|
|
.SQLITE_DONE => null,
|
|
|
|
else => error.UnknownError,
|
|
};
|
|
}
|
|
|
|
pub fn finalize(self: *PreparedStmt) void {
|
|
_ = c.sqlite3_finalize(self.stmt);
|
|
}
|
|
|
|
pub fn reset(self: *PreparedStmt) void {
|
|
_ = c.sqlite3_reset(self.stmt);
|
|
}
|
|
};
|