forked from dj-shin/connector
-
Notifications
You must be signed in to change notification settings - Fork 0
/
ircconnector.py
92 lines (74 loc) · 3.16 KB
/
ircconnector.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
# -*- coding: utf-8 -*-
import socket
import ssl
import threading
from setting import server, port, botname, botnick
from setting import DEBUG, LOG_ENABLE
from ircmessage import IRCMessage
from queue import Queue
class IRCConnector(threading.Thread):
ircsock = None
msgQueue = None
botnick = None
def __init__(self, msgQueue):
threading.Thread.__init__(self)
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.connect((server, port))
self.ircsock = ssl.wrap_socket(s)
self.ircsock.send(('USER ' + (botname + ' ') * 3 + ':' +
botnick + '\n').encode())
self.ircsock.send(('NICK ' + botnick + '\n').encode())
self.botnick = botnick
self.msgQueue = msgQueue
def ping(self):
self.ircsock.send(('PONG :pingis\n').encode())
def sendmsg(self, chan, msg):
self.ircsock.send(('PRIVMSG ' + chan + ' :' + msg + '\n').encode())
def joinchan(self, chan, key=''):
self.ircsock.send(('JOIN ' + chan + ' ' + key + '\n').encode())
def partchan(self, chan):
self.ircsock.send(('PART ' + chan + '\n').encode())
def chanlist(self):
self.ircsock.send(('WHOIS ' + botnick + '\n').encode())
def settopic(self, chan, msg):
self.ircsock.send(('TOPIC ' + chan + ' :' + msg + '\n').encode())
def gettopic(self, chan):
self.ircsock.send(('LIST ' + chan + '\n').encode())
ircmsg = self.ircsock.recv(8192)
topic = (ircmsg.decode().split('\n')[1]).split(':')[2].strip('\n\r')
return topic
def listmember(self, chan):
self.ircsock.send(('NAMES ' + chan + '\n').encode())
ircmsg = self.ircsock.recv(8192)
return ircmsg.decode().split('\n')[0].split(':')[2].strip('\n\r')
def run(self):
while True:
ircmsg = self.ircsock.recv(8192)
ircmsg = ircmsg.decode(errors = 'ignore').strip('\n\r')
if DEBUG:
print(ircmsg)
message = IRCMessage(ircmsg)
if message.isValid():
if message.msgType == 'PING':
self.ping()
elif message.msgType == 'INVITE':
if message.target == self.botnick:
self.joinchan(message.channel)
elif message.msgType == 'PRIVMSG':
if 'KeyJoin' == message.msg[:7]:
msg_split = message.msg.split()
if len(msg_split) == 3:
self.joinchan(msg_split[1], msg_split[2])
if '옵뿌려!' in message.msg:
members = self.listmember(message.channel).split(' ')
for mem in members:
if mem[0] != '@':
self.ircsock.send(('MODE ' + message.channel + ' +o ' + mem + '\n').encode())
else:
if LOG_ENABLE:
print(message)
if self.msgQueue is not None:
self.msgQueue.put({'type': 'irc', 'content': message})
if __name__ == '__main__':
connector = IRCConnector(None)
connector.run()