package bots

import (
	"fmt"
	"go.uber.org/zap"
	"gopkg.in/tucnak/telebot.v2"
	"strconv"
	"sync"
)

type Sender struct {
	done chan struct{}
	in   chan *Msg
	wg   sync.WaitGroup
}
type Msg struct {
	To      telebot.Recipient
	What    interface{}
	Options []interface{}
}

//牺牲错误处理和解耦，加快发送速度

func NewSender() *Sender {
	return &Sender{}
}
func (s *Sender) Init(goroutine int) {

	s.done = make(chan struct{})
	s.in = make(chan *Msg)

	for i := 0; i < goroutine; i++ {
		go func() {
			s.sender()
		}()
	}
}

//Stop until all messages were sent
func (s *Sender) Stop() {
	s.wg.Wait()
	close(s.done)
	fmt.Println("task finished")
}
func (s *Sender) SendMessageByID(ID int64, what interface{}, options ...interface{}) {
	s.wg.Add(1)
	go func() {
		chat, err := bot.ChatByID(strconv.FormatInt(ID, 10))
		if err != nil {
			zap.S().Errorw("failed to get chat",
				"error", err,
				"id", ID,
			)
			s.wg.Done()
			return
		}
		s.SendMessage(chat, what, options...)
	}()
}
func (s *Sender) SendMessage(to telebot.Recipient, what interface{}, options ...interface{}) {
	s.in <- &Msg{
		To:      to,
		What:    what,
		Options: options,
	}
}
func (s *Sender) sender() {
	for {
		select {
		case msg, f := <-s.in:
			if !f {
				continue
			}
			fmt.Println("send to " + msg.To.Recipient())
			if _, err := bot.Send(msg.To, msg.What, msg.Options...); err != nil {
				zap.S().Errorw("failed to send msg",
					"error", err,
					"id", msg.To.Recipient(),
				)
			}
			s.wg.Done()
		case <-s.done:
			return
		}
	}
}