136 lines
4.0 KiB
Lua
Executable File
136 lines
4.0 KiB
Lua
Executable File
local bot = {}
|
|
|
|
-- Requires are moved to init to allow for reloads.
|
|
local bindings -- Load Telegram bindings.
|
|
local utilities -- Load miscellaneous and cross-plugin functions.
|
|
|
|
bot.version = '3.7'
|
|
|
|
function bot:init() -- The function run when the bot is started or reloaded.
|
|
|
|
bindings = require('bindings')
|
|
utilities = require('utilities')
|
|
|
|
self.config = require('config') -- Load configuration file.
|
|
|
|
self.BASE_URL = 'https://api.telegram.org/bot' .. self.config.bot_api_key
|
|
if self.config.bot_api_key == '' then
|
|
error('You did not set your bot token in config.lua!')
|
|
end
|
|
|
|
-- Fetch bot information. Try until it succeeds.
|
|
repeat self.info = bindings.getMe(self) until self.info
|
|
self.info = self.info.result
|
|
|
|
-- Load the "database"! ;)
|
|
if not self.database then
|
|
self.database = utilities.load_data(self.info.username..'.db')
|
|
end
|
|
|
|
self.plugins = {} -- Load plugins.
|
|
for _,v in ipairs(self.config.plugins) do
|
|
local p = require('plugins.'..v)
|
|
table.insert(self.plugins, p)
|
|
if p.init then p.init(self) end
|
|
end
|
|
|
|
print('@' .. self.info.username .. ', AKA ' .. self.info.first_name ..' ('..self.info.id..')')
|
|
|
|
-- Generate a random seed and "pop" the first random number. :)
|
|
math.randomseed(os.time())
|
|
math.random()
|
|
|
|
self.last_update = self.last_update or 0 -- Set loop variables: Update offset,
|
|
self.last_cron = self.last_cron or os.date('%M') -- the time of the last cron job,
|
|
self.is_started = true -- and whether or not the bot should be running.
|
|
self.database.users = self.database.users or {} -- Table to cache userdata.
|
|
self.database.users[tostring(self.info.id)] = self.info
|
|
|
|
end
|
|
|
|
function bot:on_msg_receive(msg) -- The fn run whenever a message is received.
|
|
|
|
-- Create a user entry if it does not exist.
|
|
if not self.database.users[tostring(msg.from.id)] then
|
|
self.database.users[tostring(msg.from.id)] = {}
|
|
end
|
|
-- Clear things that no longer exist.
|
|
self.database.users[tostring(msg.from.id)].username = nil
|
|
self.database.users[tostring(msg.from.id)].last_name = nil
|
|
-- Wee.
|
|
for k,v in pairs(msg.from) do
|
|
self.database.users[tostring(msg.from.id)][k] = v
|
|
end
|
|
|
|
if msg.date < os.time() - 5 then return end -- Do not process old messages.
|
|
|
|
msg = utilities.enrich_message(msg)
|
|
|
|
if msg.text:match('^/start .+') then
|
|
msg.text = '/' .. utilities.input(msg.text)
|
|
msg.text_lower = msg.text:lower()
|
|
end
|
|
|
|
for _,v in ipairs(self.plugins) do
|
|
for _,w in pairs(v.triggers) do
|
|
if string.match(msg.text:lower(), w) then
|
|
local success, result = pcall(function()
|
|
return v.action(self, msg)
|
|
end)
|
|
if not success then
|
|
bindings.sendReply(self, msg, 'Sorry, an unexpected error occurred.')
|
|
utilities.handle_exception(self, result, msg.from.id .. ': ' .. msg.text)
|
|
return
|
|
end
|
|
-- If the action returns a table, make that table msg.
|
|
if type(result) == 'table' then
|
|
msg = result
|
|
-- If the action returns true, don't stop.
|
|
elseif result ~= true then
|
|
return
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
end
|
|
|
|
function bot:run()
|
|
bot.init(self) -- Actually start the script. Run the bot_init function.
|
|
|
|
while self.is_started do -- Start a loop while the bot should be running.
|
|
|
|
do
|
|
local res = bindings.getUpdates(self, self.last_update+1) -- Get the latest updates!
|
|
if res then
|
|
for _,v in ipairs(res.result) do -- Go through every new message.
|
|
self.last_update = v.update_id
|
|
bot.on_msg_receive(self, v.message)
|
|
end
|
|
else
|
|
print(self.config.errors.connection)
|
|
end
|
|
end
|
|
|
|
if self.last_cron ~= os.date('%M') then -- Run cron jobs every minute.
|
|
self.last_cron = os.date('%M')
|
|
utilities.save_data(self.info.username..'.db', self.database) -- Save the database.
|
|
for i,v in ipairs(self.plugins) do
|
|
if v.cron then -- Call each plugin's cron function, if it has one.
|
|
local res, err = pcall(function() v.cron(self) end)
|
|
if not res then
|
|
utilities.handle_exception(self, err, 'CRON: ' .. i)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
end
|
|
|
|
-- Save the database before exiting.
|
|
utilities.save_data(self.info.username..'.db', self.database)
|
|
print('Halted.')
|
|
end
|
|
|
|
return bot
|