omnibot 0.0.9 → 0.0.10

data/Rakefile

@@ -25,6 +25,11 @@ task :install => :build do
   system "gem install pkg/#{gemspec.name}-#{gemspec.version}"
 end
 
+desc "Install gem locally without documentation"
+task :install_nodoc => :build do
+  system "gem install pkg/#{gemspec.name}-#{gemspec.version} --no-rdoc --no-ri"
+end
+
 desc "Push gem"
 task :push => :build do
   system "gem push pkg/#{gemspec.name}-#{gemspec.version}.gem"

data/examples/config.yaml

@@ -5,3 +5,16 @@ config:
   notifyresource: Adium.*
   periodiccommands: [ 'vnstat' ]
 	logpath: '/tmp/omnibot.log'
+  mails:
+    - user: foobar@example.com
+      password: secret
+      host: pop.example.com
+      port: 995
+      ssl: true
+  mailtriggers:
+    - action: unpack
+      for: omni.notify@mail.ru
+      if:
+        subject: '^Weekly archive.*'
+      unpack_to: '/tmp/'
+      command_post: 'post-mail'

data/lib/omnibot.rb

@@ -6,9 +6,15 @@ require 'eventmachine'
 require 'xmpp4r'
 require 'xmpp4r/client'
 require 'xmpp4r/roster'
+require 'mail'
 require 'socket'
 require 'date'
+require 'tmpdir'
+
+module OmniBot
+
+	%w[ helpers jabberbot amqpconsumer omnisend launcher loggedcommand periodiccommand mailchecker ].each do |file|
+		require "omnibot/#{file}.rb"
+	end
 
-%w[ helpers jabberbot amqpconsumer omnisend launcher periodiccommand ].each do |file|
-  require "omnibot/#{file}.rb"
 end

data/lib/omnibot/amqpconsumer.rb

@@ -3,6 +3,8 @@ module OmniBot
 	# AMQP consumer class
 	class AMQPConsumer
 
+		attr_accessor :handlers
+
 		def send_message message
 			begin
 				@omnibot.add_message [Time.now, message]
@@ -29,14 +31,12 @@ module OmniBot
 				@omnibot.set_subscriber Jabber::JID::new(@config['notifyjid']), @config['notifyresource']
 				@omnibot.connect
 
-				pause = 0
-				[@config['periodiccommands']].flatten.each do |command|
-					OmniLog::info "Setup command #{command}..."
-					periodic_command = PeriodicCommand.new command, pause
-					periodic_command.timer_provider = EM
-					periodic_command.set_jabber_messenger { |message| send_message message }
-					periodic_command.start
-					pause += 20
+				@handlers.each_with_index do |handler, index|
+					OmniLog::info "Setup handler #{handler.to_s}..."
+					handler.timer_provider = EM
+					handler.set_jabber_messenger { |message| send_message message }
+					handler.startup_pause = index*10
+					handler.start
 				end
 
 			rescue Object => e

data/lib/omnibot/launcher.rb

@@ -15,6 +15,37 @@ module OmniBot
 			return 'omnibot.log'
 		end
 
+		def get_store config
+			store_path = config['storepath']
+			unless File.directory? store_path
+				store_path = ENV['HOME']+'/.omnibot'
+				FileUtils.mkdir store_path unless File.directory? store_path
+			end
+			store_path + '/omnibot.store'
+		end
+
+		def provide_handlers config
+			periodic_commands = ([config['periodiccommands']].flatten or [])
+			
+			mails = ([config['mails']].flatten or [])
+			mail_triggers = ([config['mailtriggers']].flatten or [])
+			mail_triggers.each do |mt|
+				raise 'No mail found for a trigger' unless mails.find { |m| m['user'] == mt['for'] }
+				raise 'Not supported action' unless mt['action'] == 'unpack'
+			end
+			used_mails = mails.select { |m| mail_triggers.find { |mt| m['user'] == mt['for'] } }
+			raise 'Sorry but multiple mail addresses is not supported yet' if used_mails.size > 1
+
+			[] + 
+				periodic_commands.map do |command|
+					PeriodicCommand.new command
+				end +
+				mail_triggers.map do |trigger|
+					mail = mails.find { |m| m['user'] == trigger['for'] }
+					MailChecker.new mail, trigger
+				end
+		end
+
 		def start args
 			config_path = get_config_path args
 			puts "Using config at #{config_path}"
@@ -26,6 +57,7 @@ module OmniBot
 			OmniLog::log.level = Logger::DEBUG
 
 			consumer = AMQPConsumer.new config
+			consumer.handlers = provide_handlers(config)
 			consumer.start 
 
 			OmniLog::log.close

data/lib/omnibot/loggedcommand.rb

@@ -0,0 +1,18 @@
+module OmniBot
+	module LoggedCommand
+
+		def jabber_logged_command banner, command
+			body = `#{command}`
+			if $? != 0
+				@jabber_messenger.call "#{banner} command #{command} failed with an error #{$?}:\n" + body
+			end
+			if body.strip != '' 
+				@jabber_messenger.call "#{banner} command #{command_str} succeeded with:\n" + body
+			end			
+		end
+
+		def set_jabber_messenger &block
+			@jabber_messenger = block
+		end
+	end
+end

data/lib/omnibot/mailchecker.rb

@@ -0,0 +1,96 @@
+module OmniBot
+
+	class MailChecker
+
+		include LoggedCommand
+
+		def on_first_timer
+			OmniLog::debug "Okay, it's near of midnight"
+			on_periodic_timer
+			@timer_provider.add_periodic_timer(3600) { on_periodic_timer }
+		end
+
+		def match_condition m, condition_name, mail_name = condition_name
+			values = m.send(mail_name)
+			[values].flatten.any? do |value| 
+				@conditions[condition_name] && Regexp.new(@conditions[condition_name]).match(value.to_s)
+			end
+		end
+
+		def match_conditions m
+			%w{ subject from to cc date}.any? { |condition| match_condition m, condition }
+		end
+
+		def on_periodic_timer
+			OmniLog::info "Checking mail #{@mail_config['address']}"
+			Mail.all.each do |m|
+				OmniLog::info "   look at message from #{m.from} about #{m.subject}"
+				handle_message(m) if match_conditions m
+			end
+		end
+
+		def handle_message m
+			OmniLog::info "Matched " + m.inspect.to_s
+			attached = m.attachments.find { |a| a.mime_type =~ /application\/x-zip.*/ }
+			if attached
+				Dir.mktmpdir('omniatt') do |tmpdir|
+					filename = tmpdir + '/' + attached.filename
+					OmniLog::info "Writing attachment to #{filename}"
+					File.open(filename,'w') { |f| f.write attached.read }
+					Dir.chdir(@unpack_to) do
+						system("unzip -oq '#{filename}'")
+						raise "Error extracting file #{filename} to #{@unpack_to}" if $? != 0
+					end
+					  
+					message_body = "Received an email '#{m.subject}' from '#{m.from.join(',')}' with "+
+							"an attachment #{attached.filename}. Successfully extracted an attachment to #{@unpack_to}."
+					@jabber_messenger.call message_body
+
+					jabber_logged_command 'Mail post-receive ', "#{@command_post} #{filename} #{@unpack_to}"
+				end
+			else
+				OmniLog::info "No attachment found" 
+			end
+		end
+		
+		def yaml_to_mailhash yaml_config
+		 { :address => yaml_config['host'],
+		 	 :port => yaml_config['port'],
+		 	 :user_name => yaml_config['user'],
+		 	 :password => yaml_config['password'],
+		 	 :enable_ssl => yaml_config['ssl']
+			}
+		end
+
+	public
+		attr_writer :timer_provider
+		attr_writer :startup_pause
+
+		def initialize mail_config, trigger_config 
+			@startup_pause = 0
+			@mail_config = mail_config
+			@conditions = trigger_config['if']
+			@unpack_to = trigger_config['unpack_to']
+			@command_post = trigger_config['command_post']
+			@send_to = trigger_config['send_to']
+
+			mailhash = yaml_to_mailhash(mail_config)
+			Mail.defaults do
+				retriever_method :pop3, mailhash
+			end
+
+			raise 'Wrong command' if (@command_post or '') == ''
+			raise 'No dir to extract to' unless File.directory? @unpack_to
+		end
+		
+		def to_s
+			"Mail checker for #{@send_to}"
+		end
+
+		def start
+			@timer_provider.add_timer(@startup_pause) { on_first_timer }
+		end
+	end
+
+end
+

data/lib/omnibot/periodiccommand.rb

@@ -3,6 +3,8 @@ module OmniBot
 	# Send to jabber user result of a daily command
 	class PeriodicCommand
 
+		include OmniBot::LoggedCommand
+
 		def on_first_timer
 			OmniLog::debug "Okay, it's near of midnight"
 			on_periodic_timer
@@ -11,35 +13,31 @@ module OmniBot
 
 		def on_periodic_timer
 			OmniLog::info "Reporting command #{@command}"
-			body = `#{@command}`
-			raise 'Error launching command ' if $? != 0
-			if body.strip != '' 
-			  message_body = "Results of daily executed command #{@command}:\n" + body
-			  @jabber_messenger.call message_body
-			end			
+			jabber_logged_command 'Periodic command', @command
 		end
 
 	public
 		attr_writer :timer_provider
+		attr_writer :startup_pause
 
-		def initialize command, pause
+		def initialize command 
 			@command = command
-			@pause = pause
+			@startup_pause = 0
+
+			raise 'Wrong command' if (@command or '') == ''
+		end
 
-			raise 'Wrong command' if (command == nil or command == '')
+		def to_s
+			"Periodic command '#{@command}'"
 		end
 
 		def start
 			now = Time.now
 			tomorrow = DateTime.now+1
 			next_report_time = Time.local(tomorrow.year, tomorrow.month, tomorrow.day, 1, 0, 0)
-			next_report_time = next_report_time + @pause
+			next_report_time = next_report_time + @startup_pause
 			@timer_provider.add_timer(next_report_time - now) { on_first_timer }
 		end
-
-		def set_jabber_messenger &block
-			@jabber_messenger = block
-		end
 	end
 
 end

data/lib/omnibot/version.rb

@@ -1,3 +1,3 @@
 module OmniBot
-	VERSION = '0.0.9'
+	VERSION = '0.0.10'
 end

metadata

@@ -1,109 +1,99 @@
---- !ruby/object:Gem::Specification 
+--- !ruby/object:Gem::Specification
 name: omnibot
-version: !ruby/object:Gem::Version 
-  prerelease: false
-  segments: 
-  - 0
-  - 0
-  - 9
-  version: 0.0.9
+version: !ruby/object:Gem::Version
+  version: 0.0.10
+  prerelease: 
 platform: ruby
-authors: 
+authors:
 - theirix
 autorequire: 
 bindir: bin
 cert_chain: []
-
-date: 2011-08-02 00:00:00 +04:00
-default_executable: 
-dependencies: 
-- !ruby/object:Gem::Dependency 
-  version_requirements: &id001 !ruby/object:Gem::Requirement 
-    requirements: 
-    - - ">="
-      - !ruby/object:Gem::Version 
-        segments: 
-        - 0
-        version: "0"
-  requirement: *id001
+date: 2011-10-28 00:00:00.000000000Z
+dependencies:
+- !ruby/object:Gem::Dependency
   name: xmpp4r
-  prerelease: false
+  requirement: &70197902207540 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
   type: :runtime
-- !ruby/object:Gem::Dependency 
-  version_requirements: &id002 !ruby/object:Gem::Requirement 
-    requirements: 
-    - - ">="
-      - !ruby/object:Gem::Version 
-        segments: 
-        - 0
-        version: "0"
-  requirement: *id002
-  name: eventmachine
   prerelease: false
+  version_requirements: *70197902207540
+- !ruby/object:Gem::Dependency
+  name: eventmachine
+  requirement: &70197902206760 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
   type: :runtime
-- !ruby/object:Gem::Dependency 
-  version_requirements: &id003 !ruby/object:Gem::Requirement 
-    requirements: 
-    - - ">="
-      - !ruby/object:Gem::Version 
-        segments: 
-        - 0
-        version: "0"
-  requirement: *id003
-  name: amqp
   prerelease: false
+  version_requirements: *70197902206760
+- !ruby/object:Gem::Dependency
+  name: amqp
+  requirement: &70197902206040 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
   type: :runtime
-description: Works with AMQP for sending messages at server side.Sends notifications to an user via XMPP.Can monitor system by performing daily commands.
+  prerelease: false
+  version_requirements: *70197902206040
+description: Works with AMQP for sending messages at server side.Sends notifications
+  to an user via XMPP.Can monitor system by performing daily commands.
 email: theirix@gmail.com
-executables: 
+executables:
 - omnibot
 - omnisend
 extensions: []
-
 extra_rdoc_files: []
-
-files: 
+files:
 - examples/config.yaml
 - lib/omnibot/amqpconsumer.rb
 - lib/omnibot/helpers.rb
 - lib/omnibot/jabberbot.rb
 - lib/omnibot/launcher.rb
+- lib/omnibot/loggedcommand.rb
+- lib/omnibot/mailchecker.rb
 - lib/omnibot/omnisend.rb
 - lib/omnibot/periodiccommand.rb
 - lib/omnibot/version.rb
 - lib/omnibot.rb
 - Rakefile
 - README.md
-has_rdoc: true
+- bin/omnibot
+- bin/omnisend
 homepage: http://github.com/theirix/omnibot
 licenses: []
-
 post_install_message: 
 rdoc_options: []
-
-require_paths: 
+require_paths:
 - lib
-required_ruby_version: !ruby/object:Gem::Requirement 
-  requirements: 
-  - - ">="
-    - !ruby/object:Gem::Version 
-      segments: 
-      - 1
-      - 8
-      version: "1.8"
-required_rubygems_version: !ruby/object:Gem::Requirement 
-  requirements: 
-  - - ">="
-    - !ruby/object:Gem::Version 
-      segments: 
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '1.8'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+      segments:
       - 0
-      version: "0"
-requirements: 
+      hash: -556400770793257517
+requirements:
 - AMQP-compatible server (for example, RabbitMQ)
 rubyforge_project: nowarning
-rubygems_version: 1.3.6
+rubygems_version: 1.8.6
 signing_key: 
 specification_version: 3
 summary: Simple XMPP bot for server monitoring
 test_files: []
-