From dc048daf547b199d3b826b1ef2ac926f69a8114d Mon Sep 17 00:00:00 2001
From: DictumMortuum <theio.vrefos@gmail.com>
Date: Mon, 20 May 2024 21:25:54 +0300
Subject: [PATCH] [auth] added password reset function overload

---
 cmd/servus-auth/config.go | 35 +++++++++++++++++++++++++++--------
 cmd/servus-auth/main.go   | 15 ++++++++++-----
 pkg/db/cache.go           |  8 ++++----
 3 files changed, 41 insertions(+), 17 deletions(-)

diff --git a/cmd/servus-auth/config.go b/cmd/servus-auth/config.go
index 0753ad9..42c427f 100644
--- a/cmd/servus-auth/config.go
+++ b/cmd/servus-auth/config.go
@@ -2,7 +2,9 @@ package main
 
 import (
 	"net/http"
+	"strings"
 
+	"github.com/supertokens/supertokens-golang/ingredients/emaildelivery"
 	"github.com/supertokens/supertokens-golang/recipe/dashboard"
 	"github.com/supertokens/supertokens-golang/recipe/session"
 	"github.com/supertokens/supertokens-golang/recipe/thirdparty"
@@ -16,7 +18,7 @@ var SuperTokensConfig = supertokens.TypeInput{
 		ConnectionURI: "http://sol.dictummortuum.com:3567",
 	},
 	AppInfo: supertokens.AppInfo{
-		AppName:   "Tables",
+		AppName:   "DictumMortuum",
 		APIDomain: "https://auth.dictummortuum.com",
 		// WebsiteDomain: "https://tables.dictummortuum.com",
 		GetOrigin: func(request *http.Request, userContext supertokens.UserContext) (string, error) {
@@ -27,12 +29,10 @@ var SuperTokensConfig = supertokens.TypeInput{
 					// there is a privacy setting on the frontend which doesn't send
 					// the origin
 				} else {
-					if origin == "https://tables.dictummortuum.com" {
-						// query from the test site
-						return "https://tables.dictummortuum.com", nil
-					} else if origin == "http://localhost:3000" {
-						// query from local development
-						return "http://localhost:3000", nil
+					for _, item := range Origins {
+						if origin == item {
+							return item, nil
+						}
 					}
 				}
 			}
@@ -42,7 +42,26 @@ var SuperTokensConfig = supertokens.TypeInput{
 		},
 	},
 	RecipeList: []supertokens.Recipe{
-		thirdpartyemailpassword.Init(&tpepmodels.TypeInput{}),
+		thirdpartyemailpassword.Init(&tpepmodels.TypeInput{
+			EmailDelivery: &emaildelivery.TypeInput{
+				Override: func(originalImplementation emaildelivery.EmailDeliveryInterface) emaildelivery.EmailDeliveryInterface {
+					ogSendEmail := *originalImplementation.SendEmail
+
+					(*originalImplementation.SendEmail) = func(input emaildelivery.EmailType, userContext supertokens.UserContext) error {
+						// You can change the path, domain of the reset password link,
+						// or even deep link it to your mobile app
+						// This is: `${websiteDomain}${websiteBasePath}/reset-password`
+						input.PasswordReset.PasswordResetLink = strings.Replace(
+							input.PasswordReset.PasswordResetLink,
+							"/auth/reset-password",
+							"/#/auth/reset-password", 1,
+						)
+						return ogSendEmail(input, userContext)
+					}
+					return originalImplementation
+				},
+			},
+		}),
 		session.Init(nil),
 		dashboard.Init(nil),
 		thirdparty.Init(nil),
diff --git a/cmd/servus-auth/main.go b/cmd/servus-auth/main.go
index 3027275..0434898 100644
--- a/cmd/servus-auth/main.go
+++ b/cmd/servus-auth/main.go
@@ -14,9 +14,17 @@ import (
 	"github.com/supertokens/supertokens-golang/supertokens"
 )
 
+var (
+	Origins = []string{
+		"https://tables.dictummortuum.com",
+		"https://tables.dictummortuum.com",
+		"http://localhost:3000",
+	}
+)
+
 func Version(c *gin.Context) {
 	rs := map[string]any{
-		"version": "v0.0.3",
+		"version": "v0.0.4",
 	}
 	c.AbortWithStatusJSON(200, rs)
 }
@@ -35,10 +43,7 @@ func main() {
 	r := gin.Default()
 
 	r.Use(cors.New(cors.Config{
-		AllowOrigins: []string{
-			"http://localhost:3000",
-			"https://tables.dictummortuum.com",
-		},
+		AllowOrigins: Origins,
 		AllowMethods: []string{"GET", "POST", "DELETE", "PUT", "OPTIONS"},
 		AllowHeaders: append([]string{"content-type"},
 			supertokens.GetAllCORSHeaders()...),
diff --git a/pkg/db/cache.go b/pkg/db/cache.go
index 10b04d3..b65be5e 100644
--- a/pkg/db/cache.go
+++ b/pkg/db/cache.go
@@ -9,7 +9,7 @@ import (
 	"github.com/redis/go-redis/v9"
 )
 
-func set(rdb *redis.Client, key string, val any) error {
+func Set(rdb *redis.Client, key string, val any) error {
 	p, err := json.Marshal(val)
 	if err != nil {
 		return err
@@ -23,7 +23,7 @@ func set(rdb *redis.Client, key string, val any) error {
 	return nil
 }
 
-func get(rdb *redis.Client, key string, dest any) error {
+func Get(rdb *redis.Client, key string, dest any) error {
 	p, err := rdb.Get(context.Background(), key).Result()
 	if err != nil {
 		return err
@@ -33,14 +33,14 @@ func get(rdb *redis.Client, key string, dest any) error {
 }
 
 func CachedSelect(DB *sqlx.DB, RDB *redis.Client, key string, dest any, query string, args ...any) error {
-	err := get(RDB, key, dest)
+	err := Get(RDB, key, dest)
 	if err == redis.Nil {
 		err = DB.Select(dest, query, args...)
 		if err != nil {
 			return err
 		}
 
-		err = set(RDB, key, dest)
+		err = Set(RDB, key, dest)
 		if err != nil {
 			return err
 		}