diff mod_http_upload/mod_http_upload.lua @ 2677:6daaa1ad2559

mod_http_upload: Add option for expiry of files after a configurable time (closes #557)
author Kim Alvefur <zash@zash.se>
date Thu, 13 Apr 2017 16:31:00 +0200
parents 0c7802b95820
children 2dec7cad9218
line wrap: on
line diff
--- a/mod_http_upload/mod_http_upload.lua	Mon Apr 10 21:09:44 2017 +0200
+++ b/mod_http_upload/mod_http_upload.lua	Thu Apr 13 16:31:00 2017 +0200
@@ -1,6 +1,6 @@
 -- mod_http_upload
 --
--- Copyright (C) 2015 Kim Alvefur
+-- Copyright (C) 2015-2017 Kim Alvefur
 --
 -- This file is MIT/X11 licensed.
 --
@@ -13,6 +13,7 @@
 local url = require "socket.url";
 local dataform = require "util.dataforms".new;
 local datamanager = require "util.datamanager";
+local array = require "util.array";
 local t_concat = table.concat;
 local t_insert = table.insert;
 local s_upper = string.upper;
@@ -28,6 +29,7 @@
 
 -- config
 local file_size_limit = module:get_option_number(module.name .. "_file_size_limit", 1024 * 1024); -- 1 MB
+local max_age = module:get_option_number(module.name .. "_expire_after");
 
 --- sanity
 local parser_body_limit = module:context("*"):get_option_number("http_max_content_size", 10*1024*1024);
@@ -66,6 +68,26 @@
 local storage_path = module:get_option_string(module.name .. "_path", join_path(prosody.paths.data, module.name));
 lfs.mkdir(storage_path);
 
+local function expire(username, host)
+	if not max_age then return true; end
+	local uploads, err = datamanager.list_load(username, host, module.name);
+	if not uploads then return true; end
+	uploads = array(uploads);
+	local expiry = os.time() - max_age;
+	uploads:filter(function (item)
+		if item.time < expiry then
+			local deleted, whynot = os.remove(item.filename);
+			if deleted then
+				return false;
+			else
+				module:log("warn", "Could not delete expired upload %s: %s", item.filename, whynot or "delete failed");
+			end
+		end
+		return true;
+	end);
+	return datamanager.list_store(username, host, module.name, uploads);
+end
+
 local function handle_request(origin, stanza, xmlns, filename, filesize)
 	-- local clients only
 	if origin.type ~= "c2s" then
@@ -79,6 +101,7 @@
 		origin.send(st.error_reply(stanza, "modify", "bad-request", "Invalid filename"));
 		return true;
 	end
+	expire(origin.username, origin.host);
 	if not filesize then
 		module:log("debug", "Missing file size");
 		origin.send(st.error_reply(stanza, "modify", "bad-request", "Missing or invalid file size"));