package app

import (
	"context"

	"github.com/tidbcloud/AgentTeam/internal/store/agent"
	"github.com/tidbcloud/AgentTeam/internal/store/message"
	"github.com/tidbcloud/AgentTeam/internal/store/message/testdelivery"
)

func (store *Store) ReceiveAgentMessages(ctx context.Context, authenticated agent.AuthenticatedAgent, limit int) ([]message.ReceiveMessage, error) {
	return testdelivery.ReceiveAgentMessagesFromFanout(ctx, store.db, authenticated, limit)
}

func (store *Store) AgentHasUnreadMessages(ctx context.Context, authenticated agent.AuthenticatedAgent) (bool, error) {
	return testdelivery.AgentHasUnreadFanoutMessages(ctx, store.db, authenticated)
}