156 lines
4.0 KiB
Go
156 lines
4.0 KiB
Go
package api
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"git.solsynth.dev/hydrogen/interactive/pkg/internal/database"
|
|
"git.solsynth.dev/hydrogen/interactive/pkg/internal/gap"
|
|
"git.solsynth.dev/hydrogen/interactive/pkg/internal/models"
|
|
"git.solsynth.dev/hydrogen/interactive/pkg/internal/services"
|
|
"github.com/gofiber/fiber/v2"
|
|
"github.com/samber/lo"
|
|
)
|
|
|
|
func listRecommendationNews(c *fiber.Ctx) error {
|
|
take := c.QueryInt("take", 0)
|
|
offset := c.QueryInt("offset", 0)
|
|
realm := c.Query("realm")
|
|
|
|
tx := services.FilterPostDraft(database.C)
|
|
|
|
if user, authenticated := c.Locals("user").(models.Account); authenticated {
|
|
tx = services.FilterPostWithUserContext(tx, &user)
|
|
} else {
|
|
tx = services.FilterPostWithUserContext(tx, nil)
|
|
}
|
|
|
|
if len(realm) > 0 {
|
|
if realm, err := services.GetRealmWithAlias(realm); err != nil {
|
|
return fiber.NewError(fiber.StatusBadRequest, fmt.Sprintf("realm was not found: %v", err))
|
|
} else {
|
|
tx = services.FilterPostWithRealm(tx, realm.ID)
|
|
}
|
|
}
|
|
|
|
if c.QueryBool("noReply", true) {
|
|
tx = services.FilterPostReply(tx)
|
|
}
|
|
|
|
countTx := tx
|
|
count, err := services.CountPost(countTx)
|
|
if err != nil {
|
|
return fiber.NewError(fiber.StatusInternalServerError, err.Error())
|
|
}
|
|
|
|
order := "published_at DESC"
|
|
if c.QueryBool("featured", false) {
|
|
order = "published_at DESC, (COALESCE(total_upvote, 0) - COALESCE(total_downvote, 0)) DESC"
|
|
}
|
|
|
|
items, err := services.ListPost(tx, take, offset, order)
|
|
if err != nil {
|
|
return fiber.NewError(fiber.StatusBadRequest, err.Error())
|
|
}
|
|
|
|
return c.JSON(fiber.Map{
|
|
"count": count,
|
|
"data": items,
|
|
})
|
|
}
|
|
|
|
func listRecommendationFriends(c *fiber.Ctx) error {
|
|
if err := gap.H.EnsureAuthenticated(c); err != nil {
|
|
return err
|
|
}
|
|
user := c.Locals("user").(models.Account)
|
|
|
|
take := c.QueryInt("take", 0)
|
|
offset := c.QueryInt("offset", 0)
|
|
realm := c.Query("realm")
|
|
|
|
tx := services.FilterPostDraft(database.C)
|
|
tx = services.FilterPostWithUserContext(tx, &user)
|
|
|
|
friends, _ := services.ListAccountFriends(user)
|
|
friendList := lo.Map(friends, func(item models.Account, index int) uint {
|
|
return item.ID
|
|
})
|
|
|
|
tx = tx.Where("author_id IN ?", friendList)
|
|
|
|
if len(realm) > 0 {
|
|
if realm, err := services.GetRealmWithAlias(realm); err != nil {
|
|
return fiber.NewError(fiber.StatusBadRequest, fmt.Sprintf("realm was not found: %v", err))
|
|
} else {
|
|
tx = services.FilterPostWithRealm(tx, realm.ID)
|
|
}
|
|
}
|
|
|
|
if c.QueryBool("noReply", true) {
|
|
tx = services.FilterPostReply(tx)
|
|
}
|
|
|
|
countTx := tx
|
|
count, err := services.CountPost(countTx)
|
|
if err != nil {
|
|
return fiber.NewError(fiber.StatusInternalServerError, err.Error())
|
|
}
|
|
|
|
order := "published_at DESC"
|
|
if c.QueryBool("featured", false) {
|
|
order = "published_at DESC, (COALESCE(total_upvote, 0) - COALESCE(total_downvote, 0)) DESC"
|
|
}
|
|
|
|
items, err := services.ListPost(tx, take, offset, order)
|
|
if err != nil {
|
|
return fiber.NewError(fiber.StatusBadRequest, err.Error())
|
|
}
|
|
|
|
return c.JSON(fiber.Map{
|
|
"count": count,
|
|
"data": items,
|
|
})
|
|
}
|
|
|
|
func listRecommendationShuffle(c *fiber.Ctx) error {
|
|
take := c.QueryInt("take", 0)
|
|
offset := c.QueryInt("offset", 0)
|
|
realm := c.Query("realm")
|
|
|
|
tx := services.FilterPostDraft(database.C)
|
|
|
|
if user, authenticated := c.Locals("user").(models.Account); authenticated {
|
|
tx = services.FilterPostWithUserContext(tx, &user)
|
|
} else {
|
|
tx = services.FilterPostWithUserContext(tx, nil)
|
|
}
|
|
|
|
if len(realm) > 0 {
|
|
if realm, err := services.GetRealmWithAlias(realm); err != nil {
|
|
return fiber.NewError(fiber.StatusBadRequest, fmt.Sprintf("realm was not found: %v", err))
|
|
} else {
|
|
tx = services.FilterPostWithRealm(tx, realm.ID)
|
|
}
|
|
}
|
|
|
|
if c.QueryBool("noReply", true) {
|
|
tx = services.FilterPostReply(tx)
|
|
}
|
|
|
|
countTx := tx
|
|
count, err := services.CountPost(countTx)
|
|
if err != nil {
|
|
return fiber.NewError(fiber.StatusInternalServerError, err.Error())
|
|
}
|
|
|
|
items, err := services.ListPost(tx, take, offset, "RANDOM()")
|
|
if err != nil {
|
|
return fiber.NewError(fiber.StatusBadRequest, err.Error())
|
|
}
|
|
|
|
return c.JSON(fiber.Map{
|
|
"count": count,
|
|
"data": items,
|
|
})
|
|
}
|