feat: Remove '+' from all email addresses before handling

zkemail · Aug 14, 2024 · 3a34b95 · 3a34b95
1 parent bd9552e
commit 3a34b95
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/packages/prover/local.py b/packages/prover/local.py
@@ -69,13 +69,21 @@ 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,
     )
     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
@@ -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}"