66 lines
1.8 KiB
Lua
66 lines
1.8 KiB
Lua
local luarun = {}
|
|
|
|
local utilities = require('otouto.utilities')
|
|
local URL = require('socket.url')
|
|
local JSON, serpent
|
|
|
|
function luarun:init(config)
|
|
luarun.triggers = utilities.triggers(self.info.username, config.cmd_pat):t('lua', true):t('return', true).table
|
|
if config.luarun_serpent then
|
|
serpent = require('serpent')
|
|
luarun.serialize = function(t)
|
|
return serpent.block(t, {comment=false})
|
|
end
|
|
else
|
|
JSON = require('dkjson')
|
|
luarun.serialize = function(t)
|
|
return JSON.encode(t, {indent=true})
|
|
end
|
|
end
|
|
end
|
|
|
|
function luarun:action(msg, config)
|
|
|
|
if msg.from.id ~= config.admin then
|
|
return true
|
|
end
|
|
|
|
local input = utilities.input(msg.text)
|
|
if not input then
|
|
utilities.send_reply(self, msg, 'Please enter a string to load.')
|
|
return
|
|
end
|
|
|
|
if msg.text_lower:match('^'..config.cmd_pat..'return') then
|
|
input = 'return ' .. input
|
|
end
|
|
|
|
local output = loadstring( [[
|
|
local bot = require('otouto.bot')
|
|
local bindings = require('otouto.bindings')
|
|
local utilities = require('otouto.utilities')
|
|
local drua = require('otouto.drua-tg')
|
|
local JSON = require('dkjson')
|
|
local URL = require('socket.url')
|
|
local HTTP = require('socket.http')
|
|
local HTTPS = require('ssl.https')
|
|
return function (self, msg, config) ]] .. input .. [[ end
|
|
]] )()(self, msg, config)
|
|
if output == nil then
|
|
output = 'Done!'
|
|
else
|
|
if type(output) == 'table' then
|
|
local s = luarun.serialize(output)
|
|
if URL.escape(s):len() < 4000 then
|
|
output = s
|
|
end
|
|
end
|
|
output = '```\n' .. tostring(output) .. '\n```'
|
|
end
|
|
utilities.send_message(self, msg.chat.id, output, true, msg.message_id, true)
|
|
|
|
end
|
|
|
|
return luarun
|
|
|