status-go/protocol/messenger_auto_message.go

package protocol

import (
	"context"
	"fmt"
	"time"

	"go.uber.org/zap"

	gocommon "github.com/status-im/status-go/common"
	"github.com/status-im/status-go/protocol/common"
	"github.com/status-im/status-go/protocol/protobuf"
	"github.com/status-im/status-go/signal"
)

// autoMessageInterval is how often we should send a message
const autoMessageInterval = 120 * time.Second

const autoMessageChatID = "status-bot"

func (m *Messenger) AutoMessageEnabled() (bool, error) {
	return m.settings.AutoMessageEnabled()
}

func (m *Messenger) startAutoMessageLoop() error {
	enabled, err := m.AutoMessageEnabled()
	if err != nil {
		m.logger.Error("[auto message] failed to start auto message loop", zap.Error(err))
		return err
	}

	if !enabled {
		return nil
	}

	m.logger.Info("[auto message] starting auto message loop")
	ticker := time.NewTicker(autoMessageInterval)
	count := 0
	go func() {
		defer gocommon.LogOnPanic()
		for {
			select {
			case <-ticker.C:
				count++
				timestamp := time.Now().Format(time.RFC3339)

				msg := common.NewMessage()
				msg.Text = fmt.Sprintf("%d\n%s", count, timestamp)
				msg.ChatId = autoMessageChatID
				msg.LocalChatID = autoMessageChatID
				msg.ContentType = protobuf.ChatMessage_TEXT_PLAIN
				resp, err := m.SendChatMessage(context.Background(), msg)
				if err != nil {
					m.logger.Error("[auto message] failed to send message", zap.Error(err))
					continue
				}
				signal.SendNewMessages(resp)

				err = m.UpdateMessageOutgoingStatus(msg.ID, common.OutgoingStatusDelivered)
				if err != nil {
					m.logger.Error("[auto message] failed to mark message as delivered", zap.Error(err))
					continue
				}

				//send signal to client that message status updated
				if m.config.messengerSignalsHandler != nil {
					m.config.messengerSignalsHandler.MessageDelivered(autoMessageChatID, msg.ID)
				}
			case <-m.quit:
				ticker.Stop()
				return
			}
		}
	}()

	return nil
}