115 lines
3.0 KiB
Go
115 lines
3.0 KiB
Go
package api
|
|
|
|
import (
|
|
"context"
|
|
"github.com/TicketsBot/GoPanel/app"
|
|
"github.com/TicketsBot/GoPanel/botcontext"
|
|
dbclient "github.com/TicketsBot/GoPanel/database"
|
|
"github.com/TicketsBot/GoPanel/utils"
|
|
"github.com/TicketsBot/database"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/rxdn/gdl/objects/channel"
|
|
"github.com/rxdn/gdl/objects/channel/embed"
|
|
"github.com/rxdn/gdl/objects/user"
|
|
"github.com/rxdn/gdl/rest"
|
|
"net/http"
|
|
"regexp"
|
|
"strconv"
|
|
"time"
|
|
)
|
|
|
|
var MentionRegex, _ = regexp.Compile("<@(\\d+)>")
|
|
|
|
func GetTicket(c *gin.Context) {
|
|
guildId := c.Keys["guildid"].(uint64)
|
|
userId := c.Keys["userid"].(uint64)
|
|
|
|
botContext, err := botcontext.ContextForGuild(guildId)
|
|
if err != nil {
|
|
_ = c.AbortWithError(http.StatusInternalServerError, app.NewServerError(err))
|
|
return
|
|
}
|
|
|
|
ticketId, err := strconv.Atoi(c.Param("ticketId"))
|
|
if err != nil {
|
|
c.JSON(http.StatusBadRequest, utils.ErrorStr("Invalid ticket ID"))
|
|
return
|
|
}
|
|
|
|
// Get the ticket struct
|
|
ticket, err := dbclient.Client.Tickets.Get(c, ticketId, guildId)
|
|
if err != nil {
|
|
_ = c.AbortWithError(http.StatusInternalServerError, app.NewServerError(err))
|
|
return
|
|
}
|
|
|
|
if ticket.GuildId != guildId {
|
|
c.JSON(http.StatusForbidden, utils.ErrorStr("Ticket does not belong to guild"))
|
|
return
|
|
}
|
|
|
|
if !ticket.Open {
|
|
c.JSON(http.StatusNotFound, utils.ErrorStr("Ticket is closed"))
|
|
return
|
|
}
|
|
|
|
hasPermission, requestErr := utils.HasPermissionToViewTicket(c, guildId, userId, ticket)
|
|
if requestErr != nil {
|
|
// TODO
|
|
c.JSON(requestErr.StatusCode, utils.ErrorJson(requestErr))
|
|
return
|
|
}
|
|
|
|
if !hasPermission {
|
|
c.JSON(http.StatusForbidden, utils.ErrorStr("You do not have permission to view this ticket"))
|
|
return
|
|
}
|
|
|
|
if ticket.ChannelId == nil {
|
|
c.JSON(http.StatusNotFound, utils.ErrorStr("Ticket channel not found"))
|
|
return
|
|
}
|
|
|
|
messages, err := fetchMessages(botContext, ticket)
|
|
if err != nil {
|
|
_ = c.AbortWithError(http.StatusInternalServerError, app.NewServerError(err))
|
|
return
|
|
}
|
|
|
|
c.JSON(200, gin.H{
|
|
"success": true,
|
|
"ticket": ticket,
|
|
"messages": messages,
|
|
})
|
|
}
|
|
|
|
type StrippedMessage struct {
|
|
Author user.User `json:"author"`
|
|
Content string `json:"content"`
|
|
Timestamp time.Time `json:"timestamp"`
|
|
Attachments []channel.Attachment `json:"attachments"`
|
|
Embeds []embed.Embed `json:"embeds"`
|
|
}
|
|
|
|
func fetchMessages(botContext *botcontext.BotContext, ticket database.Ticket) ([]StrippedMessage, error) {
|
|
// Get messages
|
|
messages, err := rest.GetChannelMessages(context.Background(), botContext.Token, botContext.RateLimiter, *ticket.ChannelId, rest.GetChannelMessagesData{Limit: 100})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// Format messages, exclude unneeded data
|
|
stripped := make([]StrippedMessage, len(messages))
|
|
for i, message := range utils.Reverse(messages) {
|
|
stripped[i] = StrippedMessage{
|
|
Author: message.Author,
|
|
Content: message.Content,
|
|
Timestamp: message.Timestamp,
|
|
Attachments: message.Attachments,
|
|
Embeds: message.Embeds,
|
|
}
|
|
}
|
|
|
|
return stripped, nil
|
|
}
|