2004-01-19 06:41:30 +01:00
|
|
|
-- make sure LuaSocket is loaded
|
|
|
|
if not LUASOCKET_LIBNAME then error('module requires LuaSocket') end
|
|
|
|
-- get LuaSocket namespace
|
|
|
|
local socket = _G[LUASOCKET_LIBNAME]
|
|
|
|
if not socket then error('module requires LuaSocket') end
|
|
|
|
-- create code namespace inside LuaSocket namespace
|
|
|
|
local mime = socket.mime or {}
|
|
|
|
socket.mime = mime
|
|
|
|
-- make all module globals fall into mime namespace
|
|
|
|
setmetatable(mime, { __index = _G })
|
|
|
|
setfenv(1, mime)
|
|
|
|
|
2004-01-21 02:09:50 +01:00
|
|
|
-- encode, decode and wrap algorithm tables
|
|
|
|
local et = {}
|
|
|
|
local dt = {}
|
|
|
|
local wt = {}
|
2004-01-19 06:41:30 +01:00
|
|
|
|
2004-02-04 15:29:11 +01:00
|
|
|
-- creates a function that chooses a filter from a given table
|
2004-01-21 02:09:50 +01:00
|
|
|
local function choose(table)
|
2004-02-04 15:29:11 +01:00
|
|
|
return function(filter, ...)
|
|
|
|
local f = table[filter or "nil"]
|
|
|
|
if not f then error("unknown filter (" .. tostring(filter) .. ")", 3)
|
2004-01-21 02:09:50 +01:00
|
|
|
else return f(unpack(arg)) end
|
2004-01-19 06:41:30 +01:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2004-02-04 15:29:11 +01:00
|
|
|
-- define the encoding filters
|
2004-01-21 02:09:50 +01:00
|
|
|
et['base64'] = function()
|
2004-02-04 15:29:11 +01:00
|
|
|
return socket.cicle(b64, "")
|
2004-01-19 06:41:30 +01:00
|
|
|
end
|
|
|
|
|
2004-01-21 02:09:50 +01:00
|
|
|
et['quoted-printable'] = function(mode)
|
2004-02-04 15:29:11 +01:00
|
|
|
return socket.cicle(qp, "", (mode == "binary") and "=0D=0A" or "\13\10")
|
2004-01-21 02:09:50 +01:00
|
|
|
end
|
|
|
|
|
2004-02-04 15:29:11 +01:00
|
|
|
-- define the decoding filters
|
2004-01-21 02:09:50 +01:00
|
|
|
dt['base64'] = function()
|
2004-02-04 15:29:11 +01:00
|
|
|
return socket.cicle(unb64, "")
|
2004-01-19 06:41:30 +01:00
|
|
|
end
|
|
|
|
|
2004-01-21 02:09:50 +01:00
|
|
|
dt['quoted-printable'] = function()
|
2004-02-04 15:29:11 +01:00
|
|
|
return socket.cicle(unqp, "")
|
2004-01-21 02:09:50 +01:00
|
|
|
end
|
|
|
|
|
2004-02-04 15:29:11 +01:00
|
|
|
-- define the line-wrap filters
|
|
|
|
wt['text'] = function(length)
|
2004-01-19 06:41:30 +01:00
|
|
|
length = length or 76
|
2004-02-04 15:29:11 +01:00
|
|
|
return socket.cicle(wrp, length, length)
|
2004-01-19 06:41:30 +01:00
|
|
|
end
|
2004-02-04 15:29:11 +01:00
|
|
|
wt['base64'] = wt['text']
|
2004-01-19 06:41:30 +01:00
|
|
|
|
2004-02-04 15:29:11 +01:00
|
|
|
wt['quoted-printable'] = function()
|
|
|
|
return socket.cicle(qpwrp, 76, 76)
|
|
|
|
end
|
|
|
|
|
|
|
|
-- function that choose the encoding, decoding or wrap algorithm
|
|
|
|
encode = choose(et)
|
|
|
|
decode = choose(dt)
|
|
|
|
-- there is a default wrap filter
|
|
|
|
local cwt = choose(wt)
|
|
|
|
function wrap(...)
|
|
|
|
if type(arg[1]) ~= "string" then table.insert(arg, 1, "text") end
|
|
|
|
return cwt(unpack(arg))
|
2004-01-19 06:41:30 +01:00
|
|
|
end
|
|
|
|
|
2004-02-04 15:29:11 +01:00
|
|
|
-- define the end-of-line translation filter
|
2004-01-19 06:41:30 +01:00
|
|
|
function canonic(marker)
|
2004-02-04 15:29:11 +01:00
|
|
|
return socket.cicle(eol, "", marker)
|
2004-01-19 06:41:30 +01:00
|
|
|
end
|
|
|
|
|
2004-01-21 02:09:50 +01:00
|
|
|
-- chains several filters together
|
2004-01-19 06:41:30 +01:00
|
|
|
function chain(...)
|
|
|
|
local layers = table.getn(arg)
|
|
|
|
return function (chunk)
|
|
|
|
if not chunk then
|
|
|
|
local parts = {}
|
|
|
|
for i = 1, layers do
|
|
|
|
for j = i, layers do
|
|
|
|
chunk = arg[j](chunk)
|
|
|
|
end
|
|
|
|
table.insert(parts, chunk)
|
|
|
|
chunk = nil
|
|
|
|
end
|
|
|
|
return table.concat(parts)
|
|
|
|
else
|
|
|
|
for j = 1, layers do
|
|
|
|
chunk = arg[j](chunk)
|
|
|
|
end
|
|
|
|
return chunk
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2004-02-04 15:29:11 +01:00
|
|
|
return mime
|