feat: new endpoints for posts and change in the get users to get by usernames.

This commit is contained in:
Nabil Ould Hamou 2024-03-04 17:07:01 +01:00
parent 2f99c09bfe
commit 3f5eeda369
No known key found for this signature in database
GPG key ID: 2B1AE73E9B6014EF
4 changed files with 124 additions and 15 deletions

View file

@ -7,13 +7,112 @@ import (
"github.com/gin-gonic/gin" "github.com/gin-gonic/gin"
"github.com/google/uuid" "github.com/google/uuid"
"github.com/h2non/filetype" "github.com/h2non/filetype"
"gorm.io/gorm"
"mime/multipart" "mime/multipart"
"net/http" "net/http"
"os" "os"
"path/filepath" "path/filepath"
"strconv"
"strings" "strings"
) )
type PostData struct {
PostId uuid.UUID `json:"id"`
Text string `json:"text"`
Files []models.File `json:"files"`
User UserData `json:"user"`
}
func GetPosts(c *gin.Context) {
page, err := strconv.Atoi(c.Query("page"))
if err != nil || page <= 0 {
page = 1
}
var posts []models.Post
offset := (page - 1) * 20
result := initializers.DB.Offset(offset).Limit(20).Order("created_at desc").Find(&posts)
if result.Error != nil {
c.Status(http.StatusBadRequest)
return
}
var response []PostData
for _, post := range posts {
var user models.User
result = initializers.DB.Preload("Avatar").First(&user, "id = ?", post.Author)
if result.Error != nil {
c.JSON(http.StatusInternalServerError, gin.H{"error": "Database error"})
return
}
respUser := PostData{
PostId: post.ID,
Text: post.Text,
Files: post.Files,
User: UserData{
ID: user.ID,
Username: user.Username,
DisplayName: user.DisplayName,
Avatar: user.Avatar.FileName,
},
}
response = append(response, respUser)
}
c.JSON(http.StatusAccepted, gin.H{
"posts": response,
})
}
func GetPostById(c *gin.Context) {
postId := c.Param("id")
uniqueId, err := uuid.Parse(postId)
if err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid id format"})
return
}
var postData PostData
var post models.Post
result := initializers.DB.Preload("Files").First(&post, "id = ?", uniqueId)
if result.Error != nil {
if errors.Is(result.Error, gorm.ErrRecordNotFound) {
c.JSON(http.StatusNotFound, gin.H{"error": "Post not found"})
} else {
c.JSON(http.StatusInternalServerError, gin.H{"error": "Database error"})
}
return
}
var user models.User
result = initializers.DB.Preload("Avatar").First(&user, "id = ?", post.Author)
if result.Error != nil {
c.JSON(http.StatusInternalServerError, gin.H{"error": "Database error"})
return
}
postData = PostData{
PostId: uniqueId,
Text: post.Text,
Files: post.Files,
User: UserData{
ID: user.ID,
Username: user.Username,
DisplayName: user.DisplayName,
Avatar: user.Avatar.FileName,
},
}
c.JSON(http.StatusOK, postData)
}
func CreatePost(c *gin.Context) { func CreatePost(c *gin.Context) {
form, _ := c.MultipartForm() form, _ := c.MultipartForm()
uploadedFiles := form.File["files"] uploadedFiles := form.File["files"]

View file

@ -11,33 +11,41 @@ import (
"strconv" "strconv"
) )
type UserData struct {
ID uuid.UUID `json:"id"`
Username string `json:"username"`
DisplayName string `json:"display_name"`
Avatar string `json:"profile_picture"`
}
// GetUsers Returns all users // GetUsers Returns all users
func GetUsers(c *gin.Context) { func GetUsers(c *gin.Context) {
page, err := strconv.Atoi(c.Query("page")) page, err := strconv.Atoi(c.Query("page"))
if err != nil || page <= 0 { if err != nil || page <= 0 {
page = 1 page = 1
} }
username := c.Query("username")
type userResponse struct {
ID uuid.UUID `json:"id"`
Username string `json:"username"`
DisplayName string `json:"display_name"`
Avatar string `json:"profile_picture"`
}
var users []models.User var users []models.User
offset := (page - 1) * 50 offset := (page - 1) * 20
var result *gorm.DB
if username != "" {
result = initializers.DB.Where("username LIKE ?", username+"%").Offset(offset).Limit(20).Find(&users)
} else {
result = initializers.DB.Offset(offset).Limit(20).Find(&users)
}
result := initializers.DB.Offset(offset).Limit(50).Find(&users)
if result.Error != nil { if result.Error != nil {
c.Status(http.StatusBadRequest) c.Status(http.StatusBadRequest)
return return
} }
var response []userResponse var response []UserData
for _, user := range users { for _, user := range users {
respUser := userResponse{ respUser := UserData{
ID: user.ID, ID: user.ID,
Username: user.Username, Username: user.Username,
DisplayName: user.DisplayName, DisplayName: user.DisplayName,

View file

@ -34,8 +34,8 @@ func main() {
api.Api.GET("/validate", middleware.RequireAuth, controllers.ValidateToken) api.Api.GET("/validate", middleware.RequireAuth, controllers.ValidateToken)
// Users // Users
api.Api.GET("/users", controllers.GetUsers) api.Api.GET("/users", middleware.RequireAuth, controllers.GetUsers)
api.Api.GET("/users/:id", controllers.GetUserById) api.Api.GET("/users/:id", middleware.RequireAuth, controllers.GetUserById)
// Files // Files
if currentMode == "debug" { if currentMode == "debug" {
@ -46,6 +46,8 @@ func main() {
// Posts // Posts
api.Api.POST("/posts", middleware.RequireAuth, controllers.CreatePost) api.Api.POST("/posts", middleware.RequireAuth, controllers.CreatePost)
api.Api.GET("/posts", middleware.RequireAuth, controllers.GetPosts)
api.Api.GET("/posts/:id", middleware.RequireAuth, controllers.GetPostById)
// Starting // Starting
err := api.Router.Run() err := api.Router.Run()

View file

@ -5,6 +5,6 @@ import "github.com/google/uuid"
type File struct { type File struct {
ID uint `gorm:"primaryKey;autoIncrement;"` ID uint `gorm:"primaryKey;autoIncrement;"`
FileName string FileName string
PostID *uuid.UUID PostID *uuid.UUID `json:"-"`
UserID *uuid.UUID UserID *uuid.UUID `json:"-"`
} }