message.go 7.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339
  1. package controller
  2. import (
  3. "encoding/json"
  4. "fmt"
  5. "github.com/gin-gonic/gin"
  6. "message-pusher/channel"
  7. "message-pusher/common"
  8. "message-pusher/model"
  9. "net/http"
  10. "strconv"
  11. "time"
  12. )
  13. func GetPushMessage(c *gin.Context) {
  14. message := model.Message{
  15. Title: c.Query("title"),
  16. Description: c.Query("description"),
  17. Content: c.Query("content"),
  18. URL: c.Query("url"),
  19. Channel: c.Query("channel"),
  20. Token: c.Query("token"),
  21. To: c.Query("to"),
  22. }
  23. if message.Description == "" {
  24. // Keep compatible with ServerChan
  25. message.Description = c.Query("desp")
  26. }
  27. if message.Channel == "" {
  28. // Keep compatible with old version
  29. message.Channel = c.Query("type")
  30. }
  31. pushMessageHelper(c, &message)
  32. }
  33. func PostPushMessage(c *gin.Context) {
  34. message := model.Message{
  35. Title: c.PostForm("title"),
  36. Description: c.PostForm("description"),
  37. Content: c.PostForm("content"),
  38. URL: c.PostForm("url"),
  39. Channel: c.PostForm("channel"),
  40. Token: c.PostForm("token"),
  41. Desp: c.PostForm("desp"),
  42. To: c.PostForm("to"),
  43. }
  44. if message == (model.Message{}) {
  45. // Looks like the user is using JSON
  46. err := json.NewDecoder(c.Request.Body).Decode(&message)
  47. if err != nil {
  48. c.JSON(http.StatusOK, gin.H{
  49. "success": false,
  50. "message": "无法解析请求体,请检查其是否为合法 JSON",
  51. })
  52. return
  53. }
  54. }
  55. if message.Description == "" {
  56. message.Description = message.Desp
  57. }
  58. pushMessageHelper(c, &message)
  59. }
  60. func pushMessageHelper(c *gin.Context, message *model.Message) {
  61. user := model.User{Username: c.Param("username")}
  62. err := user.FillUserByUsername()
  63. if err != nil {
  64. c.JSON(http.StatusForbidden, gin.H{
  65. "success": false,
  66. "message": err.Error(),
  67. })
  68. return
  69. }
  70. if user.Status == common.UserStatusNonExisted {
  71. c.JSON(http.StatusForbidden, gin.H{
  72. "success": false,
  73. "message": "用户不存在",
  74. })
  75. return
  76. }
  77. if user.Status == common.UserStatusDisabled {
  78. c.JSON(http.StatusForbidden, gin.H{
  79. "success": false,
  80. "message": "用户已被封禁",
  81. })
  82. return
  83. }
  84. if user.Token != "" && user.Token != " " {
  85. if message.Token == "" {
  86. message.Token = c.Request.Header.Get("Authorization")
  87. if message.Token == "" {
  88. c.JSON(http.StatusForbidden, gin.H{
  89. "success": false,
  90. "message": "token 为空",
  91. })
  92. return
  93. }
  94. }
  95. if user.Token != message.Token {
  96. c.JSON(http.StatusForbidden, gin.H{
  97. "success": false,
  98. "message": "无效的 token",
  99. })
  100. return
  101. }
  102. }
  103. if message.Title == "" {
  104. message.Title = common.SystemName
  105. }
  106. if message.Channel == "" {
  107. message.Channel = user.Channel
  108. if message.Channel == "" {
  109. message.Channel = channel.TypeEmail
  110. }
  111. }
  112. err = saveAndSendMessage(&user, message)
  113. if err != nil {
  114. c.JSON(http.StatusOK, gin.H{
  115. "success": false,
  116. "message": err.Error(),
  117. })
  118. return
  119. }
  120. c.JSON(http.StatusOK, gin.H{
  121. "success": true,
  122. "message": "",
  123. })
  124. }
  125. func saveAndSendMessage(user *model.User, message *model.Message) error {
  126. message.Link = common.GetUUID()
  127. if message.URL == "" {
  128. message.URL = fmt.Sprintf("%s/message/%s", common.ServerAddress, message.Link)
  129. }
  130. success := false
  131. if common.MessagePersistenceEnabled {
  132. defer func() {
  133. // Update the status of the message
  134. status := common.MessageSendStatusFailed
  135. if success {
  136. status = common.MessageSendStatusSent
  137. }
  138. err := message.UpdateStatus(status)
  139. if err != nil {
  140. common.SysError("failed to update the status of the message: " + err.Error())
  141. }
  142. }()
  143. err := message.UpdateAndInsert(user.Id)
  144. if err != nil {
  145. return err
  146. }
  147. } else {
  148. message.Link = "unsaved" // This is for user to identify whether the message is saved
  149. }
  150. err := channel.SendMessage(message, user)
  151. if err != nil {
  152. return err
  153. }
  154. success = true
  155. return nil // After this line, the message status will be updated
  156. }
  157. func GetStaticFile(c *gin.Context) {
  158. path := c.Param("file")
  159. c.FileFromFS("public/static/"+path, http.FS(common.FS))
  160. }
  161. func RenderMessage(c *gin.Context) {
  162. if !common.MessageRenderEnabled {
  163. c.HTML(http.StatusOK, "message.html", gin.H{
  164. "title": "无法渲染",
  165. "time": time.Now().Format("2006-01-02 15:04:05"),
  166. "description": "超级管理员禁用了消息渲染",
  167. "content": "很抱歉,您所使用的消息推送服务的管理员禁用了消息渲染功能,因此您的消息无法渲染。",
  168. })
  169. return
  170. }
  171. link := c.Param("link")
  172. if link == "unsaved" {
  173. c.HTML(http.StatusOK, "message.html", gin.H{
  174. "title": "无法渲染",
  175. "time": time.Now().Format("2006-01-02 15:04:05"),
  176. "description": "超级管理员禁用了消息持久化",
  177. "content": "很抱歉,您所使用的消息推送服务的管理员禁用了消息持久化功能,您的消息并没有存储到数据库中,因此无法渲染。",
  178. })
  179. return
  180. }
  181. message, err := model.GetMessageByLink(link)
  182. if err != nil {
  183. c.Status(http.StatusNotFound)
  184. return
  185. }
  186. if message.Description != "" {
  187. message.Description, err = common.Markdown2HTML(message.Description)
  188. if err != nil {
  189. common.SysLog(err.Error())
  190. }
  191. }
  192. if message.Content != "" {
  193. message.HTMLContent, err = common.Markdown2HTML(message.Content)
  194. if err != nil {
  195. common.SysLog(err.Error())
  196. }
  197. }
  198. c.HTML(http.StatusOK, "message.html", gin.H{
  199. "title": message.Title,
  200. "time": time.Unix(message.Timestamp, 0).Format("2006-01-02 15:04:05"),
  201. "description": message.Description,
  202. "content": message.HTMLContent,
  203. })
  204. return
  205. }
  206. func GetUserMessages(c *gin.Context) {
  207. userId := c.GetInt("id")
  208. p, _ := strconv.Atoi(c.Query("p"))
  209. if p < 0 {
  210. p = 0
  211. }
  212. messages, err := model.GetMessagesByUserId(userId, p*common.ItemsPerPage, common.ItemsPerPage)
  213. if err != nil {
  214. c.JSON(http.StatusOK, gin.H{
  215. "success": false,
  216. "message": err.Error(),
  217. })
  218. return
  219. }
  220. c.JSON(http.StatusOK, gin.H{
  221. "success": true,
  222. "message": "",
  223. "data": messages,
  224. })
  225. return
  226. }
  227. func GetMessage(c *gin.Context) {
  228. messageId, _ := strconv.Atoi(c.Param("id"))
  229. userId := c.GetInt("id")
  230. message, err := model.GetMessageById(messageId, userId)
  231. if err != nil {
  232. c.JSON(http.StatusOK, gin.H{
  233. "success": false,
  234. "message": err.Error(),
  235. })
  236. return
  237. }
  238. c.JSON(http.StatusOK, gin.H{
  239. "success": true,
  240. "message": "",
  241. "data": message,
  242. })
  243. return
  244. }
  245. func SearchMessages(c *gin.Context) {
  246. keyword := c.Query("keyword")
  247. messages, err := model.SearchMessages(keyword)
  248. if err != nil {
  249. c.JSON(http.StatusOK, gin.H{
  250. "success": false,
  251. "message": err.Error(),
  252. })
  253. return
  254. }
  255. c.JSON(http.StatusOK, gin.H{
  256. "success": true,
  257. "message": "",
  258. "data": messages,
  259. })
  260. return
  261. }
  262. func ResendMessage(c *gin.Context) {
  263. messageId, _ := strconv.Atoi(c.Param("id"))
  264. userId := c.GetInt("id")
  265. helper := func() error {
  266. message, err := model.GetMessageById(messageId, userId)
  267. message.Id = 0
  268. if err != nil {
  269. return err
  270. }
  271. user, err := model.GetUserById(userId, true)
  272. if err != nil {
  273. return err
  274. }
  275. err = saveAndSendMessage(user, message)
  276. if err != nil {
  277. return err
  278. }
  279. return nil
  280. }
  281. err := helper()
  282. if err != nil {
  283. c.JSON(http.StatusOK, gin.H{
  284. "success": false,
  285. "message": err.Error(),
  286. })
  287. return
  288. }
  289. c.JSON(http.StatusOK, gin.H{
  290. "success": true,
  291. "message": "",
  292. })
  293. return
  294. }
  295. func DeleteMessage(c *gin.Context) {
  296. messageId, _ := strconv.Atoi(c.Param("id"))
  297. userId := c.GetInt("id")
  298. err := model.DeleteMessageById(messageId, userId)
  299. if err != nil {
  300. c.JSON(http.StatusOK, gin.H{
  301. "success": false,
  302. "message": err.Error(),
  303. })
  304. return
  305. }
  306. c.JSON(http.StatusOK, gin.H{
  307. "success": true,
  308. "message": "",
  309. })
  310. return
  311. }
  312. func DeleteAllMessages(c *gin.Context) {
  313. err := model.DeleteAllMessages()
  314. if err != nil {
  315. c.JSON(http.StatusOK, gin.H{
  316. "success": false,
  317. "message": err.Error(),
  318. })
  319. return
  320. }
  321. c.JSON(http.StatusOK, gin.H{
  322. "success": true,
  323. "message": "",
  324. })
  325. return
  326. }