feat: make compatible with v3 (#39)

* feat: add sanitize input

* feat move middleware -> rest

src/internal/rest/send_rest.go

@@ -31,6 +31,7 @@ func (controller *Send) SendText(c *fiber.Ctx) error {
 
 	// add validation send message
 	validations.ValidateSendMessage(request)
+	utils.SanitizePhone(&request.Phone)
 
 	response, err := controller.Service.SendText(c.UserContext(), request)
 	utils.PanicIfNeeded(err)
@@ -56,6 +57,7 @@ func (controller *Send) SendImage(c *fiber.Ctx) error {
 
 	//add validation send image
 	validations.ValidateSendImage(request)
+	utils.SanitizePhone(&request.Phone)
 
 	response, err := controller.Service.SendImage(c.UserContext(), request)
 	utils.PanicIfNeeded(err)
@@ -79,6 +81,7 @@ func (controller *Send) SendFile(c *fiber.Ctx) error {
 
 	//add validation send image
 	validations.ValidateSendFile(request)
+	utils.SanitizePhone(&request.Phone)
 
 	response, err := controller.Service.SendFile(c.UserContext(), request)
 	utils.PanicIfNeeded(err)
@@ -102,6 +105,7 @@ func (controller *Send) SendVideo(c *fiber.Ctx) error {
 
 	//add validation send image
 	validations.ValidateSendVideo(request)
+	utils.SanitizePhone(&request.Phone)
 
 	response, err := controller.Service.SendVideo(c.UserContext(), request)
 	utils.PanicIfNeeded(err)
@@ -120,6 +124,7 @@ func (controller *Send) SendContact(c *fiber.Ctx) error {
 
 	// add validation send contect
 	validations.ValidateSendContact(request)
+	utils.SanitizePhone(&request.Phone)
 
 	response, err := controller.Service.SendContact(c.UserContext(), request)
 	utils.PanicIfNeeded(err)
@@ -138,6 +143,7 @@ func (controller *Send) SendLink(c *fiber.Ctx) error {
 
 	err = validations.ValidateSendLink(request)
 	utils.PanicIfNeeded(err)
+	utils.SanitizePhone(&request.Phone)
 
 	response, err := controller.Service.SendLink(c.UserContext(), request)
 	utils.PanicIfNeeded(err)
@@ -157,6 +163,7 @@ func (controller *Send) RevokeMessage(c *fiber.Ctx) error {
 
 	err = validations.ValidateRevokeMessage(request)
 	utils.PanicIfNeeded(err)
+	utils.SanitizePhone(&request.Phone)
 
 	response, err := controller.Service.Revoke(c.UserContext(), request)
 	utils.PanicIfNeeded(err)
@@ -176,6 +183,7 @@ func (controller *Send) UpdateMessage(c *fiber.Ctx) error {
 
 	err = validations.ValidateUpdateMessage(request)
 	utils.PanicIfNeeded(err)
+	utils.SanitizePhone(&request.Phone)
 
 	response, err := controller.Service.UpdateMessage(c.UserContext(), request)
 	utils.PanicIfNeeded(err)

src/internal/rest/user_rest.go

@@ -35,6 +35,7 @@ func (controller *User) UserInfo(c *fiber.Ctx) error {
 
 	// add validation send message
 	validations.ValidateUserInfo(request)
+	utils.SanitizePhone(&request.Phone)
 
 	response, err := controller.Service.Info(c.Context(), request)
 	utils.PanicIfNeeded(err)
@@ -53,6 +54,7 @@ func (controller *User) UserAvatar(c *fiber.Ctx) error {
 
 	// add validation send message
 	validations.ValidateUserAvatar(request)
+	utils.SanitizePhone(&request.Phone)
 
 	response, err := controller.Service.Avatar(c.Context(), request)
 	utils.PanicIfNeeded(err)

src/utils/whatsapp.go

@@ -31,6 +31,16 @@ var (
 	startupTime   = time.Now().Unix()
 )
 
+func SanitizePhone(phone *string) {
+	if phone != nil && !strings.Contains(*phone, "@") {
+		if len(*phone) <= 15 {
+			*phone = fmt.Sprintf("%s@.whatsapp.net", *phone)
+		} else {
+			*phone = fmt.Sprintf("%s@g.us", *phone)
+		}
+	}
+}
+
 func GetPlatformName(deviceID int) string {
 	switch deviceID {
 	case 0: