MQTT timeout/disconnect processing.
[openhab-process.git] / mqtt-bt / scan-beacons
1 #!/usr/bin/lua
2
3 function getConfig(configname)
4
5   local uci=require("uci")
6   local cur=uci.cursor()
7   local config
8   if configname then
9     config=configname
10   else
11     config="beacon"
12   end
13
14   logging = cur.get(config,"logging","enabled") 
15
16   mqtt_host = cur.get(config,"mqtt","host")
17   mqtt_port = cur.get(config,"mqtt","port")
18   mqtt_id = cur.get(config,"mqtt","id")
19   mqtt_topic = cur.get(config,"mqtt","topic")
20
21   mqtt_user = cur.get(config,"mqtt","user")
22   mqtt_passwd = cur.get(config,"mqtt","password")
23
24   if mqtt_host and not mqtt_id then
25     mqtt_id="beaconmon"
26   end
27
28   if mqtt_host and not mqtt_port then
29     mqtt_port = 1883
30   end
31
32   if mqtt_host and not mqtt_topic then
33     mqtt_topic = 'beaconmon/{type}/{details}'
34   end
35
36 end
37
38 function capture(cmd, raw)
39   local f = assert(io.popen(cmd, 'r'))
40   local s = assert(f:read('*a'))
41   f:close()
42   if raw then return s end
43   s = string.gsub(s, '^%s+', '')
44   s = string.gsub(s, '%s+$', '')
45   s = string.gsub(s, '[\n\r]+', ' ')
46   return s
47 end
48
49 function mqtt_encode(str)
50   if (str) then
51     str = string.gsub (str, "\n", "")
52     str = string.gsub (str, "/", "-")
53   end
54   return str    
55 end
56
57 function printLog(str)
58   if logging=="on" then
59     capture("logger -t beaconmon "..str)
60   else 
61     print(str)  
62   end 
63 end
64
65 function run_command(cmd)
66
67   local file = assert(io.popen(cmd, 'r'))
68   local output = file:read('*all')
69   file:close()
70
71 end
72
73 function open_dump()
74
75   f = assert(io.popen ("hcidump --raw"))
76   run_command("kill `pgrep hcitool`")
77   run_command("hciconfig hci0 down")
78   run_command("hciconfig hci0 up")
79   f_null = assert(io.popen ("hcitool lescan --duplicates"))
80
81   return f
82
83 end
84
85 function dump(o)
86    if type(o) == 'table' then
87       local s = '{ '
88       for k,v in pairs(o) do
89          if type(k) ~= 'number' then k = '"'..k..'"' end
90 --         s = s .. '['..k..'] = ' .. dump(v) .. ','
91          s = s .. dump(v) .. ','
92       end
93       return s .. '} '
94    else
95       return tostring(o)
96    end
97 end
98
99 function trim(s)
100   return (s:gsub("^%s*(.-)%s*$", "%1"))
101 end
102
103 function process_packet(packet)
104
105   local bytes={}
106   local idx=1
107   local len
108   local mac
109   local flags
110   local power
111   local tx
112   local type
113   local paysublen
114   local uuid
115   local major
116   local minor
117   local details
118
119   if packet:len()>1 then
120
121     while packet:len()>2 do
122       bytes[idx]=trim(packet:sub(1,3))
123       idx=idx+1
124       packet=packet:sub(4)
125     end
126     len = idx-1
127
128     if bytes[1]=='04' and bytes[2]=='3E' then
129       -- BLE Beacon?
130       type=""
131       mac=bytes[13]..':'..bytes[12]..':'..bytes[11]..':'..bytes[10]..':'..bytes[9]..':'..bytes[8]
132       flags=bytes[14]
133       power=tonumber("0x"..bytes[len-1])-256
134       tx=tonumber("0x"..bytes[len])-256
135       local j = 15
136       while j<len-2 do
137         paysublen=tonumber('0x'..bytes[j])
138         if bytes[j+1]=="FF" and bytes[j+2]=="4C" and bytes[j+3]=="00" and bytes[j+4]=="02" and bytes[j+5]=="15" then
139           -- Standard UUID iBeacon
140           type="ibeacon"
141           local uuid1=bytes[j+6]..bytes[j+7]..bytes[j+8]..bytes[j+9]
142           local uuid2=bytes[j+10]..bytes[j+11]
143           local uuid3=bytes[j+12]..bytes[j+13]
144           local uuid4=bytes[j+14]..bytes[j+15]
145           local uuid5=bytes[j+16]..bytes[j+17]..bytes[j+18]..bytes[j+19]..bytes[j+20]..bytes[j+21]
146           uuid=string.lower(uuid1..'-'..uuid2..'-'..uuid3..'-'..uuid4..'-'..uuid5)
147           major=bytes[j+23]..bytes[j+22]
148           minor=bytes[j+25]..bytes[j+24]
149         end
150         j=j+1+paysublen
151       end
152       if type=="ibeacon" then
153         printLog(string.format("{type:'ibeacon',mac:'%s',uuid:'%s',major:'%s',minor:'%s',power:%d,tx:%d}",mac,uuid,major,minor,power,tx))
154         details=uuid..'/'..major..'/'..minor
155       else
156         type='unknown'
157         details=dump(bytes)
158         printLog(details)
159       end
160       if mqtt_client then
161         if not mqtt_client.connected then
162           mqtt_client:connect(mqtt_id)
163         end
164         mqtt_path=string.gsub(mqtt_topic,"{(.-)}", 
165           function (name) 
166             if name=="type" then
167               return mqtt_encode(type)
168             elseif name=="details" then
169               return mqtt_encode(details)
170             else
171               return '{'..name..'}'
172             end      
173           end)
174         mqtt_client:publish(mqtt_path,tx)
175       end        
176     end
177   end
178
179 end
180
181 function read_loop()
182
183   packet = ""
184
185   for line in inp:lines() do
186
187     lchr=line:sub(1,1)
188     line=trim(line)
189
190     if lchr=="<" or lchr==">" then
191       line = trim(line:sub(2))
192     end
193
194     if not (lchr == " ") then
195       process_packet(packet)
196       packet = ""
197     end
198
199     llen=line:len()
200
201     if llen<59 then
202       packet = packet .. " " .. line
203       process_packet(packet)
204       packet=""
205     else
206       packet = packet .. " " .. line
207     end    
208     
209   end
210
211 end
212
213 getConfig(arg[1])
214
215 if mqtt_host then
216   MQTT = require "paho.mqtt"
217   mqtt_client = MQTT.client.create(mqtt_host, mqtt_port)
218   if mqtt_user then
219     mqtt_client:auth(mqtt_user, mqtt_passwd)
220   end
221 end
222
223 inp = open_dump()
224 read_loop()