package server

import (
	"code.smartsheep.studio/hydrogen/interactive/pkg/database"
	"code.smartsheep.studio/hydrogen/interactive/pkg/models"
	"code.smartsheep.studio/hydrogen/interactive/pkg/services"
	"github.com/gofiber/fiber/v2"
	"github.com/samber/lo"
	"time"
)

func getOwnPost(c *fiber.Ctx) error {
	user := c.Locals("principal").(models.Account)

	id := c.Params("postId")
	take := c.QueryInt("take", 0)
	offset := c.QueryInt("offset", 0)

	tx := database.C.Where(&models.Post{
		Alias:    id,
		AuthorID: user.ID,
	})

	post, err := services.GetPost(tx)
	if err != nil {
		return fiber.NewError(fiber.StatusNotFound, err.Error())
	}

	tx = database.C.
		Where(&models.Post{ReplyID: &post.ID}).
		Where("published_at <= ? OR published_at IS NULL", time.Now()).
		Order("created_at desc")

	var count int64
	if err := tx.
		Model(&models.Post{}).
		Count(&count).Error; err != nil {
		return fiber.NewError(fiber.StatusInternalServerError, err.Error())
	}

	posts, err := services.ListPost(tx, take, offset)
	if err != nil {
		return fiber.NewError(fiber.StatusBadRequest, err.Error())
	}

	return c.JSON(fiber.Map{
		"data":    post,
		"count":   count,
		"related": posts,
	})
}

func listOwnPost(c *fiber.Ctx) error {
	take := c.QueryInt("take", 0)
	offset := c.QueryInt("offset", 0)
	realmId := c.QueryInt("realmId", 0)

	user := c.Locals("principal").(models.Account)

	tx := database.C.
		Where(&models.Post{AuthorID: user.ID}).
		Where("published_at <= ? OR published_at IS NULL", time.Now()).
		Order("created_at desc")

	if realmId > 0 {
		tx = tx.Where(&models.Post{RealmID: lo.ToPtr(uint(realmId))})
	} else {
		tx = tx.Where("realm_id IS NULL")
	}

	if len(c.Query("category")) > 0 {
		tx = services.FilterPostWithCategory(tx, c.Query("category"))
	}

	if len(c.Query("tag")) > 0 {
		tx = services.FilterPostWithTag(tx, c.Query("tag"))
	}

	var count int64
	if err := tx.
		Model(&models.Post{}).
		Count(&count).Error; err != nil {
		return fiber.NewError(fiber.StatusInternalServerError, err.Error())
	}

	posts, err := services.ListPost(tx, take, offset)
	if err != nil {
		return fiber.NewError(fiber.StatusBadRequest, err.Error())
	}

	return c.JSON(fiber.Map{
		"count": count,
		"data":  posts,
	})
}