DuckCorp » CyborgHood

logger.info "Bot starting"

imap = IMAP.new(@config.imap, @config.imap.min_check_interval)

imap.check_mails do |msg|

process_message(msg)

# wakeup sleeps

Process.kill("ALRM", $$)

def process_message(msg)

if @stop_asap

logger.info "Bot was asked to stop..."

return false

end

mail = Mail.new(msg.content)

logger.info "Received mail with ID '#{mail.message_id}': #{mail.from_pretty} -> #{mail.to_pretty} (#{mail.subject_pretty})"

# ignore mails not signed or encrypted

unless mail.is_pgp_signed? or mail.is_pgp_encrypted?

logger.info "Mail not signed/encrypted or not RFC3156 compliant, ignoring..."

msg.delete

return true

end

logger.debug "RFC3156 content detected"

begin

report = mail.process

rescue CyberError => e

case e.severity

when :grave

logger.fatal "Fatal processing error, exiting (#{e.message})"

exit 2

when :unrecoverable

logger.error "Internal processing error, skipping mail (#{e.message})"

return true

when :processable

logger.error "Untreated processing problem, skipping mail (#{e.message})"

return true

when :ignorable

logger.warn "Internal processing warning, continuing (#{e.message})"

end

result_tag = report.ok? ? "SUCCESS" : "FAILURE"

result_msg = "Processing result: #{result_tag}"

result_msg += " (#{report.error.untranslated})" unless report.ok?

logger.info result_msg

i18n = I18nController.instance

i18n.set_language_for_user(report.user)

unless report.ok?

if report.warn_sender

logger.info "Sending reply for rejected message"

reply_intro = report.user ? _("Hello %{cn},", :cn =>report.user.cn) : _("Hello,")

mail_reply = mail.create_simple_reject_reply(reply_intro.to_s + "\n\n" +

_("A message (ID: %{id}), apparently from you, was rejected for the following reason:",

:id => mail.message_id).to_s + "\n " + report.error.to_s + "\n" + mail_signature())

mail_reply.deliver

msg.delete

return true

end

order = MailOrder.parse(report.user, report.message)

result_tag = order.valid? ? "SUCCESS" : "FAILURE"

result_msg = "Processing result: #{result_tag}"

result_msg += " (#{order.error.untranslated})" unless order.valid?

logger.info result_msg

reply_intro = _("Hello %{cn},", :cn => order.user.cn)

unless order.valid?

logger.info "Sending reply for rejected order"

mail_reply = mail.create_simple_reject_reply(reply_intro.to_s + "\n\n" +

_("An order, in a message (ID: %{id}) from you, was rejected for the following reason:",

:id => mail.message_id).to_s + "\n " + order.error.to_s + "\n" + mail_signature())

mail_reply.sign_and_crypt(order.user.keyFingerPrint)

mail_reply.deliver

msg.delete

return true

end

logger.debug "Message accepted, processing orders..."

result_list = CommandRunner.run(order)

# create transcript

logger.debug "Preparing reply"

reply_txt = reply_intro.to_s + "\n\n"

reply_txt += _("Follows the transcript of your commands:").to_s + "\n"

reply_attachments = []

result_list.each do |result|

reply_txt += "> #{result.cmd}\n"

reply_txt += "#{result.message}\n"

reply_attachments += result.refs unless result.refs.nil?

end

reply_txt += "\n" + mail_signature()

# create mail

logger.debug "Preparing mail"

mail_reply = mail.create_reply

if reply_attachments.empty?

transcript_part = mail_reply

else

mail_reply.to_multipart!

p = CyborgHood::Mail.blank

transcript_part = p

mail_reply.parts << p

reply_attachments.each do |attachment|

p.set_content_type("text", "plain", {'charset' => "utf-8"})

p.set_disposition("attachment", {'filename' => attachment.filename})

p.quoted_printable_body = attachment.content

end

# insert transcript

transcript_part.set_content_type("text", "plain", {'charset' => 'utf-8', 'format' => 'flowed'})

transcript_part.set_disposition("inline")

transcript_part.quoted_printable_body = reply_txt

# send reply

logger.debug "Sending mail"

mail_reply.sign_and_crypt(order.user.keyFingerPrint)

mail_reply.deliver

logger.info "Message processed completely, deleting"

msg.delete

true

def initialize(params, min_check_interval)

@min_check_interval = min_check_interval

@stop_mail_check = false

@imap.store(@message_id, "+FLAGS", [:Deleted])

return false

@stop_mail_check = false

disconnect

return false

def check_mails_once(&message_handler)

authenticate &&

check_inbox(&message_handler)

ensure

begin

disconnect

rescue

end

end

def check_mails(&message_handler)

until @stop_mail_check

t = Time.now.to_i

r = check_mails_once(&message_handler)

t2 = Time.now.to_i

sleep_time = @min_check_interval - (t2 - t)

if sleep_time > 0

logger.debug "Having a break before new check..."

begin

sleep(sleep_time)

rescue

end

end

end

logger.debug "Starting mail check"

unless yield IMAPMessage.new(@imap, message_id)

@stop_mail_check = true

break

end

logger.debug "Mail check finished"