mirror of
https://github.com/jlu5/PyLink.git
synced 2024-11-01 09:19:23 +01:00
9b4fb50f25
- PLUGIN SUPPORT and COMMAND HANDLING, wow!!!!!!! - Restructuring of files so that there's only one protocol module (anything else is too much to maintain for now) - Split protocol things into utils.py - Bugfixes: don't go into an endless loop of text spamming when the remote host closes the connection!
80 lines
2.2 KiB
Python
Executable File
80 lines
2.2 KiB
Python
Executable File
#!/usr/bin/python3
|
|
|
|
import yaml
|
|
import imp
|
|
import os
|
|
import threading
|
|
import socket
|
|
import multiprocessing
|
|
import time
|
|
import sys
|
|
|
|
import proto
|
|
print('PyLink starting...')
|
|
|
|
with open("config.yml", 'r') as f:
|
|
conf = yaml.load(f)
|
|
|
|
# if conf['login']['password'] == 'changeme':
|
|
# print("You have not set the login details correctly! Exiting...")
|
|
|
|
class Irc():
|
|
def __init__(self):
|
|
# Initialize some variables
|
|
self.socket = socket.socket()
|
|
self.connected = False
|
|
self.users = {}
|
|
self.channels = {}
|
|
self.name = conf['server']['netname']
|
|
self.conf = conf
|
|
|
|
self.serverdata = conf['server']
|
|
ip = self.serverdata["ip"]
|
|
port = self.serverdata["port"]
|
|
self.sid = self.serverdata["sid"]
|
|
print("Connecting to network %r on %s:%s" % (self.name, ip, port))
|
|
|
|
self.socket = socket.socket()
|
|
self.socket.connect((ip, port))
|
|
proto.connect(self)
|
|
self.connected = True
|
|
self.loaded = []
|
|
self.load_plugins()
|
|
self.run()
|
|
|
|
def run(self):
|
|
buf = ""
|
|
data = ""
|
|
while self.connected:
|
|
try:
|
|
data = self.socket.recv(4096).decode("utf-8")
|
|
buf += data
|
|
if not data:
|
|
break
|
|
while '\n' in buf:
|
|
line, buf = buf.split('\n', 1)
|
|
print("<- {}".format(line))
|
|
proto.handle_events(self, line)
|
|
except socket.error as e:
|
|
print('Received socket.error: %s, exiting.' % str(e))
|
|
break
|
|
sys.exit(1)
|
|
|
|
def send(self, data):
|
|
data = data.encode("utf-8") + b"\n"
|
|
print("-> {}".format(data.decode("utf-8").strip("\n")))
|
|
self.socket.send(data)
|
|
|
|
def load_plugins(self):
|
|
to_load = conf['plugins']
|
|
plugins_folder = [os.path.join(os.getcwd(), 'plugins')]
|
|
# Here, we override the module lookup and import the plugins
|
|
# dynamically depending on which were configured.
|
|
for plugin in to_load:
|
|
moduleinfo = imp.find_module(plugin, plugins_folder)
|
|
self.loaded.append(imp.load_source(plugin, moduleinfo[1]))
|
|
print("loaded plugins: %s" % self.loaded)
|
|
|
|
|
|
irc_obj = Irc()
|