2024-05-17 16:01:33 +00:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
2024-05-29 14:55:49 +00:00
|
|
|
"errors"
|
2024-05-28 19:19:30 +00:00
|
|
|
"log"
|
2024-05-28 17:17:30 +00:00
|
|
|
"net/http"
|
2024-05-28 19:19:30 +00:00
|
|
|
|
2024-05-28 17:17:30 +00:00
|
|
|
"github.com/gin-contrib/cors"
|
|
|
|
"github.com/gin-gonic/gin"
|
2024-05-31 14:45:28 +00:00
|
|
|
"github.com/joho/godotenv"
|
2024-06-10 04:50:15 +00:00
|
|
|
"gorm.io/driver/sqlite"
|
|
|
|
"gorm.io/gorm"
|
2024-05-30 15:42:33 +00:00
|
|
|
|
|
|
|
"log101-blog-services/middleware"
|
2024-05-17 16:01:33 +00:00
|
|
|
)
|
|
|
|
|
2024-06-10 04:50:15 +00:00
|
|
|
var db *gorm.DB
|
2024-05-28 19:19:30 +00:00
|
|
|
|
2024-05-29 14:55:49 +00:00
|
|
|
// Emoji counts for each post are retrieved in this format
|
2024-06-10 04:50:15 +00:00
|
|
|
type PostReaction struct {
|
2024-05-29 07:25:59 +00:00
|
|
|
Emoji string
|
|
|
|
TotalCount int
|
|
|
|
}
|
|
|
|
|
2024-06-10 04:50:15 +00:00
|
|
|
type EmojiReaction struct {
|
|
|
|
UserAnonIp string
|
|
|
|
PostId string
|
|
|
|
Emoji string
|
|
|
|
}
|
2024-05-29 14:55:49 +00:00
|
|
|
|
|
|
|
// Get the emoji counts foe a given post id
|
2024-06-10 04:50:15 +00:00
|
|
|
func countEmojis(postId string) (map[string]int, error) {
|
|
|
|
postReactions := []PostReaction{}
|
2024-05-29 14:55:49 +00:00
|
|
|
|
|
|
|
// Get the emoji counts for each emoji
|
2024-06-10 04:50:15 +00:00
|
|
|
// rows, err := db.Query("SELECT emoji, COUNT(*) FROM emoji_clicks WHERE post_id = ? GROUP BY emoji;", postId)
|
|
|
|
rows := db.Model(&EmojiReaction{}).Where("post_id = ?", postId).Select([]string{"emoji", "COUNT(*) as total_count"}).Group("emoji").Find(&postReactions)
|
|
|
|
if rows.Error != nil {
|
|
|
|
return nil, errors.New("couldn't get emoji counts")
|
2024-05-29 14:55:49 +00:00
|
|
|
}
|
|
|
|
|
2024-06-10 04:50:15 +00:00
|
|
|
// Transform rows to map
|
|
|
|
result := make(map[string]int)
|
|
|
|
for _, reaction := range postReactions {
|
|
|
|
result[reaction.Emoji] = reaction.TotalCount
|
2024-05-29 14:55:49 +00:00
|
|
|
}
|
|
|
|
|
2024-06-10 04:50:15 +00:00
|
|
|
return result, nil
|
2024-05-29 14:55:49 +00:00
|
|
|
}
|
|
|
|
|
2024-05-17 16:01:33 +00:00
|
|
|
func main() {
|
2024-06-02 08:44:39 +00:00
|
|
|
// Load environment variables
|
2024-05-31 14:45:28 +00:00
|
|
|
err := godotenv.Load()
|
|
|
|
if err != nil {
|
|
|
|
log.Fatal("Error loading .env file")
|
|
|
|
}
|
|
|
|
|
2024-06-10 04:50:15 +00:00
|
|
|
db, err = gorm.Open(sqlite.Open("test.db"), &gorm.Config{})
|
2024-05-28 19:19:30 +00:00
|
|
|
if err != nil {
|
2024-06-10 04:50:15 +00:00
|
|
|
panic("failed to connect database")
|
2024-05-28 19:19:30 +00:00
|
|
|
}
|
|
|
|
|
2024-06-10 04:50:15 +00:00
|
|
|
db.AutoMigrate(&EmojiReaction{})
|
2024-05-28 19:19:30 +00:00
|
|
|
|
2024-05-28 17:17:30 +00:00
|
|
|
r := gin.Default()
|
|
|
|
r.LoadHTMLGlob("templates/*")
|
2024-05-17 16:01:33 +00:00
|
|
|
|
2024-05-29 14:55:49 +00:00
|
|
|
// CORS configuration
|
2024-05-28 17:17:30 +00:00
|
|
|
corsConfig := cors.DefaultConfig()
|
2024-05-30 10:53:20 +00:00
|
|
|
corsConfig.AllowOrigins = []string{"https://log101.dev"}
|
|
|
|
|
|
|
|
ginMode := gin.Mode()
|
|
|
|
if ginMode == gin.DebugMode {
|
|
|
|
corsConfig.AllowOrigins = append(corsConfig.AllowOrigins, "http://localhost:4321")
|
|
|
|
}
|
|
|
|
|
2024-06-10 04:50:15 +00:00
|
|
|
corsConfig.AllowHeaders = []string{"hx-current-url", "hx-request", "hx-target", "hx-trigger"}
|
2024-05-30 15:42:33 +00:00
|
|
|
|
|
|
|
// Middlewares
|
2024-05-28 17:17:30 +00:00
|
|
|
r.Use(cors.New(corsConfig))
|
2024-05-30 15:42:33 +00:00
|
|
|
r.Use(middleware.AnonymizeIPMiddleware())
|
2024-05-17 16:01:33 +00:00
|
|
|
|
2024-05-30 11:43:50 +00:00
|
|
|
var hxPostUrl string
|
|
|
|
if ginMode == gin.DebugMode {
|
|
|
|
hxPostUrl = "http://localhost:8000/blog/api/forms/emoji/post"
|
|
|
|
} else {
|
|
|
|
hxPostUrl = "https://log101.dev/blog/api/forms/emoji/post"
|
|
|
|
}
|
|
|
|
|
2024-05-30 10:30:32 +00:00
|
|
|
blogForm := r.Group("/blog/api/")
|
|
|
|
{
|
|
|
|
// Get the emoji form, you must provide postId query parameter
|
|
|
|
blogForm.GET("/forms/emoji", func(c *gin.Context) {
|
|
|
|
postId := c.Query("postId")
|
|
|
|
|
|
|
|
// get emoji counts for each emoji
|
|
|
|
emojiCounter, err := countEmojis(postId)
|
|
|
|
if err != nil {
|
2024-05-30 11:43:50 +00:00
|
|
|
c.HTML(http.StatusOK, "emoji_form.tmpl", gin.H{"error": "error getting the emoji counts", "postId": postId, "hxPostUrl": hxPostUrl})
|
2024-05-30 10:30:32 +00:00
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
// Return the html template
|
|
|
|
c.HTML(http.StatusOK, "emoji_form.tmpl", gin.H{
|
2024-05-30 11:43:50 +00:00
|
|
|
"postId": c.Query("postId"),
|
|
|
|
"results": emojiCounter,
|
|
|
|
"hxPostUrl": hxPostUrl,
|
2024-05-30 10:30:32 +00:00
|
|
|
})
|
2024-05-28 19:19:30 +00:00
|
|
|
})
|
|
|
|
|
2024-05-30 10:30:32 +00:00
|
|
|
// Update the emoji counter, this handler will
|
|
|
|
// add a new entry to the database with anonymized ip
|
|
|
|
// then sums the results to get the emoji counter
|
|
|
|
blogForm.POST("/forms/emoji/post", func(c *gin.Context) {
|
2024-06-10 04:50:15 +00:00
|
|
|
reactedPostId := c.PostForm("postId")
|
|
|
|
reaction := c.PostForm("emojiInput")
|
2024-05-30 10:30:32 +00:00
|
|
|
|
|
|
|
// Check if parameters are missing
|
2024-06-10 04:50:15 +00:00
|
|
|
if reactedPostId == "" || reaction == "" {
|
|
|
|
c.HTML(http.StatusOK, "emoji_form.tmpl", gin.H{"error": "missing parameters", "postId": reactedPostId, "hxPostUrl": hxPostUrl})
|
2024-05-30 10:30:32 +00:00
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
// Add the new emoji entry to the database
|
2024-06-10 04:50:15 +00:00
|
|
|
// _, err := db.Exec("INSERT INTO emoji_clicks (user_anon_data, post_id, emoji) VALUES (?, ?, ?) ON DUPLICATE KEY UPDATE emoji = ?", c.Request.RemoteAddr, postId, emoji, emoji)
|
|
|
|
result := db.Create(&EmojiReaction{UserAnonIp: c.Request.RemoteAddr, Emoji: reaction, PostId: reactedPostId})
|
|
|
|
if result.Error != nil {
|
|
|
|
c.HTML(http.StatusOK, "emoji_form.tmpl", gin.H{"error": "error writing to database", "postId": reactedPostId, "hxPostUrl": hxPostUrl})
|
2024-05-30 10:30:32 +00:00
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
// get emoji counts for each emoji
|
2024-06-10 04:50:15 +00:00
|
|
|
emojiCounter, err := countEmojis(reactedPostId)
|
2024-05-30 10:30:32 +00:00
|
|
|
if err != nil {
|
2024-06-10 04:50:15 +00:00
|
|
|
c.HTML(http.StatusOK, "emoji_form.tmpl", gin.H{"error": "error getting the emoji counts", "postId": reactedPostId, "hxPostUrl": hxPostUrl})
|
2024-05-30 10:30:32 +00:00
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
// Return the html with the updated emoji counter
|
2024-06-10 04:50:15 +00:00
|
|
|
c.HTML(http.StatusOK, "emoji_form.tmpl", gin.H{"postId": reactedPostId, "results": emojiCounter, "hxPostUrl": hxPostUrl})
|
2024-05-30 10:30:32 +00:00
|
|
|
})
|
|
|
|
}
|
2024-05-17 16:01:33 +00:00
|
|
|
|
2024-05-30 09:52:18 +00:00
|
|
|
r.Run(":8000")
|
2024-05-17 16:01:33 +00:00
|
|
|
}
|