improve error handling, add set to monitor to initializer

Author: boveus

app/services/set_device_to_monitor_service.rb

@@ -1,6 +1,9 @@
 class SetDeviceToMonitorService
     def self.set(interface=nil)
-    	interface ||= YAML.load(File.read('./config/config.yml'))['device']
+    	config_file = YAML.load(File.read('./config/config.yml'))
+    	return false if !config_file
+    	interface = config_file['device']
+    	return false if !interface
     	set_to_monitor(interface)
     	check_device(interface)
 	end
@@ -13,12 +16,8 @@ def self.check_device(interface)
 	end
 
 	def self.set_to_monitor(interface)
-	  unless interface.empty?
-	  	`sudo ip link set #{interface} down`
-	    `sudo iwconfig #{interface} mode monitor`
-	    `sudo ip link set #{interface} up`
-	  else
-	    "ERROR: No device specified! \n"
-	  end
+  	  `sudo ip link set #{interface} down`
+      `sudo iwconfig #{interface} mode monitor`
+      `sudo ip link set #{interface} up`
 	end
 end

config/environments/production.rb

@@ -73,6 +73,10 @@
   # Use a different logger for distributed setups.
   # require 'syslog/logger'
   # config.logger = ActiveSupport::TaggedLogging.new(Syslog::Logger.new 'app-name')
+  config.after_initialize do
+    Rails.application.load_tasks 
+    Rake::Task['set_to_monitor'].invoke
+  end
 
   if ENV["RAILS_LOG_TO_STDOUT"].present?
     logger           = ActiveSupport::Logger.new(STDOUT)

lib/tasks/set_to_monitor.rake

@@ -4,5 +4,7 @@ task :set_to_monitor do
   specified_interface = ENV["interface"]
   if SetDeviceToMonitorService.set(specified_interface)
     printf "Success\n"
+  else
+  	printf "Failure!  Please ensure you ran 'rake set_interface'\n"
   end
 end