Revision f0a75e9c
Added by Marc Dequènes over 15 years ago
- ID f0a75e9c7874167708380ded63640b0f3f0229ea
lib/cyborghood/mail.rb | ||
---|---|---|
# This class handles RFC3156 signed messages, validates them, and extract orders properly.
|
||
# Encrypted content are not implemented yet.
|
||
module CyborgHood
|
||
class Command
|
||
attr_reader :cmdline, :cmdsplit
|
||
|
||
def initialize(cmdline, cmdsplit)
|
||
@cmdline = cmdline
|
||
@cmdsplit = cmdsplit
|
||
end
|
||
end
|
||
|
||
class SharedParameter
|
||
attr_reader :type, :content
|
||
|
||
def initialize(content, type = nil)
|
||
@content = content
|
||
@type = type
|
||
end
|
||
end
|
||
|
||
class ParameterReference
|
||
attr_reader :reference
|
||
|
||
def initialize(reference)
|
||
@reference = reference
|
||
end
|
||
end
|
||
|
||
class Order
|
||
attr_accessor :ok, :message, :system_message, :user, :commands, :references
|
||
attr_accessor :ok, :message, :system_message, :user, :commands, :shared_parameters
|
||
attr_writer = :warn_sender
|
||
|
||
def initialize(ok, message = nil, system_message = nil)
|
||
... | ... | |
|
||
def parse_plain
|
||
command_txt = nil
|
||
refs = nil
|
||
shared_params = nil
|
||
if multipart?
|
||
if parts[0].content_type == "text/plain"
|
||
command_txt = self.parts[0].body
|
||
refs = self.parts.collect{|p| p.dup }
|
||
shared_params = {}
|
||
i = 1
|
||
self.parts.each do |p|
|
||
shared_params[i] = SharedParameter.new(p.body, p.content_type)
|
||
filename = p.header['content-type'].params('filename')
|
||
shared_params[filename] = ParameterReference.new(i) if filename
|
||
i += 1
|
||
end
|
||
end
|
||
else
|
||
command_txt = self.body if self.content_type == "text/plain"
|
||
refs = []
|
||
shared_params = {}
|
||
end
|
||
unless command_txt
|
||
order = Order.new(false, N_("Mail does not contain a proper text part for commands."))
|
||
... | ... | |
# stop processing when detecting message signature
|
||
break if line == "-- "
|
||
|
||
commands << sline
|
||
used_refs = []
|
||
cmd_parts = sline.shellsplit.collect do |word|
|
||
if =~ /^@([a-zA-Z0-9._-]+)$/
|
||
ref = $1
|
||
d_ref, d_param = dereference_param(shared_params, param)
|
||
# TODO: should add error message for attachment not found in the Command
|
||
used_refs << d_ref
|
||
d_param
|
||
else
|
||
word
|
||
end
|
||
end
|
||
|
||
commands << Command.new(sline, cmd_parts)
|
||
end
|
||
|
||
shared_params.delete_if{|ref, param| not used_refs.include?(ref) }
|
||
|
||
logger.debug "Mail OK"
|
||
mark_processed(self.signature_timestamp)
|
||
|
||
order = Order.new(true)
|
||
order.user = self.user
|
||
order.commands = commands
|
||
order.references = refs
|
||
order.references = shared_params
|
||
order
|
||
end
|
||
|
||
def dereference_param(shared_params, param)
|
||
if param.is_a? SharedParameter
|
||
[ref, ParameterReference(ref)]
|
||
elsif param.is_a? ParameterReference
|
||
d_ref = param.reference
|
||
d_param = shared_params[d_ref]
|
||
return dereference_param(shared_params, d_param)
|
||
else
|
||
nil
|
||
end
|
||
end
|
||
|
||
def parse_signed
|
||
sigs_check = verify_pgp_signature()
|
||
return Order.new(false, N_("Mail not formatted correctly (signed part).")) if sigs_check.nil? or sigs_check.size != 1
|
Also available in: Unified diff
[evol] commands management rework #1: moved command parsing into the Mail class, so the CommandParser is now a CommandRunner, and does not have to understand mail parts at all