Mercurial > prosody-modules
annotate mod_storage_appendmap/mod_storage_appendmap.lua @ 6299:5cf5ee23b361
mod_voipms: New Module to send and receive SMS/MMS via VoIP.ms APIs.
diff --git a/mod_voipms/README.md b/mod_voipms/README.md
new file mode 100644
--- /dev/null
+++ b/mod_voipms/README.md
@@ -0,0 +1,51 @@
+---
+labels:
+- 'Stage-Alpha'
+- 'Type-Web'
+summary: Send and receive SMS/MMS via VoIP.ms APIs.
+rockspec:
+ build:
+ modules:
+ mod_voipms: mod_voipms.lua
+...
+
+Introduction
+============
+
+This is a Prosody module to map JIDs to DIDs on VoIP.ms and support sending/receiving SMS/MMS.
+
+Configuration
+=============
+
+| option | type | default |
+|-----------------------|--------|---------|
+| voipms\_api\_username | string | nil |
+| voipms\_api\_password | string | nil |
+| voipms\_query\_key | string | nil |
+| voipms\_jid\_map | table | nil |
+```
+VirtualHost "sms.example.com"
+modules_enabled = {
+ "voipms";
+}
+voipms_api_username = john@example.com -- E-mail registered with VoIP.ms
+voipms_api_password = abcd1234 -- API password configured in VoIP.ms
+voipms_query_key = some_query_key -- query param 'key' part of your URL callback
+voipms_jid_map = {
+ ["your_jid@your_domain.com"] = "+1234567890"
+}
+```
+
+HTTP
+====
+
+The module is served on Prosody's default HTTP ports at the path /voipms. More details on configuring HTTP modules in Prosody can be found in the HTTP documentation.
+
+VoIP.ms Webhook URL
+===================
+
+This module receives the VoIP.ms Webhook URL (POST) at the /voipms endpoint. It uses the sendSMS/sendMMS GET methods against the VoIP.ms APIs. This is an example webhook to use in VoIP.ms:
+
+```
+https://sms.example.com/voipms?key=some_query_key
+```
diff --git a/mod_voipms/mod_voipms.lua b/mod_voipms/mod_voipms.lua
new file mode 100644
--- /dev/null
+++ b/mod_voipms/mod_voipms.lua
@@ -0,0 +1,166 @@
+local http = require "net.http"
+local json = require "util.json"
+local st = require "util.stanza"
+
+local api_username = module:get_option_string("voipms_api_username")
+local api_password = module:get_option_string("voipms_api_password")
+local query_key = module:get_option("voipms_query_key")
+local jid_map = module:get_option("voipms_jid_map") or {}
+local rest_endpoint = "https://voip.ms/api/v1/rest.php"
+
+if not api_username or not api_password or not query_key then
+ module:log("error", "Missing required config values (voipms_api_username, voipms_api_password, voipms_query_key)")
+ return
+end
+
+module:depends("http")
+
+local function normalize_number(num)
+ if not num then return nil end
+ if num:sub(1, 1) ~= "+" then
+ return "+1" .. num
+ end
+ return num
+end
+
+local function extract_query_key(event)
+ return (event.request.url.query or ""):match("key=([^&]+)")
+end
+
+module:provides("http", {
+ route = {
+ ["POST"] = function(event)
+ local req = event.request
+ local body = req.body or ""
+
+ if extract_query_key(event) ~= query_key then
+ module:log("warn", "Unauthorized webhook: missing or invalid key")
+ return { status_code = 403 }
+ end
+
+ local json_payload, err = json.decode(body)
+ if not json_payload then
+ module:log("warn", "Invalid JSON: %s", err or "unknown error")
+ return { status_code = 400 }
+ end
+
+ local payload = json_payload.data and json_payload.data.payload
+ if not payload then
+ module:log("warn", "Missing payload in JSON")
+ return { status_code = 400 }
+ end
+
+ local from = payload.from and payload.from.phone_number
+ local to_list = payload.to or {}
+ local to = #to_list > 0 and to_list[1].phone_number or nil
+
+ if not from or not to then
+ module:log("warn", "Missing phone numbers (from: %s, to: %s)", tostring(from), tostring(to))
+ return { status_code = 400 }
+ end
+
+ local normalized_to = normalize_number(to)
+ local target_jid = nil
+
+ for jid, did in pairs(jid_map) do
+ if normalize_number(did) == normalized_to then
+ target_jid = jid
+ break
+ end
+ end
+
+ if not target_jid then
+ module:log("warn", "No JID mapping for DID %s", normalized_to)
+ return { status_code = 404 }
+ end
+
+ local normalized_from = normalize_number(from)
+ local message_text = payload.text or ""
+
+ local message = st.message({
+ from = normalized_from .. "@" .. module.host,
+ to = target_jid,
+ type = "chat"
+ }):tag("body"):text(message_text):up()
+
+ if payload.media and #payload.media > 0 then
+ for _, media_item in ipairs(payload.media) do
+ if media_item.url then
+ message_text = message_text .. "\n" .. media_item.url
+ message:tag("x", { xmlns = "jabber:x:oob" })
+ :tag("url"):text(media_item.url):up():up()
+ end
+ end
+ end
+
+ module:send(message)
+ module:log("info", "Delivered SMS from %s to %s", normalized_from, target_jid)
+
+ return { status_code = 204 }
+ end
+ }
+})
+
+module:hook("message/bare", function(event)
+ local stanza = event.stanza
+ if stanza.attr.type ~= "chat" then return end
+
+ local from_jid = stanza.attr.from
+ local to_jid = stanza.attr.to
+ local body = stanza:get_child_text("body")
+ if not body or body == "" then return end
+
+ local from_number = jid_map[from_jid]
+ if not from_number then
+ module:log("warn", "No DID mapping for JID %s", from_jid)
+ return
+ end
+
+ local to_number = to_jid:match("^([^@]+)")
+ if not to_number then
+ module:log("warn", "Malformed JID in message.to: %s", to_jid)
+ return
+ end
+
+ to_number = normalize_number(to_number)
+
+ local media_urls = {}
+ for line in body:gmatch("[^\r\n]+") do
+ if line:match("^https?://") then
+ table.insert(media_urls, line)
+ end
+ end
+
+ local method = (#media_urls > 0) and "sendMMS" or "sendSMS"
+ local query = {
+ api_username = api_username,
+ api_password = api_password,
+ method = method,
+ did = from_number,
+ dst = to_number,
+ message = body
+ }
+
+ if method == "sendMMS" then
+ for i, url in ipairs(media_urls) do
+ query["media_url[" .. (i - 1) .. "]"] = url
+ end
+ end
+
+ local query_str = http.formencode(query)
+
+ http.request(rest_endpoint .. "?" .. query_str, {
+ method = "GET";
+ }, function(response_body, code)
+ if code == 200 then
+ local resp, err = json.decode(response_body)
+ if not resp or resp.status ~= "success" then
+ module:log("error", "Failed to send %s: %s", method, err or (resp and resp.status) or "unknown")
+ else
+ module:log("info", "Sent %s from %s to %s", method, from_number, to_number)
+ end
+ else
+ module:log("error", "HTTP error sending %s: code %s", method, tostring(code))
+ end
+ end)
+end)
| author | Chaz <menel@snikket.de> |
|---|---|
| date | Mon, 21 Jul 2025 23:57:37 +0200 |
| parents | 78f766372e2c |
| children | f61564e11d3b |
| rev | line source |
|---|---|
|
2061
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
1 local dump = require "util.serialization".serialize; |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
2 local load = require "util.envload".envloadfile; |
|
5691
78f766372e2c
mod_storage_appendmap: Include timestamps when appending data
Kim Alvefur <zash@zash.se>
parents:
5690
diff
changeset
|
3 local datetime = require "util.datetime".datetime; |
|
2061
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
4 local dm = require "core.storagemanager".olddm; |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
5 |
|
2796
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
6 local REMOVE = {}; -- Special value for removing keys |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
7 |
|
2061
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
8 local driver = {}; |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
9 |
|
5691
78f766372e2c
mod_storage_appendmap: Include timestamps when appending data
Kim Alvefur <zash@zash.se>
parents:
5690
diff
changeset
|
10 local timestamps = module:get_option_boolean("appendmap_timestamps", true); |
|
2796
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
11 |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
12 local keywords = { |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
13 ["do"] = true; ["and"] = true; ["else"] = true; ["break"] = true; |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
14 ["if"] = true; ["end"] = true; ["goto"] = true; ["false"] = true; |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
15 ["in"] = true; ["for"] = true; ["then"] = true; ["local"] = true; |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
16 ["or"] = true; ["nil"] = true; ["true"] = true; ["until"] = true; |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
17 ["elseif"] = true; ["function"] = true; ["not"] = true; |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
18 ["repeat"] = true; ["return"] = true; ["while"] = true; |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
19 |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
20 -- _ENV is not technically a keyword but we need to treat it as such |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
21 ["_ENV"] = true; |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
22 }; |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
23 |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
24 local function is_usable_identifier(s) |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
25 return type(s) == "string" and not keywords[s] and s:find("^[%a_][%w_]*$"); |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
26 end |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
27 |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
28 local function serialize_key(key) |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
29 if is_usable_identifier(key) then |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
30 return key; |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
31 else |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
32 return "_ENV[" .. dump(key) .. "]"; |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
33 end |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
34 end |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
35 |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
36 local function serialize_value(value) |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
37 if value == REMOVE then |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
38 return "nil"; |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
39 else |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
40 return dump(value); |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
41 end |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
42 end |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
43 |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
44 local function serialize_pair(key, value) |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
45 key = serialize_key(key); |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
46 value = serialize_value(value); |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
47 return key .. " = " .. value .. ";\n"; |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
48 end |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
49 |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
50 local function serialize_map(keyvalues) |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
51 local keys, values = {}, {}; |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
52 for key, value in pairs(keyvalues) do |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
53 key = serialize_key(key); |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
54 value = serialize_value(value); |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
55 table.insert(keys, key); |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
56 table.insert(values, value); |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
57 end |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
58 return table.concat(keys, ", ") .. " = " .. table.concat(values, ", ") .. ";\n"; |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
59 end |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
60 |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
61 local map = { remove = REMOVE }; |
|
2061
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
62 local map_mt = { __index = map }; |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
63 |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
64 function map:get(user, key) |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
65 module:log("debug", "map:get(%s, %s)", tostring(user), tostring(key)) |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
66 local filename = dm.getpath(user, module.host, self.store, "map"); |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
67 module:log("debug", "File is %s", filename); |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
68 local env = {}; |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
69 if _VERSION == "Lua 5.1" then -- HACK |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
70 env._ENV = env; -- HACK |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
71 end -- SO MANY HACKS |
|
2635
4548c3d685b4
mod_storage_appendmap: Successfully return nothing on ENOENT
Kim Alvefur <zash@zash.se>
parents:
2431
diff
changeset
|
72 local chunk, err, errno = load(filename, env); |
|
4548c3d685b4
mod_storage_appendmap: Successfully return nothing on ENOENT
Kim Alvefur <zash@zash.se>
parents:
2431
diff
changeset
|
73 if not chunk then if errno == 2 then return end return chunk, err; end |
|
2061
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
74 local ok, err = pcall(chunk); |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
75 if not ok then return ok, err; end |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
76 if _VERSION == "Lua 5.1" then -- HACK |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
77 env._ENV = nil; -- HACK |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
78 end -- HACKS EVERYWHERE |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
79 if key == nil then |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
80 return env; |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
81 end |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
82 return env[key]; |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
83 end |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
84 |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
85 function map:set_keys(user, keyvalues) |
|
2796
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
86 local data = serialize_map(keyvalues); |
|
5691
78f766372e2c
mod_storage_appendmap: Include timestamps when appending data
Kim Alvefur <zash@zash.se>
parents:
5690
diff
changeset
|
87 if timestamps then |
|
78f766372e2c
mod_storage_appendmap: Include timestamps when appending data
Kim Alvefur <zash@zash.se>
parents:
5690
diff
changeset
|
88 data = "-- " .. datetime() .. "\n" .. data; |
|
78f766372e2c
mod_storage_appendmap: Include timestamps when appending data
Kim Alvefur <zash@zash.se>
parents:
5690
diff
changeset
|
89 end |
|
2061
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
90 return dm.append_raw(user, module.host, self.store, "map", data); |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
91 end |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
92 |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
93 function map:set(user, key, value) |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
94 if _VERSION == "Lua 5.1" then |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
95 assert(key ~= "_ENV", "'_ENV' is a restricted key"); |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
96 end |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
97 if key == nil then |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
98 local filename = dm.getpath(user, module.host, self.store, "map"); |
|
2796
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
99 return os.remove(filename); |
|
2061
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
100 end |
|
2796
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
101 local data = serialize_pair(key, value); |
|
5691
78f766372e2c
mod_storage_appendmap: Include timestamps when appending data
Kim Alvefur <zash@zash.se>
parents:
5690
diff
changeset
|
102 if timestamps then |
|
78f766372e2c
mod_storage_appendmap: Include timestamps when appending data
Kim Alvefur <zash@zash.se>
parents:
5690
diff
changeset
|
103 data = "-- " .. datetime() .. "\n" .. data; |
|
78f766372e2c
mod_storage_appendmap: Include timestamps when appending data
Kim Alvefur <zash@zash.se>
parents:
5690
diff
changeset
|
104 end |
|
2061
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
105 return dm.append_raw(user, module.host, self.store, "map", data); |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
106 end |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
107 |
|
5690
ea6c18ec0669
mod_storage_appendmap: Implement item/user iteration methods
Kim Alvefur <zash@zash.se>
parents:
5061
diff
changeset
|
108 function map:items() |
|
ea6c18ec0669
mod_storage_appendmap: Implement item/user iteration methods
Kim Alvefur <zash@zash.se>
parents:
5061
diff
changeset
|
109 return dm.users(module.host, self.store, "map"); |
|
ea6c18ec0669
mod_storage_appendmap: Implement item/user iteration methods
Kim Alvefur <zash@zash.se>
parents:
5061
diff
changeset
|
110 end |
|
ea6c18ec0669
mod_storage_appendmap: Implement item/user iteration methods
Kim Alvefur <zash@zash.se>
parents:
5061
diff
changeset
|
111 |
|
2796
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
112 local keyval = { remove = REMOVE }; |
|
2061
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
113 local keyval_mt = { __index = keyval }; |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
114 |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
115 function keyval:get(user) |
|
2796
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
116 return map.get(self, user, nil); |
|
2061
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
117 end |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
118 |
|
2796
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
119 function keyval:set(user, keyvalues) |
|
6a7b7cb7148e
mod_storage_appendmap: Factor out serialization into reusable functions
Kim Alvefur <zash@zash.se>
parents:
2635
diff
changeset
|
120 local data = serialize_map(keyvalues); |
|
5691
78f766372e2c
mod_storage_appendmap: Include timestamps when appending data
Kim Alvefur <zash@zash.se>
parents:
5690
diff
changeset
|
121 if timestamps then |
|
78f766372e2c
mod_storage_appendmap: Include timestamps when appending data
Kim Alvefur <zash@zash.se>
parents:
5690
diff
changeset
|
122 data = "-- " .. datetime() .. "\n" .. data; |
|
78f766372e2c
mod_storage_appendmap: Include timestamps when appending data
Kim Alvefur <zash@zash.se>
parents:
5690
diff
changeset
|
123 end |
|
5061
e44b868cc575
mod_storage_appendmap: Fix keyvalue writes
Kim Alvefur <zash@zash.se>
parents:
2796
diff
changeset
|
124 return dm.store_raw(dm.getpath(user, module.host, self.store, "map"), data); |
|
2062
8f7083b980cf
mod_storage_appendmap: Fix keyvalue:set()
Kim Alvefur <zash@zash.se>
parents:
2061
diff
changeset
|
125 end |
|
2061
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
126 |
|
5690
ea6c18ec0669
mod_storage_appendmap: Implement item/user iteration methods
Kim Alvefur <zash@zash.se>
parents:
5061
diff
changeset
|
127 function keyval:users() |
|
ea6c18ec0669
mod_storage_appendmap: Implement item/user iteration methods
Kim Alvefur <zash@zash.se>
parents:
5061
diff
changeset
|
128 return dm.users(module.host, self.store, "map"); |
|
ea6c18ec0669
mod_storage_appendmap: Implement item/user iteration methods
Kim Alvefur <zash@zash.se>
parents:
5061
diff
changeset
|
129 end |
|
ea6c18ec0669
mod_storage_appendmap: Implement item/user iteration methods
Kim Alvefur <zash@zash.se>
parents:
5061
diff
changeset
|
130 |
|
2061
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
131 -- TODO some kind of periodic compaction thing? |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
132 function map:_compact(user) |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
133 local data = self:get(user); |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
134 return keyval.set(self, user, data); |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
135 end |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
136 |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
137 function driver:open(store, typ) |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
138 if typ == "map" then |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
139 return setmetatable({ store = store, }, map_mt); |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
140 elseif typ == nil or typ == "keyval" then |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
141 return setmetatable({ store = store, }, keyval_mt); |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
142 end |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
143 return nil, "unsupported-store"; |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
144 end |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
145 |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
146 module:provides("storage", driver); |
|
b84284144e21
mod_storage_appendmap: Experimental storage module optimized for map stores
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
147 |
