toribio-go.lua

--- Toribio application.
-- This application starts the different tasks and device loaders.
-- It is controlled trough a configuration file (default toribio-go.conf).
-- @usage	lua toribio-go.lua [-h] [-d] [-c conffile|'none'] 
--		-h Print help
--		-c Use given configuration file (or none). 
--		   Defaults to 'toribio-go.conf'
--		-d NONE|ERROR|WARNING|INFO|DETAIL|DEBUG|ALL
-- @script toribio-go

package.path = package.path .. ";;;../lumen/?.lua;?/init.lua;../?.lua;../?/init.lua"

--require 'strict'

local sched = require 'lumen.sched'
local log = require 'lumen.log'
require "lumen.tasks.selector".init({service='nixio'})
--print ('>>>>>>>>>>>', )

local toribio = require 'toribio'

-- From http://lua-users.org/wiki/AlternativeGetOpt
-- getopt_alt.lua
-- getopt, POSIX style command line argument parser
-- param arg contains the command line arguments in a standard table.
-- param options is a string with the letters that expect string values.
-- returns a table where associated keys are true, nil, or a string value.
-- The following example styles are supported
--   -a one  ==> opts["a"]=="one"
--   -bone   ==> opts["b"]=="one"
--   -c      ==> opts["c"]==true
--   --c=one ==> opts["c"]=="one"
--   -cdaone ==> opts["c"]==true opts["d"]==true opts["a"]=="one"
-- note POSIX demands the parser ends at the first non option
--      this behavior isn't implemented.
local function getopt( arg, options )
  local tab = {}
  for k, v in ipairs(arg) do
    if string.sub( v, 1, 2) == "--" then
      local x = string.find( v, "=", 1, true )
      if x then tab[ string.sub( v, 3, x-1 ) ] = string.sub( v, x+1 )
      else tab[ string.sub( v, 3 ) ] = true
      end
    elseif string.sub( v, 1, 1 ) == "-" then
      local y = 2
      local l = string.len(v)
      local jopt
      while ( y <= l ) do
        jopt = string.sub( v, y, y )
        if string.find( options, jopt, 1, true ) then
          if y < l then
            tab[ jopt ] = string.sub( v, y+1 )
            y = l
          else
            tab[ jopt ] = arg[ k + 1 ]
          end
        else
          tab[ jopt ] = true
        end
        y = y + 1
      end
    end
  end
  return tab
end
-- Test code
--opts = getopt( arg, "ab" )
--for k, v in pairs(opts) do
--  print( k, v )
--end
-- End of: From http://lua-users.org/wiki/AlternativeGetOpt

local opts = getopt( _G.arg, "cd" )

--watches for task die events and prints out
sched.sigrun({sched.EVENT_DIE}, function(task, msg)
	log ('TORIBIOGO', 'DETAIL', 'Task %s died: %s', tostring(task), tostring(msg))
end)

--loads from a configuration file
local function load_configuration(file)
	local func_conf, err = loadfile(file)
	assert(func_conf,err)
	local conf = toribio.configuration
	local meta_create_on_query 
	meta_create_on_query = {
		__index = function (table, key)
			table[key]=setmetatable({}, meta_create_on_query)
			return table[key]
		end,
	}
	setmetatable(conf, meta_create_on_query)
	setfenv(func_conf, conf)
	func_conf()
	meta_create_on_query['__index']=nil
end

if opts["h"] then
	print [[Usage:
	lua toribio-go.lua [-h] [-d] [-c conffile|'none'] 
		-h Print help
		-c Use given configuration file (or none). 
		   Defaults to 'toribio-go.conf'
		-d default log level. Available levels:
		   NONE|ERROR|WARNING|INFO|DETAIL|DEBUG|ALL
	]]
	os.exit()
end
if not opts["c"] then
	load_configuration('toribio-go.conf')
elseif opts["c"] ~= "none" then
	load_configuration(opts["c"])
end
local param_log_level = opts["d"]
if param_log_level  == true then param_log_level ='DEBUG' end
if param_log_level then
	local conf = toribio.configuration
	conf.log = conf.log or {}
	conf.log.level = conf.log.level or {}
	conf.log.level.default = param_log_level
end

local function table_path_exists(t, ...)
	local p = t
	for i = 1, select('#', ...) do
		local field = select(i, ...)
		p = p[field]
		if p==nil then return end
	end
	return p
end

--set log level
local loglevel = table_path_exists(toribio.configuration, 'log', 'level')
for logmod,level in pairs(loglevel or {}) do
	print ('Setting log level for', logmod, level)
	if logmod == 'default' then
		log.setlevel(level)
	else
		log.setlevel(level, logmod)
	end
end


sched.run(function()
	for _, section in ipairs({'deviceloaders', 'tasks'}) do
		for task, conf in pairs(toribio.configuration[section] or {}) do
			log ('TORIBIOGO', 'DETAIL', 'Processing conf %s %s: %s', section, task, tostring((conf and conf.load) or false))

			if conf and conf.load==true then
				--[[
				local taskmodule = require (section..'/'..task)
				if taskmodule.start then
					local ok = pcall(taskmodule.start,conf)
				end
				--]]
				log ('TORIBIOGO', 'INFO', 'Starting %s %s', section, task)
				toribio.start(section, task)
			end
		end
	end
end)

--print('Toribio go!')
log ('TORIBIOGO', 'INFO', 'Ready')
sched.loop()