micro/agent/input/telegram/conn.go

package telegram

import (
	"errors"
	"strings"
	"sync"

	"github.com/forestgiant/sliceutil"
	"github.com/micro/go-micro/agent/input"
	"github.com/micro/go-micro/util/log"
	tgbotapi "gopkg.in/telegram-bot-api.v4"
)

type telegramConn struct {
	input *telegramInput

	recv <-chan tgbotapi.Update
	exit chan bool

	syncCond *sync.Cond
	mutex    sync.Mutex
}

func newConn(input *telegramInput) (*telegramConn, error) {
	conn := &telegramConn{
		input: input,
	}

	conn.syncCond = sync.NewCond(&conn.mutex)

	go conn.run()

	return conn, nil
}

func (tc *telegramConn) run() {
	u := tgbotapi.NewUpdate(0)
	u.Timeout = 60
	updates, err := tc.input.api.GetUpdatesChan(u)
	if err != nil {
		return
	}

	tc.recv = updates
	tc.syncCond.Signal()

	select {
	case <-tc.exit:
		return
	}
}

func (tc *telegramConn) Close() error {
	return nil
}

func (tc *telegramConn) Recv(event *input.Event) error {
	if event == nil {
		return errors.New("event cannot be nil")
	}

	for {
		if tc.recv == nil {
			tc.mutex.Lock()
			tc.syncCond.Wait()
		}

		update := <-tc.recv

		if update.Message == nil || (len(tc.input.whitelist) > 0 && !sliceutil.Contains(tc.input.whitelist, update.Message.From.UserName)) {
			continue
		}

		if event.Meta == nil {
			event.Meta = make(map[string]interface{})
		}

		event.Type = input.TextEvent
		event.From = update.Message.From.UserName
		event.To = tc.input.api.Self.UserName
		event.Data = []byte(update.Message.Text)
		event.Meta["chatId"] = update.Message.Chat.ID
		event.Meta["chatType"] = update.Message.Chat.Type
		event.Meta["messageId"] = update.Message.MessageID

		return nil
	}
}

func (tc *telegramConn) Send(event *input.Event) error {
	messageText := strings.TrimSpace(string(event.Data))

	chatId := event.Meta["chatId"].(int64)
	chatType := ChatType(event.Meta["chatType"].(string))

	msgConfig := tgbotapi.NewMessage(chatId, messageText)
	msgConfig.ParseMode = tgbotapi.ModeHTML

	if sliceutil.Contains([]ChatType{Group, Supergroup}, chatType) {
		msgConfig.ReplyToMessageID = event.Meta["messageId"].(int)
	}

	_, err := tc.input.api.Send(msgConfig)

	if err != nil {
		// probably it could be because of nested HTML tags -- telegram doesn't allow nested tags
		log.Log("[telegram][Send] error:", err)
		msgConfig.Text = "This bot couldn't send the response (Internal error)"
		tc.input.api.Send(msgConfig)
	}

	return nil
}