new plugin isup

plugins/isup.lua

@@ -0,0 +1,130 @@
+do
+local socket = require("socket")
+local cronned = load_from_file('data/isup.lua')
+
+local function save_cron(msg, url, delete)
+  local origin = get_receiver(msg)
+  if not cronned[origin] then
+    cronned[origin] = {}
+  end
+  if not delete then
+    table.insert(cronned[origin], url)
+  else
+    for k,v in pairs(cronned[origin]) do
+      if v == url then
+        table.remove(cronned[origin], k)
+      end
+    end
+  end
+  serialize_to_file(cronned, 'data/isup.lua')
+  return 'Saved!'
+end
+
+local function is_up_socket(ip, port)
+  print('Connect to', ip, port)
+  local c = socket.try(socket.tcp())
+  c:settimeout(3)
+  local conn = c:connect(ip, port)
+  if not conn then
+    return false
+  else
+    c:close()
+    return true
+  end
+end
+
+local function is_up_http(url)
+  -- Parse URL from input, default to http
+  local parsed_url = URL.parse(url,  { scheme = 'http', authority = '' })
+  -- Fix URLs without subdomain not parsed properly
+  if not parsed_url.host and parsed_url.path then
+    parsed_url.host = parsed_url.path
+    parsed_url.path = ""
+  end
+  -- Re-build URL
+  local url = URL.build(parsed_url)
+
+  local protocols = {
+    ["https"] = https,
+    ["http"] = http
+  }
+  local options =  {
+    url = url,
+    redirect = false,
+    method = "GET"
+  }
+  local response = { protocols[parsed_url.scheme].request(options) }
+  local code = tonumber(response[2])
+  if code == nil or code >= 400 then
+    return false
+  end
+  return true
+end
+
+local function isup(url)
+  local pattern = '^(%d%d?%d?%.%d%d?%d?%.%d%d?%d?%.%d%d?%d?):?(%d?%d?%d?%d?%d?)$'
+  local ip,port = string.match(url, pattern)
+  local result = nil
+
+  -- !isup 8.8.8.8:53
+  if ip then
+    port = port or '80'
+    result = is_up_socket(ip, port)
+  else
+    result = is_up_http(url)
+  end
+
+  return result
+end
+
+local function cron()
+  for chan, urls in pairs(cronned) do
+    for k,url in pairs(urls) do
+      print('Checking', url)
+      if not isup(url) then
+        local text = url..' looks DOWN from here. ??'
+        send_msg(chan, text, ok_cb, false)
+      end
+    end
+  end
+end
+
+local function run(msg, matches)
+  if matches[1] == 'remove' then
+    if not is_sudo(msg) then 
+      return 'This command requires privileged user'
+    end
+    return save_cron(msg, matches[2], true)
+
+  elseif matches[1] == 'save' then
+    if not is_sudo(msg) then 
+      return 'This command requires privileged user'
+    end
+    return save_cron(msg, matches[2])
+
+  elseif isup(matches[1]) then
+    return matches[1]..' looks UP from here. ??'
+  else
+    return matches[1]..' looks DOWN from here. ??'
+  end
+end
+
+return {
+  description = "Check if a website or server is up.",
+  usage = {
+    "/isup [host]: Performs a HTTP request or Socket (ip:port) connexion",
+    "/isup cron [host]: Every 5mins check if host is up. (Requires privileged user)",
+    "/isup cron delete [host]: Disable checking that host."
+  },
+  patterns = {
+    "^/isup (remove) (.*)$",
+    "^/isup (save) (.*)$",
+    "^/isup (.*)$",
+    "^/ping (.*)$",
+    "^/ping (remove) (.*)$",
+    "^/ping (save) (.*)$"
+  },
+  run = run
+}
+
+end