oops
This commit is contained in:
topkecleon
parent
f9094a73ca
commit
d3b0825fa0
@ -1,13 +1,13 @@
|
||||
--[[
|
||||
drua-tg
|
||||
Based on JuanPotato's lua-tg (https://github.com/juanpotato/lua-tg),
|
||||
modified to work more naturally from an API bot.
|
||||
drua-tg
|
||||
Based on JuanPotato's lua-tg (https://github.com/juanpotato/lua-tg),
|
||||
modified to work more naturally from an API bot.
|
||||
|
||||
Usage:
|
||||
drua = require('drua-tg')
|
||||
drua.IP = 'localhost' -- 'localhost' is default
|
||||
drua.PORT = 4567 -- 4567 is default
|
||||
drua.message(chat_id, text)
|
||||
Usage:
|
||||
drua = require('drua-tg')
|
||||
drua.IP = 'localhost' -- 'localhost' is default
|
||||
drua.PORT = 4567 -- 4567 is default
|
||||
drua.message(chat_id, text)
|
||||
|
||||
The MIT License (MIT)
|
||||
|
||||
@ -35,150 +35,150 @@ SOFTWARE.
|
||||
local SOCKET = require('socket')
|
||||
|
||||
local comtab = {
|
||||
add = { 'chat_add_user %s %s', 'channel_invite %s %s' },
|
||||
kick = { 'chat_del_user %s %s', 'channel_kick %s %s' },
|
||||
rename = { 'rename_chat %s "%s"', 'rename_channel %s "%s"' },
|
||||
link = { 'export_chat_link %s', 'export_channel_link %s' },
|
||||
photo_set = { 'chat_set_photo %s %s', 'channel_set_photo %s %s' },
|
||||
photo_get = { [0] = 'load_user_photo %s', 'load_chat_photo %s', 'load_channel_photo %s' },
|
||||
info = { [0] = 'user_info %s', 'chat_info %s', 'channel_info %s' }
|
||||
add = { 'chat_add_user %s %s', 'channel_invite %s %s' },
|
||||
kick = { 'chat_del_user %s %s', 'channel_kick %s %s' },
|
||||
rename = { 'rename_chat %s "%s"', 'rename_channel %s "%s"' },
|
||||
link = { 'export_chat_link %s', 'export_channel_link %s' },
|
||||
photo_set = { 'chat_set_photo %s %s', 'channel_set_photo %s %s' },
|
||||
photo_get = { [0] = 'load_user_photo %s', 'load_chat_photo %s', 'load_channel_photo %s' },
|
||||
info = { [0] = 'user_info %s', 'chat_info %s', 'channel_info %s' }
|
||||
}
|
||||
|
||||
local function format_target(target)
|
||||
target = tonumber(target)
|
||||
if target < -1000000000000 then
|
||||
target = 'channel#' .. math.abs(target) - 1000000000000
|
||||
return target, 2
|
||||
elseif target < 0 then
|
||||
target = 'chat#' .. math.abs(target)
|
||||
return target, 1
|
||||
else
|
||||
target = 'user#' .. target
|
||||
return target, 0
|
||||
end
|
||||
target = tonumber(target)
|
||||
if target < -1000000000000 then
|
||||
target = 'channel#' .. math.abs(target) - 1000000000000
|
||||
return target, 2
|
||||
elseif target < 0 then
|
||||
target = 'chat#' .. math.abs(target)
|
||||
return target, 1
|
||||
else
|
||||
target = 'user#' .. target
|
||||
return target, 0
|
||||
end
|
||||
end
|
||||
|
||||
local function escape(text)
|
||||
text = text:gsub('\\', '\\\\')
|
||||
text = text:gsub('\n', '\\n')
|
||||
text = text:gsub('\t', '\\t')
|
||||
text = text:gsub('"', '\\"')
|
||||
return text
|
||||
text = text:gsub('\\', '\\\\')
|
||||
text = text:gsub('\n', '\\n')
|
||||
text = text:gsub('\t', '\\t')
|
||||
text = text:gsub('"', '\\"')
|
||||
return text
|
||||
end
|
||||
|
||||
local drua = {
|
||||
IP = 'localhost',
|
||||
PORT = 4567
|
||||
IP = 'localhost',
|
||||
PORT = 4567
|
||||
}
|
||||
|
||||
function drua.send(command, do_receive)
|
||||
local s = SOCKET.connect(drua.IP, drua.PORT)
|
||||
assert(s, '\nUnable to connect to tg session.')
|
||||
s:send(command..'\n')
|
||||
local output
|
||||
if do_receive then
|
||||
output = string.match(s:receive('*l'), 'ANSWER (%d+)')
|
||||
output = s:receive(tonumber(output)):gsub('\n$', '')
|
||||
end
|
||||
s:close()
|
||||
return output
|
||||
local s = SOCKET.connect(drua.IP, drua.PORT)
|
||||
assert(s, '\nUnable to connect to tg session.')
|
||||
s:send(command..'\n')
|
||||
local output
|
||||
if do_receive then
|
||||
output = string.match(s:receive('*l'), 'ANSWER (%d+)')
|
||||
output = s:receive(tonumber(output)):gsub('\n$', '')
|
||||
end
|
||||
s:close()
|
||||
return output
|
||||
end
|
||||
|
||||
function drua.message(target, text)
|
||||
target = format_target(target)
|
||||
text = escape(text)
|
||||
local command = 'msg %s "%s"'
|
||||
command = command:format(target, text)
|
||||
return drua.send(command)
|
||||
target = format_target(target)
|
||||
text = escape(text)
|
||||
local command = 'msg %s "%s"'
|
||||
command = command:format(target, text)
|
||||
return drua.send(command)
|
||||
end
|
||||
|
||||
function drua.send_photo(target, photo)
|
||||
target = format_target(target)
|
||||
local command = 'send_photo %s %s'
|
||||
command = command:format(target, photo)
|
||||
return drua.send(command)
|
||||
target = format_target(target)
|
||||
local command = 'send_photo %s %s'
|
||||
command = command:format(target, photo)
|
||||
return drua.send(command)
|
||||
end
|
||||
|
||||
function drua.add_user(chat, target)
|
||||
local a
|
||||
chat, a = format_target(chat)
|
||||
target = format_target(target)
|
||||
local command = comtab.add[a]:format(chat, target)
|
||||
return drua.send(command)
|
||||
local a
|
||||
chat, a = format_target(chat)
|
||||
target = format_target(target)
|
||||
local command = comtab.add[a]:format(chat, target)
|
||||
return drua.send(command)
|
||||
end
|
||||
|
||||
function drua.kick_user(chat, target)
|
||||
-- Get the group info so tg will recognize the target.
|
||||
drua.get_info(chat)
|
||||
local a
|
||||
chat, a = format_target(chat)
|
||||
target = format_target(target)
|
||||
local command = comtab.kick[a]:format(chat, target)
|
||||
return drua.send(command)
|
||||
-- Get the group info so tg will recognize the target.
|
||||
drua.get_info(chat)
|
||||
local a
|
||||
chat, a = format_target(chat)
|
||||
target = format_target(target)
|
||||
local command = comtab.kick[a]:format(chat, target)
|
||||
return drua.send(command)
|
||||
end
|
||||
|
||||
function drua.rename_chat(chat, name)
|
||||
local a
|
||||
chat, a = format_target(chat)
|
||||
local command = comtab.rename[a]:format(chat, name)
|
||||
return drua.send(command)
|
||||
local a
|
||||
chat, a = format_target(chat)
|
||||
local command = comtab.rename[a]:format(chat, name)
|
||||
return drua.send(command)
|
||||
end
|
||||
|
||||
function drua.export_link(chat)
|
||||
local a
|
||||
chat, a = format_target(chat)
|
||||
local command = comtab.link[a]:format(chat)
|
||||
return drua.send(command, true)
|
||||
local a
|
||||
chat, a = format_target(chat)
|
||||
local command = comtab.link[a]:format(chat)
|
||||
return drua.send(command, true)
|
||||
end
|
||||
|
||||
function drua.get_photo(chat)
|
||||
local a
|
||||
chat, a = format_target(chat)
|
||||
local command = comtab.photo_get[a]:format(chat)
|
||||
local output = drua.send(command, true)
|
||||
if output:match('FAIL') then
|
||||
return false
|
||||
else
|
||||
return output:match('Saved to (.+)')
|
||||
end
|
||||
local a
|
||||
chat, a = format_target(chat)
|
||||
local command = comtab.photo_get[a]:format(chat)
|
||||
local output = drua.send(command, true)
|
||||
if output:match('FAIL') then
|
||||
return false
|
||||
else
|
||||
return output:match('Saved to (.+)')
|
||||
end
|
||||
end
|
||||
|
||||
function drua.set_photo(chat, photo)
|
||||
local a
|
||||
chat, a = format_target(chat)
|
||||
local command = comtab.photo_set[a]:format(chat, photo)
|
||||
return drua.send(command)
|
||||
local a
|
||||
chat, a = format_target(chat)
|
||||
local command = comtab.photo_set[a]:format(chat, photo)
|
||||
return drua.send(command)
|
||||
end
|
||||
|
||||
function drua.get_info(target)
|
||||
local a
|
||||
target, a = format_target(target)
|
||||
local command = comtab.info[a]:format(target)
|
||||
return drua.send(command, true)
|
||||
local a
|
||||
target, a = format_target(target)
|
||||
local command = comtab.info[a]:format(target)
|
||||
return drua.send(command, true)
|
||||
end
|
||||
|
||||
function drua.channel_set_admin(chat, user, rank)
|
||||
chat = format_target(chat)
|
||||
user = format_target(user)
|
||||
local command = 'channel_set_admin %s %s %s'
|
||||
command = command:format(chat, user, rank)
|
||||
return drua.send(command)
|
||||
chat = format_target(chat)
|
||||
user = format_target(user)
|
||||
local command = 'channel_set_admin %s %s %s'
|
||||
command = command:format(chat, user, rank)
|
||||
return drua.send(command)
|
||||
end
|
||||
|
||||
function drua.channel_set_about(chat, text)
|
||||
chat = format_target(chat)
|
||||
text = escape(text)
|
||||
local command = 'channel_set_about %s "%s"'
|
||||
command = command:format(chat, text)
|
||||
return drua.send(command)
|
||||
chat = format_target(chat)
|
||||
text = escape(text)
|
||||
local command = 'channel_set_about %s "%s"'
|
||||
command = command:format(chat, text)
|
||||
return drua.send(command)
|
||||
end
|
||||
|
||||
function drua.block(user)
|
||||
return drua.send('block_user user#' .. user)
|
||||
return drua.send('block_user user#' .. user)
|
||||
end
|
||||
|
||||
function drua.unblock(user)
|
||||
return drua.send('unblock_user user#' .. user)
|
||||
return drua.send('unblock_user user#' .. user)
|
||||
end
|
||||
|
||||
return drua
|
||||
|
||||
Reference in New Issue
Block a user