Mikubot/bot/bot.lua

http = require("socket.http")
https = require("ssl.https")
URL = require("socket.url")
json = (loadfile "./bot/JSON.lua")()
require("./bot/utils")

VERSION = 'v0.7.6'

function on_msg_receive (msg)
  vardump(msg)

  if msg_valid(msg) == false then
    return
  end

  update_user_stats(msg)
  do_action(msg)

  mark_read(get_receiver(msg), ok_cb, false)
end

function ok_cb(extra, success, result)
end

-- Callback to remove tmp files
function rmtmp_cb(file_path, success, result)
   os.remove(file_path)
end

function msg_valid(msg)
  -- if msg.from.id == our_id then
  --   return true
  -- end
  if msg.out then
    return false
  end
  if msg.date < now then
    return false
  end
  if msg.unread == 0 then
    return false
  end
end

-- Where magic happens
function do_action(msg)
  local receiver = get_receiver(msg)
  local text = msg.text
  if msg.text == nil then
     -- Not a text message, make text the same as what tg shows so
     -- we can match on it. The plugin is resposible for handling
     text = '['..msg.media.type..']'
  end
  -- print("Received msg", text)
  for name, desc in pairs(plugins) do
    -- print("Trying module", name)
    for k, pattern in pairs(desc.patterns) do
      -- print("Trying", text, "against", pattern)
      matches = { string.match(text, pattern) }
      if matches[1] then
        print("  matches",pattern)
        if desc.run ~= nil then
          result = desc.run(msg, matches)
          print("  sending", result)
          if (result) then
            _send_msg(receiver, result)
            return
          end
        end
      end
    end
  end
end

-- If text is longer than 4096 chars, send multiple msg.
-- https://core.telegram.org/method/messages.sendMessage
function _send_msg( destination, text)
  local msg_text_max = 4096
  local len = string.len(text)
  local iterations = math.ceil(len / msg_text_max)

  for i = 1, iterations, 1 do
    local inital_c = i * msg_text_max - msg_text_max
    local final_c = i * msg_text_max
    -- dont worry about if text length < msg_text_max
    local text_msg = string.sub(text,inital_c,final_c)
    send_msg(destination, text_msg, ok_cb, false)
  end
end


function load_config()
   local f = assert(io.open('./bot/config.json', "r"))
   local c = f:read "*a"
   local config = json:decode(c)
   if config.sh_enabled then 
      print ("!sh command is enabled")
      for v,user in pairs(config.sudo_users) do
         print("Allowed user: " .. user)
      end
   end
   f:close()
   return config
end

function is_sudo(msg)
   local var = false
   -- Check users id in config 
   for v,user in pairs(config.sudo_users) do 
      if user == msg.from.id then 
         var = true 
      end
   end
   return var
end

function get_name(msg)
   local name = msg.from.first_name
   if name == nil then
      name = msg.from.id
   end
   return name
end

function update_user_stats(msg)
   -- Save user to _users table
  local from_id = tostring(msg.from.id)
  local to_id = tostring(msg.to.id)
  local user_name = get_name(msg)
  -- If last name is nil dont save last_name.
  local user_last_name = msg.from.last_name
  local user_print_name = msg.from.print_name
  if _users[to_id] == nil then
    _users[to_id] = {}
  end
  if _users[to_id][from_id] == nil then
    _users[to_id][from_id] = {
      name = user_name,
      last_name = user_last_name,
      print_name = user_print_name,
      msg_num = 1
    }
  else
    local actual_num = _users[to_id][from_id].msg_num
    _users[to_id][from_id].msg_num = actual_num + 1
    -- And update last_name
    _users[to_id][from_id].last_name = user_last_name
  end
end

function load_user_stats()
  local f = io.open('res/users.json', "r+")
  -- If file doesn't exists
  if f == nil then
    f = io.open('res/users.json', "w+")
    f:write("{}") -- Write empty table
    f:close()
    return {}
  else
    local c = f:read "*a"
    f:close()
    return json:decode(c)
  end
end

function on_our_id (id)
  our_id = id
end

function on_user_update (user, what)
  --vardump (user)
end

function on_chat_update (chat, what)
  --vardump (chat)
end

function on_secret_chat_update (schat, what)
  --vardump (schat)
end

function on_get_difference_end ()
end

function on_binlog_replay_end ()
  started = 1
  load_plugins()
  -- Uncomment the line to enable cron plugins.
  -- cron_plugins()
  -- See plugins/ping.lua as an example for cron
end

-- Start and load values
config = load_config()
_users = load_user_stats()
our_id = 0
now = os.time()

-- load plugins
plugins = {}

-- load all plugins in the plugins/ directory
function load_plugins()
  for k, v in pairs(scandir("plugins")) do
    -- Load only lua files
    if (v:match(".lua$")) then
        print("Loading plugin", v)
        t = loadfile("plugins/" .. v)()
        table.insert(plugins, t)
    end 
  end
end

-- Cron all the enabled plugins
function cron_plugins()

  for name, desc in pairs(plugins) do
    if desc.cron ~= nil then
      print("croned!"..desc.description)
      desc.cron()
    end
  end

  -- Called again in 5 mins
  postpone (cron_plugins, false, 5*60.0)
end