329 lines
9.7 KiB
Python
329 lines
9.7 KiB
Python
from threading import Thread
|
|
from time import sleep
|
|
import select
|
|
import socket
|
|
import socks
|
|
import ssl
|
|
import re
|
|
|
|
from adapt.intent import IntentBuilder
|
|
from mycroft.audio import wait_while_speaking
|
|
from mycroft import MycroftSkill, intent_handler
|
|
from mycroft.skills.core import MycroftSkill
|
|
from mycroft.util.log import getLogger
|
|
from mycroft.util import normalize
|
|
|
|
#########################################
|
|
# RFC #
|
|
# https://tools.ietf.org/html/rfc1459 #
|
|
#########################################
|
|
|
|
LOGGER = getLogger(__name__)
|
|
|
|
class IRCSkill(MycroftSkill):
|
|
def __init__(self):
|
|
super(IRCSkill, self).__init__()
|
|
# TODO make them configureable
|
|
# TODO make them into lists
|
|
# options
|
|
self.settings['proxy'] = "127.0.0.1"
|
|
self.settings['proxy-port'] = 9050
|
|
self.settings['proxy-user'] = ""
|
|
self.settings['proxy-passwd'] = ""
|
|
self.settings['server'] = "irc.flightgear.org"
|
|
self.settings['server-password'] = ""
|
|
self.settings['port'] = 6667
|
|
self.settings['channel'] = "mycroft"
|
|
self.settings['channel-password'] = ""
|
|
self.settings['user'] = "dummy|m"
|
|
self.settings['password'] = ""
|
|
self.settings['ssl'] = False
|
|
self.settings['debug'] = False
|
|
|
|
# IPC for comunicating between threads
|
|
self.irc_lock = False
|
|
self.irc_cmd = ""
|
|
self.irc_str = ""
|
|
|
|
def initialize(self):
|
|
if self.settings['proxy'] != "":
|
|
if self.settings['debug']:
|
|
self.speak("Using proxy: " + self.settings['proxy'])
|
|
self.speak("Port: " + str(self.settings['proxy-port']))
|
|
socks.set_default_proxy(socks.SOCKS5, self.settings['proxy'], self.settings['proxy-port'], True, 'user','passwd')
|
|
socket.socket = socks.socksocket
|
|
|
|
self.con_thread = Thread(target=self._main_loop)
|
|
self.con_thread.setDaemon(False)
|
|
self.con_thread.start()
|
|
|
|
@intent_handler(IntentBuilder('ConnectIntent').require('connect'))
|
|
def handle_connect_intent(self, message):
|
|
# TODO ability to connect to different server
|
|
if self.con_thread.isAlive() == False:
|
|
self.speak("Restart thread")
|
|
self.con_thread.start()
|
|
if self.irc_lock == False:
|
|
self.speak("Connecting")
|
|
self.irc_lock == True
|
|
self.irc_cmd = "connect"
|
|
self.irc_str = ""
|
|
self.irc_lock = False
|
|
|
|
@intent_handler(IntentBuilder('JoinIntent').require('join'))
|
|
def handle_join_intent(self, message):
|
|
# TODO ability to join to different channels
|
|
if self.con_thread.isAlive() == False:
|
|
self.speak("Restart thread")
|
|
self.con_thread.start()
|
|
if self.irc_lock == False:
|
|
self.speak("Joining")
|
|
self.irc_lock == True
|
|
self.irc_cmd = "join"
|
|
self.irc_str = ""
|
|
self.irc_lock = False
|
|
|
|
@intent_handler(IntentBuilder('PartIntent').require('part'))
|
|
def handle_part_intent(self, message):
|
|
# TODO ability to join to different channels
|
|
if self.con_thread.isAlive() == False:
|
|
self.speak("Restart thread")
|
|
self.con_thread.start()
|
|
if self.irc_lock == False:
|
|
self.speak("Parting")
|
|
self.irc_lock == True
|
|
self.irc_cmd = "part"
|
|
self.irc_str = ""
|
|
self.irc_lock = False
|
|
|
|
@intent_handler(IntentBuilder('DisconnectIntent').require('disconnect'))
|
|
def handle_disconnect_intent(self, message):
|
|
# TODO ability to disconnect from different server
|
|
if self.con_thread.isAlive() == False:
|
|
self.speak("Restart thread")
|
|
self.con_thread.start()
|
|
if self.irc_lock == False:
|
|
self.speak("Disconnecting")
|
|
self.irc_lock == True
|
|
self.irc_cmd = "disconnect"
|
|
self.irc_str = ""
|
|
self.irc_lock = False
|
|
|
|
@intent_handler(IntentBuilder('SendIntent').require('send'))
|
|
def handle_send_intent(self, message):
|
|
# TODO ability to send to different users and channels
|
|
if self.con_thread.isAlive() == False:
|
|
self.speak("Restart thread")
|
|
self.con_thread.start()
|
|
if self.irc_lock == False:
|
|
response = self.get_response("get_msg")
|
|
if response != None:
|
|
self.irc_lock == True
|
|
self.irc_cmd = "send"
|
|
self.irc_str = response
|
|
self.irc_lock = False
|
|
else:
|
|
self.speak("I didn't understand a message")
|
|
|
|
def _main_loop(self):
|
|
connected = False
|
|
joined = False
|
|
while True:
|
|
sleep(2)
|
|
if connected:
|
|
text = ""
|
|
try:
|
|
ready = select.select([irc], [], [], 2)
|
|
if ready[0]:
|
|
text = irc.recv(2040)
|
|
except Exception:
|
|
continue
|
|
|
|
if text != "":
|
|
if self.settings['debug']:
|
|
self.speak(text)
|
|
|
|
# Prevent Timeout
|
|
match = re.search("PING (.*)", text)
|
|
if match != None:
|
|
irc.send('PONG ' + match.group(1) + '\r\n')
|
|
|
|
match = re.search(":(.*)!.*@\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3} QUIT", text)
|
|
if match != None:
|
|
self.speak(match.group(1) + " has disconnected")
|
|
|
|
match = re.search(":(.*)!.*@\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3} JOIN", text)
|
|
if match != None:
|
|
if match.group(1) != self.settings['user']:
|
|
self.speak(match.group(1) + " has joined the channel")
|
|
|
|
match = re.search(":(.*)!.*@\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3} PART", text)
|
|
if match != None:
|
|
self.speak(match.group(1) + " has left the channel")
|
|
|
|
match = re.search(":(.*)!.*@\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3} QUIT", text)
|
|
if match != None:
|
|
self.speak(match.group(1) + " has disconnected")
|
|
|
|
match = re.search(":(.*)!.*@\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3} PRIVMSG #.* :(.*)", text)
|
|
if match != None:
|
|
self.speak(match.group(1) + " has written in " + match.group(2) + ": " + match.group(3))
|
|
|
|
# TODO fix
|
|
# match = re.search(":(.*)!.*@\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3} PRIVMSG " + self.settings['user'] + " :(.*)", text)
|
|
# if match != None:
|
|
# self.speak(match.group(1) + " has written you a private message: " + match.group(2))
|
|
|
|
match = re.search(":(.*)!.*@\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3} NOTICE #.* :(.*)", text)
|
|
if match != None:
|
|
self.speak(match.group(1) + " has written a notice to " + match.group(2) + ". The notice is: " + match.group(3))
|
|
|
|
# TODO fix
|
|
# match = re.search(":(.*)!.*@\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3} NOTICE " + self.settings['user'] + " :(.*)", text)
|
|
# if match != None:
|
|
# self.speak(match.group(1) + " has written a private notice to you. The notice is: " + match.group(2))
|
|
|
|
cmd = ""
|
|
string = ""
|
|
|
|
if self.irc_cmd != "":
|
|
if self.irc_lock == False:
|
|
self.irc_lock = True
|
|
cmd = self.irc_cmd
|
|
string = self.irc_str
|
|
self.irc_cmd = ""
|
|
self.irc_str = ""
|
|
self.irc_lock = False
|
|
|
|
if cmd != "":
|
|
# check cmd and take action
|
|
if cmd == "connect":
|
|
# TODO add ability to connect to more than one server
|
|
if connected == False:
|
|
connected, irc = self._irc_connect(self.settings['server'], self.settings['port'], self.settings['ssl'], self.settings['server-password'], self.settings['user'], self.settings['password'])
|
|
else:
|
|
self.speak("Already connected")
|
|
|
|
elif cmd == "join":
|
|
if connected:
|
|
joined = self._irc_join(irc, self.settings['channel'], self.settings['channel-password'])
|
|
else:
|
|
self.speak("Please connect to a server first")
|
|
|
|
elif cmd == "part":
|
|
joined = self._irc_part(irc, self.settings['channel'])
|
|
|
|
elif cmd == "disconnect":
|
|
if connected:
|
|
connected = self._irc_disconnect(irc)
|
|
else:
|
|
self.speak("I'm to no server connected")
|
|
|
|
if connected == False:
|
|
joined = False
|
|
|
|
elif cmd == "send":
|
|
self._irc_send(irc, "#" + self.settings['channel'], string)
|
|
|
|
def _irc_connect(self, server, port, ssl_req, server_password, user, password):
|
|
if ssl_req:
|
|
if self.settings['debug']:
|
|
self.speak("Use SSL")
|
|
irc_C = socket.socket(socket.AF_INET, socket.SOCK_STREAM) #defines the socket
|
|
irc = ssl.wrap_socket(irc_C, cert_reqs=ssl.CERT_NONE)
|
|
else:
|
|
if self.settings['debug']:
|
|
self.speak("Connect without ssl")
|
|
irc = socket.socket(socket.AF_INET, socket.SOCK_STREAM) #defines the socket
|
|
|
|
# Connect
|
|
try:
|
|
if self.settings['debug']:
|
|
self.speak("Server: " + server)
|
|
self.speak("Port: " + str(port))
|
|
irc.settimeout(60)
|
|
irc.connect((server, port))
|
|
except Exception as e:
|
|
self.speak("Unable to connect to server.")
|
|
if self.settings['debug']:
|
|
self.speak("Error: " + str(e))
|
|
return False, irc
|
|
|
|
irc.setblocking(0)
|
|
if server_password != "":
|
|
irc.send("PASS %s\n" % (password))
|
|
irc.send("USER " + user + " " + user + " " + user + " :IRC via VOICE -> Mycroft\n")
|
|
irc.send("NICK " + user + "\n")
|
|
if password != "":
|
|
irc.send("PRIVMSG nickserv :identify %s %s\r\n" % (user, password))
|
|
|
|
self.speak("Connected")
|
|
return True, irc
|
|
|
|
def _irc_join(self, irc, channel, channel_password):
|
|
# TODO add steps to be able to use a pw
|
|
irc.send("JOIN #"+ channel +"\n")
|
|
self.speak("Channel " + channel + " joined")
|
|
return True
|
|
|
|
def _irc_part(self, irc, channel):
|
|
irc.send("PART #" + channel)
|
|
self.speak("Parted")
|
|
return False # this is the value that's written in `joined`
|
|
|
|
def _irc_disconnect(self, irc):
|
|
irc.send("QUIT :Disconnected my mycroft\n")
|
|
irc.close()
|
|
self.speak("Disconnected")
|
|
return False # this is the value that's written in `connected`
|
|
|
|
def _irc_send(self, irc, to, msg):
|
|
irc.send("PRIVMSG " + to + " :" + msg)
|
|
self.speak("Message sent")
|
|
|
|
def stop(self):
|
|
pass
|
|
|
|
def create_skill():
|
|
return IRCSkill()
|
|
"""
|
|
### Tail
|
|
tail_files = [
|
|
'/tmp/file-to-tail.txt'
|
|
]
|
|
|
|
|
|
print "Establishing connection to [%s]" % (server)
|
|
|
|
|
|
tail_line = []
|
|
for i, tail in enumerate(tail_files):
|
|
tail_line.append('')
|
|
|
|
|
|
while True:
|
|
time.sleep(2)
|
|
|
|
# Tail Files
|
|
for i, tail in enumerate(tail_files):
|
|
try:
|
|
f = open(tail, 'r')
|
|
line = f.readlines()[-1]
|
|
f.close()
|
|
if tail_line[i] != line:
|
|
tail_line[i] = line
|
|
irc.send("PRIVMSG %s :%s" % (channel, line))
|
|
except Exception as e:
|
|
print "Error with file %s" % (tail)
|
|
print e
|
|
|
|
try:
|
|
text=irc.recv(2040)
|
|
print text
|
|
|
|
# Prevent Timeout
|
|
if text.find('PING') != -1:
|
|
irc.send('PONG ' + text.split() [1] + '\r\n')
|
|
except Exception:
|
|
continue
|
|
"""
|