From 5c21f4b4eea750aa97887236dbe518ceea101e06 Mon Sep 17 00:00:00 2001
From: Aditya Bisht <adityabisht64@gmail.com>
Date: Thu, 12 Sep 2024 17:05:16 +0530
Subject: [PATCH] fix: subject template parsing

---
 packages/relayer/.env.example                 |  2 +-
 packages/relayer/src/core.rs                  |  2 ++
 .../relayer/src/utils/subject_templates.rs    | 26 ++++++++++++++-----
 3 files changed, 22 insertions(+), 8 deletions(-)

diff --git a/packages/relayer/.env.example b/packages/relayer/.env.example
index 5b1428cb..0f696907 100644
--- a/packages/relayer/.env.example
+++ b/packages/relayer/.env.example
@@ -6,7 +6,7 @@ CHAIN_ID=11155111                    # Chain ID of the testnet.
 
 SMTP_SERVER=
 
-PROVER_ADDRESS="https://zkemail--email-auth-prover-body-parsing-v1-0-0-flask-app.modal.run"
+PROVER_ADDRESS="https://zkemail--email-auth-prover-v1-3-0-flask-app.modal.run"
 
 DATABASE_URL= "postgres://test@localhost/emailauth_test"
 RELAYER_EMAIL_ADDR=
diff --git a/packages/relayer/src/core.rs b/packages/relayer/src/core.rs
index bc28757e..9939ab04 100644
--- a/packages/relayer/src/core.rs
+++ b/packages/relayer/src/core.rs
@@ -289,6 +289,8 @@ pub async fn handle_email(email: String) -> Result<EmailAuthEvent> {
                 proof: email_proof.clone(),
             };
 
+            println!("Email Auth Msg: {:?}", email_auth_msg);
+
             match CLIENT
                 .handle_recovery(
                     &request.controller_eth_addr,
diff --git a/packages/relayer/src/utils/subject_templates.rs b/packages/relayer/src/utils/subject_templates.rs
index fbffde21..e8ccd734 100644
--- a/packages/relayer/src/utils/subject_templates.rs
+++ b/packages/relayer/src/utils/subject_templates.rs
@@ -102,12 +102,13 @@ pub fn extract_template_vals(input: &str, templates: Vec<String>) -> Result<Vec<
                     .unwrap()
                     .find(input_decomposed[input_idx])
                     .ok_or(anyhow!("No string found"))?;
-                if string_match.start() != 0
-                    || string_match.end() != input_decomposed[input_idx].len()
-                {
+                if string_match.start() != 0 {
                     return Err(anyhow!("String must be the whole word"));
                 }
-                let string = string_match.as_str().to_string();
+                let mut string = string_match.as_str().to_string();
+                if string.contains("</div>") {
+                    string = string.split("</div>").collect::<Vec<&str>>()[0].to_string();
+                }
                 template_vals.push(TemplateValue::String(string));
             }
             "{uint}" => {
@@ -119,7 +120,11 @@ pub fn extract_template_vals(input: &str, templates: Vec<String>) -> Result<Vec<
                 {
                     return Err(anyhow!("Uint must be the whole word"));
                 }
-                let uint = U256::from_dec_str(uint_match.as_str()).unwrap();
+                let mut uint_match = uint_match.as_str();
+                if uint_match.contains("</div>") {
+                    uint_match = uint_match.split("</div>").collect::<Vec<&str>>()[0];
+                }
+                let uint = U256::from_dec_str(uint_match).unwrap();
                 template_vals.push(TemplateValue::Uint(uint));
             }
             "{int}" => {
@@ -130,7 +135,11 @@ pub fn extract_template_vals(input: &str, templates: Vec<String>) -> Result<Vec<
                 if int_match.start() != 0 || int_match.end() != input_decomposed[input_idx].len() {
                     return Err(anyhow!("Int must be the whole word"));
                 }
-                let int = I256::from_dec_str(int_match.as_str()).unwrap();
+                let mut int_match = int_match.as_str();
+                if int_match.contains("</div>") {
+                    int_match = int_match.split("</div>").collect::<Vec<&str>>()[0];
+                }
+                let int = I256::from_dec_str(int_match).unwrap();
                 template_vals.push(TemplateValue::Int(int));
             }
             "{decimals}" => {
@@ -143,7 +152,10 @@ pub fn extract_template_vals(input: &str, templates: Vec<String>) -> Result<Vec<
                 {
                     return Err(anyhow!("Decimals must be the whole word"));
                 }
-                let decimals = decimals_match.as_str().to_string();
+                let mut decimals = decimals_match.as_str().to_string();
+                if decimals.contains("</div>") {
+                    decimals = decimals.split("</div>").collect::<Vec<&str>>()[0].to_string();
+                }
                 template_vals.push(TemplateValue::Decimals(decimals));
             }
             "{ethAddr}" => {