-- mod_http_upload_external

--

-- Copyright (C) 2015-2016 Kim Alvefur

--

-- This file is MIT/X11 licensed.

--

-- imports

local st = require"util.stanza";

local uuid = require"util.uuid".generate;

local http = require "util.http";

local dataform = require "util.dataforms".new;

local HMAC = require "util.hashes".hmac_sha256;

-- config

local file_size_limit = module:get_option_number(module.name .. "_file_size_limit", 100 * 1024 * 1024); -- 100 MB

local base_url = assert(module:get_option_string(module.name .. "_base_url"), module.name .. "_base_url is a required option");

local secret = assert(module:get_option_string(module.name .. "_secret"), module.name .. "_secret is a required option");

-- depends

module:depends("disco");

-- namespace

-- identity and feature advertising

module:add_identity("store", "file", module:get_option_string("name", "HTTP File Upload"))

module:add_extension(dataform {

	{ name = "max-file-size", type = "text-single" },

}:form({ ["max-file-size"] = tostring(file_size_limit) }, "result"));

local function magic_crypto_dust(random, filename, filesize)

	local message = string.format("%s/%s %d", random, filename, filesize);

	local digest = HMAC(secret, message, true);

	random, filename = http.urlencode(random), http.urlencode(filename);

	return base_url .. random .. "/" .. filename, "?v=" .. digest;

end

	-- local clients only

	if origin.type ~= "c2s" then

		module:log("debug", "Request for upload slot from a %s", origin.type);

		origin.send(st.error_reply(stanza, "cancel", "not-authorized"));

	end

	-- validate

	if not filename or filename:find("/") then

		module:log("debug", "Filename %q not allowed", filename or "");

		origin.send(st.error_reply(stanza, "modify", "bad-request", "Invalid filename"));

	end

	if not filesize then

		module:log("debug", "Missing file size");

		origin.send(st.error_reply(stanza, "modify", "bad-request", "Missing or invalid file size"));

	elseif filesize > file_size_limit then

		module:log("debug", "File too large (%d > %d)", filesize, file_size_limit);

		origin.send(st.error_reply(stanza, "modify", "not-acceptable", "File too large",

				:tag("max-size"):text(tostring(file_size_limit))));

		return true;

	end

	origin.send(reply);

	return true;

end);