From 3a34b95e0121f9f57ddc09c935b75e747ef4faae Mon Sep 17 00:00:00 2001
From: Gautam Kumar <129857715+whoisgautxm@users.noreply.github.com>
Date: Wed, 14 Aug 2024 17:54:56 +0530
Subject: [PATCH] feat: Remove '+' from all email addresses before handling

---
 packages/prover/local.py        | 8 ++++++++
 packages/prover/modal_server.py | 9 +++++++++
 2 files changed, 17 insertions(+)

diff --git a/packages/prover/local.py b/packages/prover/local.py
index eb08e46d..2f431a25 100644
--- a/packages/prover/local.py
+++ b/packages/prover/local.py
@@ -69,6 +69,10 @@ def prove_claim():
 def prove_email_sender():
     req = request.get_json()
     input = req["input"]
+    # Clean the email address
+    if "email" in input:
+        input["email"] = clean_email(input["email"])
+
     nonce = random.randint(
         0,
         sys.maxsize,
@@ -76,6 +80,10 @@ def prove_email_sender():
     proof = gen_email_sender_proof(str(nonce), True, input)
     return jsonify(proof)
 
+def clean_email(email):
+    local_part, domain_part = email.split('@', 1)
+    cleaned_local_part = local_part.split('+', 1)[0]
+    return f"{cleaned_local_part}@{domain_part}"
 
 if __name__ == "__main__":
     from waitress import serve
diff --git a/packages/prover/modal_server.py b/packages/prover/modal_server.py
index e8ed898e..0bfbdc1e 100644
--- a/packages/prover/modal_server.py
+++ b/packages/prover/modal_server.py
@@ -79,6 +79,10 @@ def prove_claim():
     def prove_email_sender():
         req = request.get_json()
         input = req["input"]
+        # Clean the email address
+        if "email" in input:
+            input["email"] = clean_email(input["email"])
+        
         nonce = random.randint(
             0,
             sys.maxsize,
@@ -87,3 +91,8 @@ def prove_email_sender():
         return jsonify(proof)
 
     return app
+
+def clean_email(email):
+    local_part, domain_part = email.split('@', 1)
+    cleaned_local_part = local_part.split('+', 1)[0]
+    return f"{cleaned_local_part}@{domain_part}"
\ No newline at end of file