2015-07-03 00:15:52 +02:00
|
|
|
|
-- utilities.lua
|
|
|
|
|
-- Functions shared among plugins.
|
|
|
|
|
|
2016-04-08 23:12:02 +02:00
|
|
|
|
local utilities = {}
|
|
|
|
|
|
2016-07-31 21:29:44 +02:00
|
|
|
|
ltn12 = require('ltn12')
|
|
|
|
|
http = require('socket.http')
|
|
|
|
|
https = require('ssl.https')
|
2016-07-31 21:35:05 +02:00
|
|
|
|
socket = require('socket')
|
2016-07-31 21:29:44 +02:00
|
|
|
|
URL = require('socket.url')
|
2016-08-03 19:21:07 +02:00
|
|
|
|
json = require("dkjson")
|
|
|
|
|
pcall(json.use_lpeg)
|
2016-07-31 21:29:44 +02:00
|
|
|
|
serpent = require("serpent")
|
|
|
|
|
redis = (loadfile "./otouto/redis.lua")()
|
|
|
|
|
mimetype = (loadfile "./otouto/mimetype.lua")()
|
|
|
|
|
OAuth = require "OAuth"
|
|
|
|
|
helpers = require "OAuth.helpers"
|
|
|
|
|
|
|
|
|
|
http.timeout = 5
|
|
|
|
|
https.timeout = 5
|
2016-07-12 00:04:42 +02:00
|
|
|
|
|
2016-05-29 19:08:39 +02:00
|
|
|
|
-- For the sake of ease to new contributors and familiarity to old contributors,
|
|
|
|
|
-- we'll provide a couple of aliases to real bindings here.
|
2016-06-27 15:44:11 +02:00
|
|
|
|
function utilities:send_message(chat_id, text, disable_web_page_preview, reply_to_message_id, use_markdown, reply_markup)
|
2016-08-04 13:02:47 +02:00
|
|
|
|
if use_markdown == true then
|
|
|
|
|
use_markdown = 'Markdown'
|
|
|
|
|
elseif not use_markdown then
|
|
|
|
|
use_markdown = nil
|
|
|
|
|
end
|
2016-05-29 19:08:39 +02:00
|
|
|
|
return bindings.request(self, 'sendMessage', {
|
|
|
|
|
chat_id = chat_id,
|
|
|
|
|
text = text,
|
|
|
|
|
disable_web_page_preview = disable_web_page_preview,
|
|
|
|
|
reply_to_message_id = reply_to_message_id,
|
2016-08-04 13:02:47 +02:00
|
|
|
|
parse_mode = use_markdown,
|
2016-06-27 15:44:11 +02:00
|
|
|
|
reply_markup = reply_markup
|
2016-05-29 19:08:39 +02:00
|
|
|
|
} )
|
|
|
|
|
end
|
|
|
|
|
|
2016-07-02 00:13:17 +02:00
|
|
|
|
-- https://core.telegram.org/bots/api#editmessagetext
|
|
|
|
|
function utilities:edit_message(chat_id, message_id, text, disable_web_page_preview, use_markdown, reply_markup)
|
2016-08-04 13:02:47 +02:00
|
|
|
|
if use_markdown == true then
|
|
|
|
|
use_markdown = 'Markdown'
|
|
|
|
|
elseif not use_markdown then
|
|
|
|
|
use_markdown = nil
|
|
|
|
|
end
|
2016-07-02 00:13:17 +02:00
|
|
|
|
return bindings.request(self, 'editMessageText', {
|
|
|
|
|
chat_id = chat_id,
|
|
|
|
|
message_id = message_id,
|
|
|
|
|
text = text,
|
|
|
|
|
disable_web_page_preview = disable_web_page_preview,
|
2016-08-04 13:41:33 +02:00
|
|
|
|
parse_mode = use_markdown,
|
2016-07-02 00:13:17 +02:00
|
|
|
|
reply_markup = reply_markup
|
|
|
|
|
} )
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
function utilities:send_reply(old_msg, text, use_markdown, reply_markup)
|
2016-08-04 13:02:47 +02:00
|
|
|
|
if use_markdown == true then
|
|
|
|
|
use_markdown = 'Markdown'
|
|
|
|
|
elseif not use_markdown then
|
|
|
|
|
use_markdown = nil
|
|
|
|
|
end
|
2016-05-29 19:08:39 +02:00
|
|
|
|
return bindings.request(self, 'sendMessage', {
|
|
|
|
|
chat_id = old_msg.chat.id,
|
|
|
|
|
text = text,
|
|
|
|
|
disable_web_page_preview = true,
|
|
|
|
|
reply_to_message_id = old_msg.message_id,
|
2016-08-04 13:02:47 +02:00
|
|
|
|
parse_mode = use_markdown,
|
2016-07-02 00:13:17 +02:00
|
|
|
|
reply_markup = reply_markup
|
2016-05-29 19:08:39 +02:00
|
|
|
|
} )
|
|
|
|
|
end
|
2016-06-12 20:53:20 +02:00
|
|
|
|
|
|
|
|
|
-- NOTE: Telegram currently only allows file uploads up to 50 MB
|
|
|
|
|
-- https://core.telegram.org/bots/api#sendphoto
|
2016-07-02 12:35:14 +02:00
|
|
|
|
function utilities:send_photo(chat_id, file, text, reply_to_message_id, reply_markup)
|
2016-07-04 01:29:51 +02:00
|
|
|
|
if not file then return false end
|
2016-06-12 20:53:20 +02:00
|
|
|
|
local output = bindings.request(self, 'sendPhoto', {
|
|
|
|
|
chat_id = chat_id,
|
|
|
|
|
caption = text or nil,
|
2016-07-02 12:35:14 +02:00
|
|
|
|
reply_to_message_id = reply_to_message_id,
|
|
|
|
|
reply_markup = reply_markup
|
2016-06-12 20:53:20 +02:00
|
|
|
|
}, {photo = file} )
|
2016-07-03 01:20:04 +02:00
|
|
|
|
if string.match(file, '/tmp/') then
|
|
|
|
|
os.remove(file)
|
|
|
|
|
print("Deleted: "..file)
|
|
|
|
|
end
|
2016-06-12 20:53:20 +02:00
|
|
|
|
return output
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
-- https://core.telegram.org/bots/api#sendaudio
|
2016-06-20 17:58:46 +02:00
|
|
|
|
function utilities:send_audio(chat_id, file, reply_to_message_id, duration, performer, title)
|
2016-07-04 01:29:51 +02:00
|
|
|
|
if not file then return false end
|
2016-06-12 20:53:20 +02:00
|
|
|
|
local output = bindings.request(self, 'sendAudio', {
|
|
|
|
|
chat_id = chat_id,
|
|
|
|
|
duration = duration or nil,
|
|
|
|
|
performer = performer or nil,
|
|
|
|
|
title = title or nil,
|
|
|
|
|
reply_to_message_id = reply_to_message_id
|
|
|
|
|
}, {audio = file} )
|
2016-07-03 01:20:04 +02:00
|
|
|
|
if string.match(file, '/tmp/') then
|
|
|
|
|
os.remove(file)
|
|
|
|
|
print("Deleted: "..file)
|
|
|
|
|
end
|
2016-06-12 20:53:20 +02:00
|
|
|
|
return output
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
-- https://core.telegram.org/bots/api#senddocument
|
2016-07-02 12:35:14 +02:00
|
|
|
|
function utilities:send_document(chat_id, file, text, reply_to_message_id, reply_markup)
|
2016-07-04 01:29:51 +02:00
|
|
|
|
if not file then return false end
|
2016-06-12 20:53:20 +02:00
|
|
|
|
local output = bindings.request(self, 'sendDocument', {
|
|
|
|
|
chat_id = chat_id,
|
|
|
|
|
caption = text or nil,
|
2016-07-02 12:35:14 +02:00
|
|
|
|
reply_to_message_id = reply_to_message_id,
|
|
|
|
|
reply_markup = reply_markup
|
2016-06-12 20:53:20 +02:00
|
|
|
|
}, {document = file} )
|
2016-07-03 01:20:04 +02:00
|
|
|
|
if string.match(file, '/tmp/') then
|
|
|
|
|
os.remove(file)
|
|
|
|
|
print("Deleted: "..file)
|
|
|
|
|
end
|
2016-06-12 20:53:20 +02:00
|
|
|
|
return output
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
-- https://core.telegram.org/bots/api#sendvideo
|
|
|
|
|
function utilities:send_video(chat_id, file, text, reply_to_message_id, duration, width, height)
|
2016-07-04 01:29:51 +02:00
|
|
|
|
if not file then return false end
|
2016-06-12 20:53:20 +02:00
|
|
|
|
local output = bindings.request(self, 'sendVideo', {
|
|
|
|
|
chat_id = chat_id,
|
|
|
|
|
caption = text or nil,
|
|
|
|
|
duration = duration or nil,
|
|
|
|
|
width = width or nil,
|
|
|
|
|
height = height or nil,
|
|
|
|
|
reply_to_message_id = reply_to_message_id
|
|
|
|
|
}, {video = file} )
|
2016-07-03 01:20:04 +02:00
|
|
|
|
if string.match(file, '/tmp/') then
|
|
|
|
|
os.remove(file)
|
|
|
|
|
print("Deleted: "..file)
|
|
|
|
|
end
|
2016-06-12 20:53:20 +02:00
|
|
|
|
return output
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
-- NOTE: Voice messages are .ogg files encoded with OPUS
|
|
|
|
|
-- https://core.telegram.org/bots/api#sendvoice
|
|
|
|
|
function utilities:send_voice(chat_id, file, text, reply_to_message_id, duration)
|
2016-07-04 01:29:51 +02:00
|
|
|
|
if not file then return false end
|
2016-06-12 20:53:20 +02:00
|
|
|
|
local output = bindings.request(self, 'sendVoice', {
|
|
|
|
|
chat_id = chat_id,
|
|
|
|
|
duration = duration or nil,
|
|
|
|
|
reply_to_message_id = reply_to_message_id
|
|
|
|
|
}, {voice = file} )
|
2016-07-03 01:20:04 +02:00
|
|
|
|
if string.match(file, '/tmp/') then
|
|
|
|
|
os.remove(file)
|
|
|
|
|
print("Deleted: "..file)
|
|
|
|
|
end
|
2016-06-12 20:53:20 +02:00
|
|
|
|
return output
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
-- https://core.telegram.org/bots/api#sendlocation
|
|
|
|
|
function utilities:send_location(chat_id, latitude, longitude, reply_to_message_id)
|
|
|
|
|
return bindings.request(self, 'sendLocation', {
|
|
|
|
|
chat_id = chat_id,
|
|
|
|
|
latitude = latitude,
|
|
|
|
|
longitude = longitude,
|
|
|
|
|
reply_to_message_id = reply_to_message_id
|
|
|
|
|
} )
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
-- NOTE: Venue is different from location: it shows information, such as the street adress or
|
|
|
|
|
-- title of the location with it.
|
|
|
|
|
-- https://core.telegram.org/bots/api#sendvenue
|
|
|
|
|
function utilities:send_venue(chat_id, latitude, longitude, reply_to_message_id, title, address)
|
|
|
|
|
return bindings.request(self, 'sendVenue', {
|
|
|
|
|
chat_id = chat_id,
|
|
|
|
|
latitude = latitude,
|
|
|
|
|
longitude = longitude,
|
|
|
|
|
title = title,
|
|
|
|
|
address = address,
|
|
|
|
|
reply_to_message_id = reply_to_message_id
|
|
|
|
|
} )
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
-- https://core.telegram.org/bots/api#sendchataction
|
|
|
|
|
function utilities:send_typing(chat_id, action)
|
|
|
|
|
return bindings.request(self, 'sendChatAction', {
|
|
|
|
|
chat_id = chat_id,
|
|
|
|
|
action = action
|
|
|
|
|
} )
|
|
|
|
|
end
|
|
|
|
|
|
2016-07-02 12:35:14 +02:00
|
|
|
|
-- https://core.telegram.org/bots/api#answercallbackquery
|
|
|
|
|
function utilities:answer_callback_query(callback, text, show_alert)
|
|
|
|
|
return bindings.request(self, 'answerCallbackQuery', {
|
|
|
|
|
callback_query_id = callback.id,
|
|
|
|
|
text = text,
|
|
|
|
|
show_alert = show_alert
|
|
|
|
|
} )
|
|
|
|
|
end
|
|
|
|
|
|
2016-07-05 22:26:46 +02:00
|
|
|
|
-- https://core.telegram.org/bots/api#getchat
|
|
|
|
|
function utilities:get_chat_info(chat_id)
|
|
|
|
|
return bindings.request(self, 'getChat', {
|
|
|
|
|
chat_id = chat_id
|
|
|
|
|
} )
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
-- https://core.telegram.org/bots/api#getchatadministrators
|
|
|
|
|
function utilities:get_chat_administrators(chat_id)
|
|
|
|
|
return bindings.request(self, 'getChatAdministrators', {
|
|
|
|
|
chat_id = chat_id
|
|
|
|
|
} )
|
|
|
|
|
end
|
|
|
|
|
|
2016-07-13 01:00:32 +02:00
|
|
|
|
-- https://core.telegram.org/bots/api#answerinlinequery
|
|
|
|
|
function utilities:answer_inline_query(inline_query, results, cache_time, is_personal, next_offset, switch_pm_text, switch_pm_parameter)
|
|
|
|
|
return bindings.request(self, 'answerInlineQuery', {
|
|
|
|
|
inline_query_id = inline_query.id,
|
|
|
|
|
results = results,
|
|
|
|
|
cache_time = cache_time,
|
|
|
|
|
is_personal = is_personal,
|
|
|
|
|
next_offset = next_offset,
|
|
|
|
|
switch_pm_text = switch_pm_text,
|
|
|
|
|
switch_pm_parameter = switch_pm_parameter
|
|
|
|
|
} )
|
|
|
|
|
end
|
|
|
|
|
|
2016-01-12 11:22:28 +01:00
|
|
|
|
-- get the indexed word in a string
|
2016-04-08 23:12:02 +02:00
|
|
|
|
function utilities.get_word(s, i)
|
2016-01-08 14:44:37 +01:00
|
|
|
|
s = s or ''
|
|
|
|
|
i = i or 1
|
2015-12-13 22:31:22 +01:00
|
|
|
|
local t = {}
|
|
|
|
|
for w in s:gmatch('%g+') do
|
|
|
|
|
table.insert(t, w)
|
2015-07-03 00:15:52 +02:00
|
|
|
|
end
|
2015-12-13 22:31:22 +01:00
|
|
|
|
return t[i] or false
|
2016-03-31 13:53:12 +02:00
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
-- Like get_word(), but better.
|
|
|
|
|
-- Returns the actual index.
|
2016-04-08 23:12:02 +02:00
|
|
|
|
function utilities.index(s)
|
2016-03-31 13:53:12 +02:00
|
|
|
|
local t = {}
|
|
|
|
|
for w in s:gmatch('%g+') do
|
|
|
|
|
table.insert(t, w)
|
|
|
|
|
end
|
|
|
|
|
return t
|
2015-11-25 03:22:04 +01:00
|
|
|
|
end
|
2015-07-29 00:13:46 +02:00
|
|
|
|
|
2016-01-12 11:22:28 +01:00
|
|
|
|
-- Returns the string after the first space.
|
2016-04-08 23:12:02 +02:00
|
|
|
|
function utilities.input(s)
|
|
|
|
|
if not s:find(' ') then
|
2015-07-29 00:13:46 +02:00
|
|
|
|
return false
|
|
|
|
|
end
|
2016-04-08 23:12:02 +02:00
|
|
|
|
return s:sub(s:find(' ')+1)
|
2015-07-29 00:13:46 +02:00
|
|
|
|
end
|
|
|
|
|
|
2016-05-27 17:49:58 +02:00
|
|
|
|
-- Calculates the length of the given string as UTF-8 characters
|
|
|
|
|
function utilities.utf8_len(s)
|
|
|
|
|
local chars = 0
|
|
|
|
|
for i = 1, string.len(s) do
|
|
|
|
|
local b = string.byte(s, i)
|
|
|
|
|
if b < 128 or b >= 192 then
|
|
|
|
|
chars = chars + 1
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
return chars
|
|
|
|
|
end
|
|
|
|
|
|
2016-07-05 13:14:22 +02:00
|
|
|
|
-- Trims whitespace from a string.
|
otouto 3.11
"things occurred"
Added some utilities (id_from_username, id_from_message), removed some utilities (latcyr, others?).
Removed cycle-wasting "shortcuts" -- no more automatic id_str or name; text_lower remains.
Moved userdata (nicknames, lastfm, etc) to a different tree in the database (automatic migration will occur). /me now returns userdata.
Speaking of migration, database now stores the latest version run to make future automigration easy.
Database now saves hourly rather than minutely.
Changed readme and some plugins to reflect above changes.
Removed broken rockspec (Brayden, feel free to re-add once it's working).
Added option to automatically block people (via drua) when blacklisted.
Fixed about.lua trigger problems.
administration 1.11 - Removed /kickme and /broadcast. Users should leave manually, and announcements should be made via channel rather than spam. /setqotd now handles forwarded messages correctly. /kick, /ban, /hammer,
/mod, /admin now support multiple arguments. Added get_targets function. No migration is necessary.
2016-07-05 09:29:11 +02:00
|
|
|
|
function utilities.trim(str)
|
2016-04-08 23:12:02 +02:00
|
|
|
|
local s = str:gsub('^%s*(.-)%s*$', '%1')
|
2015-11-25 03:22:04 +01:00
|
|
|
|
return s
|
2015-07-03 00:15:52 +02:00
|
|
|
|
end
|
|
|
|
|
|
2016-08-02 01:01:06 +02:00
|
|
|
|
-- Returns true if the string is empty
|
2016-06-14 13:14:09 +02:00
|
|
|
|
function string:isempty()
|
|
|
|
|
return self == nil or self == ''
|
|
|
|
|
end
|
|
|
|
|
|
2016-08-03 19:21:07 +02:00
|
|
|
|
-- Returns true if the string is blank
|
2016-06-14 13:14:09 +02:00
|
|
|
|
function string:isblank()
|
|
|
|
|
self = self:trim()
|
|
|
|
|
return self:isempty()
|
|
|
|
|
end
|
|
|
|
|
|
2016-06-14 12:21:16 +02:00
|
|
|
|
function get_name(msg)
|
|
|
|
|
local name = msg.from.first_name
|
|
|
|
|
if name == nil then
|
|
|
|
|
name = msg.from.id
|
|
|
|
|
end
|
|
|
|
|
return name
|
|
|
|
|
end
|
|
|
|
|
|
2016-06-11 14:46:41 +02:00
|
|
|
|
-- http://www.lua.org/manual/5.2/manual.html#pdf-io.popen
|
|
|
|
|
function run_command(str)
|
|
|
|
|
local cmd = io.popen(str)
|
|
|
|
|
local result = cmd:read('*all')
|
|
|
|
|
cmd:close()
|
|
|
|
|
return result
|
|
|
|
|
end
|
|
|
|
|
|
2016-06-15 01:16:27 +02:00
|
|
|
|
function convert_timestamp(timestamp, format)
|
|
|
|
|
local converted_date = run_command('date -d @'..timestamp..' +"'..format..'"')
|
|
|
|
|
local converted_date = string.gsub(converted_date, '%\n', '')
|
|
|
|
|
return converted_date
|
|
|
|
|
end
|
|
|
|
|
|
2016-06-11 14:46:41 +02:00
|
|
|
|
function string.starts(String, Start)
|
|
|
|
|
return Start == string.sub(String,1,string.len(Start))
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
-- Saves file to $HOME/tmp/. If file_name isn't provided,
|
|
|
|
|
-- will get the text after the last "/" for filename
|
|
|
|
|
-- and content-type for extension
|
|
|
|
|
function download_to_file(url, file_name)
|
2016-06-19 21:53:24 +02:00
|
|
|
|
print('url to download: '..url)
|
|
|
|
|
if not file_name then
|
|
|
|
|
file_name = '/tmp/' .. url:match('.+/(.-)$') or '/tmp/' .. os.time()
|
|
|
|
|
else
|
|
|
|
|
file_name = '/tmp/' .. file_name
|
|
|
|
|
end
|
|
|
|
|
local body = {}
|
2016-07-31 21:29:44 +02:00
|
|
|
|
local doer = http
|
2016-06-19 21:53:24 +02:00
|
|
|
|
local do_redir = true
|
|
|
|
|
if url:match('^https') then
|
2016-07-31 21:29:44 +02:00
|
|
|
|
doer = https
|
2016-06-19 21:53:24 +02:00
|
|
|
|
do_redir = false
|
|
|
|
|
end
|
|
|
|
|
local _, res = doer.request{
|
|
|
|
|
url = url,
|
|
|
|
|
sink = ltn12.sink.table(body),
|
|
|
|
|
redirect = do_redir
|
|
|
|
|
}
|
|
|
|
|
if res ~= 200 then return false end
|
|
|
|
|
local file = io.open(file_name, 'w+')
|
|
|
|
|
file:write(table.concat(body))
|
|
|
|
|
file:close()
|
|
|
|
|
print('Saved to: '..file_name)
|
|
|
|
|
return file_name
|
2016-06-11 14:46:41 +02:00
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
function vardump(value)
|
|
|
|
|
print(serpent.block(value, {comment=false}))
|
|
|
|
|
end
|
|
|
|
|
|
2016-01-12 11:22:28 +01:00
|
|
|
|
-- Loads a JSON file as a table.
|
2016-04-08 23:12:02 +02:00
|
|
|
|
function utilities.load_data(filename)
|
2015-07-29 00:13:46 +02:00
|
|
|
|
local f = io.open(filename)
|
|
|
|
|
if not f then
|
|
|
|
|
return {}
|
|
|
|
|
end
|
|
|
|
|
local s = f:read('*all')
|
|
|
|
|
f:close()
|
2016-07-31 21:29:44 +02:00
|
|
|
|
local data = json.decode(s)
|
2015-07-29 00:13:46 +02:00
|
|
|
|
return data
|
|
|
|
|
end
|
|
|
|
|
|
2016-01-12 11:22:28 +01:00
|
|
|
|
-- Saves a table to a JSON file.
|
2016-04-08 23:12:02 +02:00
|
|
|
|
function utilities.save_data(filename, data)
|
2016-07-31 21:29:44 +02:00
|
|
|
|
local s = json.encode(data)
|
2015-07-29 00:13:46 +02:00
|
|
|
|
local f = io.open(filename, 'w')
|
|
|
|
|
f:write(s)
|
|
|
|
|
f:close()
|
2015-11-25 03:22:04 +01:00
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
-- Gets coordinates for a location. Used by gMaps.lua, time.lua, weather.lua.
|
2016-05-27 02:59:45 +02:00
|
|
|
|
function utilities.get_coords(input, config)
|
2016-08-06 23:11:53 +02:00
|
|
|
|
local url = 'https://maps.googleapis.com/maps/api/geocode/json?address='..URL.escape(input)..'&language=de'
|
|
|
|
|
local jstr, res = https.request(url)
|
|
|
|
|
if res ~= 200 then
|
|
|
|
|
return config.errors.connection
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
local jdat = json.decode(jstr)
|
|
|
|
|
if jdat.status == 'ZERO_RESULTS' then
|
|
|
|
|
return config.errors.results
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
return {
|
|
|
|
|
lat = jdat.results[1].geometry.location.lat,
|
|
|
|
|
lon = jdat.results[1].geometry.location.lng,
|
|
|
|
|
addr = jdat.results[1].formatted_address
|
|
|
|
|
}
|
2016-01-12 11:22:28 +01:00
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
-- Get the number of values in a key/value table.
|
2016-04-08 23:12:02 +02:00
|
|
|
|
function utilities.table_size(tab)
|
2016-01-12 11:22:28 +01:00
|
|
|
|
local i = 0
|
2016-04-08 23:12:02 +02:00
|
|
|
|
for _,_ in pairs(tab) do
|
2016-01-12 11:22:28 +01:00
|
|
|
|
i = i + 1
|
|
|
|
|
end
|
|
|
|
|
return i
|
|
|
|
|
end
|
|
|
|
|
|
2016-04-03 01:20:28 +02:00
|
|
|
|
-- Just an easy way to get a user's full name.
|
2016-05-13 19:22:10 +02:00
|
|
|
|
-- Alternatively, abuse it to concat two strings like I do.
|
2016-04-08 23:12:02 +02:00
|
|
|
|
function utilities.build_name(first, last)
|
2016-04-03 01:20:28 +02:00
|
|
|
|
if last then
|
|
|
|
|
return first .. ' ' .. last
|
|
|
|
|
else
|
|
|
|
|
return first
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
|
2016-04-08 23:12:02 +02:00
|
|
|
|
function utilities:resolve_username(input)
|
2016-03-22 11:16:26 +01:00
|
|
|
|
input = input:gsub('^@', '')
|
otouto 3.11
"things occurred"
Added some utilities (id_from_username, id_from_message), removed some utilities (latcyr, others?).
Removed cycle-wasting "shortcuts" -- no more automatic id_str or name; text_lower remains.
Moved userdata (nicknames, lastfm, etc) to a different tree in the database (automatic migration will occur). /me now returns userdata.
Speaking of migration, database now stores the latest version run to make future automigration easy.
Database now saves hourly rather than minutely.
Changed readme and some plugins to reflect above changes.
Removed broken rockspec (Brayden, feel free to re-add once it's working).
Added option to automatically block people (via drua) when blacklisted.
Fixed about.lua trigger problems.
administration 1.11 - Removed /kickme and /broadcast. Users should leave manually, and announcements should be made via channel rather than spam. /setqotd now handles forwarded messages correctly. /kick, /ban, /hammer,
/mod, /admin now support multiple arguments. Added get_targets function. No migration is necessary.
2016-07-05 09:29:11 +02:00
|
|
|
|
for _, user in pairs(self.database.users) do
|
|
|
|
|
if user.username and user.username:lower() == input:lower() then
|
|
|
|
|
local t = {}
|
|
|
|
|
for key, val in pairs(user) do
|
|
|
|
|
t[key] = val
|
|
|
|
|
end
|
|
|
|
|
return t
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
-- Simpler than above function; only returns an ID.
|
|
|
|
|
-- Returns nil if no ID is available.
|
|
|
|
|
function utilities:id_from_username(input)
|
|
|
|
|
input = input:gsub('^@', '')
|
|
|
|
|
for _, user in pairs(self.database.users) do
|
|
|
|
|
if user.username and user.username:lower() == input:lower() then
|
|
|
|
|
return user.id
|
2016-03-22 11:16:26 +01:00
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
|
otouto 3.11
"things occurred"
Added some utilities (id_from_username, id_from_message), removed some utilities (latcyr, others?).
Removed cycle-wasting "shortcuts" -- no more automatic id_str or name; text_lower remains.
Moved userdata (nicknames, lastfm, etc) to a different tree in the database (automatic migration will occur). /me now returns userdata.
Speaking of migration, database now stores the latest version run to make future automigration easy.
Database now saves hourly rather than minutely.
Changed readme and some plugins to reflect above changes.
Removed broken rockspec (Brayden, feel free to re-add once it's working).
Added option to automatically block people (via drua) when blacklisted.
Fixed about.lua trigger problems.
administration 1.11 - Removed /kickme and /broadcast. Users should leave manually, and announcements should be made via channel rather than spam. /setqotd now handles forwarded messages correctly. /kick, /ban, /hammer,
/mod, /admin now support multiple arguments. Added get_targets function. No migration is necessary.
2016-07-05 09:29:11 +02:00
|
|
|
|
-- Simpler than below function; only returns an ID.
|
|
|
|
|
-- Returns nil if no ID is available.
|
|
|
|
|
function utilities:id_from_message(msg)
|
|
|
|
|
if msg.reply_to_message then
|
|
|
|
|
return msg.reply_to_message.from.id
|
|
|
|
|
else
|
|
|
|
|
local input = utilities.input(msg.text)
|
|
|
|
|
if input then
|
|
|
|
|
if tonumber(input) then
|
|
|
|
|
return tonumber(input)
|
|
|
|
|
elseif input:match('^@') then
|
|
|
|
|
return utilities.id_from_username(self, input)
|
|
|
|
|
end
|
2016-03-22 11:16:26 +01:00
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
|
2016-05-15 14:22:31 +02:00
|
|
|
|
function utilities:user_from_message(msg, no_extra)
|
2016-04-08 23:12:02 +02:00
|
|
|
|
local input = utilities.input(msg.text_lower)
|
2016-03-22 11:16:26 +01:00
|
|
|
|
local target = {}
|
|
|
|
|
if msg.reply_to_message then
|
2016-05-15 14:22:31 +02:00
|
|
|
|
for k,v in pairs(self.database.users[msg.reply_to_message.from.id_str]) do
|
|
|
|
|
target[k] = v
|
|
|
|
|
end
|
2016-03-22 11:16:26 +01:00
|
|
|
|
elseif input and tonumber(input) then
|
2016-04-12 11:24:56 +02:00
|
|
|
|
target.id = tonumber(input)
|
2016-04-08 23:12:02 +02:00
|
|
|
|
if self.database.users[input] then
|
|
|
|
|
for k,v in pairs(self.database.users[input]) do
|
2016-03-22 11:16:26 +01:00
|
|
|
|
target[k] = v
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
elseif input and input:match('^@') then
|
2016-01-12 11:22:28 +01:00
|
|
|
|
local uname = input:gsub('^@', '')
|
2016-04-08 23:12:02 +02:00
|
|
|
|
for _,v in pairs(self.database.users) do
|
2016-03-22 11:16:26 +01:00
|
|
|
|
if v.username and uname == v.username:lower() then
|
|
|
|
|
for key, val in pairs(v) do
|
|
|
|
|
target[key] = val
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
if not target.id then
|
|
|
|
|
target.err = 'Sorry, I don\'t recognize that username.'
|
|
|
|
|
end
|
2016-01-12 11:22:28 +01:00
|
|
|
|
else
|
2016-03-22 11:16:26 +01:00
|
|
|
|
target.err = 'Please specify a user via reply, ID, or username.'
|
|
|
|
|
end
|
|
|
|
|
|
2016-05-15 14:22:31 +02:00
|
|
|
|
if not no_extra then
|
|
|
|
|
if target.id then
|
|
|
|
|
target.id_str = tostring(target.id)
|
|
|
|
|
end
|
|
|
|
|
if not target.first_name then
|
|
|
|
|
target.first_name = 'User'
|
|
|
|
|
end
|
|
|
|
|
target.name = utilities.build_name(target.first_name, target.last_name)
|
2016-01-12 11:22:28 +01:00
|
|
|
|
end
|
|
|
|
|
|
2016-03-22 11:16:26 +01:00
|
|
|
|
return target
|
|
|
|
|
|
2015-07-29 00:13:46 +02:00
|
|
|
|
end
|
2016-01-13 19:00:17 +01:00
|
|
|
|
|
2016-05-27 02:26:30 +02:00
|
|
|
|
function utilities:handle_exception(err, message, config)
|
2016-01-13 19:00:17 +01:00
|
|
|
|
|
2016-02-20 11:07:20 +01:00
|
|
|
|
if not err then err = '' end
|
|
|
|
|
|
2016-04-08 23:12:02 +02:00
|
|
|
|
local output = '\n[' .. os.date('%F %T', os.time()) .. ']\n' .. self.info.username .. ': ' .. err .. '\n' .. message .. '\n'
|
2016-01-13 19:00:17 +01:00
|
|
|
|
|
2016-05-27 02:26:30 +02:00
|
|
|
|
if config.log_chat then
|
2016-01-13 19:00:17 +01:00
|
|
|
|
output = '```' .. output .. '```'
|
2016-05-27 02:26:30 +02:00
|
|
|
|
utilities.send_message(self, config.log_chat, output, true, nil, true)
|
2016-01-13 19:00:17 +01:00
|
|
|
|
else
|
|
|
|
|
print(output)
|
|
|
|
|
end
|
|
|
|
|
|
2016-01-15 04:39:24 +01:00
|
|
|
|
end
|
|
|
|
|
|
2016-06-19 21:53:24 +02:00
|
|
|
|
-- MOVED TO DOWNLOAD_TO_FILE
|
2016-04-08 23:12:02 +02:00
|
|
|
|
function utilities.download_file(url, filename)
|
2016-06-19 21:53:24 +02:00
|
|
|
|
return download_to_file(url, filename)
|
2016-01-13 19:00:17 +01:00
|
|
|
|
end
|
2016-03-22 11:16:26 +01:00
|
|
|
|
|
2016-04-08 23:12:02 +02:00
|
|
|
|
function utilities.markdown_escape(text)
|
2016-03-22 11:16:26 +01:00
|
|
|
|
text = text:gsub('_', '\\_')
|
2016-07-17 20:58:06 +02:00
|
|
|
|
text = text:gsub('%[', '\\[')
|
2016-07-16 22:30:51 +02:00
|
|
|
|
text = text:gsub('%*', '\\*')
|
|
|
|
|
text = text:gsub('`', '\\`')
|
|
|
|
|
return text
|
|
|
|
|
end
|
|
|
|
|
|
2016-04-26 07:40:31 +02:00
|
|
|
|
utilities.md_escape = utilities.markdown_escape
|
2016-04-08 23:12:02 +02:00
|
|
|
|
|
2016-04-14 05:48:20 +02:00
|
|
|
|
utilities.triggers_meta = {}
|
|
|
|
|
utilities.triggers_meta.__index = utilities.triggers_meta
|
|
|
|
|
function utilities.triggers_meta:t(pattern, has_args)
|
2016-04-15 21:07:23 +02:00
|
|
|
|
local username = self.username:lower()
|
2016-05-27 05:28:44 +02:00
|
|
|
|
table.insert(self.table, '^'..self.cmd_pat..pattern..'$')
|
|
|
|
|
table.insert(self.table, '^'..self.cmd_pat..pattern..'@'..username..'$')
|
2016-04-08 23:12:02 +02:00
|
|
|
|
if has_args then
|
2016-05-27 05:28:44 +02:00
|
|
|
|
table.insert(self.table, '^'..self.cmd_pat..pattern..'%s+[^%s]*')
|
|
|
|
|
table.insert(self.table, '^'..self.cmd_pat..pattern..'@'..username..'%s+[^%s]*')
|
2016-04-08 23:12:02 +02:00
|
|
|
|
end
|
|
|
|
|
return self
|
|
|
|
|
end
|
|
|
|
|
|
2016-05-27 05:28:44 +02:00
|
|
|
|
function utilities.triggers(username, cmd_pat, trigger_table)
|
2016-04-14 05:48:20 +02:00
|
|
|
|
local self = setmetatable({}, utilities.triggers_meta)
|
2016-04-08 23:12:02 +02:00
|
|
|
|
self.username = username
|
2016-05-27 05:28:44 +02:00
|
|
|
|
self.cmd_pat = cmd_pat
|
2016-04-11 06:04:47 +02:00
|
|
|
|
self.table = trigger_table or {}
|
2016-04-08 23:12:02 +02:00
|
|
|
|
return self
|
2016-04-01 19:29:00 +02:00
|
|
|
|
end
|
2016-04-11 06:04:47 +02:00
|
|
|
|
|
|
|
|
|
function utilities.with_http_timeout(timeout, fun)
|
2016-07-31 21:29:44 +02:00
|
|
|
|
local original = http.TIMEOUT
|
|
|
|
|
http.TIMEOUT = timeout
|
2016-04-11 06:04:47 +02:00
|
|
|
|
fun()
|
2016-07-31 21:29:44 +02:00
|
|
|
|
http.TIMEOUT = original
|
2016-04-11 06:04:47 +02:00
|
|
|
|
end
|
2016-04-12 11:24:56 +02:00
|
|
|
|
|
2016-04-12 15:47:30 +02:00
|
|
|
|
function utilities.enrich_user(user)
|
2016-04-12 11:24:56 +02:00
|
|
|
|
user.id_str = tostring(user.id)
|
2016-04-12 15:47:30 +02:00
|
|
|
|
user.name = utilities.build_name(user.first_name, user.last_name)
|
2016-04-12 11:24:56 +02:00
|
|
|
|
return user
|
|
|
|
|
end
|
|
|
|
|
|
2016-04-12 15:47:30 +02:00
|
|
|
|
function utilities.enrich_message(msg)
|
2016-04-12 11:24:56 +02:00
|
|
|
|
if not msg.text then msg.text = msg.caption or '' end
|
|
|
|
|
msg.text_lower = msg.text:lower()
|
2016-04-12 15:47:30 +02:00
|
|
|
|
msg.from = utilities.enrich_user(msg.from)
|
2016-04-12 11:24:56 +02:00
|
|
|
|
msg.chat.id_str = tostring(msg.chat.id)
|
|
|
|
|
if msg.reply_to_message then
|
|
|
|
|
if not msg.reply_to_message.text then
|
|
|
|
|
msg.reply_to_message.text = msg.reply_to_message.caption or ''
|
|
|
|
|
end
|
|
|
|
|
msg.reply_to_message.text_lower = msg.reply_to_message.text:lower()
|
2016-04-12 15:47:30 +02:00
|
|
|
|
msg.reply_to_message.from = utilities.enrich_user(msg.reply_to_message.from)
|
2016-04-12 11:24:56 +02:00
|
|
|
|
msg.reply_to_message.chat.id_str = tostring(msg.reply_to_message.chat.id)
|
|
|
|
|
end
|
|
|
|
|
if msg.forward_from then
|
2016-04-12 15:47:30 +02:00
|
|
|
|
msg.forward_from = utilities.enrich_user(msg.forward_from)
|
2016-04-12 11:24:56 +02:00
|
|
|
|
end
|
|
|
|
|
if msg.new_chat_participant then
|
2016-04-12 15:47:30 +02:00
|
|
|
|
msg.new_chat_participant = utilities.enrich_user(msg.new_chat_participant)
|
2016-04-12 11:24:56 +02:00
|
|
|
|
end
|
|
|
|
|
if msg.left_chat_participant then
|
2016-04-12 15:47:30 +02:00
|
|
|
|
msg.left_chat_participant = utilities.enrich_user(msg.left_chat_participant)
|
2016-04-12 11:24:56 +02:00
|
|
|
|
end
|
|
|
|
|
return msg
|
|
|
|
|
end
|
2016-04-14 05:48:20 +02:00
|
|
|
|
|
2016-04-15 21:07:23 +02:00
|
|
|
|
function utilities.pretty_float(x)
|
|
|
|
|
if x % 1 == 0 then
|
|
|
|
|
return tostring(math.floor(x))
|
|
|
|
|
else
|
|
|
|
|
return tostring(x)
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
|
2016-05-20 09:38:43 +02:00
|
|
|
|
-- This table will store unsavory characters that are not properly displayed,
|
|
|
|
|
-- or are just not fun to type.
|
|
|
|
|
utilities.char = {
|
|
|
|
|
zwnj = '',
|
|
|
|
|
arabic = '[\216-\219][\128-\191]',
|
2016-05-22 22:08:45 +02:00
|
|
|
|
rtl_override = '',
|
2016-05-25 15:01:54 +02:00
|
|
|
|
rtl_mark = '',
|
2016-06-17 20:44:28 +02:00
|
|
|
|
em_dash = '—'
|
2016-05-20 09:38:43 +02:00
|
|
|
|
}
|
|
|
|
|
|
2016-06-18 12:51:13 +02:00
|
|
|
|
-- taken from http://stackoverflow.com/a/11130774/3163199
|
|
|
|
|
function scandir(directory)
|
|
|
|
|
local i, t, popen = 0, {}, io.popen
|
|
|
|
|
for filename in popen('ls -a "'..directory..'"'):lines() do
|
|
|
|
|
i = i + 1
|
|
|
|
|
t[i] = filename
|
|
|
|
|
end
|
|
|
|
|
return t
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
-- Returns at table of lua files inside plugins
|
|
|
|
|
function plugins_names()
|
|
|
|
|
local files = {}
|
|
|
|
|
for k, v in pairs(scandir("otouto/plugins")) do
|
|
|
|
|
-- Ends with .lua
|
|
|
|
|
if (v:match(".lua$")) then
|
2016-08-07 20:45:51 +02:00
|
|
|
|
files[#files+1] = v
|
2016-06-18 12:51:13 +02:00
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
return files
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
-- Function name explains what it does.
|
|
|
|
|
function file_exists(name)
|
|
|
|
|
local f = io.open(name,"r")
|
|
|
|
|
if f ~= nil then
|
|
|
|
|
io.close(f)
|
|
|
|
|
return true
|
|
|
|
|
else
|
|
|
|
|
return false
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
|
2016-06-13 22:18:36 +02:00
|
|
|
|
-- Returns a table with matches or nil
|
|
|
|
|
function match_pattern(pattern, text)
|
|
|
|
|
if text then
|
|
|
|
|
local matches = { string.match(text, pattern) }
|
|
|
|
|
if next(matches) then
|
|
|
|
|
return matches
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
-- nil
|
|
|
|
|
end
|
|
|
|
|
|
2016-06-17 20:44:28 +02:00
|
|
|
|
function is_sudo(msg, config)
|
|
|
|
|
local var = false
|
|
|
|
|
-- Check if user id is sudoer
|
|
|
|
|
if config.admin == msg.from.id then
|
|
|
|
|
var = true
|
|
|
|
|
end
|
|
|
|
|
return var
|
|
|
|
|
end
|
|
|
|
|
|
2016-08-01 21:51:37 +02:00
|
|
|
|
function service_modify_msg(msg)
|
|
|
|
|
if msg.new_chat_member then
|
|
|
|
|
msg.text = '//tgservice new_chat_member'
|
|
|
|
|
msg.text_lower = msg.text
|
|
|
|
|
elseif msg.left_chat_member then
|
|
|
|
|
msg.text = '//tgservice left_chat_member'
|
|
|
|
|
msg.text_lower = msg.text
|
|
|
|
|
elseif msg.new_chat_title then
|
|
|
|
|
msg.text = '//tgservice new_chat_title'
|
|
|
|
|
msg.text_lower = msg.text
|
|
|
|
|
elseif msg.new_chat_photo then
|
|
|
|
|
msg.text = '//tgservice new_chat_photo'
|
|
|
|
|
msg.text_lower = msg.text
|
|
|
|
|
elseif msg.group_chat_created then
|
|
|
|
|
msg.text = '//tgservice group_chat_created'
|
|
|
|
|
msg.text_lower = msg.text
|
|
|
|
|
elseif msg.supergroup_chat_created then
|
|
|
|
|
msg.text = '//tgservice supergroup_chat_created'
|
|
|
|
|
msg.text_lower = msg.text
|
|
|
|
|
elseif msg.channel_chat_created then
|
|
|
|
|
msg.text = '//tgservice channel_chat_created'
|
|
|
|
|
msg.text_lower = msg.text
|
|
|
|
|
elseif msg.migrate_to_chat_id then
|
|
|
|
|
msg.text = '//tgservice migrate_to_chat_id'
|
|
|
|
|
msg.text_lower = msg.text
|
|
|
|
|
elseif msg.migrate_from_chat_id then
|
|
|
|
|
msg.text = '//tgservice migrate_from_chat_id'
|
|
|
|
|
msg.text_lower = msg.text
|
|
|
|
|
end
|
|
|
|
|
return msg
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
function is_service_msg(msg)
|
|
|
|
|
local var = false
|
|
|
|
|
if msg.new_chat_member then
|
|
|
|
|
var = true
|
|
|
|
|
elseif msg.left_chat_member then
|
|
|
|
|
var = true
|
|
|
|
|
elseif msg.new_chat_title then
|
|
|
|
|
var = true
|
|
|
|
|
elseif msg.new_chat_photo then
|
|
|
|
|
var = true
|
|
|
|
|
elseif msg.group_chat_created then
|
|
|
|
|
var = true
|
|
|
|
|
elseif msg.supergroup_chat_created then
|
|
|
|
|
var = true
|
|
|
|
|
elseif msg.channel_chat_created then
|
|
|
|
|
var = true
|
|
|
|
|
elseif msg.migrate_to_chat_id then
|
|
|
|
|
var = true
|
|
|
|
|
elseif msg.migrate_from_chat_id then
|
|
|
|
|
var = true
|
|
|
|
|
end
|
|
|
|
|
return var
|
|
|
|
|
end
|
|
|
|
|
|
2016-06-14 13:14:09 +02:00
|
|
|
|
function post_petition(url, arguments, headers)
|
|
|
|
|
local url, h = string.gsub(url, "http://", "")
|
|
|
|
|
local url, hs = string.gsub(url, "https://", "")
|
|
|
|
|
local post_prot = "http"
|
|
|
|
|
if hs == 1 then
|
|
|
|
|
post_prot = "https"
|
|
|
|
|
end
|
|
|
|
|
local response_body = {}
|
|
|
|
|
local request_constructor = {
|
|
|
|
|
url = post_prot..'://'..url,
|
|
|
|
|
method = "POST",
|
|
|
|
|
sink = ltn12.sink.table(response_body),
|
|
|
|
|
headers = headers or {},
|
|
|
|
|
redirect = false
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
local source = arguments
|
|
|
|
|
if type(arguments) == "table" then
|
2016-07-26 16:39:18 +02:00
|
|
|
|
source = helpers.url_encode_arguments(arguments)
|
2016-06-14 13:14:09 +02:00
|
|
|
|
end
|
2016-07-26 16:39:18 +02:00
|
|
|
|
|
2016-06-14 13:14:09 +02:00
|
|
|
|
if not headers then
|
|
|
|
|
request_constructor.headers["Content-Type"] = "application/x-www-form-urlencoded; charset=UTF8"
|
|
|
|
|
request_constructor.headers["X-Accept"] = "application/json"
|
|
|
|
|
request_constructor.headers["Accept"] = "application/json"
|
|
|
|
|
end
|
|
|
|
|
request_constructor.headers["Content-Length"] = tostring(#source)
|
|
|
|
|
request_constructor.source = ltn12.source.string(source)
|
|
|
|
|
|
|
|
|
|
if post_prot == "http" then
|
|
|
|
|
ok, response_code, response_headers, response_status_line = http.request(request_constructor)
|
|
|
|
|
else
|
2016-07-31 21:29:44 +02:00
|
|
|
|
ok, response_code, response_headers, response_status_line = https.request(request_constructor)
|
2016-06-14 13:14:09 +02:00
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
if not ok then
|
|
|
|
|
return nil
|
|
|
|
|
end
|
|
|
|
|
|
2016-07-31 21:29:44 +02:00
|
|
|
|
response_body = json.decode(table.concat(response_body))
|
2016-06-14 13:14:09 +02:00
|
|
|
|
|
|
|
|
|
return response_body, response_headers
|
|
|
|
|
end
|
|
|
|
|
|
2016-06-11 14:46:41 +02:00
|
|
|
|
function get_redis_hash(msg, var)
|
|
|
|
|
if msg.chat.type == 'group' or msg.chat.type == 'supergroup' then
|
|
|
|
|
return 'chat:'..msg.chat.id..':'..var
|
|
|
|
|
end
|
|
|
|
|
if msg.chat.type == 'private' then
|
|
|
|
|
return 'user:'..msg.from.id..':'..var
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
-- remove whitespace
|
|
|
|
|
function all_trim(s)
|
|
|
|
|
return s:match( "^%s*(.-)%s*$" )
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
function tablelength(T)
|
|
|
|
|
local count = 0
|
|
|
|
|
for _ in pairs(T) do count = count + 1 end
|
|
|
|
|
return count
|
|
|
|
|
end
|
|
|
|
|
|
2016-06-15 01:16:27 +02:00
|
|
|
|
function round(num, idp)
|
|
|
|
|
if idp and idp>0 then
|
|
|
|
|
local mult = 10^idp
|
|
|
|
|
return math.floor(num * mult + 0.5) / mult
|
|
|
|
|
end
|
|
|
|
|
return math.floor(num + 0.5)
|
|
|
|
|
end
|
|
|
|
|
|
2016-06-11 14:46:41 +02:00
|
|
|
|
function comma_value(amount)
|
|
|
|
|
local formatted = amount
|
|
|
|
|
while true do
|
|
|
|
|
formatted, k = string.gsub(formatted, "^(-?%d+)(%d%d%d)", '%1.%2')
|
|
|
|
|
if (k==0) then
|
|
|
|
|
break
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
return formatted
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
function string.ends(str, fin)
|
|
|
|
|
return fin=='' or string.sub(str,-string.len(fin)) == fin
|
|
|
|
|
end
|
|
|
|
|
|
2016-06-14 16:17:13 +02:00
|
|
|
|
function get_location(user_id)
|
|
|
|
|
local hash = 'user:'..user_id
|
|
|
|
|
local set_location = redis:hget(hash, 'location')
|
|
|
|
|
if set_location == 'false' or set_location == nil then
|
|
|
|
|
return false
|
|
|
|
|
else
|
|
|
|
|
return set_location
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
|
2016-06-11 14:46:41 +02:00
|
|
|
|
function cache_data(plugin, query, data, timeout, typ)
|
|
|
|
|
-- How to: cache_data(pluginname, query_name, data_to_cache, expire_in_seconds)
|
|
|
|
|
local hash = 'telegram:cache:'..plugin..':'..query
|
|
|
|
|
if timeout then
|
|
|
|
|
print('Caching "'..query..'" from plugin '..plugin..' (expires in '..timeout..' seconds)')
|
|
|
|
|
else
|
|
|
|
|
print('Caching "'..query..'" from plugin '..plugin..' (expires never)')
|
|
|
|
|
end
|
|
|
|
|
if typ == 'key' then
|
|
|
|
|
redis:set(hash, data)
|
|
|
|
|
elseif typ == 'set' then
|
|
|
|
|
-- make sure that you convert your data into a table:
|
|
|
|
|
-- {"foo", "bar", "baz"} instead of
|
|
|
|
|
-- {"bar" = "foo", "foo" = "bar", "bar" = "baz"}
|
|
|
|
|
-- because other formats are not supported by redis (or I haven't found a way to store them)
|
|
|
|
|
for _,str in pairs(data) do
|
|
|
|
|
redis:sadd(hash, str)
|
|
|
|
|
end
|
|
|
|
|
else
|
|
|
|
|
redis:hmset(hash, data)
|
|
|
|
|
end
|
|
|
|
|
if timeout then
|
|
|
|
|
redis:expire(hash, timeout)
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
|
2016-07-05 19:21:37 +02:00
|
|
|
|
-- Caches file_id and last_modified
|
|
|
|
|
-- result = result of send_X() (see media.lua)
|
|
|
|
|
function cache_file(result, url, last_modified)
|
|
|
|
|
local hash = 'telegram:cache:sent_file'
|
|
|
|
|
if result.result.video then
|
|
|
|
|
file_id = result.result.video.file_id
|
|
|
|
|
elseif result.result.audio then
|
|
|
|
|
file_id = result.result.audio.file_id
|
|
|
|
|
elseif result.result.voice then
|
|
|
|
|
file_id = result.result.voice.file_id
|
|
|
|
|
elseif result.result.document then
|
|
|
|
|
file_id = result.result.document.file_id
|
|
|
|
|
elseif result.result.photo then
|
|
|
|
|
local lv = #result.result.photo
|
|
|
|
|
file_id = result.result.photo[lv].file_id
|
|
|
|
|
end
|
|
|
|
|
print('Caching File...')
|
|
|
|
|
redis:hset(hash..':'..url, 'file_id', file_id)
|
|
|
|
|
redis:hset(hash..':'..url, 'last_modified', last_modified)
|
|
|
|
|
-- Why do we set a TTL? Because Telegram recycles outgoing file_id's
|
|
|
|
|
-- See: https://core.telegram.org/bots/faq#can-i-count-on-file-ids-to-be-persistent
|
|
|
|
|
redis:expire(hash..':'..url, 5259600) -- 2 months
|
|
|
|
|
end
|
|
|
|
|
|
2016-07-23 13:58:15 +02:00
|
|
|
|
function get_http_header(url)
|
2016-08-01 01:20:45 +02:00
|
|
|
|
local doer = http
|
2016-07-05 19:21:37 +02:00
|
|
|
|
local do_redir = true
|
|
|
|
|
if url:match('^https') then
|
2016-07-31 21:29:44 +02:00
|
|
|
|
doer = https
|
2016-07-05 19:21:37 +02:00
|
|
|
|
do_redir = false
|
|
|
|
|
end
|
|
|
|
|
local _, code, header = doer.request {
|
|
|
|
|
method = "HEAD",
|
|
|
|
|
url = url,
|
|
|
|
|
redirect = do_redir
|
|
|
|
|
}
|
2016-07-07 00:31:19 +02:00
|
|
|
|
if not header then return end
|
2016-07-23 13:58:15 +02:00
|
|
|
|
return header, code
|
2016-07-05 19:21:37 +02:00
|
|
|
|
end
|
|
|
|
|
|
2016-08-03 18:35:58 +02:00
|
|
|
|
-- checks with If-Modified-Since header, if url has been changed
|
|
|
|
|
-- URL and Last-Modified heder are required
|
|
|
|
|
function was_modified_since(url, last_modified)
|
|
|
|
|
local doer = http
|
|
|
|
|
local do_redir = true
|
|
|
|
|
if url:match('^https') then
|
|
|
|
|
doer = https
|
|
|
|
|
do_redir = false
|
|
|
|
|
end
|
|
|
|
|
local _, code, header = doer.request {
|
|
|
|
|
url = url,
|
|
|
|
|
method = "HEAD",
|
|
|
|
|
redirect = do_redir,
|
|
|
|
|
headers = {
|
|
|
|
|
["If-Modified-Since"] = last_modified
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
if code == 304 then
|
|
|
|
|
return false, nil, code
|
|
|
|
|
else
|
|
|
|
|
if header["last-modified"] then
|
|
|
|
|
new_last_modified = header["last-modified"]
|
|
|
|
|
elseif header["Last-Modified"] then
|
|
|
|
|
new_last_modified = header["Last-Modified"]
|
|
|
|
|
end
|
|
|
|
|
return true, new_last_modified, code
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
|
2016-07-05 19:21:37 +02:00
|
|
|
|
-- only url is needed!
|
|
|
|
|
function get_cached_file(url, file_name, receiver, chat_action, self)
|
|
|
|
|
local hash = 'telegram:cache:sent_file'
|
|
|
|
|
local cached_file_id = redis:hget(hash..':'..url, 'file_id')
|
|
|
|
|
local cached_last_modified = redis:hget(hash..':'..url, 'last_modified')
|
|
|
|
|
|
2016-08-03 18:35:58 +02:00
|
|
|
|
if cached_last_modified then
|
|
|
|
|
was_modified, new_last_modified, code = was_modified_since(url, cached_last_modified)
|
|
|
|
|
if not was_modified then
|
|
|
|
|
print('File wasn\'t modified, skipping download...')
|
|
|
|
|
return cached_file_id, nil, true
|
|
|
|
|
else
|
|
|
|
|
if code ~= 200 then
|
|
|
|
|
redis:del(hash..':'..url)
|
|
|
|
|
return
|
|
|
|
|
end
|
|
|
|
|
print('File was modified, redownloading...')
|
|
|
|
|
if receiver and chat_action and self then
|
|
|
|
|
utilities.send_typing(self, receiver, chat_action)
|
|
|
|
|
end
|
|
|
|
|
file = download_to_file(url, file_name)
|
|
|
|
|
return file, new_last_modified, false
|
2016-07-05 19:21:37 +02:00
|
|
|
|
end
|
|
|
|
|
end
|
2016-07-23 13:58:15 +02:00
|
|
|
|
|
2016-08-03 18:35:58 +02:00
|
|
|
|
-- get last-modified and Content-Length header
|
|
|
|
|
local header, code = get_http_header(url)
|
|
|
|
|
|
2016-07-23 13:58:15 +02:00
|
|
|
|
-- file size limit is 50 MB
|
|
|
|
|
if header["Content-Length"] then
|
|
|
|
|
if tonumber(header["Content-Length"]) > 52420000 then
|
|
|
|
|
print('file is too large, won\'t send!')
|
|
|
|
|
return nil
|
|
|
|
|
end
|
|
|
|
|
elseif header["content-length"] then
|
|
|
|
|
if tonumber(header["content-length"]) > 52420000 then
|
|
|
|
|
print('file is too large, won\'t send!')
|
|
|
|
|
return nil
|
|
|
|
|
end
|
|
|
|
|
end
|
2016-08-03 18:35:58 +02:00
|
|
|
|
|
2016-07-23 13:58:15 +02:00
|
|
|
|
if header["last-modified"] then
|
|
|
|
|
last_modified = header["last-modified"]
|
|
|
|
|
elseif header["Last-Modified"] then
|
|
|
|
|
last_modified = header["Last-Modified"]
|
|
|
|
|
end
|
2016-07-05 19:21:37 +02:00
|
|
|
|
|
|
|
|
|
if not last_modified then
|
|
|
|
|
nocache = true
|
|
|
|
|
else
|
|
|
|
|
nocache = false
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
if receiver and chat_action and self then
|
|
|
|
|
utilities.send_typing(self, receiver, chat_action)
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
if not nocache then
|
2016-08-03 18:35:58 +02:00
|
|
|
|
file = download_to_file(url, file_name)
|
2016-07-05 19:21:37 +02:00
|
|
|
|
else
|
|
|
|
|
print('No Last-Modified header!')
|
|
|
|
|
file = download_to_file(url, file_name)
|
|
|
|
|
end
|
|
|
|
|
return file, last_modified, nocache
|
|
|
|
|
end
|
|
|
|
|
|
2016-06-11 14:46:41 +02:00
|
|
|
|
-- converts total amount of seconds (e.g. 65 seconds) to human redable time (e.g. 1:05 minutes)
|
|
|
|
|
function makeHumanTime(totalseconds)
|
|
|
|
|
local seconds = totalseconds % 60
|
|
|
|
|
local minutes = math.floor(totalseconds / 60)
|
|
|
|
|
local minutes = minutes % 60
|
|
|
|
|
local hours = math.floor(totalseconds / 3600)
|
|
|
|
|
if minutes == 00 and hours == 00 then
|
2016-08-02 16:19:20 +02:00
|
|
|
|
if seconds == 1 then
|
|
|
|
|
return seconds..' Sekunde'
|
|
|
|
|
else
|
|
|
|
|
return seconds..' Sekunden'
|
|
|
|
|
end
|
2016-06-11 14:46:41 +02:00
|
|
|
|
elseif hours == 00 and minutes ~= 00 then
|
2016-08-02 16:19:20 +02:00
|
|
|
|
if minutes == 1 then
|
|
|
|
|
return string.format("%02d:%02d", minutes, seconds)..' Minute'
|
|
|
|
|
else
|
|
|
|
|
return string.format("%02d:%02d", minutes, seconds)..' Minuten'
|
|
|
|
|
end
|
2016-06-11 14:46:41 +02:00
|
|
|
|
elseif hours ~= 00 then
|
2016-08-02 16:19:20 +02:00
|
|
|
|
if hours == 1 then
|
|
|
|
|
return string.format("%02d:%02d:%02d", hours, minutes, seconds)..' Stunde'
|
|
|
|
|
else
|
|
|
|
|
return string.format("%02d:%02d:%02d", hours, minutes, seconds)..' Stunden'
|
|
|
|
|
end
|
2016-06-11 14:46:41 +02:00
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
function is_blacklisted(msg)
|
|
|
|
|
_blacklist = redis:smembers("telegram:img_blacklist")
|
|
|
|
|
local var = false
|
|
|
|
|
for v,word in pairs(_blacklist) do
|
|
|
|
|
if string.find(string.lower(msg), string.lower(word)) then
|
|
|
|
|
print("Wort steht auf der Blacklist!")
|
|
|
|
|
var = true
|
|
|
|
|
break
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
return var
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
function unescape(str)
|
|
|
|
|
str = string.gsub( str, '<', '<' )
|
|
|
|
|
str = string.gsub( str, '>', '>' )
|
|
|
|
|
str = string.gsub( str, '"', '"' )
|
|
|
|
|
str = string.gsub( str, ''', "'" )
|
|
|
|
|
str = string.gsub( str, "Ä", "Ä")
|
|
|
|
|
str = string.gsub( str, "ä", "ä")
|
|
|
|
|
str = string.gsub( str, "Ö", "Ö")
|
|
|
|
|
str = string.gsub( str, "ö", "ö")
|
|
|
|
|
str = string.gsub( str, "Uuml;", "Ü")
|
|
|
|
|
str = string.gsub( str, "ü", "ü")
|
|
|
|
|
str = string.gsub( str, "ß", "ß")
|
|
|
|
|
str = string.gsub( str, '&#(%d+);', function(n) return string.char(n) end )
|
|
|
|
|
str = string.gsub( str, '&#x(%d+);', function(n) return string.char(tonumber(n,16)) end )
|
|
|
|
|
str = string.gsub( str, '&', '&' ) -- Be sure to do this after all others
|
|
|
|
|
return str
|
|
|
|
|
end
|
|
|
|
|
|
2016-06-15 18:00:59 +02:00
|
|
|
|
function url_encode(str)
|
|
|
|
|
if (str) then
|
|
|
|
|
str = string.gsub (str, "\n", "\r\n")
|
|
|
|
|
str = string.gsub (str, "([^%w %-%_%.%~])",
|
|
|
|
|
function (c) return string.format ("%%%02X", string.byte(c)) end)
|
|
|
|
|
str = string.gsub (str, " ", "+")
|
|
|
|
|
end
|
|
|
|
|
return str
|
|
|
|
|
end
|
|
|
|
|
|
2016-07-05 22:26:46 +02:00
|
|
|
|
function table.contains(table, element)
|
|
|
|
|
for _, value in pairs(table) do
|
|
|
|
|
if value == element then
|
|
|
|
|
return true
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
return false
|
|
|
|
|
end
|
|
|
|
|
|
2016-07-14 11:04:39 +02:00
|
|
|
|
return utilities
|