From d4075bd0e7ba0eb7232a670cf9114adb428f1406 Mon Sep 17 00:00:00 2001
From: gpkvt <gpkvt@protonmail.com>
Date: Thu, 11 Aug 2022 12:14:58 +0200
Subject: [PATCH] Bugfix: Whitelist config values, Added favicon

---
 favicon.ico | Bin 0 -> 318 bytes
 tts.py      |  28 +++++++++++++++++++++-------
 2 files changed, 21 insertions(+), 7 deletions(-)
 create mode 100644 favicon.ico

diff --git a/favicon.ico b/favicon.ico
new file mode 100644
index 0000000000000000000000000000000000000000..200a6a49165bb7db9226e20bafb25667040d8ef9
GIT binary patch
literal 318
zcmb_W$qj%o409z0Xg|Dpb_-TRf<1Y{3kE=<*lB|pfV6g;qpE?x6JsQI$N}g9WFis+
zp}t}=uB4eqSzP^`ga1NPvY{(d-%QuajH>p&ZH1KTwivB4{iRZX7aR2j+yeEnHb`@C
LAAys3Cj89<Q?(&A

literal 0
HcmV?d00001

diff --git a/tts.py b/tts.py
index ad92621..1c3f666 100644
--- a/tts.py
+++ b/tts.py
@@ -20,11 +20,11 @@ class IRC:
     def __init__(self):
         self.irc = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
         self.tts_denied = []
+        self.tts_allowed = []
+        self.tts_status = True
+
         if 'WHITELIST_USER' in conf:
             self.tts_allowed = conf['WHITELIST_USER']
-        else:
-            self.tts_allowed = []
-        self.tts_status = True
 
     def connect(self, server, port, channel, botnick, botpass):
         logging.info("Connecting to: " + server)
@@ -144,6 +144,7 @@ class IRC:
                         logging.debug('Removing "@" from username')
                         user = user.replace('@', '')
 
+                    logging.info("Adding "+str(user)+" to whitelist")
                     self.tts_allowed.append(user)
 
                     if user in self.tts_denied:
@@ -197,9 +198,14 @@ class IRC:
                         return False
 
                 if conf['WHITELIST']:
-                    if not user in self.tts_allowed:
-                        logging.info('User is not on whitelist')
-                        logging.info(self.tts_allowed)
+                    if self.tts_allowed:
+                        if not user in self.tts_allowed:
+                            logging.info('User is not on whitelist')
+                            logging.info(self.tts_allowed)
+                            self.sendpriv(conf['IRC_CHANNEL'], "@"+str(user), conf['MESSAGE']['WHITELISTONLY'])
+                            return False
+                    else:
+                        logging.info('Nobody is on the whitelist.')
                         self.sendpriv(conf['IRC_CHANNEL'], "@"+str(user), conf['MESSAGE']['WHITELISTONLY'])
                         return False
 
@@ -225,6 +231,14 @@ class HTTPserv(BaseHTTPRequestHandler):
                 html = fh.read()
                 self.wfile.write(html)
 
+        elif self.path == '/favicon.ico':
+            self.send_response(200)
+            self.send_header('Content-type', 'image/x-icon')
+            self.end_headers()
+            with open("favicon.ico", "rb") as fh:
+                icon = fh.read()
+                self.wfile.write(icon)
+
         elif self.path == '/tts.js':
             self.send_response(200)
             self.send_header('Content-type', 'text/javascript')
@@ -368,7 +382,7 @@ msg_queue_raw = []
 msg_queue = {}
 
 logging.basicConfig(level=logging.DEBUG, format='%(asctime)s %(module)s %(threadName)s %(levelname)s: %(message)s')
-sys.tracebacklimit = 0
+sys.tracebacklimit = 1
 
 def main():
     conf = load_config()