#!/usr/bin/lua

json = require("json")
socket = require("socket")

function getConfig(configname)

  local status,uci = pcall(require,"uci")

  local config

  if status then

    if configname then
      config=configname
    else
      config="beacon"
    end

    local cur=uci.cursor()

    logging = cur.get(config,"logging","enabled") 

    mqtt_host = cur.get(config,"mqtt","host")
    mqtt_port = cur.get(config,"mqtt","port")
    mqtt_id = cur.get(config,"mqtt","id")
    mqtt_topic = cur.get(config,"mqtt","topic")
  
    mqtt_user = cur.get(config,"mqtt","user")
    mqtt_passwd = cur.get(config,"mqtt","password")

  else
  
    local status,ini = pcall(require,"ini")
    if not status then
      os.exit(1)
    end 

    if configname then
      config=configname
    else
      config="/etc/beacon.ini"
    end

    local cur=ini.parse_file(config)
    
    logging = cur["logging"]["enabled"] 

    mqtt_host = cur["mqtt"]["host"]
    mqtt_port = cur["mqtt"]["port"]
    mqtt_id = cur["mqtt"]["id"]
    mqtt_topic = cur["mqtt"]["topic"]
  
    mqtt_user = cur["mqtt"]["user"]
    mqtt_passwd = cur["mqtt"]["password"]
  
  end

  hostname = socket.dns.gethostname()
  if mqtt_host and not mqtt_id then
    socket = require("socket")
    posix = require("posix")
    pid = posix.getpid()
    mqtt_id="beaconmon-"..hostname.."-"..pid
  end

  if mqtt_host and not mqtt_port then
    mqtt_port = 1883
  end

  if mqtt_host and not mqtt_topic then
    mqtt_topic = 'beaconmon/{type}/{details}'
  end

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, ":", "-")
    str = string.gsub (str, "/", "-")
    str = string.gsub (str, " ", "_")
  end
  return str	
end

function printLog(str)
  if logging=="yes" then
    capture("logger -t beaconmon \""..str.."\"")
  else 
    print(str)  
  end 
end

function run_command(cmd)

  local file = assert(io.popen(cmd, 'r'))
  local output = file:read('*all')
  file:close()

end

function open_dump()

  f = assert(io.popen ("/usr/bin/btmon"))
  run_command("hciconfig hci0 down")
  run_command("hciconfig hci0 up")
  f_null = assert(io.popen ("hcitool lescan --duplicates"))

  return f

end

function dump(o)
  return json.encode(o)
end

function trim(s)
  return (s:gsub("^%s*(.-)%s*$", "%1"))
end

local function starts_with(str, start)
   return str:sub(1, #start) == start
end

function mqtt_pub(path,value)
  res=mqtt_client:publish(path,value,0,false)
  printLog("Pub "..path.." returned "..res);
  return res
end

function process_packet(packet)

  local mac
  local uuid
  local details
  local type
  local name

  packet['origin'] = hostname

  mac = packet['Address']
  uuid = packet['UUID']
  type = packet['Type']
  name = packet['Name (complete)']

  if type and starts_with(type,'iBeacon') then
    details=uuid
  elseif name then
    if not(type) then
      type="name"
    end  
    details=name
  else
    if not type then 
      type='unknown'
    end  
    details=mac
  end

  if not (type=="unknown") then
    mqtt_path=string.gsub(mqtt_topic,"{(.-)}",
      function (name) 
        if name=="type" then
          return mqtt_encode(type)
        elseif name=="details" then
          return mqtt_encode(details)
        else
          return '{'..name..'}'
        end      
      end)
        
    if not pcall(mqtt_pub,mqtt_path,dump(packet)) then
      printLog('Reconnecting MQTT...')
      mqtt_client:connect(mqtt_host,mqtt_port)
    end

  end

end

function starts(String,Start)
   return string.sub(String,1,string.len(Start))==Start
end

function split(inputstr, sep)
        if sep == nil then
                sep = "%s"
        end
        local t={} ; i=1
        for str in string.gmatch(inputstr, "([^"..sep.."]+)") do
                t[i] = str
                i = i + 1
        end
        return t
end

function read_loop()

  packet={}
  inbound=false

  while true do

    str=inp:read("*l")

    if str then 

      str=trim(str)

      if inbound then
        t = split(str,':')
        if #t>=2 then
          name=t[1]
          value=trim(table.concat(t,':',2))
          if name=="Address" then
            value=split(value)[1]
          end
          packet[name]=value
        elseif #t==1 and name then
          if not(packet[name..'.list']) then
            packet[name..'.list']={}
          end  
          table.insert(packet[name..'.list'],(trim(t[1])))
        end  
      end  

      if starts(str,'> HCI Event: LE Meta Event (0x3e)') then
        inbound=true
        name=nil
      elseif starts(str,'RSSI:') then
        inbound=false
        process_packet(packet)
        packet={}
      end
      
    end  
    
  end

end

io.stdout:setvbuf('no')
io.stdin:setvbuf('no') 

getConfig(arg[1])

if mqtt_host then
  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

inp = open_dump()

read_loop()