feat: save messages to database (close #37)

controller/message.go

@@ -1,16 +1,21 @@
 package controller
 
 import (
+	"bytes"
 	"encoding/json"
+	"fmt"
 	"github.com/gin-gonic/gin"
+	"github.com/yuin/goldmark"
 	"message-pusher/channel"
 	"message-pusher/common"
 	"message-pusher/model"
 	"net/http"
+	"strconv"
+	"time"
 )
 
 func GetPushMessage(c *gin.Context) {
-	message := channel.Message{
+	message := model.Message{
 		Title:       c.Query("title"),
 		Description: c.Query("description"),
 		Content:     c.Query("content"),
@@ -30,7 +35,7 @@ func GetPushMessage(c *gin.Context) {
 }
 
 func PostPushMessage(c *gin.Context) {
-	message := channel.Message{
+	message := model.Message{
 		Title:       c.PostForm("title"),
 		Description: c.PostForm("description"),
 		Content:     c.PostForm("content"),
@@ -39,7 +44,7 @@ func PostPushMessage(c *gin.Context) {
 		Token:       c.PostForm("token"),
 		Desp:        c.PostForm("desp"),
 	}
-	if message == (channel.Message{}) {
+	if message == (model.Message{}) {
 		// Looks like the user is using JSON
 		err := json.NewDecoder(c.Request.Body).Decode(&message)
 		if err != nil {
@@ -56,7 +61,7 @@ func PostPushMessage(c *gin.Context) {
 	pushMessageHelper(c, &message)
 }
 
-func pushMessageHelper(c *gin.Context, message *channel.Message) {
+func pushMessageHelper(c *gin.Context, message *model.Message) {
 	user := model.User{Username: c.Param("username")}
 	err := user.FillUserByUsername()
 	if err != nil {
@@ -108,7 +113,16 @@ func pushMessageHelper(c *gin.Context, message *channel.Message) {
 			message.Channel = channel.TypeEmail
 		}
 	}
-	err = message.Send(&user)
+	err = message.UpdateAndInsert(user.Id)
+	message.URL = fmt.Sprintf("%s/message/%s", common.ServerAddress, message.Link)
+	if err != nil {
+		c.JSON(http.StatusOK, gin.H{
+			"success": false,
+			"message": err.Error(),
+		})
+		return
+	}
+	err = channel.SendMessage(message, &user)
 	if err != nil {
 		c.JSON(http.StatusOK, gin.H{
 			"success": false,
@@ -122,3 +136,108 @@ func pushMessageHelper(c *gin.Context, message *channel.Message) {
 	})
 	return
 }
+
+func GetStaticFile(c *gin.Context) {
+	path := c.Param("file")
+	c.FileFromFS("public/static/"+path, http.FS(common.FS))
+}
+
+func RenderMessage(c *gin.Context) {
+	link := c.Param("link")
+	message, err := model.GetMessageByLink(link)
+	if err != nil {
+		c.Status(http.StatusNotFound)
+		return
+	}
+	if message.Content != "" {
+		var buf bytes.Buffer
+		err := goldmark.Convert([]byte(message.Content), &buf)
+		if err != nil {
+			common.SysLog(err.Error())
+		} else {
+			message.HTMLContent = buf.String()
+		}
+	}
+	c.HTML(http.StatusOK, "message.html", gin.H{
+		"title":       message.Title,
+		"time":        time.Unix(message.Timestamp, 0).Format("2006-01-02 15:04:05"),
+		"description": message.Description,
+		"content":     message.HTMLContent,
+	})
+	return
+}
+
+func GetUserMessages(c *gin.Context) {
+	userId := c.GetInt("id")
+	p, _ := strconv.Atoi(c.Query("p"))
+	if p < 0 {
+		p = 0
+	}
+	messages, err := model.GetMessagesByUserId(userId, p*common.ItemsPerPage, common.ItemsPerPage)
+	if err != nil {
+		c.JSON(http.StatusOK, gin.H{
+			"success": false,
+			"message": err.Error(),
+		})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{
+		"success": true,
+		"message": "",
+		"data":    messages,
+	})
+	return
+}
+
+func GetMessage(c *gin.Context) {
+	messageId, _ := strconv.Atoi(c.Param("id"))
+	userId := c.GetInt("id")
+	message, err := model.GetMessageById(messageId, userId)
+	if err != nil {
+		c.JSON(http.StatusOK, gin.H{
+			"success": false,
+			"message": err.Error(),
+		})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{
+		"success": true,
+		"message": "",
+		"data":    message,
+	})
+	return
+}
+
+func DeleteMessage(c *gin.Context) {
+	messageId, _ := strconv.Atoi(c.Param("id"))
+	userId := c.GetInt("id")
+	err := model.DeleteMessageById(messageId, userId)
+	if err != nil {
+		c.JSON(http.StatusOK, gin.H{
+			"success": false,
+			"message": err.Error(),
+		})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{
+		"success": true,
+		"message": "",
+	})
+	return
+}
+
+func DeleteAllMessages(c *gin.Context) {
+	err := model.DeleteAllMessages()
+	if err != nil {
+		c.JSON(http.StatusOK, gin.H{
+			"success": false,
+			"message": err.Error(),
+		})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{
+		"success": true,
+		"message": "",
+	})
+	return
+}