Rewrite the logic for 'before-connect' key

Signed-off-by: Pratyush Desai <pratyush.desai@liberta.casa>

irc_register.py

@@ -5,101 +5,55 @@ import os
 webircpass = os.getenv("WEBIRC_PASS")
 
 def ircregister(userip, username, password, email="*"):
-
     d = irctokens.StatefulDecoder()
     e = irctokens.StatefulEncoder()
     s = socket.socket()
 
+    
     # Here we assume using this only on localhost i.e. loopback
     s.connect(("127.0.0.1", 6667))
 
-    # define the send function
     def _send(line):
         print(f"> {line.format()}")
         e.push(line)
         while e.pending():
             e.pop(s.send(e.pending()))
-
-    # Registering connection
-    
-    # WEBIRC
     _send(irctokens.build("WEBIRC", [ webircpass, "WebregGateway", userip, userip, "secure"]))
-
-    # Check for "ERROR :Invalid WebIRC password"
-    # Should all of this this be under a try except block?
     lines = d.push(s.recv(1024))
     if lines == None:
         print("!disconnected")
         return "disconnected"
     elif lines.command == "ERROR" and lines.params == "Invalid WebIRC password":
         return "WebIRC bad password"
-    
-    # Inform the server that we support
-    # CAP 3.2
     _send(irctokens.build("CAP", ["LS", "302"]))
-    # lines = d.push(s.recv(1024))
-    if 'draft/account-registration=before-connnect' in lines.params:
-        # add extra conditional for custom account and email-required
-        _send(irctokens.build("REGISTER" ["*", password]))
-    else:
-
-    # REGISTER can be attempted before-connect if server supports
-    # but if the server responds with the corresponding FAIL we
-    # need to try again. We can also handle email-required using
-    # the same keys. How to access these key-value pairs?
-    # reference: https://ircv3.net/specs/extensions/account-registration.html
-    
-    # NICK and USER
-        _send(irctokens.build("USER", ["u", "0", "*", username]))
-        _send(irctokens.build("NICK", [username]))
-
-        # go through all cases
-
-        while True:
-            for line in lines:
-                print(f"< {line.format()}")
-                if line.command == "432":
-                    return "ERR_ERRONEUSNICKNAME"
-                elif line.command == "433":
-                    return "ERR_NICKNAMEINUSE"
-                _send(irctokens.build("CAP", ["REQ", "draft/account-registration"]))
-                if line.command == "CAP" and ("NAK" in line.params):
-                    return "cap refused"
-                elif line.command == "CAP" and ("ACK" in line.params):
-                    to_send = irctokens.build("CAP", ["END"])
-                    _send(to_send)
-                if line.command == "PING":
-                    to_send = irctokens.build("PONG", [line.params[0]])
-                    _send(to_send)
-                if line.command == "001":
-                    # assuming no verif reqd.
-                    to_send = irctokens.build("REGISTER", ["*", email, password])
-                    _send(to_send)
-                if line.command == "REGISTER" and ("SUCCESS" in line.params):
-                    to_send = irctokens.build("QUIT")
-                    _send(to_send)
-                    return "SUCCESS"
     while True:
+        lines = d.push(s.recv(1024))
+        if lines == None:
+            print("! disconnected")
+            break        
         for line in lines:
             print(f"< {line.format()}")
-            if line.command == "432":
+            # REGISTER can be attempted before-connect if server supports
-                return "ERR_ERRONEUSNICKNAME"
+            # but if the server responds with the corresponding FAIL we
-            elif line.command == "433":
+            # need to try again. We can also handle email-required using
-                return "ERR_NICKNAMEINUSE"
+            # the same keys. How to access these key-value pairs?
-            _send(irctokens.build("CAP", ["REQ", "draft/account-registration"]))
+            # reference: https://ircv3.net/specs/extensions/account-registration.html
+            # do we handle if email-req but not before-connect
+            # also how about when `custom-account-name` may be a value
+            if 'draft/account-registration=before-connect' in line.params:
+                _send(irctokens.build("CAP", ["REQ", "draft/account-registration"]))
             if line.command == "CAP" and ("NAK" in line.params):
                 return "CAP_REFUSED"
             elif line.command == "CAP" and ("ACK" in line.params):
-                to_send = irctokens.build("CAP", ["END"])
+                _send(irctokens.build("CAP", ["END"]))
-                _send(to_send)
+                _send(irctokens.build("USER", ["u", "0", "*", username]))
-            if line.command == "PING":
+                _send(irctokens.build("NICK", [username]))
-                to_send = irctokens.build("PONG", [line.params[0]])
+                if line.command == "432":
-                _send(to_send)
+                    return "ERR_ERRONEUSNICKNAME"
-            if line.command == "001":
+                if line.command == "433":
-                # assuming no verif reqd.
+                    return "ERR_NICKNAMEINUSE"
-                to_send = irctokens.build("REGISTER", ["*", email, password])
+                _send(irctokens.build("REGISTER", [username, "*", password]))
-                _send(to_send)
             if line.command == "REGISTER" and ("SUCCESS" in line.params):
-                to_send = irctokens.build("QUIT")
+                _send(irctokens.build("QUIT"))
-                _send(to_send)
                 return "SUCCESS"
+	

requirements.txt

@@ -1,10 +0,0 @@
-click==8.0.1
-Flask==2.0.1
-Flask-WTF==0.15.1
-gunicorn==20.1.0
-irctokens==2.0.0
-itsdangerous==2.0.1
-Jinja2==3.0.1
-MarkupSafe==2.0.1
-Werkzeug==2.0.1
-WTForms==2.3.3