fix: aguardar sync de chaves E2E no envio (v1.0.7 update)

Author: luanmykel

CHANGELOG.md

@@ -3,10 +3,11 @@
 ## [v1.0.7] - 2026-01-19
 ### Corrigido
 - **Race condition no pareamento**: salvamento do JID agora é síncrono, garantindo que a sessão esteja persistida antes de liberar para uso.
-- Mensagens ficando em "Aguardando" após conectar instância: adicionado controle `sessionReady` que indica quando a sessão está pronta para enviar mensagens (após presence enviado).
+- **Mensagens "Aguardando"**: A API de envio de mensagens agora aguarda inteligentemente (até 10s) que a sincronização das chaves de criptografia (prekeys) esteja concluída antes de despachar a mensagem. Isso evita que as primeiras mensagens fiquem pendentes no destinatário.
 
 ### Adicionado
-- Método `IsSessionReady()` no Manager para verificar se a sessão está completamente pronta para operações de envio.
+- Método `IsSessionReady()` no Manager para verificar se a sessão está completamente pronta.
+- Lógica de polling no `MessageService` para garantir estabilidade no envio inicial.
 
 ### Internals
 - Limpeza automática do estado `sessionReady` e `pairingSuccess` ao deletar sessão.

cmd/api/main.go

@@ -135,7 +135,7 @@ func main() {
 	}
 
 	logr.Debug("inicializando serviços")
-	messageService := message.NewServiceWithSession(repos.Message, sessionManager, repos.Instance)
+	messageService := message.NewServiceWithSession(repos.Message, sessionManager, repos.Instance, logr)
 	apiTokenService := api_token.NewService(repos.APIToken)
 	userService := user.NewService(repos.User, apiTokenService, instanceService)
 	authService := auth.NewService(cfg.JWT.Secret, cfg.JWT.ExpHours, repos.User)

internal/service/message/service.go

@@ -14,6 +14,7 @@ import (
 	"go.mau.fi/whatsmeow"
 	"go.mau.fi/whatsmeow/proto/waE2E"
 	"go.mau.fi/whatsmeow/types"
+	"go.uber.org/zap"
 	"google.golang.org/protobuf/proto"
 
 	"github.com/open-apime/apime/internal/storage"
@@ -76,21 +77,27 @@ type Service struct {
 	repo         storage.MessageRepository
 	sessionMgr   SessionManager
 	instanceRepo storage.InstanceRepository
+	log          *zap.Logger
 }
 
 type SessionManager interface {
 	GetClient(instanceID string) (*whatsmeow.Client, error)
+	IsSessionReady(instanceID string) bool
 }
 
-func NewService(repo storage.MessageRepository) *Service {
-	return &Service{repo: repo}
+func NewService(repo storage.MessageRepository, log *zap.Logger) *Service {
+	return &Service{
+		repo: repo,
+		log:  log,
+	}
 }
 
-func NewServiceWithSession(repo storage.MessageRepository, sessionMgr SessionManager, instanceRepo storage.InstanceRepository) *Service {
+func NewServiceWithSession(repo storage.MessageRepository, sessionMgr SessionManager, instanceRepo storage.InstanceRepository, log *zap.Logger) *Service {
 	return &Service{
 		repo:         repo,
 		sessionMgr:   sessionMgr,
 		instanceRepo: instanceRepo,
+		log:          log,
 	}
 }
 
@@ -166,6 +173,27 @@ func (s *Service) Send(ctx context.Context, input SendInput) (model.Message, err
 		return model.Message{}, ErrInstanceNotConnected
 	}
 
+	// Aguardar sessão ficar pronta (sincronização de chaves E2E)
+	// Timeout de 10 segundos
+	readyStart := time.Now()
+	isReady := false
+	for time.Since(readyStart) < 10*time.Second {
+		if s.sessionMgr.IsSessionReady(input.InstanceID) {
+			isReady = true
+			break
+		}
+		time.Sleep(500 * time.Millisecond)
+	}
+
+	if !isReady {
+		// Logar warning mas tentar enviar mesmo assim (melhor que falhar)
+		// Em alguns casos o evento critical_block pode não vir ou vir diferente
+		s.log.Warn("Sessão não reportou pronta após timeout, enviando mensagem mesmo assim",
+			zap.String("instance_id", input.InstanceID),
+			zap.Duration("timeout", 10*time.Second),
+		)
+	}
+
 	normalizedTo := normalizeJID(input.To)
 
 	toJID, err := types.ParseJID(normalizedTo)

internal/session/whatsmeow/manager.go

@@ -129,7 +129,6 @@ func (m *Manager) GetSessionStorageInfo(instanceID string) SessionStorageInfo {
 	}
 }
 
-
 func (m *Manager) IsSessionReady(instanceID string) bool {
 	m.mu.RLock()
 	defer m.mu.RUnlock()
@@ -340,7 +339,6 @@ func (m *Manager) monitorQRChannel(instanceID string, client *whatsmeow.Client,
 			m.pairingSuccess[instanceID] = time.Now()
 			m.mu.Unlock()
 
-			
 			if client != nil && client.Store != nil && client.Store.ID != nil {
 				if m.instanceRepo != nil {
 					ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
@@ -388,15 +386,15 @@ func (m *Manager) monitorQRChannel(instanceID string, client *whatsmeow.Client,
 							}
 						} else {
 							m.log.Info("presence enviado com sucesso", zap.String("instance_id", instanceID))
-							
+
 							m.mu.Lock()
 							m.sessionReady[instanceID] = true
 							m.mu.Unlock()
 							m.log.Info("sessão marcada como pronta para mensagens", zap.String("instance_id", instanceID))
 							return
 						}
 					}
-					
+
 					m.mu.Lock()
 					m.sessionReady[instanceID] = true
 					m.mu.Unlock()
@@ -980,7 +978,7 @@ func (m *Manager) restoreSessionIfExists(ctx context.Context, instanceID string)
 				return
 			}
 		}
-		
+
 		m.mu.Lock()
 		m.sessionReady[instanceID] = true
 		m.mu.Unlock()
@@ -1321,14 +1319,14 @@ func (m *Manager) handleEvent(instanceID string, evt any) {
 						}
 					} else {
 						m.log.Info("presence enviado - instância totalmente ativa", zap.String("instance_id", instanceID))
-						
+
 						m.mu.Lock()
 						m.sessionReady[instanceID] = true
 						m.mu.Unlock()
 						return
 					}
 				}
-				
+
 				m.mu.Lock()
 				m.sessionReady[instanceID] = true
 				m.mu.Unlock()
@@ -1472,9 +1470,14 @@ func (m *Manager) handleEvent(instanceID string, evt any) {
 			zap.String("name", string(v.Name)),
 		)
 
-		// Quando critical_block sync completa, a instância está pronta para uso
 		if v.Name == "critical_block" {
-			m.log.Info("sincronização crítica concluída - instância pronta para receber mensagens",
+			m.log.Info("sincronização crítica concluída - prekeys E2E prontas",
+				zap.String("instance_id", instanceID))
+
+			m.mu.Lock()
+			m.sessionReady[instanceID] = true
+			m.mu.Unlock()
+			m.log.Info("sessão pronta para enviar mensagens (critical_block sync completo)",
 				zap.String("instance_id", instanceID))
 		}
 	default: