#!/usr/bin/lua
--[[

	Part of AREDN -- Used for creating Amateur Radio Emergency Data Networks
	Copyright (C) 2016 Darryl Quinn
	See Contributors file for additional contributors

	This program 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 version 3 of the License.

	This program 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 this program.  If not, see <http://www.gnu.org/licenses/>.

	Additional Terms:

	Additional use restrictions exist on the AREDN(TM) trademark and logo.
		See AREDNLicense.txt for more info.

	Attributions to the AREDN Project must be retained in the source code.
	If importing this code into a new or existing project attribution
	to the AREDN project must be added to the source code.

	You must not misrepresent the origin of the material contained within.

	Modified versions must be modified to attribute to the original source
	and be marked in reasonable ways as differentiate it from the original
	version

--]]

require("aredn.utils")
require("uci")
require("aredn.uci")
local nxo=require("nixio")
require("iwinfo")
require("luci.sys")


-- setup extensions
os.capture=capture
string.print=print_r


-- delay just after rssi_monitor has a chance to run noise floor calibration
sleep(5)
local MAXLINES=2880 -- 2 days worth
local AGETIME=43200
local INACTIVETIMEOUT=10000
local tmpdir="/tmp/snrlog"
local lastdat="/tmp/snr.dat"
local lasttime={}
local lines={}
local arpcache={}
local nscache={}
local defnoise=-95
local tmpdirlist={}
local neighbors={}
local stations={}
local wifiiface=""
local bandwidth=""
local nulledout={}

-- Neighbor Class
Neighbor={}
Neighbor.__index=Neighbor
function Neighbor.create(macaddress)
	local n={}
	setmetatable(n,Neighbor)
	n.mac=macaddress:lower()
	n.hostname=nil
	n.sigfile=nil
	n.datafile=nil
	n.ip=nil
	n.lastseen=nil
	n.level=nil

	n.ip=n:findIp()
	n.hostname=n:findHostname()
	n.signal=n:findSignal()
	n.noise=n:findNoise()
	n.tx_mcs=n:findTxMcs()
	n.tx_rate=n:findTxRate()
	n.rx_mcs=n:findRxMcs()
	n.rx_rate=n:findRxRate()
	n.lastseen=n:findLastTime()
	n.datafile=n:generateDataFileName()
	return n
end

function Neighbor:hasIP()
	return not (self.ip==nil)
end

function Neighbor:hasHostName()
	return not (self.hostname==nil)
end

function Neighbor:getExistingDataFileName()
	local efn=nil
	for fn in nxo.fs.glob(tmpdir.."/"..self.mac.."-*") do
		efn=nxo.fs.basename(fn)
	end
	return efn
end

function Neighbor:generateDataFileName()
	local tmpfn=nil
	local efn=self:getExistingDataFileName()
	if efn~=nil then
		-- existing file found, improve or use
		tmpfn=self.mac.."-"
		m,x=string.match(efn,"^([%x2:]*)%-(.*)")
		if x==nil or x=="" then	-- "00:11:22:33:44:55-"
			if self:hasHostName() then
				tmpfn=tmpfn..self.hostname
				nxo.fs.rename(tmpdir.."/"..efn,tmpdir.."/"..tmpfn)
			elseif self:hasIP() then
				tmpfn=tmpfn..self.ip
				nxo.fs.rename(tmpdir.."/"..efn,tmpdir.."/"..tmpfn)
			end
		elseif get_ip_type(x)==1 then -- "00:11:22:33:44:55-10.11.22.33"
			if self:hasHostName() then
				tmpfn=tmpfn..self.hostname
				nxo.fs.rename(tmpdir.."/"..efn,tmpdir.."/"..tmpfn)
			else
				tmpfn=efn
			end
		else 	-- "00:11:22:33:44:55-MAYBE-GOOD-HOST"
			if x==self.hostname then	-- "00:11:22:33:44:55-SAME-HOST"
				tmpfn=efn
			else	-- "00:11:22:33:44:55-NEW-HOST-NAME"
				if self.hostname~=nil then
					tmpfn=tmpfn..self.hostname
					nxo.fs.rename(tmpdir.."/"..efn,tmpdir.."/"..tmpfn)
				else
					tmpfn=efn
				end
			end
		end
	else
		-- no prior file, let's generate one
		tmpfn=self.mac.."-"
		if self:hasHostName() then
			tmpfn=tmpfn..self.hostname
		elseif self:hasIP() then
			tmpfn=tmpfn..self.ip
		end
	end
	return tmpfn
end

function Neighbor:findLastTime()
	self.lastseen=lasttime[mac]
	return self.lastseen
end

function Neighbor:findHostname()
	if self:hasIP() then
		self.hostname=nslookup(self.ip)
	end
	return self.hostname
end

function Neighbor:findIp()
	-- lookup IP from arpcache
	local myarp=arpcache[self.mac]
	if myarp~=nil then
		self.ip=myarp["IP address"]
	end
	return self.ip
end

function Neighbor:findSignal()
	-- lookup from iwinfo
	self.signal=stations[self.mac:upper()].signal
	return self.signal
end

function Neighbor:findNoise()
	-- lookup from iwinfo
	self.noise=stations[self.mac:upper()].noise
	return self.noise
end

function Neighbor:findTxMcs()
	self.tx_mcs=stations[self.mac:upper()].tx_mcs
	return self.tx_mcs
end

function Neighbor:findTxRate()
	local r=(stations[self.mac:upper()].tx_rate)/1000
	self.tx_rate=adjust_rate(r,bandwidth)
	return self.tx_rate
end

function Neighbor:findRxMcs()
	self.rx_mcs=stations[self.mac:upper()].rx_mcs
	return self.rx_mcs
end

function Neighbor:findRxRate()
	local r=(stations[self.mac:upper()].rx_rate)/1000
	self.rx_rate=adjust_rate(r,bandwidth)
	return self.rx_rate
end

function Neighbor:log()
	local oktolog=true
	if self.lastseen ~= nil then
		if stations[self.mac:upper()].inactive >= INACTIVETIMEOUT then
			-- beacons expired
			self.signal="null"
			if nulledout[self.mac] == "true" then
				-- No need to double log inactive null's
				oktolog=false
			end
		end
	end

	if self.signal==0 then
		oktolog=false
		if nulledout[self.mac] == nil then
			-- First time we have seen this show up
			-- but it is at 0 wont be logged but will
			-- end up in snrcache
			nulledout[self.mac]="true"
		end
	end

	if oktolog then
		nulledout[self.mac]="false"
		-- log neighbor data to datafile
		local f, err=assert(io.open(tmpdir.."/"..self.datafile, "a"),"Cannot open file ("..tmpdir.."/"..self.datafile..") for appending!")
		if f~=nil then
			local now=os.date("%m/%d %H:%M:%S",os.time())
			local outline=string.format("%s,%s,%s,%s,%s,%s,%s\n",now,self.signal,self.noise,self.tx_mcs,self.tx_rate,self.rx_mcs,self.rx_rate)
			f:write(outline)
			f:close()
			if self.signal == "null" then
				nulledout[self.mac]="true"
			end
		else
			print(err)
		end
	end
	return oktolog
end
-- Neighbor Class END

-- MAIN() -------------------------------------------------------------------------------------
-- get wifi interface name
wifiiface=get_ifname("wifi")

-- load the lasttime table
if file_exists(lastdat) then
	local f,err=io.open(lastdat,"r")
	if f~=nil then
		for line in f:lines() do
			mac,last,nulled=string.match(line, "(.*)|(.*)|(.*)")
			lasttime[mac]=last
			nulledout[mac]=nulled
		end
	end
	f:close()
end

-- get radio noise floor
local nf=iwinfo["nl80211"].noise(wifiiface)
if (nf < -101 or nf > -50) then
	nf=defnoise
end

-- create tmp dir if needed
if not dir_exists(tmpdir) then
	nxo.fs.mkdir(tmpdir)
end

-- get system uptime
--now=os.time()
now=luci.sys.uptime()

-- get all stations
stations=iwinfo["nl80211"].assoclist(wifiiface)


-- load up arpcache
luci.sys.net.arptable(
	function(a)
		arpcache[a["HW address"]:lower()]=a
	end
)

-- get the current bandwidth setting
bandwidth=get_bandwidth()

-- iterate over all the stations
for mstation in pairs(stations) do
	local n=Neighbor.create(mstation)
	table.insert(neighbors,n)
end

-- get all the existing files in tmpdir
local tmpdirlist={}
for maclist in nxo.fs.dir(tmpdir) do
	maclistbase=nxo.fs.basename(maclist)
	table.insert(tmpdirlist,maclistbase)
end

-- neighbors loop START
for _,n in pairs(neighbors) do
	-- trim datafile
	file_trim(tmpdir.."/"..n.datafile,MAXLINES)
	if n:log()==true then
		lasttime[n.mac]=now
	end
end
-- neighbors loop END

-- update snr.dat
local snrdatcache={}
if not file_exists(lastdat) then
	-- create file
	local f,err=assert(io.open(lastdat,"w+"),"Cannot create "..lastdat)
	f:close()
end
for line in io.lines(lastdat) do
	local mac,lastt,nulledoutprev=string.match(line,"(.*)|(.*)|(.*)")
	if ((now-lastt) < AGETIME) then
		-- keep it
		snrdatcache[mac]=lastt

		-- check if in neighbors table
		local foundneighbor=false
		for _,n in pairs(neighbors) do
			if n.mac == mac then
				foundneighbor=true
			end
		end

		-- If wasn't previously nulled out and we don't have it in the neigbors table write a null
		if nulledoutprev[mac] == "false" and foundneigbor == false then

			-- find the log file name
			for maclist in nxo.fs.glob(tmpdir.."/"..mac.."*") do
				logdatafile=macclist
			end

			-- Write a null to the log file
			local f, err=assert(io.open(logdatafile, "a"),"Cannot open file ("..logdatafile..") for appending!")
			if f~=nil then
				local now=os.date("%m/%d %H:%M:%S",os.time())
				local outline=string.format("%s,%s,%s,%s,%s,%s,%s\n",now,'null',nf,'0','0','0','0')
				f:write(outline)
				f:close()
				nulledout[mac]="true"
			else
				-- Don't log the null into SNRLog cause we were not successful
				-- Though the assert() above should cause this too.
				nulledout[mac]="false"
			end

		end

	else
		-- find the file and purge it
		for maclist in nxo.fs.glob(tmpdir.."/"..mac.."*") do
			nxo.fs.remove(maclist)
		end
	end
end

for _,n in pairs(neighbors) do
		snrdatcache[n.mac]=now
end

-- re-write snr.dat file
local f,err=assert(io.open(lastdat,"w+"),"Cannot overwrite "..lastdat)
for k,v in pairs(snrdatcache) do
	f:write(string.format("%s|%s|%s\n",k,v,nulledout[k]))
end
f:close()

-- END MAIN