Merge branch 'master' of github.com:yagop/telegram-bot
This commit is contained in:
yagop
commit
fcb9fd6168
243
bot/bot.lua
243
bot/bot.lua
|
|
@ -1,128 +1,150 @@
|
|||
http = require("socket.http")
|
||||
json = (loadfile "./bot/JSON.lua")()
|
||||
|
||||
VERSION = 'v0.1'
|
||||
VERSION = 'v0.2'
|
||||
|
||||
|
||||
function on_msg_receive (msg)
|
||||
-- vardump(msg)
|
||||
|
||||
if msg_valid(msg) == false then
|
||||
return
|
||||
end
|
||||
-- Check if command starts with ! eg !echo
|
||||
if msg.text:sub(0,1) == '!' then
|
||||
do_action(msg)
|
||||
else
|
||||
if is_image_url(msg.text) then
|
||||
send_image_from_url (msg)
|
||||
end
|
||||
end
|
||||
|
||||
print('mark_read: '..get_receiver(msg))
|
||||
mark_read(get_receiver(msg), ok_cb, false)
|
||||
-- write_log_file(msg)
|
||||
end
|
||||
|
||||
|
||||
function ok_cb(extra, success, result)
|
||||
end
|
||||
|
||||
function msg_valid(msg)
|
||||
if msg.out then
|
||||
return
|
||||
return false
|
||||
end
|
||||
if msg.date < now then
|
||||
return
|
||||
return false
|
||||
end
|
||||
if msg.text == nil then
|
||||
return
|
||||
return false
|
||||
end
|
||||
if msg.unread == 0 then
|
||||
return
|
||||
end
|
||||
-- Check if command starts with ! eg !echo
|
||||
if msg.text:sub(0,1) == '!' then
|
||||
msg.text = msg.text:sub(2,-1)
|
||||
do_action(msg)
|
||||
return false
|
||||
end
|
||||
end
|
||||
|
||||
-- This is very unestable ...
|
||||
if is_image_url(msg.text) then
|
||||
last = string.get_last_word(msg.text)
|
||||
last = last:match("[%w_:/.%%&-]+") -- Lets sanitize!
|
||||
extension = string.get_extension_from_filename(last)
|
||||
function send_image_from_url (msg)
|
||||
last = string.get_last_word(msg.text)
|
||||
last = last:match("[%w_:/.%%&-]+") -- Lets sanitize!
|
||||
extension = string.get_extension_from_filename(last)
|
||||
|
||||
file_name = string.random(5)
|
||||
file = "/tmp/"..file_name.."."..extension
|
||||
sh = "curl --insecure -o '"..file.."' "..last
|
||||
run_bash(sh)
|
||||
send_photo(get_receiver(msg), file)
|
||||
end
|
||||
|
||||
mark_read(get_receiver(msg))
|
||||
|
||||
-- write_log_file(msg)
|
||||
file_name = string.random(5)
|
||||
file = "/tmp/"..file_name.."."..extension
|
||||
sh = "curl --insecure -o '"..file.."' "..last
|
||||
run_bash(sh)
|
||||
send_photo(get_receiver(msg), file, ok_cb, false)
|
||||
end
|
||||
|
||||
function is_image_url(text)
|
||||
print ('IS image ' .. text ..'?')
|
||||
last = string.get_last_word(text)
|
||||
print ('Last is: ' .. last)
|
||||
extension = string.get_extension_from_filename(last)
|
||||
if extension == 'jpg' or extension == 'png' then
|
||||
print('Is image :D')
|
||||
return true
|
||||
end
|
||||
print 'Not image D:'
|
||||
return false
|
||||
end
|
||||
|
||||
-- Where magic happens
|
||||
function do_action(msg)
|
||||
receiver = get_receiver(msg)
|
||||
local receiver = get_receiver(msg)
|
||||
|
||||
if string.starts(msg.text, 'sh') then
|
||||
if string.starts(msg.text, '!sh') then
|
||||
text = run_sh(msg)
|
||||
send_msg(receiver, text)
|
||||
end
|
||||
|
||||
if string.starts(msg.text, 'torrent') then
|
||||
text = save_torrent(msg)
|
||||
send_msg(receiver, text)
|
||||
send_msg(receiver, text, ok_cb, false)
|
||||
return
|
||||
end
|
||||
|
||||
if string.starts(msg.text, 'uc3m') then
|
||||
if string.starts(msg.text, '!uc3m') then
|
||||
text = get_fortunes_uc3m()
|
||||
send_msg(receiver, text)
|
||||
send_msg(receiver, text, ok_cb, false)
|
||||
return
|
||||
end
|
||||
if string.starts(msg.text, '9gag') then
|
||||
|
||||
if string.starts(msg.text, '!9gag') then
|
||||
url, title = get_9GAG()
|
||||
file_name = url:match("([^/]+)$")
|
||||
file = "/tmp/"..file_name
|
||||
sh = "curl -o '"..file.."' "..url
|
||||
run_bash(sh)
|
||||
send_photo(receiver, file)
|
||||
send_msg(receiver, title)
|
||||
end
|
||||
if string.starts(msg.text, 'fortune') then
|
||||
send_photo(receiver, file, ok_cb, false)
|
||||
send_msg(receiver, title, ok_cb, false)
|
||||
return
|
||||
end
|
||||
|
||||
if string.starts(msg.text, '!fortune') then
|
||||
text = run_bash('fortune')
|
||||
send_msg(receiver, text)
|
||||
send_msg(receiver, text, ok_cb, false)
|
||||
return
|
||||
end
|
||||
if string.starts(msg.text, 'forni') then
|
||||
text = msg.text:sub(7,-1)
|
||||
send_msg('Fornicio_2.0', text)
|
||||
|
||||
if string.starts(msg.text, '!forni') then
|
||||
text = msg.text:sub(8,-1)
|
||||
send_msg('Fornicio_2.0', text, ok_cb, false)
|
||||
return
|
||||
end
|
||||
if string.starts(msg.text, 'hackers') then
|
||||
text = msg.text:sub(9,-1)
|
||||
send_msg('Juankers._Dios_existe_y_es_<span_class=', text)
|
||||
end
|
||||
if string.starts(msg.text, 'fwd') then
|
||||
fwd_msg (receiver, msg.id)
|
||||
end
|
||||
if string.starts(msg.text, 'cpu') then
|
||||
|
||||
if string.starts(msg.text, '!fwd') then
|
||||
fwd_msg (receiver, msg.id, ok_cb, false)
|
||||
return
|
||||
end
|
||||
|
||||
if string.starts(msg.text, '!cpu') then
|
||||
text = run_bash('uname -snr') .. ' ' .. run_bash('whoami')
|
||||
text = text .. '\n' .. run_bash('top -b |head -2')
|
||||
send_msg(receiver, text)
|
||||
send_msg(receiver, text, ok_cb, false)
|
||||
return
|
||||
end
|
||||
if string.starts(msg.text, 'ping') then
|
||||
send_msg(receiver, "pong")
|
||||
|
||||
if string.starts(msg.text, '!ping') then
|
||||
print('receiver: '..receiver)
|
||||
send_msg (receiver, 'pong', ok_cb, false)
|
||||
return
|
||||
end
|
||||
if string.starts(msg.text, 'weather') then
|
||||
|
||||
if string.starts(msg.text, '!weather') then
|
||||
if string.len(msg.text) <= 9 then
|
||||
city = 'Madrid,ES'
|
||||
else
|
||||
city = msg.text:sub(9,-1)
|
||||
city = msg.text:sub(10,-1)
|
||||
end
|
||||
text = get_weather(city)
|
||||
send_msg(receiver, text)
|
||||
send_msg(receiver, text, ok_cb, false)
|
||||
return
|
||||
end
|
||||
if string.starts(msg.text, 'echo') then
|
||||
-- Removes echo from the string
|
||||
echo = msg.text:sub(6,-1)
|
||||
send_msg(receiver, echo)
|
||||
|
||||
if string.starts(msg.text, '!echo') then
|
||||
echo = msg.text:sub(7,-1)
|
||||
send_msg(receiver, echo, ok_cb, false)
|
||||
return
|
||||
end
|
||||
if string.starts(msg.text, 'version') then
|
||||
|
||||
if string.starts(msg.text, '!version') then
|
||||
text = 'James Bot '.. VERSION
|
||||
send_msg(receiver, text)
|
||||
send_msg(receiver, text, ok_cb, false)
|
||||
return
|
||||
end
|
||||
if string.starts(msg.text, 'help') then
|
||||
|
||||
if string.starts(msg.text, '!help') then
|
||||
text = [[!help : print this help
|
||||
!ping : bot sends pong
|
||||
!sh (text) : send commands to bash (only privileged users)
|
||||
|
|
@ -135,37 +157,10 @@ function do_action(msg)
|
|||
!weather [city] : weather in that city (Madrid if not city)
|
||||
!9gag : send random url image from 9gag
|
||||
!uc3m : fortunes from Universidad Carlos III]]
|
||||
send_msg(receiver, text)
|
||||
send_msg(receiver, text, ok_cb, false)
|
||||
return
|
||||
end
|
||||
end
|
||||
|
||||
function get_receiver(msg)
|
||||
if msg.to.type == 'user' then
|
||||
return msg.from.print_name
|
||||
end
|
||||
if msg.to.type == 'chat' then
|
||||
return msg.to.print_name
|
||||
end
|
||||
end
|
||||
|
||||
|
||||
function on_our_id (id)
|
||||
our_id = id
|
||||
end
|
||||
|
||||
function on_secret_chat_created (peer)
|
||||
end
|
||||
|
||||
function on_user_update (user)
|
||||
end
|
||||
|
||||
function on_chat_update (user)
|
||||
end
|
||||
|
||||
function on_get_difference_end ()
|
||||
end
|
||||
|
||||
function on_binlog_replay_end ()
|
||||
end
|
||||
|
||||
function string.starts(String,Start)
|
||||
|
|
@ -244,13 +239,6 @@ function run_bash(str)
|
|||
return result
|
||||
end
|
||||
|
||||
function readAll(file)
|
||||
local f = io.open(file, "rb")
|
||||
local content = f:read("*all")
|
||||
f:close()
|
||||
return content
|
||||
end
|
||||
|
||||
function get_fortunes_uc3m()
|
||||
math.randomseed(os.time())
|
||||
local i = math.random(0,178) -- max 178
|
||||
|
|
@ -292,18 +280,6 @@ function get_weather(location)
|
|||
return temp .. '\n' .. conditions
|
||||
end
|
||||
|
||||
function sanitize_html(txt)
|
||||
local replacements = {
|
||||
['&' ] = '&',
|
||||
['<' ] = '<',
|
||||
['>' ] = '>',
|
||||
['\n'] = '<br/>'
|
||||
}
|
||||
return txt
|
||||
:gsub('[&<>\n]', replacements)
|
||||
:gsub(' +', function(s) return ' '..(' '):rep(#s-1) end)
|
||||
end
|
||||
|
||||
function string.random(length)
|
||||
math.randomseed(os.time())
|
||||
local str = "";
|
||||
|
|
@ -374,3 +350,36 @@ end
|
|||
config = load_config()
|
||||
our_id = 0
|
||||
now = os.time()
|
||||
|
||||
function get_receiver(msg)
|
||||
if msg.to.type == 'user' then
|
||||
return 'user#id'..msg.from.id
|
||||
end
|
||||
if msg.to.type == 'chat' then
|
||||
return 'chat#id'..msg.to.id
|
||||
end
|
||||
end
|
||||
|
||||
|
||||
function on_our_id (id)
|
||||
our_id = id
|
||||
end
|
||||
|
||||
function on_user_update (user, what)
|
||||
--vardump (user)
|
||||
end
|
||||
|
||||
function on_chat_update (chat, what)
|
||||
--vardump (chat)
|
||||
end
|
||||
|
||||
function on_secret_chat_update (schat, what)
|
||||
--vardump (schat)
|
||||
end
|
||||
|
||||
function on_get_difference_end ()
|
||||
end
|
||||
|
||||
function on_binlog_replay_end ()
|
||||
started = 1
|
||||
end
|
||||
2
tg
2
tg
|
|
@ -1 +1 @@
|
|||
Subproject commit 598f1ab92770b266758fbf38021d1dbc4fe4fef4
|
||||
Subproject commit ce61adac12fc18e2f5c8165e10f6ba91e1cb0928
|
||||
Reference in New Issue
Block a user