Software /
code /
prosody-modules
Changeset
4468:5d8f9cc5c6fb
mod_migrate_http_upload: Upload data converter to mod_http_file_share
author | Kim Alvefur <zash@zash.se> |
---|---|
date | Wed, 24 Feb 2021 16:55:46 +0100 |
parents | 4467:6d595857164a |
children | 4469:6844733ad2f5 |
files | mod_migrate_http_upload/README.markdown mod_migrate_http_upload/mod_migrate_http_upload.lua |
diffstat | 2 files changed, 87 insertions(+), 0 deletions(-) [+] |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/mod_migrate_http_upload/README.markdown Wed Feb 24 16:55:46 2021 +0100 @@ -0,0 +1,31 @@ +--- +summary: mod_http_upload to mod_http_file_share migrator +labels: +- Stage-Alpha +--- + + +This is a migration script for converting records of [mod_http_upload] +into the format used by the new [mod_http_file_share] which will be +available with Prosody 0.12 (currently in trunk). + +# Usage + +If your main `VirtualHost` is called "example.com" and your HTTP upload +`Component` is called "upload.example.com", then this command would +convert records of existing uploads via [mod_http_upload] to +[mod_http_file_share][doc:modules:mod_http_file_share]: + +```bash +sudo prosodyctl mod_migrate_http_upload upload.example.com example.com +``` + +In order to preserve URLs you will need to configure the +[path][doc:http#path_configuration] to be the same as mod_http_upload: + +```lua +Component "upload.example.com" "http_file_share" +http_paths = { + file_share = "/upload" +} +```
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/mod_migrate_http_upload/mod_migrate_http_upload.lua Wed Feb 24 16:55:46 2021 +0100 @@ -0,0 +1,56 @@ +-- Copyright (C) 2021 Kim Alvefur +-- +-- This file is MIT licensed. + +local lfs = require "lfs"; +local st = require "util.stanza"; +local jid = require "util.jid"; +local paths = require "util.paths"; +local unpack = table.unpack or _G.unpack; + +function module.command(arg) + local sm = require "core.storagemanager"; + local dm = sm.olddm; + + local component, user_host = unpack(arg); + + sm.initialize_host(component); + + local new_uploads = sm.open(component, "uploads", "archive"); + + local legacy_uploads = {}; + for user in assert(dm.users(user_host, "http_upload", "list")) do + legacy_uploads[user] = dm.list_load(user, user_host, "http_upload"); + end + while true do + local oldest_uploads, uploader; + for user, uploads in pairs(legacy_uploads) do + if uploads[1] and (not oldest_uploads or uploads[1].time < oldest_uploads[1].time) then + oldest_uploads, uploader = uploads, jid.join(user, user_host); + end + end + if not oldest_uploads then break end + local item = table.remove(oldest_uploads, 1); + local source_directory = paths.join(prosody.paths.data, "http_upload", item.dir); + local source_filename = paths.join(prosody.paths.data, "http_upload", item.dir, item.filename); + local target_filename = dm.getpath(item.dir, component, "http_file_share", "bin", true); + if not lfs.attributes(source_filename, "mode") then + print("Not migrating missing file " .. source_filename); + else + print("Moving " .. source_filename .. " to " .. target_filename .. " for " .. uploader); + local upload = st.stanza("request", { + xmlns = "urn:xmpp:http:upload:0"; + filename = item.filename; + size = string.format("%d", item.size); + -- content-type not included with mod_http_upload + }); + assert(new_uploads:append(nil, item.dir, upload, item.time, uploader)); + assert(os.rename(source_filename, target_filename)); + end + os.remove(source_directory); -- failure not fatal + end + for user, uploads in pairs(legacy_uploads) do + assert(dm.list_store(user, user_host, "http_upload", uploads)); + end + return 0; +end