Больше не поддерживается вариант на python.

[weathermon.git] / weathermon.lua

diff --git a/weathermon.lua b/weathermon.lua
index 3e940cd8b89baa0962c1cde14284305fa894a8f2..3208e304e0d2643f07021edca2b9b9cd7f8c1e8f 100755 (executable)
--- a/weathermon.lua
+++ b/weathermon.lua
@@ -1,10 +1,55 @@
 #!/usr/bin/lua
 
 #!/usr/bin/lua
 

-function getConfig()
+json = require("json")
+socket = require("socket")
+
+function startswith(String,Start)
+   if String then
+     return string.sub(String,1,string.len(Start))==Start
+   else
+     return False
+   end    
+end
+
+function url_encode(str)
+  if (str) then
+    str = string.gsub (str, "\n", "\r\n")
+    str = string.gsub (str, "([^%w %-%_%.%~])",
+        function (c) return string.format ("%%%02X", string.byte(c)) end)
+    str = string.gsub (str, " ", "+")
+  end
+  return str   
+end
+
+function capture(cmd, raw)
+  local f = assert(io.popen(cmd, 'r'))
+  local s = assert(f:read('*a'))
+  f:close()
+  if raw then return s end
+  s = string.gsub(s, '^%s+', '')
+  s = string.gsub(s, '%s+$', '')
+  s = string.gsub(s, '[\n\r]+', ' ')
+  return s
+end
+
+function mqtt_encode(str)
+  if (str) then
+    str = string.gsub (str, "\n", "")
+    str = string.gsub (str, "/", "-")
+  end
+  return str   
+end
+
+function getConfig(configname)

 
   local uci=require("uci")
   local cur=uci.cursor()
 
   local uci=require("uci")
   local cur=uci.cursor()

-  local config="weathermon"
+  local config
+  if configname then
+    config=configname
+  else
+    config="weathermon"
+  end

 
   web_url  = cur.get(config,"web","url")
   web_user = cur.get(config,"web","user")
 
   web_url  = cur.get(config,"web","url")
   web_user = cur.get(config,"web","user")


@@ -13,6 +58,14 @@ function getConfig()
 
   web_iface = cur.get(config,"web","iface")
 
 
   web_iface = cur.get(config,"web","iface")
 

+  if web_iface then
+  
+    command = '/sbin/ifconfig '..web_iface..' | grep \'\\<inet\\>\' | sed -n \'1p\' | tr -s \' \' | cut -d \' \' -f3 | cut -d \':\' -f2'
+    f=io.popen(command,'r')
+    ip_addr=f:read()
+  
+  end
+

   if not web_devid then
   
     if web_iface then
   if not web_devid then
   
     if web_iface then


@@ -30,14 +83,19 @@ function getConfig()
   end
 
   logging = cur.get(config,"logging","enabled") 
   end
 
   logging = cur.get(config,"logging","enabled") 

+  touch_file = cur.get(config,"logging","touch_file") 

 
   serial_port = cur.get(config,"serial","port")
   serial_baud = cur.get(config,"serial","baud")
 
 
   serial_port = cur.get(config,"serial","port")
   serial_baud = cur.get(config,"serial","baud")
 

+  input_file = cur.get(config,"input","file")
+  input_exec = cur.get(config,"input","exec")
+  alarm_exec = cur.get(config,"alarm","exec")
+

   if serial_port then
 
     command = "stty -F  "..serial_port.." "..serial_baud
   if serial_port then
 
     command = "stty -F  "..serial_port.." "..serial_baud

-    os.execute(command)
+    capture(command)

 
   end
 
 
   end
 


@@ -45,6 +103,10 @@ function getConfig()
   mqtt_port = cur.get(config,"mqtt","port")
   mqtt_id = cur.get(config,"mqtt","id")
   mqtt_topic = cur.get(config,"mqtt","topic")
   mqtt_port = cur.get(config,"mqtt","port")
   mqtt_id = cur.get(config,"mqtt","id")
   mqtt_topic = cur.get(config,"mqtt","topic")

+  mqtt_alarm_topic = cur.get(config,"mqtt","alarm_topic")
+
+  mqtt_user = cur.get(config,"mqtt","user")
+  mqtt_passwd = cur.get(config,"mqtt","password")

 
   if mqtt_host and not mqtt_id then
     mqtt_id="weather-"..web_devid
 
   if mqtt_host and not mqtt_id then
     mqtt_id="weather-"..web_devid


@@ -55,12 +117,21 @@ function getConfig()
   end
 
   if mqtt_host and not mqtt_topic then
   end
 
   if mqtt_host and not mqtt_topic then

-    mqtt_topic = 'weathermon/'..web_devid
+    mqtt_topic = 'weathermon/{dev}/{type}/{id}/{param}'
+  end
+
+  if mqtt_host and not mqtt_alarm_topic then
+    mqtt_alarm_topic = 'alarm/{dev}/{type}/{id}'

   end
 
 end
 
   end
 
 end
 

-require "socket"
+function touch()
+  if touch_file then
+    local file = io.open(touch_file, 'w')
+    file:close()
+  end  
+end

 
 function sleep(sec)
   socket.select(nil, nil, sec)
 
 function sleep(sec)
   socket.select(nil, nil, sec)


@@ -71,7 +142,6 @@ function splitStr(str,char)
   local res = {}
   local idx = 1
 
   local res = {}
   local idx = 1
 

-

   while str:len()>0 do
     pos = str:find(char); 
     if pos == nil then
   while str:len()>0 do
     pos = str:find(char); 
     if pos == nil then


@@ -89,39 +159,98 @@ function splitStr(str,char)
 end
 
 function printLog(str)
 end
 
 function printLog(str)

-  print(str)

   if logging=="on" then
   if logging=="on" then

-    os.execute("logger -t weathermon "..str)
+    capture("logger -t weathermon "..str)
+    print(str)  
+  elseif logging=="syslog" then
+    capture("logger -t weathermon "..str)
+  elseif logging=="stdout" then 
+    print(str)  

   end 
 end
 
 function submitValue(type,id,param,val)
 
   end 
 end
 
 function submitValue(type,id,param,val)
 

-  url = web_url.."?stype="..type.."&sid="..id.."&param="..param.."&value="..val
+  url = web_url.."?stype="..url_encode(type).."&sid="..url_encode(id).."&param="..url_encode(param).."&value="..url_encode(val)

 
   command = "curl"
 
   if web_iface then
 
   command = "curl"
 
   if web_iface then

-    command = command.." --interface "..web_iface
+    command = command.." --interface "..ip_addr

   end
 
   if web_user then
     command = command.." -u "..web_user..":"..web_pass
   end
 
   end
 
   if web_user then
     command = command.." -u "..web_user..":"..web_pass
   end
 

-  command = command.." \""..url.."\""
+  command = command.." \""..url.."\" 2>&1"
+
+  result = capture(command)
+
+  touch()
+
+end
+
+function processJson(str)
+
+  msg=json.decode(str)
+
+  sensor={}
+
+  for key,value in pairs(msg) do
+    if value then
+      if key=="model" or key=="device" then
+        sensor_type=value
+      elseif key=="id" then
+        sensor_id=value
+      elseif key=='time' then
+        sensor_time=value
+      else
+        sensor[key]=value
+      end
+    end
+  end
+
+  if not sensor_id then
+    sensor_id = web_devid
+  end

 
 

-  print(command)
-  os.execute(command)
-  print("")
+  if not (sensor_type==nil or sensor_id==nil or sensor_type=='' or sensor_id=='') then
+    if next(sensor)==nil then
+      sensor["command"]="alarm"
+    end
+    for k,v in pairs(sensor) do
+      printLog("Type = "..sensor_type..", ID = "..sensor_id..", Param = "..k..", Value = \""..v.."\"")
+      submitValue(sensor_type,sensor_id,k,v)
+      if mqtt_client then
+        mqtt_path=string.gsub(mqtt_topic,"{(.-)}", 
+          function (name) 
+            if name=="dev" then
+              return mqtt_encode(web_devid)
+            elseif name=="type" then
+              return mqtt_encode(sensor_type)
+            elseif name=="id" then
+              return mqtt_encode(sensor_id)
+            elseif name=="param" then
+              return k
+            else
+              return '{'..name..'}'
+            end      
+          end)
+        mqtt_client:publish(mqtt_path,v)
+      end  
+    end
+  else
+    printLog("Cannot parse sensor input: "..str)
+  end

 
 end
 
 function processLine(str)
 
   msg=splitStr(line,':')
 
 end
 
 function processLine(str)
 
   msg=splitStr(line,':')

-  msg_type=msg[1] or nil
-  msg_body=msg[2] or nil
+  msg_type=msg[1] or ''
+  msg_body=msg[2] or ''

   if msg_type=="STATUS" then
     printLog("Status: "..msg_body)
   elseif msg_type=="ERROR" then
   if msg_type=="STATUS" then
     printLog("Status: "..msg_body)
   elseif msg_type=="ERROR" then


@@ -135,8 +264,8 @@ function processLine(str)
     sensor_id = web_devid
     for i,rec in ipairs(sens) do
       recrd=splitStr(rec,'=')
     sensor_id = web_devid
     for i,rec in ipairs(sens) do
       recrd=splitStr(rec,'=')

-      key=recrd[1] or nil
-      value=recrd[2] or nil
+      key=recrd[1] or ''
+      value=recrd[2] or ''

       if value then
         if key=="TYPE" then
           sensor_type=value
       if value then
         if key=="TYPE" then
           sensor_type=value


@@ -147,33 +276,113 @@ function processLine(str)
         end
       end
     end
         end
       end
     end

+    if not (sensor_type==nil or sensor_id==nil or sensor_type=='' or sensor_id=='') then
+      for k,v in pairs(sensor) do
+        printLog("Type = "..sensor_type..", ID = "..sensor_id..", Param = "..k..", Value = "..v)
+        submitValue(sensor_type,sensor_id,k,v)
+        if mqtt_client then
+          mqtt_path=string.gsub(mqtt_topic,"{(.-)}", 
+            function (name) 
+              if name=="dev" then
+                return web_devid
+              elseif name=="type" then
+                return sensor_type
+              elseif name=="id" then
+                return sensor_id
+              elseif name=="param" then
+                return k
+              else
+                return '{'..name..'}'
+              end      
+            end)
+          mqtt_client:publish(mqtt_path,v)
+        end  
+      end
+    else
+      printLog("Cannot parse sensor input: "..msg_body)
+    end
+  elseif msg_type=="ALARM" then
+    printLog("ALARM: "..msg_body)  
+    sens = splitStr(msg_body,",")
+    sensor = {}
+    idx = 1
+    sensor_type = nil
+    sensor_id = web_devid

     mqtt_param = {}
     mqtt_param = {}

-    for k,v in pairs(sensor) do
-      printLog("Type = "..sensor_type..", ID = "..sensor_id..", Param = "..k..", Value = "..v)
-      submitValue(sensor_type,sensor_id,k,v)
-      mqtt_param[k]=v
+    for i,rec in ipairs(sens) do
+      recrd=splitStr(rec,'=')
+      key=recrd[1] or ''
+      value=recrd[2] or ''
+      if value then
+        if key=="TYPE" then
+          alarm_type=value
+        elseif key=="ID" then
+          alarm_id=value
+        end
+      end

     end
     end

-    mqtt_msg = { type=sensor_type, data=mqtt_param }
-    if mqtt_client then
-      serializedString = json.encode(mqtt_msg)
-      mqtt_client:publish(mqtt_topic,serializedString)
+    if not (alarm_type==nil or alarm_id==nil or alarm_type=='' or alarm_id=='') then
+      if mqtt_client then
+        mqtt_path=string.gsub(mqtt_alarm_topic,"{(.-)}", 
+          function (name) 
+            if name=="dev" then
+              return web_devid
+            elseif name=="type" then
+              return sensor_type
+            elseif name=="id" then
+              return sensor_id
+            else
+              return '{'..name..'}'
+            end      
+          end)
+        mqtt_client:publish(mqtt_path,msg_body)
+      end
+      if alarm_exec then
+        command=alarm_exec..
+          " \""..string.gsub(alarm_type,"\"","\\\"")..
+          "\" \""..string.gsub(alarm_id,"\"","\\\"")..
+          "\" \""..string.gsub(msg_body,"\"","\\\"").."\""
+        capture(command)
+      end
+    else
+      printLog("Cannot parse alarm input: "..msg_body)

     end
   end
 
 end
 
     end
   end
 
 end
 

-getConfig()
+getConfig(arg[1])

 
 if mqtt_host then
 
 if mqtt_host then

-  MQTT = require "paho.mqtt"
-  mqtt_client = MQTT.client.create(mqtt_host, mqtt_port)
-  mqtt_client:connect(mqtt_id)
-  json = require( "json" )
+  MQTT = require "mosquitto"
+  mqtt_client = MQTT.new(mqtt_id)
+  if mqtt_user then
+    mqtt_client:login_set(mqtt_user, mqtt_passwd)
+  end
+  mqtt_client:connect(mqtt_host,mqtt_port)

 end
 
 end
 

-serialin=io.open(serial_port,"r")
+if serial_port then
+  serialin=io.open(serial_port,"r")
+elseif input_file == "-" then
+  serialin=io.stdin;
+elseif input_file then
+  serialin=io.open(input_file,"r")
+elseif input_exec then
+  serialin=io.popen(input_exec,"r")
+else
+  printLog("No input selected")
+  return
+end  

 while 1 do
 while 1 do

-  line=serialin:read()
-  print(line)
-  processLine(line)
+  line=serialin:read("*l")
+  if line == nil then
+    break
+  end
+  printLog("Received: "..line);
+  if startswith(line,'{') then
+    processJson(line)
+  else
+   processLine(line)
+  end

 end
 end