package plugins
import (
"context"
"fmt"
"io"
"io/ioutil"
"log"
"net/http"
"os"
"strings"
tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api"
"github.com/neonxp/tamtam"
"transport/lib"
)
type Telegram struct {
Routing []lib.Routing
API *tgbotapi.BotAPI
Updates chan lib.Message
Bus chan lib.Message
}
func NewTelegram(token string, updates chan lib.Message, bus chan lib.Message, routing []lib.Routing) *Telegram {
tgApi, err := tgbotapi.NewBotAPI(token)
if err != nil {
log.Panic(err)
}
return &Telegram{API: tgApi, Updates: updates, Bus: bus, Routing: routing}
}
func (t *Telegram) Run(ctx context.Context) error {
u := tgbotapi.NewUpdate(0)
u.Timeout = 60
updates, err := t.API.GetUpdatesChan(u)
if err != nil {
return err
}
for {
select {
case <-ctx.Done():
return nil
case upd := <-updates:
if upd.Message == nil { // ignore any non-Message Updates
continue
}
for _, r := range t.Routing {
if r.TgID == upd.Message.Chat.ID {
from := fmt.Sprintf("%s %s", upd.Message.From.FirstName, upd.Message.From.LastName)
if upd.Message.From.UserName != "" {
from = fmt.Sprintf("%s %s (%s)", upd.Message.From.FirstName, upd.Message.From.LastName, upd.Message.From.UserName)
}
isSticker := ""
images := make([]string, 0)
if upd.Message.Sticker != nil {
isSticker = upd.Message.Sticker.Emoji
s, _ := t.API.GetFileDirectURL(upd.Message.Sticker.Thumbnail.FileID)
images = append(images, s)
}
if upd.Message.Photo != nil && len(*upd.Message.Photo) > 0 {
p := (*upd.Message.Photo)[0]
s, _ := t.API.GetFileDirectURL(p.FileID)
images = append(images, s)
}
t.Bus <- lib.Message{
To: r.TTID,
From: from,
Text: upd.Message.Text,
Images: images,
Sticker: isSticker,
}
}
}
case msg := <-t.Updates:
att := make([]interface{}, 0)
for _, a := range msg.Images {
att = append(att, tamtam.Image{Url: a})
}
text := fmt.Sprintf("*%s*:\n%s", strings.Trim(msg.From, " "), msg.Text)
if len(msg.Text) > 0 {
res, err := t.API.Send(tgbotapi.MessageConfig{
BaseChat: tgbotapi.BaseChat{
ChatID: msg.To,
ReplyToMessageID: 0,
},
Text: text,
DisableWebPagePreview: false,
ParseMode: "markdown",
})
log.Printf("[TG] Send text: %#v %#v", res, err)
}
if len(msg.Images) > 0 {
for _, i := range msg.Images {
u, err := t.DownloadFile(i)
if err != nil {
log.Printf("[TG] Download image: %#v", err)
continue
}
m := tgbotapi.NewPhotoUpload(msg.To, u)
m.Caption = msg.Sticker
res, err := t.API.Send(m)
log.Printf("[TG] Send image: %#v %#v", res, err)
}
}
}
}
}
func (t *Telegram) DownloadFile(u string) (string, error) {
resp, err := http.Get(u)
if err != nil {
return "", err
}
defer resp.Body.Close()
// Write the body to file
f, err := ioutil.TempFile(os.TempDir(), "tg*")
if err != nil {
return "", err
}
_, err = io.Copy(f, resp.Body)
defer f.Close()
return f.Name(), nil
}
