http = require("socket.http") https = require("ssl.https") URL = require("socket.url") json = (loadfile "./libs/JSON.lua")() serpent = (loadfile "./libs/serpent.lua")() require("./bot/utils") VERSION = '0.8.1' function on_msg_receive (msg) vardump(msg) if msg_valid(msg) == false then return end do_action(msg) mark_read(get_receiver(msg), ok_cb, false) end function ok_cb(extra, success, result) end function on_binlog_replay_end () started = 1 -- Uncomment the line to enable cron plugins. -- postpone (cron_plugins, false, 5.0) -- See plugins/ping.lua as an example for cron _config = load_config() -- load plugins plugins = {} load_plugins() end function msg_valid(msg) -- Dont process outgoing messages if msg.out then return false end if msg.date < now then return false end if msg.unread == 0 then return false end end function do_lex(msg, text) local mutated = true while mutated do mutated = false for name, desc in pairs(plugins) do if (desc.lex ~= nil) then result = desc.lex(msg, text) if (result ~= nil) then -- print ("Mutating to " .. result) text = result mutated = true end end end end -- print("Text mutated to " .. text) return text 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) text = do_lex(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 local result2 = do_lex(msg, result) if (result2 == nil) then result2 = result end _send_msg(receiver, result2) 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 -- Save the content of _config to config.lua function save_config( ) file = io.open('./bot/config.lua', 'w+') local serialized = serpent.block(_config, { comment = false, name = "config" }) file:write(serialized) file:close() print ('saved config into ./bot/config.lua') end function load_config( ) local f = io.open('./bot/config.lua', "r") -- If config.lua doesnt exists if not f then print ("Created new config file: bot/config.lua") create_config() end f:close() local config = loadfile ("./bot/config.lua")() for v,user in pairs(config.sudo_users) do print("Allowed user: " .. user) end return config end -- Create a basic config.json file and saves it. function create_config( ) -- A simple config with basic plugins and ourserves as priviled user config = { enabled_plugins = { "9gag", "echo", "get", "set", "images", "img_google", "location", "media", "plugins", "stats", "time", "version", "youtube" }, sudo_users = {our_id} } file = io.open('./bot/config.lua', 'w+') local serialized = serpent.block(config, { comment = false, name = "config" }) file:write(serialized) file:close() print ('saved config into ./bot/config.lua') 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 -- Enable plugins in config.json function load_plugins() for k, v in pairs(_config.enabled_plugins) do print("Loading plugin", v) t = loadfile("plugins/"..v..'.lua')() table.insert(plugins, t) end end -- Cron all the enabled plugins function cron_plugins() for name, desc in pairs(plugins) do if desc.cron ~= nil then print(desc.description) desc.cron() end end -- Called again in 5 mins postpone (cron_plugins, false, 5*60.0) end -- Start and load values our_id = 0 now = os.time()