--- /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