diff options
Diffstat (limited to 'lualib/rspamadm/mime.lua')
| -rw-r--r-- | lualib/rspamadm/mime.lua | 30 |
1 files changed, 28 insertions, 2 deletions
diff --git a/lualib/rspamadm/mime.lua b/lualib/rspamadm/mime.lua index b44f647a0..2cdec1513 100644 --- a/lualib/rspamadm/mime.lua +++ b/lualib/rspamadm/mime.lua @@ -74,6 +74,8 @@ extract:flag "-w --words" :description "Extracts words" extract:flag "-p --part" :description "Show part info" +extract:flag "-s --structure" + :description "Show structure info (e.g. HTML tags)" local stat = parser:command "stat st s" @@ -187,6 +189,7 @@ end local function extract_handler(opts) local out_elts = {} + local process_func if opts.words then -- Enable stemming @@ -249,7 +252,30 @@ local function extract_handler(opts) if opts.words then table.insert(out_elts[fname], table.concat(part:get_words(), ' ')) else - table.insert(out_elts[fname], tostring(part:get_content(how))) + if opts.structure then + local hc = part:get_html() + local res = {} + process_func = function(k, v) + return rspamd_logger.slog("%s = %s", k, v) + end + + hc:foreach_tag('any', function(tag) + local elt = {} + local ex = tag:get_extra() + elt.tag = tag:get_type() + if ex then + elt.extra = ex + end + local content = tag:get_content() + if content then + elt.content = content + end + table.insert(res, elt) + end) + out_elts[fname] = res + else + table.insert(out_elts[fname], tostring(part:get_content(how))) + end end end end @@ -260,7 +286,7 @@ local function extract_handler(opts) task:destroy() -- No automatic dtor end - print_elts(out_elts, opts) + print_elts(out_elts, opts, process_func) end local function stat_handler(opts) |