package rest
import (
"fmt"
"github.com/sirupsen/logrus"
domainGroup "github.com/aldinokemal/go-whatsapp-web-multidevice/domains/group"
"github.com/aldinokemal/go-whatsapp-web-multidevice/pkg/utils"
"github.com/gofiber/fiber/v2"
"go.mau.fi/whatsmeow"
)
type Group struct {
Service domainGroup.IGroupUsecase
}
func InitRestGroup(app fiber.Router, service domainGroup.IGroupUsecase) Group {
rest := Group{Service: service}
app.Post("/group", rest.CreateGroup)
app.Post("/group/join-with-link", rest.JoinGroupWithLink)
app.Get("/group/info-from-link", rest.GetGroupInfoFromLink)
app.Get("/group/info", rest.GroupInfo)
app.Post("/group/leave", rest.LeaveGroup)
app.Post("/group/participants", rest.AddParticipants)
app.Post("/group/participants/remove", rest.DeleteParticipants)
app.Post("/group/participants/promote", rest.PromoteParticipants)
app.Post("/group/participants/demote", rest.DemoteParticipants)
app.Get("/group/participant-requests", rest.ListParticipantRequests)
app.Post("/group/participant-requests/approve", rest.ApproveParticipantRequests)
app.Post("/group/participant-requests/reject", rest.RejectParticipantRequests)
app.Post("/group/photo", rest.SetGroupPhoto)
app.Post("/group/name", rest.SetGroupName)
app.Post("/group/locked", rest.SetGroupLocked)
app.Post("/group/announce", rest.SetGroupAnnounce)
app.Post("/group/topic", rest.SetGroupTopic)
app.Get("/group/invite-link", rest.GetGroupInviteLink)
return rest
}
func (controller *Group) JoinGroupWithLink(c *fiber.Ctx) error {
var request domainGroup.JoinGroupWithLinkRequest
err := c.BodyParser(&request)
utils.PanicIfNeeded(err)
response, err := controller.Service.JoinGroupWithLink(c.UserContext(), request)
utils.PanicIfNeeded(err)
return c.JSON(utils.ResponseData{
Status: 200,
Code: "SUCCESS",
Message: "Success joined group",
Results: map[string]string{
"group_id": response,
},
})
}
func (controller *Group) GetGroupInfoFromLink(c *fiber.Ctx) error {
var request domainGroup.GetGroupInfoFromLinkRequest
err := c.QueryParser(&request)
utils.PanicIfNeeded(err)
response, err := controller.Service.GetGroupInfoFromLink(c.UserContext(), request)
utils.PanicIfNeeded(err)
return c.JSON(utils.ResponseData{
Status: 200,
Code: "SUCCESS",
Message: "Success get group info from link",
Results: response,
})
}
func (controller *Group) LeaveGroup(c *fiber.Ctx) error {
var request domainGroup.LeaveGroupRequest
err := c.BodyParser(&request)
utils.PanicIfNeeded(err)
utils.SanitizePhone(&request.GroupID)
err = controller.Service.LeaveGroup(c.UserContext(), request)
utils.PanicIfNeeded(err)
return c.JSON(utils.ResponseData{
Status: 200,
Code: "SUCCESS",
Message: "Success leave group",
})
}
func (controller *Group) CreateGroup(c *fiber.Ctx) error {
var request domainGroup.CreateGroupRequest
err := c.BodyParser(&request)
utils.PanicIfNeeded(err)
groupID, err := controller.Service.CreateGroup(c.UserContext(), request)
utils.PanicIfNeeded(err)
return c.JSON(utils.ResponseData{
Status: 200,
Code: "SUCCESS",
Message: fmt.Sprintf("Success created group with id %s", groupID),
Results: map[string]string{
"group_id": groupID,
},
})
}
func (controller *Group) AddParticipants(c *fiber.Ctx) error {
return controller.manageParticipants(c, whatsmeow.ParticipantChangeAdd, "Success add participants")
}
func (controller *Group) DeleteParticipants(c *fiber.Ctx) error {
return controller.manageParticipants(c, whatsmeow.ParticipantChangeRemove, "Success delete participants")
}
func (controller *Group) PromoteParticipants(c *fiber.Ctx) error {
return controller.manageParticipants(c, whatsmeow.ParticipantChangePromote, "Success promote participants")
}
func (controller *Group) DemoteParticipants(c *fiber.Ctx) error {
return controller.manageParticipants(c, whatsmeow.ParticipantChangeDemote, "Success demote participants")
}
func (controller *Group) ListParticipantRequests(c *fiber.Ctx) error {
var request domainGroup.GetGroupRequestParticipantsRequest
err := c.QueryParser(&request)
utils.PanicIfNeeded(err)
if request.GroupID == "" {
return c.Status(fiber.StatusBadRequest).JSON(utils.ResponseData{
Status: 400,
Code: "INVALID_GROUP_ID",
Message: "Group ID cannot be empty",
})
}
utils.SanitizePhone(&request.GroupID)
result, err := controller.Service.GetGroupRequestParticipants(c.UserContext(), request)
utils.PanicIfNeeded(err)
return c.JSON(utils.ResponseData{
Status: 200,
Code: "SUCCESS",
Message: "Success getting list requested participants",
Results: result,
})
}
func (controller *Group) ApproveParticipantRequests(c *fiber.Ctx) error {
return controller.handleRequestedParticipants(c, whatsmeow.ParticipantChangeApprove, "Success approve requested participants")
}
func (controller *Group) RejectParticipantRequests(c *fiber.Ctx) error {
return controller.handleRequestedParticipants(c, whatsmeow.ParticipantChangeReject, "Success reject requested participants")
}
// Generalized participant management handler
func (controller *Group) manageParticipants(c *fiber.Ctx, action whatsmeow.ParticipantChange, successMsg string) error {
var request domainGroup.ParticipantRequest
err := c.BodyParser(&request)
utils.PanicIfNeeded(err)
utils.SanitizePhone(&request.GroupID)
request.Action = action
result, err := controller.Service.ManageParticipant(c.UserContext(), request)
utils.PanicIfNeeded(err)
return c.JSON(utils.ResponseData{
Status: 200,
Code: "SUCCESS",
Message: successMsg,
Results: result,
})
}
// Generalized requested participants handler
func (controller *Group) handleRequestedParticipants(c *fiber.Ctx, action whatsmeow.ParticipantRequestChange, successMsg string) error {
var request domainGroup.GroupRequestParticipantsRequest
err := c.BodyParser(&request)
utils.PanicIfNeeded(err)
utils.SanitizePhone(&request.GroupID)
request.Action = action
result, err := controller.Service.ManageGroupRequestParticipants(c.UserContext(), request)
utils.PanicIfNeeded(err)
return c.JSON(utils.ResponseData{
Status: 200,
Code: "SUCCESS",
Message: successMsg,
Results: result,
})
}
func (controller *Group) SetGroupPhoto(c *fiber.Ctx) error {
var request domainGroup.SetGroupPhotoRequest
err := c.BodyParser(&request)
utils.PanicIfNeeded(err)
utils.SanitizePhone(&request.GroupID)
file, err := c.FormFile("photo")
if err == nil {
logrus.Printf("INFO: Received group photo - Filename: %s, Size: %d bytes, ContentType: %s",
file.Filename, file.Size, file.Header.Get("Content-Type"))
// Basic validation only - processing will be done in usecase
if err := utils.ValidateGroupPhotoFormat(file); err != nil {
logrus.Printf("ERROR: Group photo validation failed - %v", err)
return c.Status(fiber.StatusBadRequest).JSON(utils.ResponseData{
Status: 400,
Code: "INVALID_IMAGE_FORMAT",
Message: fmt.Sprintf("Image validation failed: %v", err),
})
}
request.Photo = file
} else {
logrus.Printf("DEBUG: No photo file provided - Error: %v", err)
}
pictureID, err := controller.Service.SetGroupPhoto(c.UserContext(), request)
if err != nil {
logrus.Printf("ERROR: WhatsApp service failed to set group photo - %v", err)
}
utils.PanicIfNeeded(err)
message := "Success update group photo"
if request.Photo == nil {
message = "Success remove group photo"
}
return c.JSON(utils.ResponseData{
Status: 200,
Code: "SUCCESS",
Message: message,
Results: domainGroup.SetGroupPhotoResponse{
PictureID: pictureID,
Message: message,
},
})
}
func (controller *Group) SetGroupName(c *fiber.Ctx) error {
var request domainGroup.SetGroupNameRequest
err := c.BodyParser(&request)
utils.PanicIfNeeded(err)
utils.SanitizePhone(&request.GroupID)
err = controller.Service.SetGroupName(c.UserContext(), request)
utils.PanicIfNeeded(err)
return c.JSON(utils.ResponseData{
Status: 200,
Code: "SUCCESS",
Message: fmt.Sprintf("Success update group name to '%s'", request.Name),
})
}
func (controller *Group) SetGroupLocked(c *fiber.Ctx) error {
var request domainGroup.SetGroupLockedRequest
err := c.BodyParser(&request)
utils.PanicIfNeeded(err)
utils.SanitizePhone(&request.GroupID)
err = controller.Service.SetGroupLocked(c.UserContext(), request)
utils.PanicIfNeeded(err)
message := "Success set group as unlocked"
if request.Locked {
message = "Success set group as locked"
}
return c.JSON(utils.ResponseData{
Status: 200,
Code: "SUCCESS",
Message: message,
})
}
func (controller *Group) SetGroupAnnounce(c *fiber.Ctx) error {
var request domainGroup.SetGroupAnnounceRequest
err := c.BodyParser(&request)
utils.PanicIfNeeded(err)
utils.SanitizePhone(&request.GroupID)
err = controller.Service.SetGroupAnnounce(c.UserContext(), request)
utils.PanicIfNeeded(err)
message := "Success disable announce mode"
if request.Announce {
message = "Success enable announce mode"
}
return c.JSON(utils.ResponseData{
Status: 200,
Code: "SUCCESS",
Message: message,
})
}
func (controller *Group) SetGroupTopic(c *fiber.Ctx) error {
var request domainGroup.SetGroupTopicRequest
err := c.BodyParser(&request)
utils.PanicIfNeeded(err)
utils.SanitizePhone(&request.GroupID)
err = controller.Service.SetGroupTopic(c.UserContext(), request)
utils.PanicIfNeeded(err)
message := "Success update group topic"
if request.Topic == "" {
message = "Success remove group topic"
}
return c.JSON(utils.ResponseData{
Status: 200,
Code: "SUCCESS",
Message: message,
})
}
// GroupInfo handles the /group/info endpoint to fetch group information
func (controller *Group) GroupInfo(c *fiber.Ctx) error {
var request domainGroup.GroupInfoRequest
err := c.QueryParser(&request)
utils.PanicIfNeeded(err)
utils.SanitizePhone(&request.GroupID)
response, err := controller.Service.GroupInfo(c.UserContext(), request)
utils.PanicIfNeeded(err)
return c.JSON(utils.ResponseData{
Status: 200,
Code: "SUCCESS",
Message: "Success get group info",
Results: response.Data,
})
}
func (controller *Group) GetGroupInviteLink(c *fiber.Ctx) error {
var request domainGroup.GetGroupInviteLinkRequest
err := c.QueryParser(&request)
utils.PanicIfNeeded(err)
utils.SanitizePhone(&request.GroupID)
response, err := controller.Service.GetGroupInviteLink(c.UserContext(), request)
utils.PanicIfNeeded(err)
return c.JSON(utils.ResponseData{
Status: 200,
Code: "SUCCESS",
Message: "Success get group invite link",
Results: response,
})
}
