package api

import (
	"context"
	"github.com/TicketsBot/GoPanel/app"
	"github.com/TicketsBot/GoPanel/database"
	"github.com/TicketsBot/GoPanel/utils"
	"github.com/gin-gonic/gin"
	"github.com/rxdn/gdl/objects/user"
	"github.com/rxdn/gdl/rest"
	"net/http"
)

type whitelabelResponse struct {
	Id       uint64 `json:"id,string"`
	Username string `json:"username"`
	statusUpdateBody
}

func WhitelabelGet(c *gin.Context) {
	userId := c.Keys["userid"].(uint64)

	// Check if this is a different token
	bot, err := database.Client.Whitelabel.GetByUserId(c, userId)
	if err != nil {
		_ = c.AbortWithError(http.StatusInternalServerError, app.NewServerError(err))
		return
	}

	if bot.BotId == 0 {
		c.JSON(404, utils.ErrorStr("No bot found"))
		return
	}

	// Get status
	status, statusType, _, err := database.Client.WhitelabelStatuses.Get(c, bot.BotId)
	if err != nil {
		_ = c.AbortWithError(http.StatusInternalServerError, app.NewServerError(err))
		return
	}

	username := getBotUsername(c, bot.Token)

	c.JSON(200, whitelabelResponse{
		Id:       bot.BotId,
		Username: username,
		statusUpdateBody: statusUpdateBody{ // Zero values if no status is fine
			Status:     status,
			StatusType: user.ActivityType(statusType),
		},
	})
}

func getBotUsername(ctx context.Context, token string) string {
	user, err := rest.GetCurrentUser(ctx, token, nil)
	if err != nil {
		// TODO: Log error
		return "Unknown User"
	}

	return user.Username
}