259 lines
8.6 KiB
Python
259 lines
8.6 KiB
Python
import sys
|
|
import socket
|
|
import string
|
|
import select
|
|
import re
|
|
import pickle
|
|
|
|
from pytox import Tox, ToxAV
|
|
|
|
from time import sleep
|
|
from os.path import exists
|
|
from threading import Thread
|
|
|
|
SERVER = ['54.199.139.199', 33445, '7F9C31FE850E97CEFD4C4591DF93FC757C7C12549DDD55F8EEAECC34FE76C029']
|
|
GROUP_BOT = '56A1ADE4B65B86BCD51CC73E2CD4E542179F47959FE3E0E21B4B0ACDADE51855D34D34D37CB5'
|
|
PWD = ''
|
|
|
|
IRC_HOST = 'irc.freenode.net'
|
|
IRC_PORT = 6667
|
|
NAME = NICK = IDENT = REALNAME = 'SyncBot'
|
|
|
|
CHANNEL = '#tox-ontopic'
|
|
MEMORY_DB = 'memory.pickle'
|
|
|
|
class AV(ToxAV):
|
|
def __init__(self, core, max_calls):
|
|
self.core = self.get_tox()
|
|
self.cs = None
|
|
self.call_type = self.TypeAudio
|
|
|
|
def on_invite(self, idx):
|
|
self.cs = self.get_peer_csettings(idx, 0)
|
|
self.call_type = self.cs['call_type']
|
|
|
|
print('Incoming %s call from %d:%s ...' % (
|
|
'video' if self.call_type == self.TypeVideo else 'audio', idx,
|
|
self.core.get_name(self.get_peer_id(idx, 0))))
|
|
|
|
self.answer(idx, self.call_type)
|
|
print('Answered, in call...')
|
|
|
|
def on_start(self, idx):
|
|
self.change_settings(idx, {'max_video_width': 1920,
|
|
'max_video_height': 1080})
|
|
self.prepare_transmission(idx, self.jbufdc * 2, self.VADd,
|
|
True if self.call_type == self.TypeVideo else False)
|
|
|
|
def on_end(self, idx):
|
|
self.kill_transmission()
|
|
|
|
print('Call ended')
|
|
|
|
def on_peer_timeout(self, idx):
|
|
self.stop_call()
|
|
|
|
def on_audio_data(self, idx, size, data):
|
|
sys.stdout.write('.')
|
|
sys.stdout.flush()
|
|
self.send_audio(idx, size, data)
|
|
|
|
def on_video_data(self, idx, width, height, data):
|
|
sys.stdout.write('*')
|
|
sys.stdout.flush()
|
|
self.send_video(idx, width, height, data)
|
|
|
|
bot_toxname = 'SyncBot'
|
|
|
|
class SyncBot(Tox):
|
|
def __init__(self):
|
|
if exists('data'):
|
|
self.load_from_file('data')
|
|
|
|
self.av = AV(self, 10)
|
|
self.connect()
|
|
self.set_name(bot_toxname)
|
|
self.set_status_message("Send me a message with the word 'invite'")
|
|
print('ID: %s' % self.get_address())
|
|
|
|
self.readbuffer = ''
|
|
self.tox_group_id = None
|
|
|
|
self.irc_init()
|
|
self.memory = {}
|
|
|
|
if exists(MEMORY_DB):
|
|
with open(MEMORY_DB, 'r') as f:
|
|
self.memory = pickle.load(f)
|
|
|
|
def irc_init(self):
|
|
self.irc = socket.socket()
|
|
self.irc.connect((IRC_HOST, IRC_PORT))
|
|
self.irc.send('NICK %s\r\n' % NICK)
|
|
self.irc.send('USER %s %s bla :%s\r\n' % (IDENT, IRC_HOST, REALNAME))
|
|
|
|
def connect(self):
|
|
print('connecting...')
|
|
self.bootstrap_from_address(SERVER[0], SERVER[1], SERVER[2])
|
|
|
|
def ensure_exe(self, func, args):
|
|
count = 0
|
|
THRESHOLD = 50
|
|
|
|
while True:
|
|
try:
|
|
return func(*args)
|
|
except:
|
|
assert count < THRESHOLD
|
|
count += 1
|
|
for i in range(10):
|
|
self.do()
|
|
sleep(0.02)
|
|
|
|
def loop(self):
|
|
checked = False
|
|
self.joined = False
|
|
self.request = False
|
|
|
|
try:
|
|
while True:
|
|
status = self.isconnected()
|
|
if not checked and status:
|
|
print('Connected to DHT.')
|
|
checked = True
|
|
try:
|
|
self.bid = self.get_friend_id(GROUP_BOT)
|
|
except:
|
|
self.ensure_exe(self.add_friend, (GROUP_BOT, 'Hi'))
|
|
self.bid = self.get_friend_id(GROUP_BOT)
|
|
|
|
if checked and not status:
|
|
print('Disconnected from DHT.')
|
|
self.connect()
|
|
checked = False
|
|
|
|
readable, _, _ = select.select([self.irc], [], [], 0.01)
|
|
|
|
if readable:
|
|
self.readbuffer += self.irc.recv(4096)
|
|
lines = self.readbuffer.split('\n')
|
|
self.readbuffer = lines.pop()
|
|
|
|
for line in lines:
|
|
rx = re.match(r':(.*?)!.*? PRIVMSG %s :(.*?)\r' %
|
|
CHANNEL, line, re.S)
|
|
if rx:
|
|
print('IRC> %s: %s' % rx.groups())
|
|
msg = '[%s]: %s' % rx.groups()
|
|
content = rx.group(2)
|
|
|
|
if content[1:].startswith('ACTION '):
|
|
action = '[%s]: %s' % (rx.group(1),
|
|
rx.group(2)[8:-1])
|
|
self.ensure_exe(self.group_action_send,
|
|
(self.tox_group_id, action))
|
|
elif self.tox_group_id != None:
|
|
self.ensure_exe(self.group_message_send,
|
|
(self.tox_group_id, msg))
|
|
|
|
if content.startswith('^'):
|
|
self.handle_command(content)
|
|
|
|
l = line.rstrip().split()
|
|
if l[0] == 'PING':
|
|
self.irc_send('PONG %s\r\n' % l[1])
|
|
if l[1] == '376':
|
|
self.irc.send('PRIVMSG NickServ :IDENTIFY %s %s\r\n'
|
|
% (NICK, PWD))
|
|
self.irc.send('JOIN %s\r\n' % CHANNEL)
|
|
|
|
self.do()
|
|
except KeyboardInterrupt:
|
|
self.save_to_file('data')
|
|
|
|
def irc_send(self, msg):
|
|
success = False
|
|
while not success:
|
|
try:
|
|
self.irc.send(msg)
|
|
success = True
|
|
break
|
|
except socket.error:
|
|
self.irc_init()
|
|
sleep(1)
|
|
|
|
def on_connection_status(self, friendId, status):
|
|
if not self.request and not self.joined \
|
|
and friendId == self.bid and status:
|
|
print('Groupbot online, trying to join group chat.')
|
|
self.request = True
|
|
self.ensure_exe(self.send_message, (self.bid, 'invite'))
|
|
|
|
def on_group_invite(self, friendid, type, data):
|
|
if not self.joined:
|
|
self.joined = True
|
|
self.tox_group_id = self.join_groupchat(friendid, data)
|
|
print('Joined groupchat.')
|
|
|
|
def on_group_message(self, groupnumber, friendgroupnumber, message):
|
|
name = self.group_peername(groupnumber, friendgroupnumber)
|
|
if len(name) and name != NAME:
|
|
print('TOX> %s: %s' % (name, message))
|
|
if message.startswith('>'):
|
|
message = '\x0309%s\x03' % message
|
|
|
|
self.irc_send('PRIVMSG %s :[%s]: %s\r\n' %
|
|
(CHANNEL, name, message))
|
|
if message.startswith('^'):
|
|
self.handle_command(message)
|
|
|
|
def on_group_action(self, groupnumber, friendgroupnumber, action):
|
|
name = self.group_peername(groupnumber, friendgroupnumber)
|
|
if len(name) and name != NAME:
|
|
print('TOX> %s: %s' % (name, action))
|
|
if action.startswith('>'):
|
|
action = '\x0309%s\x03' % action
|
|
self.irc_send('PRIVMSG %s :\x01ACTION [%s]: %s\x01\r\n' %
|
|
(CHANNEL, name, action))
|
|
|
|
def on_friend_request(self, pk, message):
|
|
print('Friend request from %s: %s' % (pk, message))
|
|
self.add_friend_norequest(pk)
|
|
print('Accepted.')
|
|
|
|
def on_friend_message(self, friendid, message):
|
|
if message == 'invite':
|
|
if not self.tox_group_id is None:
|
|
print('Inviting %s' % self.get_name(friendid))
|
|
self.invite_friend(friendid, self.tox_group_id)
|
|
return
|
|
else:
|
|
message = 'Waiting for GroupBot, please try again in 1 min.'
|
|
|
|
self.ensure_exe(self.send_message, (friendid, message))
|
|
|
|
def send_both(self, content):
|
|
self.ensure_exe(self.group_message_send, (self.tox_group_id, content))
|
|
self.irc_send('PRIVMSG %s :%s\r\n' % (CHANNEL, content))
|
|
|
|
def handle_command(self, cmd):
|
|
cmd = cmd[1:]
|
|
if cmd in ['syncbot', 'echobot']:
|
|
self.send_both(self.get_address())
|
|
elif cmd == 'resync':
|
|
sys.exit(0)
|
|
elif cmd.startswith('remember '):
|
|
args = cmd[9:].split(' ')
|
|
subject = args[0]
|
|
desc = ' '.join(args[1:])
|
|
self.memory[subject] = desc
|
|
with open(MEMORY_DB, 'w') as f:
|
|
pickle.dump(self.memory, f)
|
|
self.send_both('Remembering ^%s: %s' % (subject, desc))
|
|
elif self.memory.has_key(cmd):
|
|
self.send_both(self.memory[cmd])
|
|
|
|
|
|
t = SyncBot()
|
|
t.loop()
|