Create node tables

This commit is contained in:
Brandon Dyck 2024-07-18 02:24:28 -06:00
parent 005f754195
commit cabf230f3c
3 changed files with 107 additions and 0 deletions

59
assets/.lua/schema.lua Normal file
View File

@ -0,0 +1,59 @@
local csexp = require"csexp"
local sqlite = require"lsqlite3"
local function sqlite_ok(errcode)
return errcode == sqlite.OK or errcode == sqlite.DONE
end
local function quote_name(name)
return [["]]..string.gsub(name, [["]], [[""]])..[["]]
end
local function setup(db)
db:create_function("parse_node_name", 1, function(ctx, table_name)
local obj = csexp.parse(table_name)
if type(obj) ~= "table" or obj[1] ~= "node" then
ctx:result_null()
return
end
local node_name = obj[2]
if not node_name then
ctx:result_null()
return
end
ctx:result_text(node_name)
end)
end
local function new_node_type(db, name)
local table_name = csexp.serialize{"node", name}
local quoted_name = quote_name(table_name)
local query = string.format([[
CREATE TABLE %s (
id INTEGER PRIMARY KEY
);
]], quoted_name)
-- for _ in db:rows(query) do end
db:exec(query)
if not sqlite_ok(db:errcode()) then
error(db:errmsg())
end
end
local function get_node_types(db)
local query = [[
SELECT parse_node_name(name)
FROM sqlite_schema
WHERE parse_node_name(name) IS NOT NULL
ORDER BY parse_node_name(name) ASC;
]]
return db:urows(query)
end
return {
setup = setup,
quote_name = quote_name,
new_node_type = new_node_type,
get_node_types = get_node_types,
}

View File

@ -3,6 +3,7 @@ lester = require "lester"
lester.show_traceback = false
require "test-csexp"
require "test-schema"
lester.report()
lester.exit()

47
tests/test-schema.lua Normal file
View File

@ -0,0 +1,47 @@
local schema = require "schema"
local sqlite = require "lsqlite3"
local describe, it, expect = lester.describe, lester.it, lester.expect
describe("schema", function()
local db
lester.before(function()
db = sqlite.open_memory()
db:exec("PRAGMA foreign_keys")
schema.setup(db)
end)
lester.after(function()
db:close()
db = nil
end)
describe("new_node_type", function()
it("creates new node types", function()
local node_type_names = {"a", "b", "c", "d"}
for _, name in ipairs(node_type_names) do
schema.new_node_type(db, name)
end
local results = {}
for name in schema.get_node_types(db) do
table.insert(results, name)
end
expect.equal(node_type_names, results)
end)
it("fails when creating a duplicate node type", function()
local function create()
schema.new_node_type(db, "a")
end
create()
expect.fail(create)
end)
end)
describe("quote name", function()
it("quotes names and escapes double quotes", function()
local original = [[My "name" is "Frankie".]]
local expected = [["My ""name"" is ""Frankie""."]]
local actual = schema.quote_name(original)
expect.equal(expected, actual)
end)
end)
end)