Add remote services

1 files changed, 669 insertions, 0 deletions

diff --git a/remote/lib/support.lua b/remote/lib/support.lua
new file mode 100644
index 0000000..984d677
--- /dev/null
+++ b/remote/lib/support.lua
@@ -0,0 +1,669 @@
+require "luarocks.require"
+require "md5"
+require "lfs"
+
+home = os.getenv( "HOME" )
+
+os.execute("mkdir -p " .. home .. "/.conky/requests");
+os.execute("mkdir -p " .. home .. "/.conky/info");
+c_timer_init = -1
+
+do
+ function conky_eval(...)
+    return conky_parse(table.concat(arg, " "))
+ end
+ 
+ function conky_multiply_line(times, ...)
+    local str = table.concat(arg, " ")
+    local tbl = {}
+    for i=1,times,1 do
+	tbl[i] = string.gsub(str,"@i@", i-1)
+    end
+    local res = table.concat(tbl,"\n")
+    return res;
+ end
+
+ function split(delimiter, text)
+  local list = {}
+  local pos = 1
+  if string.find("", delimiter, 1) then -- this would result in endless loops
+    error("delimiter matches empty string!")
+  end
+  while 1 do
+    local first, last = string.find(text, delimiter, pos)
+    if first then -- found?
+      table.insert(list, string.sub(text, pos, first-1))
+      pos = last+1
+    else
+      table.insert(list, string.sub(text, pos))
+      break
+    end
+  end
+  return list
+ end
+ 
+ function conky_filter_real_ip(iface)
+    local req =  string.format('${addrs %s}', iface)
+    local ips = conky_parse(req);
+--[[
+    print(ips)
+--]]
+    local list = split(",", ips)
+    for i, item in ipairs(list) do
+	if (string.match(item, "141\.52\.64")) then
+	    return item
+	end
+    end
+    return "Unknown"
+ end
+
+ user = split('/', home)
+ user = user[#user]
+
+ result_check_server_status = {}
+ result_check_adei_source = {}
+ result_check_server_ = {}
+ result_check_service_ = {}
+ online = {}
+ status = {}
+ tested = {}
+ c_timer = c_timer_init
+ width = 80
+ update_time = 0
+ outdate_time = 0
+ timeout = 2
+
+ function conky_set_width (w)
+    width = tonumber(w)
+    return ""
+ end
+ 
+ function conky_set_timeout (t)
+    timeout = tonumber(t)
+    return ""
+ end
+
+ function cmd_popen(server, port, cmd)
+    if (c_timer_init == 0) then
+	return io.popen(cmd)
+    else 
+	local cmd_md5 = md5.sumhexa(cmd)
+        local fn = string.format("%s/.conky/info/%s:%s-%s.srv", home, server, port, cmd_md5)
+	stat = lfs.attributes(fn)
+	if (stat == nil) then
+	    return nil
+	end
+	
+	local since = os.difftime(stat['modification'], update_time)
+	if (since < 0) then
+	    return nil
+	end
+	since = os.difftime(outdate_time, os.time())
+	if (since < 0) then
+	    return nil
+	end
+	return io.open(fn, "r")
+    end
+ end
+
+ function conky_check_server_status(server, port, ...)
+    port = port or 22
+    local name = string.format("%s:%i", server, port);
+
+    if (c_timer < 0) then
+	if (c_timer == -1) then
+	    local fn = string.format("%s/.conky/requests/%s:%i.srv", home, server, port)
+	    
+	    tested[string.format("%s:%i", server, port)] = false;
+
+	    local f = assert(io.open(fn, "w"))
+
+	    local cmd = string.format("/etc/conky/scripts/ping.pl %s:%i %i", server, port, timeout)
+	    f:write(cmd .. "\n")
+
+	    if (arg[1] ~= "-") then
+		cmd = string.format("ssh -x -p %i -o ConnectTimeout=%i root@%s /opt/scripts/check_server_status.sh 2>&1 | fold -w %i", port, timeout, server, width)
+		f:write(cmd .. "\n")
+	    
+		for i, service in ipairs(arg)
+		do		
+		    cmd = string.format("ssh -x -p %i -o ConnectTimeout=%i root@%s /opt/scripts/check_%s_status.sh 2>&1 | fold -w %i", port, timeout, server, service, width)
+	    	    f:write(cmd .. "\n")
+		end
+	    end
+	    f:close()
+	end    
+	if (result_check_server_status[name] == nil)  then
+	    local res
+	    if (math.abs(c_timer)%2 > 0) then
+		res = "${color yellow}?${color}"
+	    else
+		res = "${color yellow}.${color}"
+	    end
+	    
+	    if (arg[1] ~= "-") then
+		res = res .. "  "
+		for i, service in ipairs(arg) do
+	    	    res = res .. "  "
+		end
+	    end
+	    --result_check_server_status[server] = res
+	    --status[server] = false
+	    return res
+	end
+    elseif (c_timer == 0) then
+	result_check_server_status[name] = check_server_status(server, port, arg)
+    end
+
+    return result_check_server_status[name];
+ end
+ 
+ function check_server_status(server, port, additional)
+    port = port or 22
+
+    local res
+    local cmd = string.format("/etc/conky/scripts/ping.pl %s:%i %i", server, port, timeout)
+    local cmdf = cmd_popen(server, port, cmd)
+    if (cmdf == nil) then
+	res = -1
+    else
+	local data = cmdf:read("*a")
+	res = tonumber(data)
+	cmdf:close()
+    end
+    
+    if (additional[1] == "-") then
+	if (res > 0) then
+	    online[server] = true
+	    return "${color green}*"
+	elseif (res < 0) then
+	    return "${color red}?"
+	else
+	    online[server] = false
+	    return "${color red}*"
+	end
+    end
+    
+    if (res > 0) then
+	online[server] = true
+	
+	local output
+
+	cmd = string.format("ssh -x -p %i -o ConnectTimeout=%i root@%s /opt/scripts/check_server_status.sh 2>&1 | fold -w %i", port, timeout, server, width)
+	cmdf = cmd_popen(server, port, cmd)
+	if (cmdf == nil) then
+	    output = string.format("${color green}* ${color red}?")
+	else
+	    local res = {}
+	    local line = cmdf:read("*l")
+	    while line do
+		local m1 = string.match(line, "^\(.*[^%s]\)%s*$")
+		if (m1) then
+	    	    table.insert(res, m1)
+		end
+		line = cmdf:read("*l")
+	    end
+--	    res = cmdf:read("*a")
+	    cmdf:close()
+	    
+	    res = table.concat(res, "\n")
+	    
+	    if (string.match(res, "[%a%d]")) then
+		table.insert(status, string.format("${color yellow}%s:%i${color gray}\n%s", server, port, res))
+	        output = string.format("${color green}* ${color red}*")
+	    else
+		output = string.format("${color green}* *")
+	    end
+	end
+
+	for i, service in ipairs(additional)
+	do	
+	    cmd = string.format("ssh -x -p %i -o ConnectTimeout=%i root@%s /opt/scripts/check_%s_status.sh 2>&1 | fold -w %i", port, timeout, server, service, width)
+	    cmdf = cmd_popen(server, port, cmd)
+	    if (cmdf == nil) then
+		output = output .. string.format(" ${color red}?")
+	    else 
+	        local res = {}
+		local line = cmdf:read("*l")
+	        while line do
+		    local m1 = string.match(line, "^\(.*[^%s]\)%s*$")
+		    if (m1) then
+	    		table.insert(res, m1)
+		    end
+		    line = cmdf:read("*l")
+		end
+		cmdf:close()
+
+		if (res[1]) then
+		    local info = table.remove(res)
+		    local s = 0
+		    local extra = ""
+		    
+		    local m1, m2 = string.match(info, "^\(%d+\)\(%s.+\)$")
+		    if (m1 == nil) then
+			m1 = string.match(info, "^\%d+$");
+		    end
+		    if (m1 == nil) then
+			table.insert(res, info)
+		    else
+			s = tonumber(m1)
+			if (m2 ~= nil) then
+			    extra = string.sub(m2,2)
+			end
+		    end
+	    	    
+		    if (res[1]) then
+			res = table.concat(res, "\n")
+			table.insert(status, string.format("${color yellow}%s:%i:%s${color gray}\n%s", server, port, service, res))
+		    end
+		    if (s == 0) then
+			output = output .. string.format(" ${color red}*")
+		    elseif (s == 1) then
+			output = output .. string.format(" ${color green}*")
+		    else
+			output = output .. string.format(" ${color yellow}*")
+		    end
+		    output=output .. " ${color white}" .. extra;
+		else
+		    output = output .. string.format(" ${color green}*")
+		end
+	    end
+	end
+	
+	return output
+    else
+	if (res < 0) then
+	    res = string.format("${color red}?  ")
+	else
+	    online[server] = false
+	    res = string.format("${color red}*  ")
+	end
+	for i, service in ipairs(additional) do
+	    res = res .. "  "
+	end
+
+	return res
+    end
+ end
+
+ function conky_check_adei_source(server, port, config, setup, db_server, db_name, ignore_list)
+    port = port or 22
+    local name = string.format("%s:%i__%s__%s", server, port, db_server, db_name)
+    if (c_timer < 0) then
+	if (c_timer == -1) then
+	    ignore_list = ignore_list or ""
+
+	    local fn = string.format("%s/.conky/requests/%s:%i.srv", home, server, port)
+
+	    local f = assert(io.open(fn, "a+"))
+	    local cmd = string.format('ssh -x -p %i -o ConnectTimeout=%i root@%s /opt/scripts/check_adei_source.sh %s %s %s %s "%s" 2>&1', port, timeout, server, config, setup, db_server, db_name, ignore_list)
+	    f:write(cmd .. "\n")
+	    f:close()
+	end
+	
+	if (result_check_adei_source[name] == nil) then
+	    return "${color yellow}?${color}"
+	else
+	    if (online[server]) then
+    		return result_check_adei_source[name];
+	    else
+		return "${color red}?${color}"
+	    end
+	end
+    elseif (online[server]) then
+	if (c_timer == 0) then
+	    result_check_adei_source[name] = check_adei_source(server, port, config, setup, db_server, db_name, ignore_list)
+	end
+        return result_check_adei_source[name];
+    else
+	return "${color red}?${color}"
+    end
+ end
+ 
+ function check_adei_source(server, port, config, setup, db_server, db_name, ignore_list)
+	ignore_list = ignore_list or ""
+	cmd = string.format('ssh -x -p %i -o ConnectTimeout=%i root@%s /opt/scripts/check_adei_source.sh %s %s %s %s "%s" 2>&1', port, timeout, server, config, setup, db_server, db_name, ignore_list)
+	cmdf = cmd_popen(server, port, cmd)
+	if (cmdf == nil) then
+	    return "${color red}?"
+	end
+	
+	local res = {}
+	line = cmdf:read("*l")
+	while line do
+	    local m1 = string.match(line, "^\(.*[^%s]\)%s*$")
+	    if (m1) then
+		table.insert(res, m1)
+	    end
+	    line = cmdf:read("*l")
+	end
+	cmdf:close()
+
+	local info = table.remove(res)
+
+	local t1, t2, t3 = string.match(info, "^\(%d+\)%s+\(%d+\)%s+\(%d+\)")
+	
+	local s = tonumber(t1)
+	if (s == nil) then 
+	    table.insert(res, info)
+	    s = 0 
+	end
+
+	res = table.concat(res, "\n")
+	
+	local output
+	if (s == 1) then 
+	    output="${color green}*"
+	else
+--[[
+	    table.insert(status, string.format("${color yellow}%s -- %s${color gray}\n%s\n", db_server, db_name, res))
+--]]
+	    if (s == 0) then output="${color red}*"
+	    else output="${color yellow}*" end
+	end
+	
+--	if (s ~= 0) then
+	    local groups = tonumber(t2)
+	    if (groups) then
+		output = output .. string.format("${color white} %i groups", groups);
+	    else
+		output = output .. string.format("${color red} ? groups");
+	    end
+
+	    local size = tonumber(t3)	
+	    if (groups and size) then
+		output = output .. string.format("${color white}, %i GB", size);
+	    else
+		output = output .. string.format("${color red}, ? GB");
+	    end
+--	end
+
+        if (string.match(res, "[%a%d]")) then
+	    output = output .. "${color gray}\n   ${font Bitstream Vera Sans Mono:size=7}" .. string.gsub(res,"\n", "${font}\n   ${font Bitstream Vera Sans Mono:size=7}") .. "${font}${color white}"
+	end
+	
+	return output
+ end
+
+ function conky_check_server_(service, server, port, ...)
+    port = port or 22
+    local name = string.format("%s:%i:%s", server, port, service)
+    if (c_timer < 0) then
+        if (c_timer == -1) then
+            ignore_list = ignore_list or ""
+            
+            local fn = string.format("%s/.conky/requests/%s:%i.srv", home, server, port)
+        
+            local f = assert(io.open(fn, "a+"))
+            local cmd = string.format("ssh -x -p %i -o ConnectTimeout=%i root@%s /opt/scripts/check_server_%s.sh 2>&1", port, timeout, server, service)
+            f:write(cmd .. "\n")
+            f:close()
+        end
+           
+        if (result_check_server_[name] == nil) then
+            return "${color yellow}?${color}"
+        else
+            if (online[server]) then
+                return result_check_server_[name];
+            else
+                return "${color red}?${color}"
+            end
+        end
+    elseif (online[server]) then
+        if (c_timer == 0) then
+            result_check_server_[name] = check_server_(service, server, port, arg)
+        end
+        return result_check_server_[name];
+    else
+        return "${color red}?${color}"
+    end
+ end
+
+ function conky_check_service_(service, id, ...)
+    local name = string.format("%s:%s", service, id)
+    if (c_timer < 0) then
+        if (c_timer == -1) then
+            ignore_list = ignore_list or ""
+
+	    tested[string.format("%s:%s", service, id)] = false;
+            
+            local fn = string.format("%s/.conky/requests/%s:%s.srv", home, service, id)
+        
+            local f = assert(io.open(fn, "a+"))
+            local cmd = string.format("/etc/conky/service/check_%s.sh %s %s 2>&1", service, id, table.concat(arg, " "))
+            f:write(cmd .. "\n")
+            f:close()
+        end
+           
+        if (result_check_server_[name] == nil) then
+            return "${color yellow}?${color}"
+        else
+            return result_check_server_[name];
+        end
+    else
+        if (c_timer == 0) then
+            result_check_server_[name] = check_service_(service, id, arg)
+        end
+        return result_check_server_[name];
+    end
+ end
+
+
+ function format_traffic(value, yellow, red)
+    yellow = yellow or 100
+    red = red or 1000
+    
+    local value = math.floor(tonumber(value) / 1073741824)
+    
+    local res
+    if (value > red) then res="${color red}"
+    elseif (value > yellow) then res="${color yellow}"
+    else res="${color white}" end
+
+    if (value < 10) then res = res .. "  " .. value
+    elseif (value < 100) then res = res .. " " .. value
+    else res = res .. value end
+    
+    return res .. " GB${color gray}"
+ end
+ 
+ function check_server_(service, server, port, opts)
+        port = port or 22
+	local cmd = string.format("ssh -x -p %i -o ConnectTimeout=%i root@%s /opt/scripts/check_server_%s.sh 2>&1", port, timeout, server, service)
+	cmdf = cmd_popen(server, port, cmd)
+	if (cmdf == nil) then
+	    return "${color red}?"
+	end
+
+	local res = {}
+	line = cmdf:read("*l")
+	while line do
+	    local m1 = string.match(line, "^\(.*[^%s]\)%s*$")
+	    if (m1) then
+		table.insert(res, m1)
+	    end
+	    line = cmdf:read("*l")
+	end
+	cmdf:close()
+	
+	if (service == "traffic") then
+	    if (res[1]) then
+		local t1, t2, t3 = string.match(res[1], "^\(%d+\)%s+\(%d+\)%s+\(%d+\)")
+		
+		if (t1 and t2 and t3) then
+	    	    if ((opts) and (opts[1])) then yellow = tonumber(opts[1])
+		    else yellow = nil end
+		    if ((opts) and (opts[2])) then red = tonumber(opts[2])
+		    else red = nil end
+		
+		    t1=format_traffic(t1,yellow,red)
+		    t2=format_traffic(t2,yellow,red)
+		    t3=format_traffic(t3,yellow,red)
+		    return string.format("${color gray}i: %s, o: %s, f: %s", t1, t2, t3)
+		else
+		    return "${color red}" .. res[1]
+		end
+	    end
+	end
+	
+
+	return ""
+ end
+
+
+ function check_service_(service, id, opts)
+	local cmd = string.format("/etc/conky/service/check_%s.sh %s %s 2>&1", service, id, table.concat(opts, " "))
+	cmdf = cmd_popen(service, id, cmd)
+	if (cmdf == nil) then
+	    return "${color red}?"
+	end
+
+	local res = {}
+	local local_res = {}
+	line = cmdf:read("*l")
+	while line do
+	    local t, m1 = string.match(line, "^\(\*?\)\(.*[^%s]\)%s*$")
+	    if (m1) then
+		if (t == nil) or (t == '') then
+		    table.insert(res, m1)
+		else
+		    table.insert(local_res, string.sub(m1,1))
+		end
+	    end
+	    line = cmdf:read("*l")
+	end
+	cmdf:close()
+
+	local output
+	local info = table.remove(res)
+	local t1 = 0
+	local t2 = nil
+	local msg = ''
+
+	if (info) then
+	    t1, t2, msg = string.match(info, "^\(%d+\)%s+\(%d+\)\(.*\)")
+	end
+	
+	local s = tonumber(t1)
+	if (s == nil) then 
+	    table.insert(res, info)
+	    s = 0 
+	end
+
+	if (s == 1) then 
+	    output="${color green}*"
+	else
+	    if (s == 0) then output="${color red}*"
+	    else output="${color yellow}*" end
+	end
+	
+	s = tonumber(t2)
+	if (s == nil) then 
+	    s = 0 
+	else
+	    if (s == 1) then 
+		output=output .. " ${color green}*"
+	    else
+		if (s == 0) then output=output .. " ${color red}*"
+		else output=output .. " ${color yellow}*" end
+	    end
+	end
+
+	output = output .. " ${color white}" .. msg
+
+	res = table.concat(res, "\n")
+	local_res = table.concat(local_res, "\n")
+	
+
+	if (string.match(res, "[%a%d]")) then
+	    table.insert(status, string.format("${color yellow}%s:%s${color gray}\n%s", service, id, res))
+	end
+
+	if (string.match(local_res, "[%a%d]")) then
+	    output = output .. "${color gray}\n   ${font Bitstream Vera Sans Mono:size=7}" .. string.gsub(local_res,"\n", "${font}\n   ${font Bitstream Vera Sans Mono:size=7}") .. "${font}${color white}"
+	end
+	
+	return output
+ end
+
+
+ 
+ function conky_print_server_errors(interval, tag)
+    local res = table.concat(status, "\n")
+
+    if (c_timer < 0) then
+	tag = user .. "_" .. tag
+
+	if (c_timer == -1) then
+	    update_time = os.time()
+	    
+	    local result = 0;
+	    while (result == 0) do
+	        result = os.execute("ps x | grep conky_process_requests | grep " .. tag .. " | grep -v grep > /dev/null")
+	    end
+
+	    for item, s in pairs(tested) do
+		os.execute("/etc/conky/scripts/conky_process_requests.pl \"" .. item .. ".srv\" " .. tag .. " &")
+	    end
+	end
+	
+	local result=os.execute("ps x | grep conky_process_requests | grep " .. tag .. " | grep -v grep > /dev/null")
+	if (result == 0) then
+	    c_timer = c_timer - 1;
+	    local since = os.difftime(os.time(), update_time)
+	    if (since > tonumber(interval)) then
+		result_check_server_status = {}
+		result_check_adei_source = {}
+		result_check_server_ = {}
+	    end
+	else
+	    outdate_time = os.time() + 2 * tonumber(interval)
+	    c_timer = 0
+	end
+	return ""
+    else 
+	c_timer = c_timer + conky_info.update_interval
+	if (c_timer > tonumber(interval)) then 
+	    status = {}
+	    tested = {}
+	    c_timer = c_timer_init
+	end
+    end
+
+    if (string.match(res, "[%a%d]")) then
+	return string.format("\n\n%s", res)
+    else
+	return ""
+    end
+ end
+ 
+ function conky_outcon(pos1, pos2)
+    local n = tonumber(conky_parse('${tcp_portmon 32768 65535 count}'))
+    local res = {}
+    local str = ""
+    for i=1,n do
+	local val = tostring(conky_parse(string.format('${tcp_portmon 32768 65535 rhost %i}:${tcp_portmon 32768 65535 rport %i}',i - 1,i - 1)))
+	
+	if (res[val] == nil) then
+	    res[val] = 1
+	else
+	    res[val] = res[val] + 1
+	end
+    end
+
+
+    local names = {}
+    for addr,num in pairs(res) do
+	table.insert(names, addr)
+    end
+
+    table.sort(names, function(a,b) return res[a]>res[b] end)
+
+    for i=1,#names do
+	local name = names[i]
+	str = str .. string.format("${goto %i}%s${goto %i}cnt  %3i\n",pos1,name,pos2,res[name])
+    end
+    
+    return str
+ end
+end