DuckCorp » CyborgHood

#!/usr/bin/ruby1.9.3

# CyborgHood, a distributed system management software.

# This program is free software: you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation, either version 3 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

# GNU General Public License for more details.

#

# You should have received a copy of the GNU General Public License

# along with this program. If not, see <http://www.gnu.org/licenses/>.

#++

$: << File.join(File.dirname(__FILE__), "..", "lib")

#require 'socket'

#class Config

# protected

# class PostmanValidator < CyborgHoodValidator

# def validate_hook_in(value, rule, path, msg_list)

# msg_list << "Youhou !!!"

# end

# end

#end

class Postman #< Cyborg

# load config

Config.load(self.human_name.downcase)

@config = Config.instance

unless @config.log.nil?

logger.output_level(@config.log.console_level) unless @config.log.console_level.nil?

if File.directory? @config.log.path

logger.log_to_file(File.join(@config.log.path, "ch_#{self.class.human_name}.log"))

else

logger.fatal "Log path does not exist or is not a directory, exiting"

exit 1

end

end

ldap_config = @config.ldap.marshal_dump

ldap_config[:logger] = logger

begin

ActiveLdap::Base.setup_connection(ldap_config)

# force testing a connection NOW (by default ActiveLdap is doing lazy connections)

# (the loaded schema will be useful soon anyway)

# it also tests search parameters (provided to setup_connection)

ActiveLdap::Base.find(:first)

rescue

end

logger.info "Bot '#{self.human_name}' loaded"

@imap.check_mails do |msg|

begin

process_message(msg)

rescue CyberError => e

raise

rescue

logger.error "Crash class: %s" % $!.class.to_s

logger.error "Crash trace: %s" % $!.backtrace.join("\n")

end

logger.error "Crash class: %s" % $!.class.to_s

logger.error "Crash trace: %s" % $!.backtrace.join("\n")

end

@imap.stop_mail_check

private

mail = Mail.new(msg.content)

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

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

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})"

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:",

return true

end

order = mail_parser.parse(report.message)

result_msg = "Processing result: #{result_tag}"

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

logger.info result_msg

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:",

msg.delete

return true

end

result_list = CommandRunner.run(order)

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"

logger.debug "Preparing mail"

mail_reply = mail.create_reply

if reply_attachments.empty?

transcript_part = mail_reply

else

mail_reply.to_multipart!

transcript_part = p

mail_reply.parts << p

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

p.quoted_printable_body = attachment.content

end

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

transcript_part.set_disposition("inline")

logger.debug "Sending mail"

msg.delete

true

end

end

trap('INT') do

bot.ask_to_stop

end

trap('TERM') do

bot.ask_to_stop

end

bot.run