Browse Source

feat: support custom message URL (#39)

* feat:custom message url

* feat: support custom URL

Co-authored-by: JustSong <[email protected]>
Jay.Fang 2 years ago
parent
commit
e6f47c306a
1 changed files with 5 additions and 3 deletions
  1. 5 3
      controller/message.go

+ 5 - 3
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 {