From e6f47c306a324d26fba29a2a039d0939323401dd Mon Sep 17 00:00:00 2001
From: "Jay.Fang" <jayfang1993@gmail.com>
Date: Fri, 23 Dec 2022 00:37:54 +0800
Subject: [PATCH] feat: support custom message URL (#39)

* feat:custom message url

* feat: support custom URL

Co-authored-by: JustSong <quanpengsong@gmail.com>
---
 controller/message.go | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/controller/message.go b/controller/message.go
index ba5a98c..5289ce5 100644
--- a/controller/message.go
+++ b/controller/message.go
@@ -113,6 +113,7 @@ func pushMessageHelper(c *gin.Context, message *model.Message) {
 			message.Channel = channel.TypeEmail
 		}
 	}
+	link := "unsaved"
 	if common.MessagePersistenceEnabled {
 		err = message.UpdateAndInsert(user.Id)
 		if err != nil {
@@ -122,9 +123,10 @@ func pushMessageHelper(c *gin.Context, message *model.Message) {
 			})
 			return
 		}
-		message.URL = fmt.Sprintf("%s/message/%s", common.ServerAddress, message.Link)
-	} else {
-		message.URL = fmt.Sprintf("%s/message/unsaved", common.ServerAddress)
+		link = message.Link
+	}
+	if message.URL == "" {
+		message.URL = fmt.Sprintf("%s/message/%s", common.ServerAddress, link)
 	}
 	err = channel.SendMessage(message, &user)
 	if err != nil {