From 9ccd47b2360c29e2963124ae84963d8e9e4d90c9 Mon Sep 17 00:00:00 2001
From: Mysh! <me@mysh.dev>
Date: Mon, 27 Feb 2023 11:57:42 +0300
Subject: [PATCH] Add bypass for SMTP settings and fix maddyctl call

---
 main.go | 71 +++++++++++++++++++++++++++++++++------------------------
 1 file changed, 41 insertions(+), 30 deletions(-)

diff --git a/main.go b/main.go
index 128f3d5..d909832 100644
--- a/main.go
+++ b/main.go
@@ -66,6 +66,8 @@ const (
 
 	// HTTPServerPort is an HTTP server port
 	HTTPServerPort = 1323
+
+	DebugBypassMailSending = true
 )
 
 const (
@@ -97,26 +99,34 @@ func (t *Template) Render(w io.Writer, name string, data interface{}, _ echo.Con
 }
 
 func main() {
-	log.Println("[EmailMessage const] Checking your template")
-	if !strings.Contains(EmailMessage, "$RESET_LINK") {
-		log.Fatalln("[EmailMessage const] Your message template does not contain $RESET_LINK, so user can't reset his password!")
-	}
+	var auth smtp.Auth
+	if !DebugBypassMailSending {
+		log.Println("[EmailMessage const] Checking your template")
+		if !strings.Contains(EmailMessage, "$RESET_LINK") {
+			log.Fatalln("[EmailMessage const] Your message template does not contain $RESET_LINK, so user can't reset his password!")
+		}
 
-	log.Println("[EmailTemplate const] Checking your template")
-	if !strings.Contains(EmailTemplate, "$TO") {
-		log.Fatalln("[EmailTemplate const] Your template does not contain $TO, make sure to add it.")
-	}
+		log.Println("[EmailTemplate const] Checking your template")
+		if !strings.Contains(EmailTemplate, "$TO") {
+			log.Fatalln("[EmailTemplate const] Your template does not contain $TO, make sure to add it.")
+		}
 
-	if !strings.Contains(EmailTemplate, "$FROM") {
-		log.Fatalln("[EmailTemplate const] Your template does not contain $FROM, make sure to add it.")
-	}
+		if !strings.Contains(EmailTemplate, "$FROM") {
+			log.Fatalln("[EmailTemplate const] Your template does not contain $FROM, make sure to add it.")
+		}
 
-	if !strings.Contains(EmailTemplate, "$SUBJECT") {
-		log.Fatalln("[EmailTemplate const] Your template does not contain $SUBJECT, make sure to add it, so user can see a message preview.")
-	}
+		if !strings.Contains(EmailTemplate, "$SUBJECT") {
+			log.Fatalln("[EmailTemplate const] Your template does not contain $SUBJECT, make sure to add it, so user can see a message preview.")
+		}
 
-	if !strings.Contains(EmailTemplate, "$MESSAGE") {
-		log.Fatalln("[EmailTemplate const] Your template does not contain $MESSAGE, make sure to add it.")
+		if !strings.Contains(EmailTemplate, "$MESSAGE") {
+			log.Fatalln("[EmailTemplate const] Your template does not contain $MESSAGE, make sure to add it.")
+		}
+
+		// Set up authentication information.
+		auth = smtp.PlainAuth("", SMTPMailUsername, SMTPMailPassword, SMTPMailHostname)
+	} else {
+		log.Println("[SMTP] Debug mode enabled, not checking email template")
 	}
 
 	log.Println("[Sqlite] Loading Maddy's credentials database")
@@ -125,9 +135,6 @@ func main() {
 		log.Fatalln(err)
 	}
 
-	// Set up authentication information.
-	auth := smtp.PlainAuth("", SMTPMailUsername, SMTPMailPassword, SMTPMailHostname)
-
 	log.Println("[Cache] Registering cache for password resets")
 	passwordResetCache := cache.New(CacheTime)
 
@@ -178,17 +185,21 @@ func main() {
 			// and send the email all in one step.
 			to := []string{mail}
 
-			msg := strings.ReplaceAll(EmailTemplate, "$TO", mail)
-			msg = strings.ReplaceAll(msg, "$FROM", EmailFrom)
-			msg = strings.ReplaceAll(msg, "$SUBJECT", EmailSubject)
-			msg = strings.ReplaceAll(msg, "$MESSAGE", EmailMessage)
-			msg = strings.ReplaceAll(msg, "$RESET_LINK", HostingURL+"reset/"+random)
+			if !DebugBypassMailSending {
+				msg := strings.ReplaceAll(EmailTemplate, "$TO", mail)
+				msg = strings.ReplaceAll(msg, "$FROM", EmailFrom)
+				msg = strings.ReplaceAll(msg, "$SUBJECT", EmailSubject)
+				msg = strings.ReplaceAll(msg, "$MESSAGE", EmailMessage)
+				msg = strings.ReplaceAll(msg, "$RESET_LINK", HostingURL+"reset/"+random)
 
-			//msg := []byte(fmt.Sprintf(EmailMessage, mail, SMTPMailUsername, HostingURL+"reset/"+random))
-			err := smtp.SendMail(MXServer, auth, SMTPMailUsername, to, []byte(msg))
-			if err != nil {
-				log.Println("[SMTP] Failed to send mail - ", err)
-				return
+				err := smtp.SendMail(MXServer, auth, SMTPMailUsername, to, []byte(msg))
+				if err != nil {
+					log.Println("[SMTP] Failed to send mail - ", err)
+					return
+				}
+			} else {
+				log.Println("[SMTP] Debug mode enabled, not sending email")
+				log.Println("[SMTP] Reset link:", HostingURL+"reset/"+random)
 			}
 		}()
 		return c.Render(http.StatusOK, "reset.gohtml", map[string]any{
@@ -215,7 +226,7 @@ func main() {
 			passwordResetCache.Delete(key)
 		}
 
-		maddyExecCommand := exec.Command("maddy", "creds", "password", mail.(string), "-p", password)
+		maddyExecCommand := exec.Command("maddy", "creds", "password", "-p", password, mail.(string))
 		err = maddyExecCommand.Run()
 		if err != nil {
 			log.Println("[maddyExecCommand] Failed to execute Maddy's password reset command - ", err)