util.jsonpointer: Resolve JSON Pointers per RFC 6901

3 files changed, 87 insertions, 1 deletions

diff --git a/GNUmakefile b/GNUmakefile
index 17f0ea51..a86a0b7e 100644
--- a/GNUmakefile
+++ b/GNUmakefile
@@ -110,7 +110,7 @@ vpath %.tl teal-src/
 	tl -I teal-src/ --gen-compat off --gen-target 5.1 gen $^ -o $@
 	-lua-format -i $@
 
-teal: util/jsonschema.lua util/datamapper.lua
+teal: util/jsonschema.lua util/datamapper.lua util/jsonpointer.lua
 
 util/%.so:
 	$(MAKE) install -C util-src
diff --git a/teal-src/util/jsonpointer.tl b/teal-src/util/jsonpointer.tl
new file mode 100644
index 00000000..7b9db760
--- /dev/null
+++ b/teal-src/util/jsonpointer.tl
@@ -0,0 +1,46 @@
+
+local enum ptr_error
+	"invalid-table"
+	"invalid-path"
+end
+
+local function unescape_token(escaped_token : string) : string
+	local unescaped = escaped_token:gsub("~1", "/"):gsub("~0", "~")
+	return unescaped
+end
+
+local function resolve_json_pointer(ref : table, path : string) : any, ptr_error
+	local ptr_len = #path+1
+	for part, pos in path:gmatch("/([^/]*)()") do
+		local token = unescape_token(part)
+		if not ref is table then
+			return nil
+		end
+		local idx = next(ref)
+		local new_ref : any
+
+		if idx is string then
+			new_ref = ref[token]
+		elseif idx is integer then
+			local i = tonumber(token)
+			if token == "-" then i = #ref + 1 end
+			new_ref = ref[i]
+		else
+			return nil, "invalid-table"
+		end
+
+		if pos as integer == ptr_len then
+			return new_ref
+		elseif new_ref is table then
+			ref = new_ref
+		elseif not ref is table then
+			return nil, "invalid-path"
+		end
+
+	end
+	return ref
+end
+
+return {
+	resolve = resolve_json_pointer,
+}
diff --git a/util/jsonpointer.lua b/util/jsonpointer.lua
new file mode 100644
index 00000000..7c657266
--- /dev/null
+++ b/util/jsonpointer.lua
@@ -0,0 +1,40 @@
+local function unescape_token(escaped_token)
+	local unescaped = escaped_token:gsub("~1", "/"):gsub("~0", "~")
+	return unescaped
+end
+
+local function resolve_json_pointer(ref, path)
+	local ptr_len = #path + 1
+	for part, pos in path:gmatch("/([^/]*)()") do
+		local token = unescape_token(part)
+		if not (type(ref) == "table") then
+			return nil
+		end
+		local idx = next(ref)
+		local new_ref
+
+		if type(idx) == "string" then
+			new_ref = ref[token]
+		elseif math.type(idx) == "integer" then
+			local i = tonumber(token)
+			if token == "-" then
+				i = #ref + 1
+			end
+			new_ref = ref[i]
+		else
+			return nil, "invalid-table"
+		end
+
+		if pos == ptr_len then
+			return new_ref
+		elseif type(new_ref) == "table" then
+			ref = new_ref
+		elseif not (type(ref) == "table") then
+			return nil, "invalid-path"
+		end
+
+	end
+	return ref
+end
+
+return { resolve = resolve_json_pointer }