From 9e9275834728a965eae36cee4579dbdc108dc018 Mon Sep 17 00:00:00 2001
From: Gareth Pulham <garethpulham@gmail.com>
Date: Wed, 3 Feb 2021 01:37:57 +0000
Subject: [PATCH] Add registration via XML-RPC

---
 freenode.py | 29 +++++++++++++++++++++++++++++
 web.py      |  4 +++-
 2 files changed, 32 insertions(+), 1 deletion(-)
 create mode 100644 freenode.py

diff --git a/freenode.py b/freenode.py
new file mode 100644
index 0000000..47d3437
--- /dev/null
+++ b/freenode.py
@@ -0,0 +1,29 @@
+import xmlrpc.client
+
+class RegistrationError(RuntimeError):
+    pass
+
+# ValueError returns with a reason, and permits token/username reuse
+# RegistrationError returns a reason but requires staff intervention (stuck account)
+def register_user(username, password, certfp):
+    services = xmlrpc.client.ServerProxy("http://localhost:8192/xmlrpc")
+
+    fakemail = "{}@tor.fake.mail.freenode.net".format(username)
+    feedback = services.atheme.command("*", "*", "::1337", "nickserv", "register", username, password, fakemail)
+    if "is now registered" not in feedback:
+        raise ValueError(feedback)
+
+    try:
+        cookie = services.atheme.login(username, password)
+    except xmlrpc.client.Fault as e:
+        raise RegistrationError(
+            "Unable to update account {}, fault {}. Please report this error and username to staff".format(
+                username, e.faultCode
+            )
+        )
+    
+    feedback = services.atheme.command(cookie, username, "::1337", "nickserv", "cert", "add", certfp)
+    if "Added fingerprint" not in feedback:
+        token = services.atheme.command(cookie, username, "::1337", "nickserv", "drop", username).split()[-1]
+        services.atheme.command(cookie, username, "::1337", "nickserv", "drop", username, token)
+        raise ValueError(feedback)
diff --git a/web.py b/web.py
index 3a60af1..df37db0 100644
--- a/web.py
+++ b/web.py
@@ -2,6 +2,8 @@ from flask import Flask, render_template, request, g, jsonify
 import sqlite3
 import hashcash
 
+import freenode
+
 hash_difficulty = 4
 hash_expiry = 7 # Days
 
@@ -65,7 +67,7 @@ def register():
         return jsonify({"status": "failure", "reason": str(e)})
 
     try:
-        # freenode.register_user(username, password, certfp=fingerprint, tor_only=True)
+        freenode.register_user(username, password, certfp=fingerprint)
         database.commit()
     except NotImplementedError as e: # registration_error
         database.rollback()
-- 
2.34.4