feat: update postgrest connector

Author: Olivier Bonnaure

.lua/postgrest.lua

@@ -1,5 +1,5 @@
-PGPGRestApiUrl = ""
-PGRestHeaders = {}
+local PGPGRestApiUrl = ""
+local PGRestHeaders = {}
 
 local function init(db_config)
   PGRestApiUrl = db_config["url"]
@@ -9,9 +9,14 @@ local function init(db_config)
   }
 end
 
+local function count(path)
+  local ok, h, body = Fetch(PGRestApiUrl .. path, { headers = table.merge(PGRestHeaders, { Prefer = "count=exact" }) })
+  return ok, tonumber(string.split(h["Content-Range"], "/")[2])
+end
+
 local function read(path)
   local ok, h, body = Fetch(PGRestApiUrl .. path, { headers = PGRestHeaders })
-  return DecodeJson(body)
+  return ok, DecodeJson(body)
 end
 
 local function update(path, data)
@@ -20,30 +25,32 @@ local function update(path, data)
     body = EncodeJson(data),
     headers = PGRestHeaders
   })
-  return DecodeJson(body)
+  return ok, DecodeJson(body)
 end
 
 local function insert(path, data)
   local ok, h, body = Fetch(PGRestApiUrl .. path, {
     method = "POST",
     body = EncodeJson(data),
-    headers = PGRestHeaders
+    headers = table.merge(PGRestHeaders, { Prefer = "return=representation" })
   })
-  return ok
+  return ok, body
 end
 
 local function delete(path, data)
   local ok, h, body = Fetch(PGRestApiUrl .. path, {
     method = "DELETE",
     headers = PGRestHeaders
   })
-  return DecodeJson(body)
+  return ok, DecodeJson(body)
 end
 
 return {
   init = init,
+  count = count,
   read = read,
   insert = insert,
   update = update,
-  delete = delete
+  delete = delete,
+  upsert = upsert
 }