2015-07-03 00:15:52 +02:00
|
|
|
|
-- utilities.lua
|
|
|
|
|
-- Functions shared among plugins.
|
|
|
|
|
|
2016-04-08 23:12:02 +02:00
|
|
|
|
local utilities = {}
|
|
|
|
|
|
|
|
|
|
local HTTP = require('socket.http')
|
|
|
|
|
local ltn12 = require('ltn12')
|
|
|
|
|
local HTTPS = require('ssl.https')
|
|
|
|
|
local URL = require('socket.url')
|
2016-04-15 21:07:23 +02:00
|
|
|
|
local JSON = require('dkjson')
|
2016-06-14 13:14:09 +02:00
|
|
|
|
local http = require('socket.http')
|
2016-06-11 14:46:41 +02:00
|
|
|
|
local serpent = require("serpent")
|
2016-06-07 06:31:34 +02:00
|
|
|
|
local bindings = require('otouto.bindings')
|
2016-06-11 14:46:41 +02:00
|
|
|
|
local redis = (loadfile "./otouto/redis.lua")()
|
|
|
|
|
local mimetype = (loadfile "./otouto/mimetype.lua")()
|
2016-01-15 04:39:24 +01:00
|
|
|
|
|
2016-07-12 00:04:42 +02:00
|
|
|
|
HTTP.timeout = 10
|
|
|
|
|
|
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-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-06-27 15:44:11 +02:00
|
|
|
|
parse_mode = use_markdown and 'Markdown' or nil,
|
|
|
|
|
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)
|
|
|
|
|
return bindings.request(self, 'editMessageText', {
|
|
|
|
|
chat_id = chat_id,
|
|
|
|
|
message_id = message_id,
|
|
|
|
|
text = text,
|
|
|
|
|
disable_web_page_preview = disable_web_page_preview,
|
|
|
|
|
parse_mode = use_markdown and 'Markdown' or nil,
|
|
|
|
|
reply_markup = reply_markup
|
|
|
|
|
} )
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
function utilities:send_reply(old_msg, text, use_markdown, reply_markup)
|
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-07-02 00:13:17 +02:00
|
|
|
|
parse_mode = use_markdown and 'Markdown' or nil,
|
|
|
|
|
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-06-14 13:14:09 +02:00
|
|
|
|
-- Retruns true if the string is empty
|
|
|
|
|
function string:isempty()
|
|
|
|
|
return self == nil or self == ''
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
-- Retruns true if the string is blank
|
|
|
|
|
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 = {}
|
|
|
|
|
local doer = HTTP
|
|
|
|
|
local do_redir = true
|
|
|
|
|
if url:match('^https') then
|
|
|
|
|
doer = HTTPS
|
|
|
|
|
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()
|
|
|
|
|
local data = JSON.decode(s)
|
|
|
|
|
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)
|
2015-07-29 00:13:46 +02:00
|
|
|
|
local s = JSON.encode(data)
|
|
|
|
|
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)
|
2015-11-25 03:22:04 +01:00
|
|
|
|
|
2016-06-14 16:17:13 +02:00
|
|
|
|
local url = 'https://maps.googleapis.com/maps/api/geocode/json?address=' .. URL.escape(input)
|
2015-11-25 03:22:04 +01:00
|
|
|
|
|
2016-06-14 16:17:13 +02:00
|
|
|
|
local jstr, res = HTTPS.request(url)
|
2015-11-25 03:22:04 +01:00
|
|
|
|
if res ~= 200 then
|
2016-05-27 02:26:30 +02:00
|
|
|
|
return config.errors.connection
|
2015-11-25 03:22:04 +01:00
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
local jdat = JSON.decode(jstr)
|
|
|
|
|
if jdat.status == 'ZERO_RESULTS' then
|
2016-05-27 02:26:30 +02:00
|
|
|
|
return config.errors.results
|
2015-11-25 03:22:04 +01:00
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
return {
|
|
|
|
|
lat = jdat.results[1].geometry.location.lat,
|
|
|
|
|
lon = jdat.results[1].geometry.location.lng
|
|
|
|
|
}
|
|
|
|
|
|
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('_', '\\_')
|
|
|
|
|
text = text:gsub('%[', '\\[')
|
2016-04-20 22:43:08 +02:00
|
|
|
|
text = text:gsub('%]', '\\]')
|
2016-03-22 11:16:26 +01:00
|
|
|
|
text = text:gsub('%*', '\\*')
|
|
|
|
|
text = text:gsub('`', '\\`')
|
|
|
|
|
return text
|
|
|
|
|
end
|
2016-04-01 19:29:00 +02:00
|
|
|
|
|
2016-07-16 22:30:51 +02:00
|
|
|
|
function utilities.markdown_escape_simple(text)
|
|
|
|
|
text = text:gsub('_', '\\_')
|
|
|
|
|
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)
|
|
|
|
|
local original = HTTP.TIMEOUT
|
|
|
|
|
HTTP.TIMEOUT = timeout
|
|
|
|
|
fun()
|
|
|
|
|
HTTP.TIMEOUT = original
|
|
|
|
|
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
|
|
|
|
|
table.insert(files, v)
|
|
|
|
|
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-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
|
|
|
|
|
local source = helpers.url_encode_arguments(arguments)
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
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-06-19 21:53:24 +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
|
|
|
|
|
|
|
|
|
|
response_body = JSON.decode(table.concat(response_body))
|
|
|
|
|
|
|
|
|
|
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
|
|
|
|
|
|
|
|
|
|
function get_last_modified_header(url)
|
|
|
|
|
local doer = HTTP
|
|
|
|
|
local do_redir = true
|
|
|
|
|
if url:match('^https') then
|
|
|
|
|
doer = HTTPS
|
|
|
|
|
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-05 19:21:37 +02:00
|
|
|
|
if header["last-modified"] then
|
|
|
|
|
last_modified = header["last-modified"]
|
|
|
|
|
elseif header["Last-Modified"] then
|
|
|
|
|
last_modified = header["Last-Modified"]
|
|
|
|
|
end
|
|
|
|
|
return last_modified, code
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
-- 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')
|
|
|
|
|
|
|
|
|
|
-- get last-modified header
|
|
|
|
|
local last_modified, code = get_last_modified_header(url)
|
|
|
|
|
if code ~= 200 then
|
|
|
|
|
if cached_file_id then
|
|
|
|
|
redis:del(hash..':'..url)
|
|
|
|
|
end
|
|
|
|
|
return
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
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
|
|
|
|
|
if last_modified == cached_last_modified then
|
|
|
|
|
print('File not modified and already cached')
|
|
|
|
|
nocache = true
|
|
|
|
|
file = cached_file_id
|
|
|
|
|
else
|
|
|
|
|
print('File cached, but modified or not already cached. (Re)downloading...')
|
|
|
|
|
file = download_to_file(url, file_name)
|
|
|
|
|
end
|
|
|
|
|
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
|
|
|
|
|
return seconds..' Sekunden'
|
|
|
|
|
elseif hours == 00 and minutes ~= 00 then
|
|
|
|
|
return string.format("%02d:%02d", minutes, seconds)..' Minuten'
|
|
|
|
|
elseif hours ~= 00 then
|
|
|
|
|
return string.format("%02d:%02d:%02d", hours, minutes, seconds)..' Stunden'
|
|
|
|
|
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
|