1 files changed, 126 insertions, 0 deletions
diff --git a/mode-watch.lua b/mode-watch.lua
new file mode 100644
index 0000000..26ef870
--- /dev/null
+++ b/mode-watch.lua
@@ -0,0 +1,126 @@
+--
+--  rhctl
+--
+--  Copyright (C) 2009 Christian Pointner <equinox@spreadspace.org>
+--
+--  This file is part of rhctl.
+--
+--  rhctl 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
+--  any later version.
+--
+--  rhctl 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 rhctl. If not, see <http://www.gnu.org/licenses/>.
+--
+
+socket = require("socket")
+
+function set_master_led()
+   os.execute("/sbin/led.sh set master")                             
+   os.execute("/sbin/led.sh clear standby") 
+end
+
+function set_standby_led() 
+   os.execute("/sbin/led.sh clear master")                         
+   os.execute("/sbin/led.sh set standby")                 
+end
+
+function clear_leds()
+   os.execute("/sbin/led.sh clear master")                      
+   os.execute("/sbin/led.sh clear standby")    
+end
+
+function send_mail(address, subject, bodytext)
+   local fp = assert(io.popen("/usr/sbin/mini_sendmail -smailrelay -fnoreply@helsinki.at " .. address, "w"))
+   
+   fp:write("Subject: " .. subject .. "\n")
+   fp:write("To: " .. address .. "\n")
+   fp:write("\n")
+   fp:write(bodytext)
+   fp:write("\n")
+   fp:close()
+end
+
+current_mode = "master"
+
+function process_cmd(message)
+   log.printf(log.DEBUG, "received message: '%s'", message)
+   
+   local new_mode = nil
+   local exps = { "Current Mode: (%a+)", "new Mode: (%a+)" }
+   for _, exp in ipairs(exps) do
+      new_mode = string.match(message, exp)
+      if(new_mode) then
+         new_mode = string.lower(new_mode)
+         if(new_mode == "master") then
+            set_master_led()
+            break
+         else
+            if(new_mode == "standby") then
+               set_standby_led()
+               break
+            end
+         end
+      end
+   end
+   
+   if(new_mode and new_mode ~= current_mode) then
+      log.printf(log.NOTICE, "mode is now " .. new_mode)
+      send_mail("logs@helsinki.at", "RHCTL mode changed to " .. new_mode, 
+                "RHCTL just switched from " .. current_mode .. " to " .. new_mode)
+      current_mode = new_mode
+   end
+   
+   return 0
+end
+
+function main_loop(opt) 
+   log.printf(log.NOTICE, "main_loop started")
+   local sig = signal.init()
+   local cmdfd = cmd.init()
+   
+   cmd.send_string("listen mode");
+   cmd.send_string("status");
+   
+   clear_leds()
+   
+   local return_value = 0
+   while return_value == 0 do
+      local readable, _, err = socket.select({ sig , cmdfd }, nil)
+      if(err) then
+         log.printf(log.ERROR, "select returned with error: %s", err)
+         return_value = -1
+      else
+         for _, input in ipairs(readable) do
+            if(input == sig) then
+               return_value = signal.handle()
+               if(return_value == 1) then break end
+            else 
+               if(input == cmdfd) then
+                  return_value = cmd.recv_data(process_cmd)
+                  if(return_value == 2) then
+                     log.printf(log.ERROR, "connection to server lost, exitting")
+                     return_value = -1
+                  end
+                  if(return_value ~= 0) then break end
+               else
+                  log.printf(log.ERROR, "select returned invalid handle??")
+                  return_value = -1
+                  break;
+               end
+            end
+         end
+      end
+   end
+   
+   clear_leds() 
+   
+   signal.stop()
+   return return_value
+end