add new module: irc-bridge

modules/irc-bridge/bot.py

@@ -0,0 +1,122 @@
+import time
+import socket
+import threading
+
+login_lines_raw = [
+    "NICK bridge",
+    "USER bridge bridge * !",
+    "JOIN #io23-bridge"
+]
+
+login_lines = map( lambda x: (x + "\n").encode("utf-8"), login_lines_raw )
+
+recv_s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+
+send_buffer = []
+reconnecting = False
+
+LINE_END = "\r\n".encode("utf-8")
+
+recv_s.bind( ("127.0.0.1", 5001) )
+
+def reconnect():
+    global s, reconnecting
+    reconnecting = True
+    while True:
+        try:
+            s.close()
+        except:
+            pass
+        
+        try:
+            s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+            s.connect( ("10.1.1.1", 6667) )
+    
+            for i in login_lines:
+                s.send(i)
+                time.sleep(0.5)
+            
+            reconnecting = False
+            break
+        except Exception as e:
+            print(f"[reconnect:main] ERROR: got exception {e}")
+            t = time.time()
+            open("error.log", "a", encoding = "utf-8").write(f"[{t}] recv_thread:main ERROR: got exception {e}\n")
+
+            time.sleep(2)
+
+def recv_thread():
+    global s, send_buffer, reconnecting
+    databuffer = bytes()
+    while True:
+        try:
+            new_data = s.recv(1024)
+            print(f"[DEBUG] new_data: {new_data}")
+            if len(new_data) == 0:
+                if not reconnecting:
+                    reconnect() # bin the old socket and get a new one instead
+                continue
+
+            databuffer += new_data
+
+            if LINE_END in databuffer:
+                commands = map( lambda x: x.decode("UTF-8"), databuffer.split(LINE_END)[:-1])
+
+                for command in commands:
+                    if len(command) > 0:
+                        if command.split(" ")[0] == "PING":
+                            print("[DEBUG] recv_thread:command:ping: Responding with {}".format(command.split(" ")[1]).encode("UTF-8"))
+                            send_buffer.append(("PONG {}\n".format( command.split(" ")[1]).encode("UTF-8") ))
+                        else:
+                            pass
+
+                databuffer = databuffer.split(LINE_END)[-1]
+        except Exception as e:
+            print(f"[recv_thread:main] ERROR: got exception {e}")
+            t = time.time()
+            open("error.log", "a", encoding = "utf-8").write(f"[{t}] recv_thread:main ERROR: got exception {e}\n")
+            try:
+                if not reconnecting:
+                    reconnect()
+            except Exception as e:
+                print(f"[recv_thread:except:reconnect] ERROR: got exception {e}")
+                t = time.time()
+                open("error.log", "a", encoding = "utf-8").write(f"[{t}] recv_thread:except:reconnect ERROR: got exception {e}\n")
+
+def send_thread():
+    global s, send_buffer, reconnecting
+    while True:
+        if len(send_buffer) > 0:
+            try:
+                print(f"[DEBUG] send_thread: sending {send_buffer[0]}")
+                s.send(send_buffer[0])
+                del send_buffer[0]
+            except Exception as e:
+                print(f"[send_thread:main] ERROR: got exception {e}")
+                t = time.time()
+                open("error.log", "a", encoding = "utf-8").write(f"[{t}] send_thread:main ERROR: got exception {e}\n")
+
+                if not reconnecting:
+                    reconnect()
+
+            time.sleep(0.2)
+        else:
+            time.sleep(3)
+
+def msg_thread():
+    global recv_s, send_buffer
+    while True:
+        d = recv_s.recvfrom(16384)
+        print(f"[DEBUG] msg_thread:d: {d}")
+        send_buffer.append(d[0])
+
+st = threading.Thread(target = send_thread, args = [])
+rt = threading.Thread(target = recv_thread, args = [])
+mt = threading.Thread(target = msg_thread, args = [])
+
+reconnect()
+
+st.start()
+rt.start()
+mt.start()

modules/irc-bridge/index.py

@@ -0,0 +1,10 @@
+if self.MESSAGE["chat"]["id"] == -1001570221452:
+    l = min( len(self.MESSAGE["text"]), 300 )
+    new_msg = "{} {}: {}".format(self.MESSAGE.from_user.first_name,\
+            self.MESSAGE.from_user.last_name,\
+            self.MESSAGE.text.replace("\n", " | ")[:l])
+
+    import socket
+    s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+    s.sendto(f"PRIVMSG #io23-bridge :{new_msg}\n".encode("utf-8"), ("127.0.0.1", 5001))
+    del s

modules/irc-bridge/meta.json

@@ -0,0 +1,7 @@
+{
+    "start_on_boot": true,
+    "alias": "irc-bridge",
+    "version": 1,
+    "index_file": "index.py",
+    "predefine": "predefine.py"
+}

modules/irc-bridge/predefine.py

@@ -0,0 +1,5 @@
+import socket
+
+SEND_ADDR = ( "127.0.0.1", 5001 )
+
+s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)