fusionpbx/resources/install/scripts/ha_monitor.lua

--! @todo move `servers` table to `local.lua` file
--[[
servers = {}
servers[#servers + 1] = {
	method   = 'curl';
	username = "aaa";
	password = "***";
	hostname = "127.0.0.1";
	port     = "8080";
}
servers[#servers + 1] = {
	method   = 'ssh';
	username = "aaa";
	password = "***";
	hostname = "127.0.0.2";
	port     = "8080";
}
]]

require "resources.functions.config"

local log           = require "resources.functions.log".ha_monitor
local EventConsumer = require "resources.functions.event_consumer"
local api           = require "resources.functions.api"

local service_name = 'ha_monitor'
local pid_file = scripts_dir .. '/run/' .. service_name .. '.tmp'
local method   = 'curl' -- default method

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

local function urlencode(s)
	s = string.gsub(s, "([^%w])",function(c)
		return string.format("%%%02X", string.byte(c))
	end)
	return s
end

local function emit(new_event_name)
	return function(self, name, event)
		self:emit(new_event_name, event)
	end
end

-- execute command on remote FS using mod_xml_rpc
local function curl(server, command, args)
	local url = 'http://'
	if server.username then
		url = url .. server.username .. ':' .. server.password .. '@'
	end

	url = url .. server.hostname

	if server.port then
		url = url .. ':' .. server.port
	end

	url = url .. '/webapi/' .. command

	if args then
		url = url .. '?' .. urlencode(args)
	end

	-- log.noticef('curl %s', url)

	local response, err = api:execute('curl', url)

	if err then log.warningf('[curl %s] error [%s]', url, err)
	else log.noticef('[curl %s] pass [%s]', url, response) end
end

-- execute command on remote FS using SSH access
local function ssh(server, command, args)
	local cmd = 'ssh '..server.username..'@'..server.hostname
	if server.port then
		cmd = cmd .. ':' .. server.port
	end
	cmd = cmd .. [[ "fs_cli -x ']] .. command
	if args then
		--! todo escape args
		cmd = cmd .. ' ' .. args
	end
	cmd = cmd .. [['"]]

	-- log.notice(cmd)

	local status = os.execute(cmd)

	if status ~= 0 then log.warningf('[%s] fail [%d]', cmd, status)
	else log.noticef('[%s] pass', cmd) end
end

local remote_execute_methods = {curl = curl; ssh = ssh}
local default_remote_execute = assert(remote_execute_methods[servers.method or method], 'Unknown execute method')

local function remote_execute(server, command, args)
	local exec = default_remote_execute
	if server.method then
		exec = assert(remote_execute_methods[server.method], 'Unknown execute method')
	end
	return exec(server, command, args)
end

-- execute command on all remotes FS
local function remote_execute_all(...)
	for _, server in ipairs(servers) do
		remote_execute(server, ...)
	end
end

local events = EventConsumer.new(pid_file)

-- FS shutdown
events:bind('SHUTDOWN', function(self, name, event)
	log.notice('shutdown')
	return self:stop()
end)

-- Control commands from FusionPBX
events:bind("CUSTOM::fusion::service::control", function(self, name, event)
	if service_name ~= event:getHeader('service-name') then return end

	local command = event:getHeader('service-command')
	if command == "stop" then
		log.notice("get stop command")
		return self:stop()
	end

	log.warningf('Unknown service command: %s', command or '<NONE>')
end)

-- Lua code can not generate 'MEMCACHE' event so we use `CUSTOM::fusion::memcache`
-- and forwart MEMCACHE to it for backward compatability
events:bind('MEMCACHE', emit'CUSTOM::fusion::memcache')

-- Memcache event
events:bind('CUSTOM::fusion::memcache', function(self, name, event)
	-- log.noticef('event: %s\n%s', name, event:serialize('xml'))

	local api_command = trim(event:getHeader('API-Command'))
	if api_command ~= 'memcache' then return end

	local api_command_argument = trim(event:getHeader('API-Command-Argument'))

	local memcache_updated = api_command_argument and (
		(api_command_argument == 'flush')
		or (api_command_argument:sub(1, 6) == 'delete')
	)

	if memcache_updated then
		remote_execute_all(api_command, api_command_argument)
	end
end)

-- start gateway
events:bind('CUSTOM::sofia::gateway_add', function(self, name, event)
	-- log.noticef('event: %s\n%s', name, event:serialize('xml'))

	local profile = event:getHeader('profile-name')
	local gateway = event:getHeader('Gateway')

	local cmd = 'profile ' .. profile .. ' start ' .. gateway
	remote_execute_all('sofia',  cmd)
end)

-- stop gateway
events:bind('CUSTOM::sofia::gateway_delete', function(self, name, event)
	-- log.noticef('event: %s\n%s', name, event:serialize('xml'))

	local profile = event:getHeader('profile-name')
	local gateway = event:getHeader('Gateway')

	local cmd = 'profile ' .. profile .. ' killgw ' .. gateway
	remote_execute_all('sofia',  cmd)
end)

log.notice('start')

events:run()

log.notice('stop')